Add saveMode to CustomFS

byzer-org · Dec 24, 2021 · 83fb873 · 83fb873
1 parent 04217f9
commit 83fb873
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/streamingpro-mlsql/src/main/java/tech/mlsql/datasource/impl/CustomFS.scala b/streamingpro-mlsql/src/main/java/tech/mlsql/datasource/impl/CustomFS.scala
@@ -49,10 +49,9 @@ class CustomFS(override val uid: String) extends MLSQLSource
         ("spark.hadoop." + item._1, item._2)
       } else item
     }.foreach(item => session.conf.set(item._1, item._2))
-
 
     val format = config.config.getOrElse("implClass", fullFormat)
-    writer.options(loadFileConf).format(format).save(config.path)
+    writer.options(loadFileConf).mode(config.mode).format(format).save(config.path)
   }