pyspark读写mongo数据库

读写mongo

方式一

from pyspark.sql import SparkSession
my_spark = SparkSession \
    .builder \
    .appName("myApp") \
    .config("spark.mongodb.input.uri", "mongodb://127.0.0.1/test.coll") \
    .config("spark.mongodb.output.uri", "mongodb://127.0.0.1/test.coll") \
    .getOrCreate()

people.write.format("mongo").mode("append").save()# 默认存在test库的coll表里
people.write.format("mongo").mode("append").option("database","people").option("collection", "contacts").save()# 指定存在people库里的contact表里

df = spark.read.format("mongo").load()# 默认读test库的coll表
df = spark.read.format("mongo").option("uri","mongodb://127.0.0.1/people.contacts").load()# 指定读people库的contact表

方式二

mongo_url = "mongodb://username:password@hostIP:port/dbname"

spark.read.format("com.mongodb.spark.sql").option("uri", mongo_url).option("collection", "your_table_name").load().createOrReplaceTempView("imp_dsj_task")

dsj_endf.write.option("header", "true").format("com.mongodb.spark.sql").option("replaceDocument","true").option("spark.mongodb.output.uri",mongo_url).option("collection","your_table_name").mode("Append").save()

参考文献

Spark Connector Python Guide