to feature #7169

ouyangyewei · ouyangyewei · commit 5b3b3264d3c7 · 2021-12-04T18:16:46.000+08:00
diff --git a/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/enums/Status.java b/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/enums/Status.java
@@ -18,6 +18,7 @@
 package org.apache.dolphinscheduler.api.enums;
 
 import java.util.Locale;
+import java.util.Optional;
 
 import org.springframework.context.i18n.LocaleContextHolder;
 
@@ -382,4 +383,18 @@ public String getMsg() {
             return this.enMsg;
         }
     }
+
+    /**
+     * Retrieve Status enum entity by status code.
+     * @param code
+     * @return
+     */
+    public static Optional<Status> findStatusBy(int code) {
+        for (Status status : Status.values()) {
+            if (code == status.getCode()) {
+                return Optional.of(status);
+            }
+        }
+        return Optional.empty();
+    }
 }
diff --git a/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/enums/StatusTest.java b/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/enums/StatusTest.java
@@ -21,6 +21,7 @@
 import org.springframework.context.i18n.LocaleContextHolder;
 
 import java.util.Locale;
+import java.util.Optional;
 
 import static org.junit.Assert.*;
 
@@ -41,4 +42,15 @@ public void testGetMsg() {
         Assert.assertEquals("成功", Status.SUCCESS.getMsg());
     }
 
+    @Test
+    public void testGetStatusByCode() {
+        // FAILURE
+        Optional<Status> optional = Status.findStatusBy(1);
+        Assert.assertFalse(optional.isPresent());
+
+        // SUCCESS
+        optional = Status.findStatusBy(10018);
+        Assert.assertTrue(optional.isPresent());
+        Assert.assertEquals(Status.PROJECT_NOT_FOUNT, optional.get());
+    }
 }
