时刻模板上传修改

zlz
1 parent 8b6a0247
Showing 1 changed file with 20 additions and 2 deletions
src/main/java/com/bsth/service/impl/TrafficManageServiceImpl.java
@@ -8,9 +8,13 @@ import com.bsth.entity.schedule.SchedulePlanInfo;
 import com.bsth.entity.schedule.TTInfo;
 import com.bsth.entity.schedule.TTInfoDetail;
 import com.bsth.entity.search.CustomerSpecs;
+import com.bsth.entity.sys.SysUser;
+import com.bsth.entity.traffic.SKBUploadLogger;
 import com.bsth.repository.*;
 import com.bsth.repository.realcontrol.ScheduleRealInfoRepository;
 import com.bsth.repository.schedule.*;
+import com.bsth.repository.traffic.SKBUploadLoggerRepository;
+import com.bsth.security.util.SecurityUtils;
 import com.bsth.service.TrafficManageService;
 import com.bsth.util.TimeUtils;
 import com.bsth.util.db.DBUtils_MS;
@@ -101,6 +105,10 @@ public class TrafficManageServiceImpl  implements TrafficManageService{
 	@Autowired
 	private ScheduleRealInfoRepository scheduleRealInfoRepository;
+	// 时刻表上传记录repository
+	@Autowired
+	private SKBUploadLoggerRepository skbUploadLoggerRepository;
+
 	// 运管处接口
 	private InternalPortType portType = null;//new Internal().getInternalHttpSoap11Endpoint();
@@ -763,11 +771,12 @@ public class TrafficManageServiceImpl  implements TrafficManageService{
 			String[] idArray = ids.split(",");
 			StringBuffer sBufferA;
 			StringBuffer sBufferB;
+			// 上传的时刻表集合
+			List<TTInfo> ttinfoList = new ArrayList<>();
 			TTInfo ttInfo;
 			TTInfoDetail ttInfoDetail;
 			Iterator<TTInfoDetail> ttInfoDetailIterator;
 			HashMap<String,Object> param = new HashMap<String, Object>();
-			String lineCode ;
 			sBuffer.append("<SKBs>");
 			HashMap<String,String> paramMap;
 			HashMap<String,String> otherMap = new HashMap<String, String>();
@@ -775,6 +784,7 @@ public class TrafficManageServiceImpl  implements TrafficManageService{
 				ttInfo = ttInfoRepository.findOne(Long.valueOf(idArray[i]));
 				if(ttInfo == null)
 					continue;
+				ttinfoList.add(ttInfo); // 保存时刻表
 				param.put("ttinfo.id_eq", ttInfo.getId());
 				ttInfoDetailIterator = ttInfoDetailRepository.findAll(new CustomerSpecs<TTInfoDetail>(param),
 						new Sort(Direction.ASC, "xlDir")).iterator();
@@ -802,7 +812,6 @@ public class TrafficManageServiceImpl  implements TrafficManageService{
 							sBufferA.append("<JHYYLC>").append(ttInfoDetail.getJhlc()).append("</JHYYLC>");
 							sBuffer.append(sBufferA).append(sBufferB);
 						}
-						lineCode = ttInfoDetail.getXl().getLineCode();
 						// 如果发车时间格式错误，忽略此条
 						if(changeTimeFormat(ttInfoDetail) == null){
 							continue;
@@ -831,6 +840,15 @@ public class TrafficManageServiceImpl  implements TrafficManageService{
 			sBuffer.append("</SKBs>");
 			if(ssop.setSKB(userNameOther, passwordOther, sBuffer.toString()).isSuccess()){
 				result = "success";
+				SKBUploadLogger skbUploadLogger ;
+				SysUser user = SecurityUtils.getCurrentUser();
+				// 保存时刻表上传记录
+				for(TTInfo ttInfo1 : ttinfoList){
+					skbUploadLogger = new SKBUploadLogger();
+					skbUploadLogger.setTtInfo(ttInfo1);
+					skbUploadLogger.setUser(user);
+					skbUploadLoggerRepository.save(skbUploadLogger);
+				}
 			}
 		} catch (Exception e) {
 			logger.error("setSKB:", e);