ramesh-muthusamy commented on a change in pull request #26339: [SPARK-27194][SPARK-29302][SQL] Fix the issue that for dynamic partition overwrite a task would conflict with its speculative task URL: https://github.com/apache/spark/pull/26339#discussion_r358381751
########## File path: core/src/main/scala/org/apache/spark/internal/io/HadoopMapReduceCommitProtocol.scala ########## @@ -236,13 +271,23 @@ class HadoopMapReduceCommitProtocol( committer.setupTask(taskContext) addedAbsPathFiles = mutable.Map[String, String]() partitionPaths = mutable.Set[String]() + stagingTaskFiles = mutable.Set[Path]() } override def commitTask(taskContext: TaskAttemptContext): TaskCommitMessage = { val attemptId = taskContext.getTaskAttemptID logTrace(s"Commit task ${attemptId}") SparkHadoopMapRedUtil.commitTask( committer, taskContext, attemptId.getJobID.getId, attemptId.getTaskID.getId) + if (dynamicPartitionOverwrite) { + val fs = stagingDir.getFileSystem(taskContext.getConfiguration) + stagingTaskFiles.foreach { stagingTaskFile => + val fileName = stagingTaskFile.getName + val taskPartitionPath = getPartitionPath(stagingTaskFile) + val destFile = new Path(new Path(stagingDir, taskPartitionPath), fileName) + fs.rename(stagingTaskFile, destFile) Review comment: requesting your help in clarifying the note "If we just rename a folder, it would overwrite other task's output." I see that we are creating _temporary path including the partition info as _temporary/partitionPath/taskAttemptId/ , if we rename this to partitionPath/ I believe it would not impact other partitions and would not overwrite the task output. ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services --------------------------------------------------------------------- To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org For additional commands, e-mail: reviews-h...@spark.apache.org