Update

徐烜
1 parent 4e501eed
Showing 6 changed files with 239 additions and 13 deletions
src/main/java/com/bsth/service/schedule/impl/SchedulePlanServiceImpl.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/ttinfo/LpInfoResults_output.java
src/main/resources/rules/functions.drl
src/main/resources/rules/ttinfo.drl
@@ -15,6 +15,7 @@ import com.bsth.service.schedule.rules.shiftloop.ScheduleCalcuParam_input;
 import com.bsth.service.schedule.rules.shiftloop.ScheduleResults_output;
 import com.bsth.service.schedule.rules.shiftloop.ScheduleRule_input;
 import com.bsth.service.schedule.rules.strategy.IStrategy;
+import com.bsth.service.schedule.rules.ttinfo.LpInfoResults_output;
 import com.bsth.service.schedule.rules.ttinfo.TTInfoCalcuParam_input;
 import com.bsth.service.schedule.rules.ttinfo.TTInfoResults_output;
 import com.bsth.service.schedule.rules.ttinfo.TTInfo_input;
@@ -156,6 +157,7 @@ public class SchedulePlanServiceImpl extends BServiceImpl&lt;SchedulePlan, Long&gt; im
                 );
         // 规则输出数据
         TTInfoResults_output ttInfoResults_output = new TTInfoResults_output();
+        LpInfoResults_output lpInfoResults_output = new LpInfoResults_output();
         // 1-2、构造drools session->载入数据->启动规则->计算->销毁session
         // 创建session，内部配置的是stateful
@@ -163,8 +165,11 @@ public class SchedulePlanServiceImpl extends BServiceImpl&lt;SchedulePlan, Long&gt; im
         // 设置gloable对象，在drl中通过别名使用
         session.setGlobal("results", ttInfoResults_output);
+        session.setGlobal("lpInfoResults_output", lpInfoResults_output);
         session.setGlobal("log", logger); // 设置日志
+        session.setGlobal("tTInfoDetailService", ttInfoDetailService);
+
         // 载入数据
         session.insert(ttInfoCalcuParam_input);
         for (TTInfo ttInfo : ttInfos) {
@@ -225,18 +230,18 @@ public class SchedulePlanServiceImpl extends BServiceImpl&lt;SchedulePlan, Long&gt; im
     @Transactional(propagation = Propagation.REQUIRED, isolation = Isolation.READ_COMMITTED)
     public SchedulePlan save2(SchedulePlan schedulePlan) {
-        // 1、循环规则计算输出
-        ScheduleResults_output scheduleResults_output = loopRuleOutput(schedulePlan);
+//        // 1、循环规则计算输出
+//        ScheduleResults_output scheduleResults_output = loopRuleOutput(schedulePlan);
         // 2、时刻表选择
         TTInfoResults_output ttInfoResults_output = ttInfoOutput(schedulePlan);
-        // 3、计划输出
-        PlanResult planResult = planResultOutput(schedulePlan, scheduleResults_output, ttInfoResults_output);
-
-        schedulePlan.getSchedulePlanInfoList().addAll(planResult.getSchedulePlanInfos());
-        schedulePlan.setTtInfoId("todo"); // TODO：待修正
-        schedulePlan.setTtInfoNames("todo"); // TODO：待修正
-
-        super.save(schedulePlan);
+//        // 3、计划输出
+//        PlanResult planResult = planResultOutput(schedulePlan, scheduleResults_output, ttInfoResults_output);
+//
+//        schedulePlan.getSchedulePlanInfoList().addAll(planResult.getSchedulePlanInfos());
+//        schedulePlan.setTtInfoId("todo"); // TODO：待修正
+//        schedulePlan.setTtInfoNames("todo"); // TODO：待修正
+//
+//        super.save(schedulePlan);
         return new SchedulePlan();
     }
+package com.bsth.service.schedule.rules.ttinfo;
+
+import org.joda.time.DateTime;
+
+/**
+ * 每日路爬信息。
+ */
+public class LpInfoResult_output {
+    /** 具体日期 */
+    private DateTime dateTime;
+    /** 路牌Id */
+    private String lpId;
+    /** 线路Id */
+    private String xlId;
+    /** 时刻表Id */
+    private String ttInfoId;
+
+    public DateTime getDateTime() {
+        return dateTime;
+    }
+
+    public void setDateTime(DateTime dateTime) {
+        this.dateTime = dateTime;
+    }
+
+    public String getLpId() {
+        return lpId;
+    }
+
+    public void setLpId(String lpId) {
+        this.lpId = lpId;
+    }
+
+    public String getXlId() {
+        return xlId;
+    }
+
+    public void setXlId(String xlId) {
+        this.xlId = xlId;
+    }
+
+    public String getTtInfoId() {
+        return ttInfoId;
+    }
+
+    public void setTtInfoId(String ttInfoId) {
+        this.ttInfoId = ttInfoId;
+    }
+}
+package com.bsth.service.schedule.rules.ttinfo;
+
+import com.bsth.entity.schedule.TTInfoDetail;
+import org.kie.api.runtime.rule.AccumulateFunction;
+
+import java.io.*;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * 路牌信息统计函数。
+ */
+public class LpInfoResultsFunction implements AccumulateFunction {
+    @Override
+    public void writeExternal(ObjectOutput out) throws IOException {
+
+    }
+
+    @Override
+    public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException {
+
+    }
+
+    protected static class LpInfoResultsData implements Externalizable {
+        public List<LpInfoResult_output> lpInfoResult_outputs = new ArrayList<>();
+        public Map<Long, LpInfoResult_output> lpInfoResult_outputMap = new HashMap<>();
+        public TTInfoDetail ttInfoDetail;
+
+        public LpInfoResultsData() {}
+
+        @Override
+        public void writeExternal(ObjectOutput out) throws IOException {
+            out.writeObject(lpInfoResult_outputs);
+            out.writeObject(ttInfoDetail);
+        }
+
+        @Override
+        public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException {
+            lpInfoResult_outputs = (List<LpInfoResult_output>) in.readObject();
+            ttInfoDetail = (TTInfoDetail) in.readObject();
+        }
+    }
+
+    @Override
+    public Serializable createContext() {
+        return new LpInfoResultsData();
+    }
+
+    @Override
+    public void init(Serializable context) throws Exception {
+        LpInfoResultsData lpInfoResultsData = (LpInfoResultsData) context;
+        lpInfoResultsData.lpInfoResult_outputs = new ArrayList<>();
+        lpInfoResultsData.lpInfoResult_outputMap = new HashMap<>();
+    }
+
+    @Override
+    public void accumulate(Serializable context, Object value) {
+        LpInfoResultsData lpInfoResultsData = (LpInfoResultsData) context;
+        TTInfoDetail ttInfoDetail = (TTInfoDetail) value;
+
+        if (ttInfoDetail.getLp() != null) {
+            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.setXlId(String.valueOf(ttInfoDetail.getXl().getId()));
+                lpInfoResult_output.setTtInfoId(String.valueOf(ttInfoDetail.getTtinfo().getId()));
+
+                lpInfoResultsData.lpInfoResult_outputMap.put(
+                        ttInfoDetail.getLp().getId(),
+                        lpInfoResult_output);
+            }
+        }
+
+        lpInfoResultsData.lpInfoResult_outputs.clear();
+        lpInfoResultsData.lpInfoResult_outputs.addAll(
+                lpInfoResultsData.lpInfoResult_outputMap.values());
+    }
+
+    @Override
+    public boolean supportsReverse() {
+        return true;
+    }
+
+    @Override
+    public void reverse(Serializable context, Object value) throws Exception {
+        LpInfoResultsData lpInfoResultsData = (LpInfoResultsData) context;
+        TTInfoDetail ttInfoDetail = (TTInfoDetail) value;
+
+        lpInfoResultsData.lpInfoResult_outputMap.remove(ttInfoDetail.getLp().getId());
+
+        lpInfoResultsData.lpInfoResult_outputs.clear();
+        lpInfoResultsData.lpInfoResult_outputs.addAll(
+                lpInfoResultsData.lpInfoResult_outputMap.values());
+
+    }
+
+    @Override
+    public Object getResult(Serializable context) throws Exception {
+        LpInfoResultsData lpInfoResultsData = (LpInfoResultsData) context;
+        return lpInfoResultsData.lpInfoResult_outputs;
+    }
+
+    @Override
+    public Class<?> getResultType() {
+        return List.class;
+    }
+}
+package com.bsth.service.schedule.rules.ttinfo;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * Created by xu on 17/3/21.
+ */
+public class LpInfoResults_output {
+
+    private List<LpInfoResult_output> lpInfoResult_outputs = new ArrayList<>();
+
+    public List<LpInfoResult_output> getLpInfoResult_outputs() {
+        return lpInfoResult_outputs;
+    }
+
+    public void setLpInfoResult_outputs(List<LpInfoResult_output> lpInfoResult_outputs) {
+        this.lpInfoResult_outputs = lpInfoResult_outputs;
+    }
+}
@@ -2,4 +2,5 @@ package com.bsth.service.schedule;
 import accumulate com.bsth.service.schedule.rules.ttinfo2.ErrorBcCountFunction ecount;
 import accumulate com.bsth.service.schedule.rules.shiftloop.GidsCountFunction gidscount;
-import accumulate com.bsth.service.schedule.rules.shiftloop.GidFbTimeFunction gidfbtime;
 \ No newline at end of file
+import accumulate com.bsth.service.schedule.rules.shiftloop.GidFbTimeFunction gidfbtime;
+import accumulate com.bsth.service.schedule.rules.ttinfo.LpInfoResultsFunction lpinforesult;
 \ No newline at end of file
@@ -7,13 +7,31 @@ import com.bsth.service.schedule.rules.ttinfo.TTInfoCalcuParam_input;
 import com.bsth.service.schedule.rules.ttinfo.TTInfo_input;
 import com.bsth.service.schedule.rules.ttinfo.TTInfoResult_output;
 import com.bsth.service.schedule.rules.ttinfo.TTInfoResults_output;
+import com.bsth.service.schedule.rules.ttinfo.LpInfoResult_output;
+import com.bsth.service.schedule.rules.ttinfo.LpInfoResults_output;
-import org.slf4j.Logger;
+import com.bsth.service.schedule.TTInfoDetailService;
+
+import com.bsth.entity.schedule.TTInfo;
+import com.bsth.entity.schedule.TTInfoDetail;
+
+import org.slf4j.Logger
 // 全局日志
 global Logger log;
+// service
+global TTInfoDetailService tTInfoDetailService;
 // return输出
 global TTInfoResults_output results
+global LpInfoResults_output lpInfoResults_output
+
+function Map ttidParams(List ttinfolist) {
+    // 获取第一张时刻表id
+    TTInfo_input ttInfo_input = (TTInfo_input) ttinfolist.get(0);
+    Map param = new HashMap();
+    param.put("ttinfo.id_eq", Long.parseLong(ttInfo_input.getTtInfoId()));
+    return param;
+}
 /*
@@ -91,6 +109,7 @@ rule &quot;calcu_ttinfo_special_day&quot; // 特殊日期匹配
         $calcu_ttinfo_enable_result : Calcu_ttinfo_enable_result($xlId : xlId, $calcu_date : calcu_date, $ttInfo_input_list : ttInfo_input_list)
         $calcu_days_result : Calcu_days_result(calcu_date == $calcu_date, xlId == $xlId, $calcu_day : calcu_day)
         $ttinfolist : ArrayList(size >= 1) from collect (TTInfo_input(specialDays contains $calcu_date) from $ttInfo_input_list)
+        $lpInfoResults: List() from accumulate ($ttd: TTInfoDetail() from tTInfoDetailService.list(ttidParams($ttinfolist)), lpinforesult($ttd))
     then
         // 更新Calcu_days_result对象
         int new_calcu_day = $calcu_day + 1;
@@ -105,6 +124,13 @@ rule &quot;calcu_ttinfo_special_day&quot; // 特殊日期匹配
         Collections.sort($ttinfolist);
         results.addXlTTInfos($xlId, $calcu_date, $ttinfolist);
+        // lp输出
+        for (int i = 0; i < $lpInfoResults.size(); i++) {
+            LpInfoResult_output lpInfoResult_output = (LpInfoResult_output) $lpInfoResults.get(i);
+            lpInfoResult_output.setDateTime($calcu_date); // 设定时间
+            lpInfoResults_output.getLpInfoResult_outputs().add(lpInfoResult_output);
+        }
+
         update($calcu_days_result);
 end
@@ -114,6 +140,7 @@ rule &quot;calcu_ttinfo_normal_day&quot; // 平日匹配
         $calcu_ttinfo_enable_result : Calcu_ttinfo_enable_result($xlId : xlId, $calcu_date : calcu_date, $ttInfo_input_list : ttInfo_input_list)
         $calcu_days_result : Calcu_days_result(calcu_date == $calcu_date, xlId == $xlId, $calcu_day : calcu_day, $calcu_weekday : calcu_weekday)
         $ttinfolist : ArrayList(size >= 1) from collect (TTInfo_input(specialDays not contains $calcu_date, weekdays[$calcu_weekday - 1] == true) from $ttInfo_input_list)
+        $lpInfoResults: List() from accumulate ($ttd: TTInfoDetail() from tTInfoDetailService.list(ttidParams($ttinfolist)), lpinforesult($ttd))
     then
         // 更新Calcu_days_result对象
         int new_calcu_day = $calcu_day + 1;
@@ -122,12 +149,19 @@ rule &quot;calcu_ttinfo_normal_day&quot; // 平日匹配
         $calcu_days_result.setCalcu_date(new_calcu_date);
         $calcu_days_result.setCalcu_weekday(new_calcu_date.getDayOfWeek());
-        log.info("启用常规日期时刻表：xlId={} 时刻表个数={} 常规日期={} 星期几={}", $xlId, $ttinfolist.size(), $calcu_date, $calcu_weekday);
+        log.info("启用常规日期时刻表：xlId={} 时刻表个数={} 常规日期={} 星期几={} 路牌size={}", $xlId, $ttinfolist.size(), $calcu_date, $calcu_weekday, $lpInfoResults.size());
         // $ttinfolist按时间倒排序，result输出
         Collections.sort($ttinfolist);
         results.addXlTTInfos($xlId, $calcu_date, $ttinfolist);
+        // lp输出
+        for (int i = 0; i < $lpInfoResults.size(); i++) {
+            LpInfoResult_output lpInfoResult_output = (LpInfoResult_output) $lpInfoResults.get(i);
+            lpInfoResult_output.setDateTime($calcu_date); // 设定时间
+            lpInfoResults_output.getLpInfoResult_outputs().add(lpInfoResult_output);
+        }
+
         update($calcu_days_result);
 end
@@ -137,6 +171,7 @@ rule &quot;calcu_ttinfo_other_day&quot; // 都没有的情况下，匹配
         $calcu_ttinfo_enable_result : Calcu_ttinfo_enable_result($xlId : xlId, $calcu_date : calcu_date, $ttInfo_input_list : ttInfo_input_list)
         $calcu_days_result : Calcu_days_result(calcu_date == $calcu_date, xlId == $xlId, $calcu_day : calcu_day, $calcu_weekday : calcu_weekday)
         $ttinfolist : ArrayList(size >= 1) from collect (TTInfo_input(specialDays not contains $calcu_date, weekdays[$calcu_weekday - 1] == false) from $ttInfo_input_list)
+        $lpInfoResults: List() from accumulate ($ttd: TTInfoDetail() from tTInfoDetailService.list(ttidParams($ttinfolist)), lpinforesult($ttd))
     then
         // 更新Calcu_days_result对象
         int new_calcu_day = $calcu_day + 1;
@@ -151,6 +186,13 @@ rule &quot;calcu_ttinfo_other_day&quot; // 都没有的情况下，匹配
         Collections.sort($ttinfolist);
         results.addXlTTInfos($xlId, $calcu_date, $ttinfolist);
+        // lp输出
+        for (int i = 0; i < $lpInfoResults.size(); i++) {
+            LpInfoResult_output lpInfoResult_output = (LpInfoResult_output) $lpInfoResults.get(i);
+            lpInfoResult_output.setDateTime($calcu_date); // 设定时间
+            lpInfoResults_output.getLpInfoResult_outputs().add(lpInfoResult_output);
+        }
+
         update($calcu_days_result);
 end
 \ No newline at end of file