[Enhancement] no need to getLabelState if an existing label is finished (StarRocks#231)

Signed-off-by: shuzz <[email protected]>

Author: szza

starrocks-stream-load-sdk/src/main/java/com/starrocks/data/load/stream/DefaultStreamLoader.java

@@ -272,13 +272,13 @@ protected StreamLoadResponse send(StreamLoadTableProperties tableProperties, Tab
                     streamLoadResponse.setCostNanoTime(System.nanoTime() - startNanoTime);
                     region.complete(streamLoadResponse);
                 } else if (StreamLoadConstants.RESULT_STATUS_LABEL_EXISTED.equals(status)) {
-                    String labelState = getLabelState(host, region.getDatabase(), region.getTable(), label, Collections.singleton(StreamLoadConstants.LABEL_STATE_PREPARE));
-                    if (StreamLoadConstants.LABEL_STATE_COMMITTED.equals(labelState) || StreamLoadConstants.LABEL_STATE_VISIBLE.equals(labelState)) {
+                    String existingJobStatus = streamLoadBody.getExistingJobStatus();
+                    if (StreamLoadConstants.EXISTING_JOB_STATUS_FINISHED.equals(existingJobStatus)) {
                         streamLoadResponse.setCostNanoTime(System.nanoTime() - startNanoTime);
                         region.complete(streamLoadResponse);
                     } else {
                         String errorMsage = String.format("Stream load failed because label existed, " +
-                                "db: %s, table: %s, label: %s, label state: %s", region.getDatabase(), region.getTable(), label, labelState);
+                                "db: %s, table: %s, label: %s, existingJobStatus: %s", region.getDatabase(), region.getTable(), label, existingJobStatus);
                         throw new StreamLoadFailException(errorMsage);
                     }
                 } else {

starrocks-stream-load-sdk/src/main/java/com/starrocks/data/load/stream/StreamLoadConstants.java

@@ -20,6 +20,8 @@ public interface StreamLoadConstants {
     String RESULT_STATUS_TRANSACTION_COMMIT_TIMEOUT = "Commit Timeout";
     String RESULT_STATUS_TRANSACTION_PUBLISH_TIMEOUT = "Publish Timeout";
 
+    String EXISTING_JOB_STATUS_FINISHED = "FINISHED";
+
     String LABEL_STATE_VISIBLE = "VISIBLE";
     String LABEL_STATE_COMMITTED = "COMMITTED";
     String LABEL_STATE_PREPARED = "PREPARED";

starrocks-stream-load-sdk/src/main/java/com/starrocks/data/load/stream/StreamLoadResponse.java

@@ -174,6 +174,10 @@ public String getStatus() {
             return status;
         }
 
+        public String getExistingJobStatus() {
+            return this.existingJobStatus;
+        }
+
         public String getMsg() {
             return msg;
         }