试验获取设备信息修改

blade-service/blade-business/src/main/java/org/springblade/business/controller/TrialDetectionController.java

@@ -638,7 +638,7 @@ public class TrialDetectionController extends BladeController {
                 List<TrailDeviceUseInfoDTO> trialDeviceUseList = jdbcTemplate.query("select * from u_trial_device_use where self_inspection_record_id = " + id, new BeanPropertyRowMapper<>(TrailDeviceUseInfoDTO.class));
                 if (!trialDeviceUseList.isEmpty()) {
                     Set<Long> ids = trialDeviceUseList.stream().map(TrailDeviceUseInfoDTO::getDeviceInfoId).collect(Collectors.toSet());
-                    List<TrialDeviceInfo> deviceInfoList = jdbcTemplate.query("select * from u_device_info where id in (" + org.apache.commons.lang.StringUtils.join(ids, ",") + ")", new BeanPropertyRowMapper<>(TrialDeviceInfo.class));
+                    List<TrialDeviceInfo> deviceInfoList = jdbcTemplate.query("select * from u_trial_device_info where id in (" + org.apache.commons.lang.StringUtils.join(ids, ",") + ")", new BeanPropertyRowMapper<>(TrialDeviceInfo.class));
                     Map<Long, TrialDeviceInfo> trialDeviceInfoMap = deviceInfoList.stream().collect(Collectors.toMap(TrialDeviceInfo::getId, trailDeviceInfo -> trailDeviceInfo));
                     trialDeviceUseList.forEach(trailDeviceUseInfoDTO -> {
                         TrialDeviceInfo deviceInfo = trialDeviceInfoMap.get(trailDeviceUseInfoDTO.getDeviceInfoId());
@@ -646,11 +646,18 @@ public class TrialDetectionController extends BladeController {
                             trailDeviceUseInfoDTO.setMeasuringRange(deviceInfo.getMeasuringRange());
                             trailDeviceUseInfoDTO.setAccuracy(deviceInfo.getAccuracy());
                             // 计算使用时长
-                            trailDeviceUseInfoDTO.setUseDuration(DateUtils.dateDiff(trailDeviceUseInfoDTO.getStartDate(), trailDeviceUseInfoDTO.getEndDate()));
-                            // 从开始时间中获取日期
-                            trailDeviceUseInfoDTO.setUseDate(DateUtils.getDateStr(trailDeviceUseInfoDTO.getStartDate()));
-                            trailDeviceUseInfoDTO.setUseStartTime(DateUtils.getTimeStr(trailDeviceUseInfoDTO.getStartDate()).substring(11));
-                            trailDeviceUseInfoDTO.setUseEndTime(DateUtils.getTimeStr(trailDeviceUseInfoDTO.getStartDate()).substring(11));
+                            if (trailDeviceUseInfoDTO.getStartDate() != null && trailDeviceUseInfoDTO.getEndDate() != null) {
+                                trailDeviceUseInfoDTO.setUseDuration(DateUtils.dateDiff(trailDeviceUseInfoDTO.getStartDate(), trailDeviceUseInfoDTO.getEndDate()));
+                            }
+                            SimpleDateFormat sdf = new SimpleDateFormat("HH:mm:ss");
+                            if (trailDeviceUseInfoDTO.getStartDate() != null) {
+                                // 从开始时间中获取日期
+                                trailDeviceUseInfoDTO.setUseDate(DateUtils.getDateStr(trailDeviceUseInfoDTO.getStartDate()));
+                                trailDeviceUseInfoDTO.setUseEndTime(sdf.format(trailDeviceUseInfoDTO.getStartDate()));
+                            }
+                            if (trailDeviceUseInfoDTO.getEndDate() != null) {
+                                trailDeviceUseInfoDTO.setUseStartTime(sdf.format(trailDeviceUseInfoDTO.getEndDate()));
+                            }
                         }
                     });
                 }
@@ -770,6 +777,7 @@ public class TrialDetectionController extends BladeController {
                 dto.setDeviceNumber(deviceInfo.getDeviceNumber());
                 dto.setDeviceModel(deviceInfo.getDeviceModel());
                 dto.setRemarks(deviceInfo.getRemarks());
+                dto.setDeviceInfoId(deviceInfo.getId());
             }
             dto.setMeasuringRange(deviceInfo.getMeasuringRange());
             dto.setAccuracy(deviceInfo.getAccuracy());