添加新的排班验证规则，如果当天的时刻表有的路牌在排班数据中没有，提示错误

徐烜
1 parent e53f80dc
Showing 7 changed files with 232 additions and 6 deletions
src/main/java/com/bsth/service/schedule/plan/DroolsSchedulePlan.java
src/main/java/com/bsth/service/schedule/rules/ttinfo/LpInfoResult_output.java
src/main/java/com/bsth/service/schedule/rules/ttinfo/LpInfoResultsFunction.java
src/main/java/com/bsth/service/schedule/rules/validate/ValidWantLpFunction.java
src/main/java/com/bsth/service/schedule/rules/validate/ValidateResource.java
src/main/resources/rules/functions.drl
src/main/resources/rules/validplan.drl
@@ -475,6 +475,9 @@ public class DroolsSchedulePlan {
      * @param schedulePlan
      */
     public void validPlanResult(PlanResult planResult) {
+        // 1-0、获取路牌信息
+        LpInfoResults_output lpInfoResults_output = (LpInfoResults_output) this.ttInfoOutput(this.mainLine)[1];
+
         // 1-1、构造drools规则输入数据，输出数据
         ValidateParam validateParam = new ValidateParam(
                 new DateTime(this.from), new DateTime(this.to));
@@ -494,6 +497,9 @@ public class DroolsSchedulePlan {
         for (SchedulePlanInfo schedulePlanInfo: planResult.getSchedulePlanInfos()) {
             session.insert(schedulePlanInfo);
         }
+        for (LpInfoResult_output lpInfoResult_output: lpInfoResults_output.getLpInfoResult_outputs()) {
+            session.insert(lpInfoResult_output);
+        }
         // 执行rule
         session.fireAllRules();
@@ -10,6 +10,8 @@ public class LpInfoResult_output {
     private DateTime dateTime;
     /** 路牌Id */
     private String lpId;
+    /** 路牌名字 */
+    private String lpName;
     /** 线路Id */
     private String xlId;
     /** 时刻表Id */
@@ -56,4 +58,12 @@ public class LpInfoResult_output {
     public void setTtInfoName(String ttInfoName) {
         this.ttInfoName = ttInfoName;
     }
+
+    public String getLpName() {
+        return lpName;
+    }
+
+    public void setLpName(String lpName) {
+        this.lpName = lpName;
+    }
 }
@@ -64,6 +64,7 @@ public class LpInfoResultsFunction implements AccumulateFunction {
             if (lpInfoResultsData.lpInfoResult_outputMap.get(ttInfoDetail.getLp().getId()) == null) {
                 LpInfoResult_output lpInfoResult_output = new LpInfoResult_output();
                 lpInfoResult_output.setLpId(String.valueOf(ttInfoDetail.getLp().getId()));
+                lpInfoResult_output.setLpName(ttInfoDetail.getLp().getLpName());
                 lpInfoResult_output.setXlId(String.valueOf(ttInfoDetail.getXl().getId()));
                 lpInfoResult_output.setTtInfoId(String.valueOf(ttInfoDetail.getTtinfo().getId()));
                 lpInfoResult_output.setTtInfoName(ttInfoDetail.getTtinfo().getName());
+package com.bsth.service.schedule.rules.validate;
+
+import com.bsth.entity.schedule.SchedulePlanInfo;
+import com.bsth.service.schedule.rules.ttinfo.LpInfoResult_output;
+import org.kie.api.runtime.rule.AccumulateFunction;
+
+import java.io.*;
+import java.text.SimpleDateFormat;
+import java.util.*;
+
+/**
+ * 计算缺少路牌错误。
+ * 同一天，如果有时刻表路牌没有执行到，统计。
+ * 注意：使用这个函数时，要一天计算一次，多天计算无意义。
+ */
+public class ValidWantLpFunction implements AccumulateFunction {
+    @Override
+    public void writeExternal(ObjectOutput out) throws IOException {
+    }
+
+    @Override
+    public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException {
+
+    }
+
+    protected static class WantLpInfo implements Externalizable {
+        /** 错误描述 */
+        public List<ValidateResults_output.ValidInfo> validInfoList = new ArrayList<>();
+        /** 每天的路牌班次数量 */
+        public Map<String, Integer> lpBcCount = new HashMap<>();
+        /** 每天的路牌名字对应 */
+        public Map<String, String> lpNamesMap = new HashMap<>();
+        /** 排班日期 */
+        public Date scheduleDate;
+
+        public WantLpInfo() {
+
+        }
+
+        @Override
+        public void writeExternal(ObjectOutput out) throws IOException {
+            out.writeObject(validInfoList);
+        }
+
+        @Override
+        public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException {
+            validInfoList = (List<ValidateResults_output.ValidInfo>) in.readObject();
+        }
+    }
+
+    @Override
+    public Serializable createContext() {
+        return new WantLpInfo();
+    }
+
+    @Override
+    public void init(Serializable serializable) throws Exception {
+        // TODO：
+//        System.out.println("init");
+    }
+
+    @Override
+    public void accumulate(Serializable context, Object o) {
+        WantLpInfo wantLpInfo = (WantLpInfo) context;
+        ValidateResource validateResource = (ValidateResource) o;
+
+        SchedulePlanInfo spi = validateResource.getSpi();
+        List<LpInfoResult_output> lpiList = validateResource.getLpiList();
+
+        // 获取当天所有路牌信息，每个验证对象都带当天所有路牌信息，只需判定一次
+        if (wantLpInfo.lpNamesMap.isEmpty()) {
+            for (LpInfoResult_output lpInfoResult_output: lpiList) {
+                wantLpInfo.lpNamesMap.put(lpInfoResult_output.getLpId(), lpInfoResult_output.getLpName());
+                wantLpInfo.lpBcCount.put(lpInfoResult_output.getLpId(), 0);
+            }
+        }
+        if (wantLpInfo.scheduleDate == null) {
+            wantLpInfo.scheduleDate = spi.getScheduleDate();
+        }
+
+        // 累计记录每个排班班次出现次数
+        String lpId_spi = spi.getLp().toString();
+        wantLpInfo.lpBcCount.put(lpId_spi, wantLpInfo.lpBcCount.get(lpId_spi) + 1);
+
+    }
+
+    @Override
+    public boolean supportsReverse() {
+        return true;
+    }
+
+    @Override
+    public void reverse(Serializable context, Object o) throws Exception {
+        WantLpInfo wantLpInfo = (WantLpInfo) context;
+
+        // 清空数据，下一天继续迭代
+        wantLpInfo.lpBcCount.clear();
+        wantLpInfo.lpNamesMap.clear();
+        wantLpInfo.scheduleDate = null;
+        wantLpInfo.validInfoList.clear();
+
+//        System.out.println("reverse");
+
+    }
+
+    @Override
+    public Class<?> getResultType() {
+        return List.class;
+    }
+
+    @Override
+    public Object getResult(Serializable context) throws Exception {
+        WantLpInfo wantLpInfo = (WantLpInfo) context;
+
+        SimpleDateFormat sf = new SimpleDateFormat("yyyy年MM月dd日");
+        String infoFormat = "日期(%s),路牌(%s),没有排班班次";
+
+        for (String lpId : wantLpInfo.lpBcCount.keySet()) {
+            if (wantLpInfo.lpBcCount.get(lpId) == 0) {
+                // 排班没有班次
+                ValidateResults_output.ValidInfo validInfo = new ValidateResults_output.ValidInfo();
+                validInfo.setSd(wantLpInfo.scheduleDate);
+                validInfo.setDesc(String.format(
+                        infoFormat,
+                        sf.format(wantLpInfo.scheduleDate),
+                        wantLpInfo.lpNamesMap.get(lpId))
+                );
+                wantLpInfo.validInfoList.add(validInfo);
+            }
+        }
+
+        System.out.println("ValidWantLpFunction==>" + wantLpInfo.lpBcCount);
+
+        return wantLpInfo.validInfoList;
+    }
+
+}
+
+
+
+
+
+
+
+
+package com.bsth.service.schedule.rules.validate;
+
+import com.bsth.entity.schedule.SchedulePlanInfo;
+import com.bsth.service.schedule.rules.ttinfo.LpInfoResult_output;
+
+import java.util.Date;
+import java.util.List;
+
+/**
+ * 验证操作dsl类。
+ */
+public class ValidateResource {
+    /** 具体日期 */
+    private Date sd;
+    /** 当天排班计划 */
+    private SchedulePlanInfo spi;
+    /** 当天所有路牌信息 */
+    private List<LpInfoResult_output> lpiList;
+
+    public Date getSd() {
+        return sd;
+    }
+
+    public void setSd(Date sd) {
+        this.sd = sd;
+    }
+
+    public SchedulePlanInfo getSpi() {
+        return spi;
+    }
+
+    public void setSpi(SchedulePlanInfo spi) {
+        this.spi = spi;
+    }
+
+    public List<LpInfoResult_output> getLpiList() {
+        return lpiList;
+    }
+
+    public void setLpiList(List<LpInfoResult_output> lpiList) {
+        this.lpiList = lpiList;
+    }
+}
@@ -8,3 +8,4 @@ import accumulate com.bsth.service.schedule.rules.ttinfo.LpInfoResultsFunction l
 import accumulate com.bsth.service.schedule.rules.ttinfo.MinRuleQyrqFunction minruleqyrq;
 import accumulate com.bsth.service.schedule.rules.validate.ValidRepeatBcFunction vrb;
 import accumulate com.bsth.service.schedule.rules.validate.ValidWholeRerunBcFunction vwrb;
+import accumulate com.bsth.service.schedule.rules.validate.ValidWantLpFunction vwlp;
@@ -2,6 +2,7 @@ package com.bsth.service.schedule.rules.validate;
 import com.bsth.entity.schedule.SchedulePlanInfo;
 import com.bsth.service.schedule.rules.ttinfo.LpInfoResult_output;
+import com.bsth.service.schedule.rules.validate.ValidateResource;
 import org.joda.time.*;
 import java.util.*;
@@ -14,7 +15,22 @@ global Logger log;
 // 输出
 global ValidateResults_output validResult;
-//------------------------- 第一阶段、构造循环体 ----------------------------//
+//------------------------- 第一阶段、构造验证源 ----------------------------//
+rule "Calcu_validate_resource"
+    salience 2000
+    when
+        $spi: SchedulePlanInfo($sd: scheduleDate)
+        $lpiList: ArrayList() from collect (LpInfoResult_output(dateTime.getMillis() == $sd.getTime()))
+    then
+        ValidateResource resource = new ValidateResource();
+        resource.setSd($sd);
+        resource.setSpi($spi);
+        resource.setLpiList($lpiList);
+
+        insert(resource);
+end
+
+//------------------------- 第二阶段、构造循环体 ----------------------------//
 declare Loop_param
     start_date: DateTime // 开始日期（这个要不停的更新迭代）
@@ -39,23 +55,27 @@ rule &quot;Calcu_Loop_param&quot;
         insert(p);
 end
-//------------------------- 第二阶段、验证计算 ----------------------------//
+//------------------------- 第三阶段、验证计算 ----------------------------//
-rule "Valid_repeat_bc" // 验证是否存在重复班次
+rule "Valid_repeat_bc" // 验证是否存在重复班次，未套跑班次，未执行路牌
     salience 600
     when
         $lp: Loop_param($sd: start_date, $ed: end_date)
         eval($sd.isBefore($ed) || $sd.isEqual($ed))
-        $spiList: ArrayList() from collect (SchedulePlanInfo(scheduleDate.getTime() == $sd.getMillis()))
-        $infos: ArrayList() from accumulate ($spi: SchedulePlanInfo() from $spiList, vrb($spi))
-        $infos2: ArrayList() from accumulate ($spi: SchedulePlanInfo() from $spiList, vwrb($spi))
+        $vrList: ArrayList() from collect (ValidateResource(sd.getTime() == $sd.getMillis()))
+        $infos: ArrayList() from accumulate ($vr: ValidateResource() from $vrList, vrb($vr.getSpi()))
+        $infos2: ArrayList() from accumulate ($vr: ValidateResource() from $vrList, vwrb($vr.getSpi()))
+        $infos3: ArrayList() from accumulate ($vr: ValidateResource() from $vrList, vwlp($vr))
     then
         // TODO：
 //        log.info("日期={}，班次重复错误数={}", $sd, $infos.size());
+//        log.info("班次数={}", $vrList.size());
+
         validResult.getInfos().addAll($infos);
         validResult.getInfos().addAll($infos2);
+        validResult.getInfos().addAll($infos3);
         // 迭代
         $lp.setStart_date($sd.plusDays(1));