transferInvestmentManager

coolstore-partner-common/src/main/java/com/cool/store/enums/WorkflowStageEnum.java

@@ -1,6 +1,12 @@
 package com.cool.store.enums;
 
+import org.apache.commons.lang3.StringUtils;
+
+import java.util.Arrays;
 import java.util.HashMap;
+import java.util.Map;
+import java.util.function.Function;
+import java.util.stream.Collectors;
 
 /**
  * @Author suzhuhong
@@ -44,4 +50,13 @@ public enum WorkflowStageEnum {
         return message;
     }
 
+    protected static final Map<String, WorkflowStageEnum> map = Arrays.stream(values()).collect(
+            Collectors.toMap(WorkflowStageEnum::getCode, Function.identity()));
+
+    public static WorkflowStageEnum getWorkflowStageByCode(String code) {
+        if(StringUtils.isBlank(code)) {
+            return null;
+        }
+        return map.get(code);
+    }
 }