lanyuanxiaoyao/hudi
1
0
Fork 0
Code Activity

[HUDI-3571] Spark datasource continuous checkpoint should have own fs variable (#5265)

Browse Source
This commit is contained in:
satishm
2022-04-08 16:46:01 +05:30
committed by GitHub
parent d7cc767dbc
commit 26eb7b8183
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
hudi-integ-test/src/main/scala/org/apache/hudi/integ/testsuite/SparkDataSourceContinuousIngest.scala
View File

@@ -34,8 +34,9 @@ class SparkDataSourceContinuousIngest(val spark: SparkSession, val conf: Configu
def startIngestion(): Unit = {
val fs = sourcePath.getFileSystem(conf)
var checkPointFs = checkpointFile.getFileSystem(conf)
var orderedBatch : Array[FileStatus] = null
if (fs.exists(checkpointFile)) {
if (checkPointFs.exists(checkpointFile)) {
log.info("Checkpoint file exists. ")
val checkpoint = spark.sparkContext.textFile(checkpointFile.toString).collect()(0)
log.warn("Checkpoint to resume from " + checkpoint)
@@ -69,7 +70,7 @@ class SparkDataSourceContinuousIngest(val spark: SparkSession, val conf: Configu
val df = spark.read.format(sourceFormat).load(pathToConsume.toString)
df.write.format("hudi").options(hudiOptions).mode(SaveMode.Append).save(hudiBasePath.toString)
writeToFile(checkpointFile, entry.getPath.getName, fs)
writeToFile(checkpointFile, entry.getPath.getName, checkPointFs)
log.info("Completed batch " + entry + ". Moving to next batch. Sleeping for " + minSyncIntervalSeconds + " secs before next batch")
Thread.sleep(minSyncIntervalSeconds * 1000)
})