Add lock.

infiniflow · Dec 29, 2023 · f8d30c4 · f8d30c4
1 parent d71532e
commit f8d30c4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/scheduler/fragment_task.cpp b/src/scheduler/fragment_task.cpp
@@ -111,7 +111,10 @@ u64 FragmentTask::FragmentId() const {
 }
 
 // Finished **OR** Error
-bool FragmentTask::IsComplete() const { return sink_state_->prev_op_state_->Complete() || status_ == FragmentTaskStatus::kError; }
+bool FragmentTask::IsComplete() {
+    UniqueLock<Mutex> lock(mutex_);
+    return sink_state_->prev_op_state_->Complete() || status_ == FragmentTaskStatus::kError;
+}
 
 // Stream fragment source has no data
 bool FragmentTask::QuitFromWorkerLoop() {
@@ -142,7 +145,6 @@ TaskBinding FragmentTask::TaskBinding() const {
 }
 
 void FragmentTask::CompleteTask() {
-    UniqueLock<Mutex> lock(mutex_);
     if (status_ == FragmentTaskStatus::kRunning) {
         status_ = FragmentTaskStatus::kFinished;
     }