bug修正

bigdata-web/src/main/java/org/diagbot/bigdata/work/BigDataParamsProxy.java

@@ -8,9 +8,6 @@ import org.diagbot.nlp.feature.FeatureType;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-import java.util.ArrayList;
-import java.util.List;
-
 /**
  * @ClassName org.diagbot.bigdata.work.ParamsDataProxy
  * @Description TODO
@@ -33,7 +30,7 @@ public class BigDataParamsProxy {
      * @param searchData
      */
     public AlgorithmClassify[] createAlgorithmClassify(String sysCode, String[] featureTypes, SearchData searchData) {
-        List<AlgorithmClassify> algorithmClassifies = new ArrayList<>();
+        AlgorithmClassify[] classifies = new AlgorithmClassify[featureTypes.length];
         //下了诊断且其他信息全为空 反推标识
         boolean reverse = !StringUtils.isEmpty(searchData.getDiag()) && StringUtils.isEmpty(searchData.getSymptom());
         for (int i = 0; i < featureTypes.length; i++) {
@@ -42,64 +39,56 @@ public class BigDataParamsProxy {
                 switch (FeatureType.parse(featureTypes[i])) {
                     case SYMPTOM:
                         if (reverse) {
-                            algorithmClassifies.add(AlgorithmClassify.NEURAL_DIAG_SYMPTOM);
+                            classifies[i] = AlgorithmClassify.NEURAL_DIAG_SYMPTOM;
                         } else {
-                            algorithmClassifies.add(AlgorithmClassify.NEURAL_SYMPTOM);
+                            classifies[i] = AlgorithmClassify.NEURAL_SYMPTOM;
                         }
                         break;
                     case DIAG:
                         if (reverse) {
-                            algorithmClassifies.add(null);
+                            classifies[i] = null;
                         } else {
-                            algorithmClassifies.add(AlgorithmClassify.NEURAL_DIAG);
+                            classifies[i] = AlgorithmClassify.NEURAL_DIAG;
                         }
                         break;
                     case VITAL:
                         if (reverse) {
-                            algorithmClassifies.add(AlgorithmClassify.NEURAL_DIAG_VITAL);
-
+                            classifies[i] = AlgorithmClassify.NEURAL_DIAG_VITAL;
                         } else {
-                            algorithmClassifies.add(AlgorithmClassify.NEURAL_VITAL);
-
+                            classifies[i] = AlgorithmClassify.NEURAL_VITAL;
                         }
                         break;
                     case LIS:
                         if (reverse) {
-                            algorithmClassifies.add(AlgorithmClassify.NEURAL_DIAG_LIS);
-
+                            classifies[i] = AlgorithmClassify.NEURAL_DIAG_LIS;
                         } else {
-                            algorithmClassifies.add(AlgorithmClassify.NEURAL_LIS);
+                            classifies[i] = AlgorithmClassify.NEURAL_LIS;
                         }
                         break;
                     case PACS:
                         if (reverse) {
-                            algorithmClassifies.add(AlgorithmClassify.NEURAL_DIAG_PACS);
+                            classifies[i] = AlgorithmClassify.NEURAL_DIAG_PACS;
                         } else {
-                            algorithmClassifies.add(AlgorithmClassify.NEURAL_PACS);
+                            classifies[i] = AlgorithmClassify.NEURAL_PACS;
                         }
                         break;
                     case TREAT:
                         if (reverse) {
-                            algorithmClassifies.add(AlgorithmClassify.NEURAL_DIAG_TREAT);
-
+                            classifies[i] = AlgorithmClassify.NEURAL_DIAG_TREAT;
                         } else {
-                            algorithmClassifies.add(AlgorithmClassify.NEURAL_TREAT);
-
+                            classifies[i] = AlgorithmClassify.NEURAL_TREAT;
                         }
                         break;
                     case HISTORY:
                         if (reverse) {
-                            algorithmClassifies.add(AlgorithmClassify.NEURAL_DIAG_HISTORY);
-
+                            classifies[i] = AlgorithmClassify.NEURAL_DIAG_HISTORY;
                         } else {
-                            algorithmClassifies.add(AlgorithmClassify.NEURAL_HISTORY);
+                            classifies[i] = AlgorithmClassify.NEURAL_HISTORY;
                         }
                         break;
                 }
             }
         }
-        AlgorithmClassify[] classifies = new AlgorithmClassify[algorithmClassifies.size()];
-        algorithmClassifies.toArray(classifies);
         return classifies;
     }
 }