Có hai cách để lấy dữ liệu từ MongoDB sang Apache Spark.
Phương pháp 1:Sử dụng Casbah (Lớp trên trình điều khiển Java MongDB)
val uriRemote = MongoClientURI("mongodb://RemoteURL:27017/")
val mongoClientRemote = MongoClient(uriRemote)
val dbRemote = mongoClientRemote("dbName")
val collectionRemote = dbRemote("collectionName")
val ipMongo = collectionRemote.find
val ipRDD = sc.makeRDD(ipMongo.toList)
ipRDD.saveAsTextFile("hdfs://path/to/hdfs")
Ở đây, chúng tôi đang sử dụng Scala và Casbah để lấy dữ liệu trước rồi lưu vào HDFS.
Phương pháp 2:Spark Worker được chúng tôi sử dụng
Phiên bản mã tốt hơn:Sử dụng Spark worker và sử dụng nhiều lõi để lấy dữ liệu trong thời gian ngắn.
val config = new Configuration()
config.set("mongo.job.input.format","com.mongodb.hadoop.MongoInputFormat")
config.set("mongo.input.uri", "mongodb://RemoteURL:27017/dbName.collectionName")
val keyClassName = classOf[Object]
val valueClassName = classOf[BSONObject]
val inputFormatClassName = classOf[com.mongodb.hadoop.MongoInputFormat]
val ipRDD = sc.newAPIHadoopRDD(config,inputFormatClassName,keyClassName,valueClassName)
ipRDD.saveAsTextFile("hdfs://path/to/hdfs")