reorder checks

apache · Nov 27, 2024 · 8e58c4b · 8e58c4b
1 parent 2e9c95a
commit 8e58c4b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/be/src/vec/sink/load_stream_stub.cpp b/be/src/vec/sink/load_stream_stub.cpp
@@ -334,11 +334,11 @@ Status LoadStreamStub::wait_for_schema(int64_t partition_id, int64_t index_id, i
 
 Status LoadStreamStub::close_wait(RuntimeState* state, int64_t timeout_ms) {
     DBUG_EXECUTE_IF("LoadStreamStub::close_wait.long_wait", DBUG_BLOCK);
+    if (!_is_open.load()) {
+        // we don't need to close wait on non-open streams
+        return Status::OK();
+    }
     if (!_is_closing.load()) {
-        if (!_is_open.load()) {
-            // we don't need to close wait on non-open streams
-            return Status::OK();
-        }
         return _status;
     }
     if (_is_closed.load()) {