update

徐烜
1 parent eb7d970f
Showing 14 changed files with 613 additions and 58 deletions
src/main/java/com/bsth/entity/schedule/SchedulePlanInfo.java
src/main/java/com/bsth/service/schedule/SchedulePlanServiceImpl.java
src/main/java/com/bsth/service/schedule/rules/MyDroolsConfiguration.java
src/main/java/com/bsth/service/schedule/rules/shiftloop/ScheduleCalcuParam_input.java
src/main/java/com/bsth/service/schedule/rules/strategy/IStrategy.java
src/main/java/com/bsth/service/schedule/rules/strategy/IStrategyImpl.java
src/main/java/com/bsth/service/schedule/rules/ttinfo/TTInfoCalcuParam_input.java
src/main/java/com/bsth/service/schedule/rules/ttinfo/TTInfoResult_output.java
src/main/java/com/bsth/service/schedule/rules/ttinfo/TTInfoResults_output.java
src/main/java/com/bsth/service/schedule/rules/ttinfo/TTInfo_input.java
src/main/java/com/bsth/service/schedule/rules/ttinfo/readme.txt
src/main/resources/rules/shiftloop.drl
src/main/resources/rules/ttinfo.drl
src/test/java/com/bsth/service/schedule/rules/DroolsRulesTest.java
@@ -3,10 +3,10 @@ package com.bsth.entity.schedule;
 import com.bsth.entity.Line;
 import com.bsth.entity.sys.SysUser;
 import com.bsth.service.schedule.rules.shiftloop.ScheduleResult_output;
-import com.fasterxml.jackson.annotation.JsonIgnore;
 import javax.persistence.*;
 import java.util.Date;
+import java.util.List;
 /**
  * 排班计划明细。
@@ -150,7 +150,7 @@ public class SchedulePlanInfo {
             ScheduleResult_output scheduleResult_output,
             TTInfoDetail ttInfoDetail,
             CarConfigInfo carConfigInfo,
-            EmployeeConfigInfo employeeConfigInfo,
+            List<EmployeeConfigInfo> employeeConfigInfoList,
             SchedulePlan schedulePlan) {
         // TODO：关联的公司名称
@@ -180,6 +180,17 @@ public class SchedulePlanInfo {
         // TODO：报道时间，出场时间没有
         // 关联的驾驶员
+        EmployeeConfigInfo employeeConfigInfo = null;
+        if (ttInfoDetail.getIsFB()) {
+            if (employeeConfigInfoList.size() > 1) {
+                employeeConfigInfo = employeeConfigInfoList.get(1);
+            } else {
+                employeeConfigInfo = employeeConfigInfoList.get(0);
+            }
+        } else {
+            employeeConfigInfo = employeeConfigInfoList.get(0);
+        }
+
         this.j = employeeConfigInfo.getJsy().getId();
         this.jGh = employeeConfigInfo.getJsy().getJobCode();
         this.jName = employeeConfigInfo.getJsy().getPersonnelName();
@@ -207,15 +218,13 @@ public class SchedulePlanInfo {
             this.zdz = ttInfoDetail.getTcc().getId(); // 终点站-停车场id
             this.zdzCode = ttInfoDetail.getTcc().getParkCode(); // 终点站-停车场code
             this.zdzName = ttInfoDetail.getTcc().getParkName(); // 终点站-停车场name
-        } else if ("normal".equals(this.bcType)) { // 正常班次
+        } else { // 其他班次
             this.qdz = ttInfoDetail.getQdz().getId(); // 起点站id
             this.qdzCode = ttInfoDetail.getQdz().getStationCod(); // 起点站code
             this.qdzName = ttInfoDetail.getQdz().getStationName(); // 起点站name
             this.zdz = ttInfoDetail.getZdz().getId(); // 终点站id
             this.zdzCode = ttInfoDetail.getZdz().getStationCod(); // 终点站code
             this.zdzName = ttInfoDetail.getZdz().getStationName(); // 终点站name
-        } else {
-            throw new RuntimeException("排班计划数据，未知班次类型：" + this.bcType);
         }
         this.fcsj = ttInfoDetail.getFcsj(); // 发车时间
@@ -47,8 +47,8 @@ public class SchedulePlanServiceImpl extends BaseServiceImpl&lt;SchedulePlan, Long&gt;
         // 1-1、查找线路具体信息
         Line xl = strategy.getLine(schedulePlan.getXl().getId());
         // 1-2、查出指定线路的所有规则
-        TTInfo ttInfo = strategy.getTTInfo(xl.getId()); // 时刻表id
-        schedulePlan.setTtInfo(ttInfo); // 关联的时刻表
+        TTInfo ttInfo = strategy.getTTInfo(xl.getId()).get(0); // 时刻表id
+        schedulePlan.setTtInfo(ttInfo); // TODO：关联的时刻表，之后改掉
         // 2-1、构造drools规则输入数据，输出数据
         // 全局计算参数
@@ -89,7 +89,9 @@ public class SchedulePlanServiceImpl extends BaseServiceImpl&lt;SchedulePlan, Long&gt;
         // 3、根据规则返回，组合最后的输出数据
         // 3-1、根据注入的策略服务，获取原始数据
-        Multimap<Long, TTInfoDetail> gbdTTinfoMaps = strategy.getGuideboardXlTTInfoDetailMaps(xl.getId()); // 路牌对应时刻明细
+        Map<Date, Multimap<Long, TTInfoDetail>> gbdTTinfoMaps = strategy.getGuideboardXlTTInfoDetailMaps(
+                xl.getId(), schedulePlan.getScheduleFromTime(), schedulePlan.getScheduleToTime());
+
         Map<Long, CarConfigInfo> carConfigMaps = strategy.getCarConfigMaps(xl.getId()); // 车辆配置对应车辆信息
         Map<Long, EmployeeConfigInfo> employeeConfigMaps = strategy.getEmployeeConfigMaps(xl.getId()); // 人员配置对应的人员信息
@@ -97,18 +99,23 @@ public class SchedulePlanServiceImpl extends BaseServiceImpl&lt;SchedulePlan, Long&gt;
         List<SchedulePlanInfo> schedulePlanInfos = new ArrayList<>();
         for (ScheduleResult_output scheduleResult_output : scheduleResults_output.getResults()) {
             // 车辆配置对应的车辆
-            CarConfigInfo configInfo = carConfigMaps.get(scheduleResult_output.getCarConfigId());
-            // 人员配置对应的人员
-            EmployeeConfigInfo employeeConfigInfo = employeeConfigMaps.get(scheduleResult_output.getEmployeeConfigId());
+            CarConfigInfo configInfo = carConfigMaps.get(Long.valueOf(scheduleResult_output.getCarConfigId()));
+            // 人员配置对应的人员，这里需要分班处理的
+            List<EmployeeConfigInfo> employeeConfigInfoList = new ArrayList<>();
+            String[] eids = scheduleResult_output.getEmployeeConfigId().split("-");
+            for (String eid : eids) {
+                employeeConfigInfoList.add(employeeConfigMaps.get(Long.valueOf(eid)));
+            }
             // 排班明细（这个要迭代的）
-            Collection<TTInfoDetail> ttInfoDetails = gbdTTinfoMaps.get(Long.parseLong(scheduleResult_output.getGuideboardId()));
+            Collection<TTInfoDetail> ttInfoDetails = gbdTTinfoMaps.get(scheduleResult_output.getSd().toDate()).get(
+                    Long.parseLong(scheduleResult_output.getGuideboardId()));
             for (TTInfoDetail ttInfoDetail : ttInfoDetails) {
                 SchedulePlanInfo schedulePlanInfo = new SchedulePlanInfo(
                         xl,
                         scheduleResult_output,
                         ttInfoDetail,
                         configInfo,
-                        employeeConfigInfo,
+                        employeeConfigInfoList,
                         schedulePlan);
                 schedulePlanInfos.add(schedulePlanInfo);
             }
@@ -58,6 +58,9 @@ public class MyDroolsConfiguration {
         kfs.write("src/main/resources/shiftloop.drl", kieServices.getResources()
                 .newInputStreamResource(this.getClass().getResourceAsStream(
                         "/rules/shiftloop.drl"), "UTF-8"));
+        kfs.write("src/main/resources/ttinfo.drl", kieServices.getResources()
+                .newInputStreamResource(this.getClass().getResourceAsStream(
+                        "/rules/ttinfo.drl"), "UTF-8"));
         // TODO：还有其他drl....
         // 4、创建KieBuilder，使用KieFileSystem构建
@@ -11,15 +11,12 @@ public class ScheduleCalcuParam_input {
     private DateTime fromDate;
     /** 结束计算日期 */
     private DateTime toDate;
-    /** 时刻表id */
-    private Long ttinfoId;
     public ScheduleCalcuParam_input() {}
     public ScheduleCalcuParam_input(SchedulePlan schedulePlan) {
         this.fromDate = new DateTime((schedulePlan.getScheduleFromTime()));
         this.toDate = new DateTime((schedulePlan.getScheduleToTime()));
-        this.ttinfoId = schedulePlan.getTtInfo().getId();
     }
     public DateTime getFromDate() {
@@ -38,11 +35,4 @@ public class ScheduleCalcuParam_input {
         this.toDate = toDate;
     }
-    public Long getTtinfoId() {
-        return ttinfoId;
-    }
-
-    public void setTtinfoId(Long ttinfoId) {
-        this.ttinfoId = ttinfoId;
-    }
 }
@@ -8,6 +8,8 @@ import com.bsth.entity.schedule.TTInfoDetail;
 import com.bsth.entity.schedule.rule.ScheduleRule1Flat;
 import com.google.common.collect.Multimap;
+import java.util.Date;
+import java.util.List;
 import java.util.Map;
 /**
@@ -27,7 +29,14 @@ public interface IStrategy {
      * @param xlId 线路id
      * @return 时刻表
      */
-    TTInfo getTTInfo(Integer xlId);
+    List<TTInfo> getTTInfo(Integer xlId);
+
+    /**
+     * 获取指定线路的时刻表的明细。
+     * @param xlId 线路id
+     * @return
+     */
+    List<TTInfoDetail> getTTInfoDetail(Integer xlId);
     /**
      * 获取指定线路下，可用的排班规则。
@@ -37,11 +46,13 @@ public interface IStrategy {
     Iterable<ScheduleRule1Flat> getScheduleRule(Integer xlId);
     /**
-     * 获取指定线路下，路牌与时刻明细对应的Map。
+     * 获取指定线路下，日期与路牌与时刻明细对应的Map。
      * @param xlId 线路id
+     * @param fromDate 开始日期
+     * @param toDate 结束日期
      * @return 路牌id为key，时刻明细 Collection<TTInfoDetail> 为value
      */
-    Multimap<Long, TTInfoDetail> getGuideboardXlTTInfoDetailMaps(Integer xlId);
+    Map<Date, Multimap<Long, TTInfoDetail>> getGuideboardXlTTInfoDetailMaps(Integer xlId, Date fromDate, Date toDate);
     /**
      * 获取指定线路下，车辆配置与车辆信息对应的Map。
@@ -8,14 +8,21 @@ import com.bsth.entity.schedule.TTInfoDetail;
 import com.bsth.entity.schedule.rule.ScheduleRule1Flat;
 import com.bsth.service.LineService;
 import com.bsth.service.schedule.*;
+import com.bsth.service.schedule.rules.ttinfo.TTInfoCalcuParam_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.TTInfo_input;
 import com.google.common.collect.ArrayListMultimap;
 import com.google.common.collect.Multimap;
+import org.joda.time.DateTime;
+import org.kie.api.KieBase;
+import org.kie.api.runtime.KieSession;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
-import java.util.HashMap;
-import java.util.Iterator;
-import java.util.Map;
+import java.util.*;
 /**
  * Created by xu on 16/7/10.
@@ -35,6 +42,12 @@ public class IStrategyImpl implements IStrategy {
     @Autowired
     private ScheduleRule1FlatService scheduleRule1FlatService;
+    /** 日志记录器 */
+    private Logger logger = LoggerFactory.getLogger(IStrategyImpl.class);
+
+    @Autowired
+    private KieBase kieBase;
+
     @Override
     public Line getLine(Integer xlId) {
         Line xl = lineService.findById(xlId); // 查找线路具体信息
@@ -42,18 +55,39 @@ public class IStrategyImpl implements IStrategy {
     }
     @Override
-    public TTInfo getTTInfo(Integer xlId) {
-        // TODO：本来要使用规则判定到底使用哪张时刻表，这里选用第一张
-        Map<String, Object> param = new HashMap<>(); // 查询参数
-        param.clear();
-        param.put("xl.id_eq", xlId); // 线路id
-        param.put("isCancel_eq", false); // 没有作废
-        param.put("isEnableDisTemplate_eq", true); // 是否启用
-        Iterable<TTInfo> ttInfoIterable = ttInfoService.list(param);
-        Iterator<TTInfo> ttInfoIterator = ttInfoIterable.iterator();
-        if (!ttInfoIterator.hasNext())
-            throw new RuntimeException("线路id=" + xlId + "，下没有任何时刻表数据！");
-        return ttInfoIterator.next();
+    public List<TTInfo> getTTInfo(Integer xlId) {
+        // 查询参数
+        Map<String, Object> param = new HashMap<>();
+        param.put("xl.id_eq", xlId); // 线路Id
+        param.put("isCancel_eq", false); // 作废的过滤掉
+        Iterator<TTInfo> ttInfoIterator = ttInfoService.list(param).iterator();
+        if (!ttInfoIterator.hasNext()) {
+            throw new RuntimeException("线路id=" + xlId + " 没有时刻表！");
+        }
+        List<TTInfo> ttInfos = new ArrayList<>();
+        while (ttInfoIterator.hasNext()) {
+            TTInfo ttInfo = ttInfoIterator.next();
+            ttInfos.add(ttInfo);
+        }
+        return ttInfos;
+    }
+
+    @Override
+    public List<TTInfoDetail> getTTInfoDetail(Integer xlId) {
+        List<TTInfoDetail> ttInfoDetails = new ArrayList<>();
+
+        List<TTInfo> ttInfos = getTTInfo(xlId);
+        Map<String, Object> param = new HashMap<>();
+        for (TTInfo ttInfo : ttInfos) {
+            param.clear();
+            param.put("ttinfo.id_eq", ttInfo.getId());
+            Iterator<TTInfoDetail> ttInfoDetailIterator = ttInfoDetailService.list(param).iterator();
+            while (ttInfoDetailIterator.hasNext()) {
+                ttInfoDetails.add(ttInfoDetailIterator.next());
+            }
+        }
+
+        return ttInfoDetails;
     }
     @Override
@@ -70,23 +104,59 @@ public class IStrategyImpl implements IStrategy {
     }
     @Override
-    public Multimap<Long, TTInfoDetail> getGuideboardXlTTInfoDetailMaps(Integer xlId) {
-        TTInfo ttInfo = getTTInfo(xlId);
-        // 查询参数
+    public Map<Date, Multimap<Long, TTInfoDetail>> getGuideboardXlTTInfoDetailMaps(
+            Integer xlId, Date fromDate, Date toDate) {
+        // 获取线路的所有时刻表
+        List<TTInfo> ttInfos = getTTInfo(xlId);
+
+        // 执行规则，判定每天使用的时刻表
+        KieSession session = kieBase.newKieSession();
+
+        session.setGlobal("log", logger);
+        TTInfoResults_output ttInfoResults_output = new TTInfoResults_output();
+        session.setGlobal("results", ttInfoResults_output);
+
+        TTInfoCalcuParam_input ttInfoCalcuParam_input = new TTInfoCalcuParam_input(
+                new DateTime(fromDate), new DateTime(toDate));
+        session.insert(ttInfoCalcuParam_input);
+        for (TTInfo ttInfo : ttInfos) {
+            TTInfo_input ttInfo_input = new TTInfo_input(ttInfo);
+            session.insert(ttInfo_input);
+        }
+
+        session.fireAllRules();
+        session.dispose();
+
+        // 获取ttinfoDetail
+        List<TTInfoDetail> ttInfoDetails = getTTInfoDetail(xlId);
+
+        // 规则输出结果
+        Multimap<DateTime, TTInfoResult_output> outputMultimap =
+                ttInfoResults_output.getResults().get(String.valueOf(xlId));
+        // return结果输出
+        Map<Date, Multimap<Long, TTInfoDetail>> ttInfoDetailMultimap = new HashMap<>();
+
         Map<String, Object> param = new HashMap<>();
-        param.put("ttinfo.id_eq", ttInfo.getId());
-        Iterable<TTInfoDetail> ttInfoDetailIterable = ttInfoDetailService.list(param);
-        Iterator<TTInfoDetail> ttInfoDetailIterator = ttInfoDetailIterable.iterator();
-        if (!ttInfoDetailIterator.hasNext())
-            throw new RuntimeException("时刻表id=" + ttInfo.getId() + "，下没有明细数据！");
-
-        Multimap<Long, TTInfoDetail> gtmaps = ArrayListMultimap.create();
-        while (ttInfoDetailIterator.hasNext()) {
-            TTInfoDetail ttInfoDetail = ttInfoDetailIterator.next();
-            gtmaps.put(ttInfoDetail.getLp().getId(), ttInfoDetail);
+        for (DateTime dateTime : outputMultimap.keySet()) {
+            Collection<TTInfoResult_output> ttInfoResult_outputs = outputMultimap.get(dateTime);
+            // 如果有多个，使用第一个
+            Iterator<TTInfoResult_output> ttInfoResult_outputIterator = ttInfoResult_outputs.iterator();
+            if (ttInfoResult_outputIterator.hasNext()) {
+                // 同一天，多张时刻表只取第一张
+                TTInfoResult_output ttInfoResult_output = ttInfoResult_outputIterator.next();
+                // 查找时刻表明细
+                Multimap<Long, TTInfoDetail> ttinfodetailMap2 = ArrayListMultimap.create();
+                for (TTInfoDetail ttInfoDetail : ttInfoDetails) {
+                    if (ttInfoDetail.getTtinfo().getId() == Long.valueOf(ttInfoResult_output.getTtInfoId())) {
+                        ttinfodetailMap2.put(ttInfoDetail.getLp().getId(), ttInfoDetail);
+                    }
+                }
+
+                ttInfoDetailMultimap.put(dateTime.toDate(), ttinfodetailMap2);
+            }
         }
-        return gtmaps;
+        return ttInfoDetailMultimap;
     }
     @Override
+package com.bsth.service.schedule.rules.ttinfo;
+
+import org.joda.time.DateTime;
+
+/**
+ * 时刻表计算参数_输入。
+ */
+public class TTInfoCalcuParam_input {
+    /** 开始计算日期 */
+    private DateTime fromDate;
+    /** 结束计算日期 */
+    private DateTime toDate;
+
+    public TTInfoCalcuParam_input() {}
+
+    public TTInfoCalcuParam_input(DateTime fromDate, DateTime toDate) {
+        this.fromDate = fromDate;
+        this.toDate = toDate;
+    }
+
+    public DateTime getFromDate() {
+        return fromDate;
+    }
+
+    public void setFromDate(DateTime fromDate) {
+        this.fromDate = fromDate;
+    }
+
+    public DateTime getToDate() {
+        return toDate;
+    }
+
+    public void setToDate(DateTime toDate) {
+        this.toDate = toDate;
+    }
+}
+package com.bsth.service.schedule.rules.ttinfo;
+
+import org.joda.time.DateTime;
+
+/**
+ * 时刻表选择规则出的结果_输出。
+ */
+public class TTInfoResult_output {
+    /** 具体日期 */
+    private DateTime dateTime;
+    /** 时刻表Id */
+    private String ttInfoId;
+    /** 线路Id */
+    private String xlId;
+
+    public DateTime getDateTime() {
+        return dateTime;
+    }
+
+    public void setDateTime(DateTime dateTime) {
+        this.dateTime = dateTime;
+    }
+
+    public String getTtInfoId() {
+        return ttInfoId;
+    }
+
+    public void setTtInfoId(String ttInfoId) {
+        this.ttInfoId = ttInfoId;
+    }
+
+    public String getXlId() {
+        return xlId;
+    }
+
+    public void setXlId(String xlId) {
+        this.xlId = xlId;
+    }
+
+    @Override
+    public String toString() {
+        return String.format(
+                "<日期=%s 线路id=%s 时刻表id=%s>\n",
+                dateTime.toString("yyyy-MM-dd"),
+                xlId,
+                ttInfoId);
+    }
+}
+package com.bsth.service.schedule.rules.ttinfo;
+
+import com.google.common.collect.ArrayListMultimap;
+import com.google.common.collect.Multimap;
+import org.joda.time.DateTime;
+
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * 输出结果集合。
+ */
+public class TTInfoResults_output {
+
+    /** 输出列表，key为线路id，value是key为日期，value为排序的时刻表output列表 */
+    private Map<String, Multimap<DateTime, TTInfoResult_output>> results = new HashMap<>();
+
+    public Map<String, Multimap<DateTime, TTInfoResult_output>> getResults() {
+        return results;
+    }
+
+    public void setResults(Map<String, Multimap<DateTime, TTInfoResult_output>> results) {
+        this.results = results;
+    }
+
+    public void addXlTTInfos(String xlid, DateTime dt, List<TTInfo_input> ttInfo_inputList) {
+        Multimap<DateTime, TTInfoResult_output> map;
+        if (results.get(xlid) == null) {
+            map = ArrayListMultimap.create();
+            results.put(xlid, map);
+        } else {
+            map = results.get(xlid);
+        }
+
+        for (TTInfo_input ttInfo_input : ttInfo_inputList) {
+            TTInfoResult_output ttInfoResult_output = new TTInfoResult_output();
+            ttInfoResult_output.setDateTime(dt);
+            ttInfoResult_output.setTtInfoId(ttInfo_input.getTtInfoId());
+            ttInfoResult_output.setXlId(xlid);
+            map.put(dt, ttInfoResult_output);
+        }
+    }
+
+    /**
+     * 输出计算后的时刻表
+     * @return
+     */
+    public String showTTInfoDesc1() {
+        StringBuilder str = new StringBuilder();
+        for (String key : results.keySet()) {
+            str.append("线路id=" + key);
+            str.append("\n");
+            str.append("时刻表=" + results.get(key));
+            str.append("\n");
+        }
+
+        return str.toString();
+    }
+
+}
+package com.bsth.service.schedule.rules.ttinfo;
+
+import com.bsth.entity.schedule.TTInfo;
+import org.joda.time.DateTime;
+import org.joda.time.format.DateTimeFormat;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * 时刻表_输入
+ */
+public class TTInfo_input {
+    /** 时刻表id */
+    private String ttInfoId;
+    /** 线路Id */
+    private String xlId;
+    /** 周一到周日是否启用 */
+    private List<Boolean> weekdays = new ArrayList<>();
+    /** 特殊节假日 */
+    private List<DateTime> specialDays = new ArrayList<>();
+    /** 最新修改时间 */
+    private DateTime updateDate;
+    /** 是否启用 */
+    private Boolean isEnable;
+    /** 启用日期 */
+    private DateTime qyDate;
+
+    public TTInfo_input() {
+
+    }
+
+    public TTInfo_input(TTInfo ttInfo) {
+        this.ttInfoId = String.valueOf(ttInfo.getId());
+        this.xlId = String.valueOf(ttInfo.getXl().getId());
+        String[] days = ttInfo.getRule_days().split(",");
+        System.out.println(days.length);
+        for (int i = 0; i < 7; i++) {
+            if ("1".equals(days[i])) {
+                weekdays.add(true);
+            } else {
+                weekdays.add(false);
+            }
+        }
+        String[] sdays = ttInfo.getSpecial_days().split(",");
+        for (int i = 0; i < sdays.length; i++) {
+            specialDays.add(DateTimeFormat.forPattern("yyyy-MM-dd").
+                    parseDateTime(sdays[i]));
+        }
+        this.updateDate = new DateTime(ttInfo.getUpdateDate());
+        this.isEnable = ttInfo.getIsEnableDisTemplate();
+        this.qyDate = new DateTime(ttInfo.getQyrq());
+
+    }
+
+    public String getTtInfoId() {
+        return ttInfoId;
+    }
+
+    public void setTtInfoId(String ttInfoId) {
+        this.ttInfoId = ttInfoId;
+    }
+
+    public String getXlId() {
+        return xlId;
+    }
+
+    public void setXlId(String xlId) {
+        this.xlId = xlId;
+    }
+
+    public List<Boolean> getWeekdays() {
+        return weekdays;
+    }
+
+    public void setWeekdays(List<Boolean> weekdays) {
+        this.weekdays = weekdays;
+    }
+
+    public List<DateTime> getSpecialDays() {
+        return specialDays;
+    }
+
+    public void setSpecialDays(List<DateTime> specialDays) {
+        this.specialDays = specialDays;
+    }
+
+    public DateTime getUpdateDate() {
+        return updateDate;
+    }
+
+    public void setUpdateDate(DateTime updateDate) {
+        this.updateDate = updateDate;
+    }
+
+    public Boolean getIsEnable() {
+        return isEnable;
+    }
+
+    public void setIsEnable(Boolean isEnable) {
+        this.isEnable = isEnable;
+    }
+
+    public DateTime getQyDate() {
+        return qyDate;
+    }
+
+    public void setQyDate(DateTime qyDate) {
+        this.qyDate = qyDate;
+    }
+}
+时刻表选择规则，每天的时刻表都不一样
 \ No newline at end of file
-package com.bsth.service.schedule;
+package com.bsth.service.schedule.shiftloop;
 import org.joda.time.*;
 import java.util.*;
+package com.bsth.service.schedule.ttinfo;
+
+import org.joda.time.*;
+import java.util.*;
+
+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 org.slf4j.Logger;
+
+// 全局日志
+global Logger log;
+// return输出
+global TTInfoResults_output results
+
+
+/*
+    TODO：规则说明，以后待说明
+*/
+
+//----------------- 第一阶段、计算规则准备数据（天数）----------------//
+
+declare Calcu_days_result
+    calcu_day : Integer // 该计算第几天
+    calcu_weekday : Integer // 星期几（1到7）
+    calcu_date : DateTime // 该计算的具体日期
+    calcu_days : Integer // 总共需要计算的天数
+    calcu_start_date : DateTime // 开始计算日期
+    calcu_end_date : DateTime // 结束计算日期
+end
+
+rule "calcu_days"
+    when
+        TTInfoCalcuParam_input($fromDate : fromDate, $toDate : toDate, $fromDate.isBefore($toDate))
+    then
+        // 构造Calcu_days_result对象，进行下一阶段计算
+        Calcu_days_result cdr = new Calcu_days_result();
+        Period p = new Period($fromDate, $toDate, PeriodType.days());
+
+        cdr.setCalcu_day(1);
+        cdr.setCalcu_date($fromDate);
+        cdr.setCalcu_days(p.getDays() + 1);
+        cdr.setCalcu_weekday($fromDate.getDayOfWeek());
+        cdr.setCalcu_start_date($fromDate);
+        cdr.setCalcu_end_date(($toDate));
+
+        log.info("总共塑腰计算的天数 calcu_days={} 之后的计算从第1天开始 ", p.getDays() + 1);
+
+        insert(cdr); // 插入fact数据，进入下一个阶段
+end
+
+//----------------- 第二阶段、判定时刻表是否启用 ----------------//
+
+declare Calcu_ttinfo_enable_result
+    xlid : String // 线路id
+    ttid : String // 时刻表id
+    calcu_date : DateTime // 计算日期
+end
+
+rule "calcu_ttinfo_enable"
+    salience 900
+    when
+        $calcu_days_result : Calcu_days_result($calcu_date : calcu_date, calcu_day <= calcu_days)
+        $tTInfo_input : TTInfo_input($xlid : xlId, $ttid : ttInfoId, isEnable == true)
+    then
+        // 构造Calcu_ttinfo_enable_result对象，进行下一步计算
+        Calcu_ttinfo_enable_result cter = new Calcu_ttinfo_enable_result();
+        cter.setXlid($xlid);
+        cter.setTtid($ttid);
+        cter.setCalcu_date($calcu_date);
+
+        log.info("启用的时刻表：xlid={} ttid={} 计算日期={}", $xlid, $ttid, $calcu_date);
+
+        insert(cter);
+end
+
+//----------------- 第三阶段 -------------------//
+
+rule "calcu_ttinfo_special_day"
+    salience 800
+    when
+        $calcu_ttinfo_enable_result : Calcu_ttinfo_enable_result($xlid : xlid, $calcu_date : calcu_date)
+        $calcu_days_result : Calcu_days_result(calcu_date == $calcu_date, $calcu_day : calcu_day)
+        $ttinfolist : ArrayList(size >= 1) from collect (TTInfo_input(xlId == $xlid, isEnable == true, specialDays contains $calcu_date))
+    then
+        // 更新Calcu_days_result对象
+        int new_calcu_day = $calcu_day + 1;
+        $calcu_days_result.setCalcu_day(new_calcu_day);
+        DateTime new_calcu_date = $calcu_date.plusDays(1);
+        $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);
+
+        // result输出
+        results.addXlTTInfos($xlid, $calcu_date, $ttinfolist);
+
+        update($calcu_days_result);
+end
+
+rule "calcu_ttinfo_normal_day"
+    salience 700
+    when
+        $calcu_ttinfo_enable_result : Calcu_ttinfo_enable_result($xlid : xlid, $calcu_date : calcu_date)
+        $calcu_days_result : Calcu_days_result(calcu_date == $calcu_date, $calcu_day : calcu_day, $calcu_weekday : calcu_weekday)
+        $ttinfolist : ArrayList(size >= 1) from collect (TTInfo_input(xlId == $xlid, isEnable == true, specialDays not contains $calcu_date, weekdays[$calcu_weekday - 1] == true))
+    then
+        // 更新Calcu_days_result对象
+        int new_calcu_day = $calcu_day + 1;
+        $calcu_days_result.setCalcu_day(new_calcu_day);
+        DateTime new_calcu_date = $calcu_date.plusDays(1);
+        $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);
+
+        // result输出
+        results.addXlTTInfos($xlid, $calcu_date, $ttinfolist);
+
+        update($calcu_days_result);
+end
+
+rule "calcu_ttinfo_other_day"
+    salience 500
+    when
+        $calcu_ttinfo_enable_result : Calcu_ttinfo_enable_result($xlid : xlid, $calcu_date : calcu_date)
+        $calcu_days_result : Calcu_days_result(calcu_date == $calcu_date, $calcu_day : calcu_day, $calcu_weekday : calcu_weekday)
+        $ttinfolist : ArrayList(size >= 1) from collect (TTInfo_input(xlId == $xlid, isEnable == true, specialDays not contains $calcu_date, weekdays[$calcu_weekday - 1] == false))
+    then
+        // 更新Calcu_days_result对象
+        int new_calcu_day = $calcu_day + 1;
+        $calcu_days_result.setCalcu_day(new_calcu_day);
+        DateTime new_calcu_date = $calcu_date.plusDays(1);
+        $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);
+
+        // result输出
+        results.addXlTTInfos($xlid, $calcu_date, $ttinfolist);
+
+        update($calcu_days_result);
+
+end
 \ No newline at end of file
@@ -4,11 +4,16 @@ import com.bsth.Application;
 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.ttinfo.TTInfoCalcuParam_input;
+import com.bsth.service.schedule.rules.ttinfo.TTInfoResults_output;
+import com.bsth.service.schedule.rules.ttinfo.TTInfo_input;
 import org.joda.time.DateTime;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.kie.api.KieBase;
 import org.kie.api.runtime.KieSession;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.test.SpringApplicationConfiguration;
 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
@@ -21,10 +26,13 @@ import java.util.List;
 @SpringApplicationConfiguration(classes = {Application.class})
 public class DroolsRulesTest {
+    /** 日志记录器 */
+    private final static Logger logger = LoggerFactory.getLogger(DroolsRulesTest.class);
+
     @Autowired
     private KieBase kieBase;
-    @Test
+//    @Test
     public void helloWorldDrlTest() throws Exception {
         // 1、创建session，内部配置的是stateful
         KieSession session = kieBase.newKieSession();
@@ -51,6 +59,61 @@ public class DroolsRulesTest {
     }
     @Test
+    public void ttinfoDrlTest() throws Exception {
+        logger.info("------------ttinfoDrlTest 测试---------------");
+
+        // 1、创建session，内部配置的是stateful
+        KieSession session = kieBase.newKieSession();
+
+        // 1.1 设置gloable对象，在drl中通过别人使用
+        session.setGlobal("log", logger);
+        TTInfoResults_output ttInfoResults_output = new TTInfoResults_output();
+        session.setGlobal("results", ttInfoResults_output);
+
+        // 1.2 可以设置一些监听器，再议
+
+        // 2、创建fact对象
+        TTInfoCalcuParam_input ttInfoCalcuParam_input = new TTInfoCalcuParam_input(
+                new DateTime(2016, 8, 1, 0, 0),
+                new DateTime(2016, 8, 10, 0, 0)
+        );
+
+        TTInfo_input ttInfo_input1 = new TTInfo_input();
+        ttInfo_input1.setTtInfoId("1");
+        ttInfo_input1.setXlId("1");
+        ttInfo_input1.setWeekdays(Arrays.asList(true, true, true, true, true, false, false));
+        ttInfo_input1.getSpecialDays().add(new DateTime(2016, 8, 1, 0, 0));
+        ttInfo_input1.setUpdateDate(new DateTime(2016, 1, 1, 0, 0));
+        ttInfo_input1.setIsEnable(true);
+        ttInfo_input1.setQyDate(new DateTime(2016, 1, 1, 0, 0));
+
+        TTInfo_input ttInfo_input2 = new TTInfo_input();
+        ttInfo_input2.setTtInfoId("2");
+        ttInfo_input2.setXlId("1");
+        ttInfo_input2.setWeekdays(Arrays.asList(true, false, false, false, false, true, false));
+        ttInfo_input2.getSpecialDays().add(new DateTime(2016, 8, 11, 0, 0));
+        ttInfo_input2.setUpdateDate(new DateTime(2016, 1, 1, 0, 0));
+        ttInfo_input2.setIsEnable(true);
+        ttInfo_input2.setQyDate(new DateTime(2016, 1, 1, 0, 0));
+
+        session.insert(ttInfoCalcuParam_input);
+        session.insert(ttInfo_input1);
+        session.insert(ttInfo_input2);
+
+
+
+        // 3、执行rule
+        session.fireAllRules();
+
+        // 4、执行完毕销毁，有日志的也要关闭
+        session.dispose();
+
+        // 打印global结果
+        logger.info(ttInfoResults_output.showTTInfoDesc1());
+
+    }
+
+//    @Test
     public void shiftloopDrlTest() throws Exception {
         // 1、创建session，内部配置的是stateful
         KieSession session = kieBase.newKieSession();
@@ -66,7 +129,6 @@ public class DroolsRulesTest {
         ScheduleCalcuParam_input scheduleCalcuParam_input = new ScheduleCalcuParam_input();
         scheduleCalcuParam_input.setFromDate(new DateTime(2016, 8, 1, 0, 0));
         scheduleCalcuParam_input.setToDate(new DateTime(2016, 8, 10, 0, 0));
-        scheduleCalcuParam_input.setTtinfoId(1L);
         ScheduleRule_input scheduleRule_input1 = new ScheduleRule_input();
         scheduleRule_input1.setRuleId("1");