添加按日期上传计划班次的功能

zlz
1 parent b020e9c6
Showing 3 changed files with 31 additions and 4 deletions
src/main/java/com/bsth/controller/schedule/TrafficManageController.java
src/main/java/com/bsth/service/TrafficManageService.java
src/main/java/com/bsth/service/impl/TrafficManageServiceImpl.java
@@ -111,6 +111,15 @@ public class TrafficManageController {
         }
     }
+    @RequestMapping(value = "/setJHBCByDate", method = RequestMethod.GET)
+    public String setJHBCByDate(@RequestParam("theDate") String theDate) throws Exception {
+        try {
+            return trManageService.setJHBC(theDate);
+        } catch (Exception exp) {
+            throw new Exception(exp.getCause());
+        }
+    }
+
     @RequestMapping(value = "/setSKB", method = RequestMethod.GET)
     public String setSKB(@RequestParam("ids") String ids) throws Exception {
         try {
@@ -70,6 +70,8 @@ public interface TrafficManageService {
 	 */
 	String setJHBC();
+	String setJHBC(String theDate);
+
 	String setLD(String theDate);
 	String setLD();
@@ -812,10 +812,26 @@ public class TrafficManageServiceImpl  implements TrafficManageService{
 	}
 	/**
+	 * 上传计划班次 指定日期 yyyy-MM-dd
+	 * @param theDate
+	 * @return
+	 */
+	public String setJHBC(String theDate){
+		return uploadJHBC(theDate);
+	}
+
+	/**
+	 * 上传计划班次
+	 * @return
+	 */
+	public String setJHBC(){
+		return uploadJHBC(null);
+	}
+	/**
 	 * 上传线路计划班次表
 	 */
-	@Override
-	public String setJHBC() {
+
+	private String uploadJHBC(String theDate) {
 		String result = "failure";
 		Line line;
 		StringBuffer sBuffer =new StringBuffer();
@@ -825,8 +841,8 @@ public class TrafficManageServiceImpl  implements TrafficManageService{
 			SchedulePlanInfo schedulePlanInfo;
 			String xlbm,zbh = "";
 			Long lp = 0L;
-			// 取明天的日期
-			String tomorrow = sdfnyr.format(DateUtils.addDays(new Date(), +1));
+			// 取得计划班次时间
+			String tomorrow = theDate == null ? sdfnyr.format(DateUtils.addDays(new Date(), +1)) : theDate;
 			// 查询所有班次
 			List<SchedulePlanInfo> schedulePlanList = schedulePlanInfoRepository.findLineScheduleBc(tomorrow);
 			int j = 0; // 初始化标识