优化代码

flow-common/flow-common-flowable-starter/src/main/java/com/flow/flowable/converter/CustomServiceTaskXMLConverter.java

@@ -2,14 +2,13 @@ package com.flow.flowable.converter;
 
 import com.flow.flowable.model.CopyServiceTask;
 import com.flow.flowable.model.JumpServiceTask;
+import com.flow.flowable.utils.ProcessElementUtil;
 import org.flowable.bpmn.converter.ServiceTaskXMLConverter;
 import org.flowable.bpmn.model.BaseElement;
 import org.flowable.bpmn.model.BpmnModel;
-import org.flowable.bpmn.model.FieldExtension;
 import org.flowable.bpmn.model.ServiceTask;
 
 import javax.xml.stream.XMLStreamReader;
-import java.util.List;
 
 public class CustomServiceTaskXMLConverter extends ServiceTaskXMLConverter {
     public CustomServiceTaskXMLConverter() {
@@ -38,20 +37,12 @@ public class CustomServiceTaskXMLConverter extends ServiceTaskXMLConverter {
     }
 
     protected void convertCopyServiceTaskXMLProperties(CopyServiceTask copyServiceTask, BpmnModel bpmnModel, XMLStreamReader xtr) throws Exception {
-        List<FieldExtension> fieldExtensions = copyServiceTask.getFieldExtensions();
-        for (FieldExtension fieldExtension : fieldExtensions) {
-            if ("assignee".equals(fieldExtension.getFieldName())) {
-                copyServiceTask.setAssignee(fieldExtension.getExpression());
-            }
-        }
+        String assignee = ProcessElementUtil.getFieldExtensionExpression(copyServiceTask, "assignee");
+        copyServiceTask.setAssignee(assignee);
     }
 
     protected void convertJumpServiceTaskXMLProperties(JumpServiceTask jumpServiceTask, BpmnModel bpmnModel, XMLStreamReader xtr) throws Exception {
-        List<FieldExtension> fieldExtensions = jumpServiceTask.getFieldExtensions();
-        for (FieldExtension fieldExtension : fieldExtensions) {
-            if ("targetNode".equals(fieldExtension.getFieldName())) {
-                jumpServiceTask.setTargetNode(fieldExtension.getStringValue());
-            }
-        }
+        String targetNode = ProcessElementUtil.getFieldExtensionValue(jumpServiceTask, "targetNode");
+        jumpServiceTask.setTargetNode(targetNode);
     }
 }