使用复合密钥访问带有spark dataframe的mongodb集合

unftdfkk  于 2021-07-14  发布在  Spark
关注(0)|答案(0)|浏览(217)

我尝试在一个名为 test . 以下代码与单列一起用于执行搜索:

val dfMongo = MongoSpark.load(sparkSession, confMongoDb.getReadConfig("test"))

然后,构建一个数组以选择相应的 ID 价值观:

val colToComp = dfUsedForSearch
  .select("ID")
  .map(r =>{r.getString(0)})
  .collect()

最后,搜索 ID 在mongodb集合中执行以下操作:

val dfFilter = dfMongo.filter(dfMongo.col("ID").isInCollection(colToComp))
  .select("ID")

如何使用复合键进行此类搜索,例如: (ID_A, ID_B, ID_C) ?

暂无答案!

目前还没有任何答案,快来回答吧!

相关问题