From b5f8965016c597b7bc9811f83efe32d83d4a2627 Mon Sep 17 00:00:00 2001
From: yuan <123@>
Date: 星期一, 25 五月 2026 01:23:44 +0800
Subject: [PATCH] 1
---
/dev/null | 339 --------------------------------------------------------
1 files changed, 0 insertions(+), 339 deletions(-)
diff --git a/src/test/java/com/ruoyi/device/service/impl/MaintenanceTaskJobTest.java b/src/test/java/com/ruoyi/device/service/impl/MaintenanceTaskJobTest.java
deleted file mode 100644
index 89dcb16..0000000
--- a/src/test/java/com/ruoyi/device/service/impl/MaintenanceTaskJobTest.java
+++ /dev/null
@@ -1,449 +0,0 @@
-package com.ruoyi.device.service.impl;
-
-import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
-import com.ruoyi.device.mapper.MaintenanceTaskMapper;
-import com.ruoyi.device.pojo.DeviceMaintenance;
-import com.ruoyi.device.pojo.MaintenanceTask;
-import org.junit.jupiter.api.Test;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.boot.test.context.SpringBootTest;
-import org.springframework.jdbc.core.JdbcTemplate;
-
-import java.time.LocalDateTime;
-import java.time.LocalTime;
-import java.time.YearMonth;
-import java.time.LocalDate;
-import java.time.DayOfWeek;
-import java.util.ArrayList;
-import java.util.List;
-import java.util.Objects;
-
-/**
- * 璁惧淇濆吇浠诲姟娴嬭瘯绫�
- *
- * 鏌ヨ鏁版嵁搴撲腑鐨勪繚鍏讳换鍔★紝鏍规嵁鐧昏鏃ユ湡銆侀娆°�佸紑濮嬫棩鏈熶笌鏃堕棿鏉ョ敓鎴愪繚鍏昏褰�
- * 浠庣櫥璁版棩鏈熷埌浠婂ぉ锛屾寜棰戞鐢熸垚鎵�鏈夌鍚堢殑璁板綍
- */
-@SpringBootTest
-public class MaintenanceTaskJobTest {
-
- @Autowired
- private MaintenanceTaskMapper maintenanceTaskMapper;
-
- @Autowired
- private DeviceMaintenanceServiceImpl deviceMaintenanceService;
-
- @Autowired
- private JdbcTemplate jdbcTemplate;
-
- /**
- * 娴嬭瘯锛氭寜鏃ユ湡鍒嗘暎鐢熸垚淇濆吇璁板綍
- * 浠庣櫥璁版棩鏈熷埌浠婂ぉ锛屾寜鏃ユ湡椤哄簭锛屾瘡澶╃殑鎵�鏈夎澶囦繚鍏昏褰曚竴璧风敓鎴�
- */
- @Test
- void testGenerateOneByOne() {
- // 鏌ヨ鎵�鏈夊惎鐢ㄧ殑淇濆吇浠诲姟
- List<MaintenanceTask> tasks = maintenanceTaskMapper.selectList(
- new LambdaQueryWrapper<MaintenanceTask>()
- .eq(MaintenanceTask::getIsActive, 1)
- .eq(MaintenanceTask::getDeleted, 0)
- );
-
- if (tasks.isEmpty()) {
- System.out.println("=== 娌℃湁鎵惧埌淇濆吇浠诲姟 ===");
- return;
- }
-
- // 鎵惧嚭鏈�鏃╁拰鏈�鏅氱殑鐧昏鏃ユ湡
- LocalDate earliestDate = tasks.stream()
- .map(MaintenanceTask::getRegistrationDate)
- .filter(Objects::nonNull)
- .min(LocalDate::compareTo)
- .orElse(LocalDate.now());
- LocalDate latestDate = LocalDate.now();
-
- System.out.println("=== 鐧昏鏃ユ湡鑼冨洿: " + earliestDate + " 鑷� " + latestDate + " ===");
- System.out.println("=== 鍏辨壘鍒� " + tasks.size() + " 涓繚鍏讳换鍔� ===\n");
-
- // 鎸夋棩鏈熼『搴忕敓鎴愶紝姣忓ぉ鐢熸垚涓�娆�
- LocalDate currentDate = earliestDate;
- int totalRecords = 0;
-
- while (!currentDate.isAfter(latestDate)) {
- final LocalDate date = currentDate;
- final List<DeviceMaintenance> recordsToSave = new ArrayList<>();
-
- // 鎵惧嚭褰撳ぉ闇�瑕佷繚鍏荤殑鎵�鏈変换鍔�
- for (MaintenanceTask task : tasks) {
- LocalDate startDate = task.getRegistrationDate() != null ? task.getRegistrationDate() : LocalDate.now();
- if (date.isBefore(startDate)) {
- continue; // 璺宠繃鐧昏鏃ユ湡涔嬪墠鐨勬棩鏈�
- }
-
- // 妫�鏌ヨ浠诲姟鐨勯娆℃槸鍚﹀尮閰嶅綋澶�
- if (isExecutionDate(task.getFrequencyType(), task.getFrequencyDetail(), startDate, date)) {
- LocalTime time = getExecutionTime(task.getFrequencyDetail());
- LocalDateTime executionDateTime = LocalDateTime.of(date, time);
-
- DeviceMaintenance record = createMaintenanceRecord(task, executionDateTime);
- recordsToSave.add(record);
- }
- }
-
- // 濡傛灉褰撳ぉ鏈夎褰曪紝鎵归噺淇濆瓨
- if (!recordsToSave.isEmpty()) {
- for (DeviceMaintenance record : recordsToSave) {
- try {
- deviceMaintenanceService.save(record);
- System.out.println(" 鈫� 璁″垝鏃ユ湡: " + date + " | 璁惧: " + record.getDeviceName()
- + " | 璁板綍ID: " + record.getId());
- } catch (Exception e) {
- System.out.println(" 鉁� 璁″垝鏃ユ湡: " + date + " | 璁惧: " + record.getDeviceName()
- + " | 澶辫触: " + e.getMessage());
- }
- }
- totalRecords += recordsToSave.size();
- System.out.println(" === 鏃ユ湡 " + date + " 鍏辩敓鎴� " + recordsToSave.size() + " 鏉¤褰� ===\n");
- }
-
- currentDate = currentDate.plusDays(1);
- }
-
- System.out.println("=== 鎵ц瀹屾垚: 鍏辩敓鎴� " + totalRecords + " 鏉¤褰� ===");
- }
-
- /**
- * 妫�鏌ユ寚瀹氭棩鏈熸槸鍚︾鍚堜换鍔＄殑鎵ц棰戞
- */
- private boolean isExecutionDate(String frequencyType, String frequencyDetail, LocalDate startDate, LocalDate checkDate) {
- if (checkDate.isBefore(startDate)) {
- return false;
- }
-
- switch (frequencyType) {
- case "DAILY":
- return true; // 姣忓ぉ閮介渶瑕佹墽琛�
- case "WEEKLY":
- return isWeeklyMatch(frequencyDetail, checkDate);
- case "MONTHLY":
- return isMonthlyMatch(frequencyDetail, checkDate);
- case "QUARTERLY":
- return isQuarterlyMatch(frequencyDetail, checkDate);
- default:
- return false;
- }
- }
-
- private boolean isWeeklyMatch(String detail, LocalDate date) {
- String[] parts = detail.split(",");
- String dayOfWeekStr = parts[0];
- DayOfWeek targetDay = parseDayOfWeek(dayOfWeekStr);
- return date.getDayOfWeek() == targetDay;
- }
-
- private boolean isMonthlyMatch(String detail, LocalDate date) {
- String[] parts = detail.split(",");
- int targetDay = Integer.parseInt(parts[0]);
- return date.getDayOfMonth() == targetDay;
- }
-
- private boolean isQuarterlyMatch(String detail, LocalDate date) {
- String[] parts = detail.split(",");
- int quarterMonth = Integer.parseInt(parts[0]); // 瀛ｅ害涓殑绗嚑涓湀
- int targetDay = Integer.parseInt(parts[1]);
-
- int currentMonth = date.getMonthValue();
- int monthInQuarter = (currentMonth - 1) % 3 + 1;
-
- return monthInQuarter == quarterMonth && date.getDayOfMonth() == targetDay;
- }
-
- private LocalTime getExecutionTime(String frequencyDetail) {
- String[] parts = frequencyDetail.split(",");
- if (parts.length >= 2 && (parts.length == 2 || parts.length == 3)) {
- return LocalTime.parse(parts[parts.length - 1]);
- }
- return LocalTime.of(9, 0); // 榛樿鏃堕棿
- }
-
- /**
- * 娴嬭瘯锛氭牴鎹墍鏈変繚鍏讳换鍔＄敓鎴愯澶囦繚鍏昏褰�
- * 鎸夐『搴忓鐞嗘瘡涓换鍔★紝浠庣櫥璁版棩鏈熷埌浠婂ぉ锛屾寜棰戞渚濇鐢熸垚鎵�鏈夌鍚堢殑璁板綍
- */
- @Test
- void testGenerateMaintenanceRecordForAllTasks() {
- // 鏌ヨ鎵�鏈夊惎鐢ㄧ殑淇濆吇浠诲姟锛屾寜ID鎺掑簭
- List<MaintenanceTask> tasks = maintenanceTaskMapper.selectList(
- new LambdaQueryWrapper<MaintenanceTask>()
- .eq(MaintenanceTask::getIsActive, 1)
- .eq(MaintenanceTask::getDeleted, 0)
- .orderByAsc(MaintenanceTask::getId)
- );
-
- System.out.println("=== 鍏辨壘鍒� " + tasks.size() + " 涓繚鍏讳换鍔� ===\n");
-
- int totalRecords = 0;
- for (MaintenanceTask task : tasks) {
- try {
- // 鎸夐『搴忕敓鎴愯浠诲姟鎵�鏈夌鍚堥娆＄殑璁板綍
- int recordCount = generateRecordsForTaskSequentially(task);
- totalRecords += recordCount;
- System.out.println("鉁� 浠诲姟ID: " + task.getId() + " | " + task.getTaskName()
- + " | 鐧昏鏃ユ湡: " + task.getRegistrationDate()
- + " | 鐢熸垚璁板綍鏁�: " + recordCount);
- } catch (Exception e) {
- System.out.println("鉁� 浠诲姟ID: " + task.getId() + " | " + task.getTaskName() + " | 澶辫触: " + e.getMessage());
- }
- }
-
- System.out.println("\n=== 鎵ц瀹屾垚: 鍏辩敓鎴� " + totalRecords + " 鏉¤褰� ===");
- }
-
- /**
- * 涓哄崟涓换鍔℃寜椤哄簭鐢熸垚鎵�鏈夌鍚堥娆＄殑璁板綍
- */
- private int generateRecordsForTaskSequentially(MaintenanceTask task) {
- LocalDate startDate = task.getRegistrationDate();
- if (startDate == null) {
- startDate = LocalDate.now();
- }
- LocalDate endDate = LocalDate.now();
-
- // 鏍规嵁棰戞鑾峰彇鎵�鏈夐渶瑕佹墽琛岀殑鏃ユ湡
- List<LocalDateTime> executionDates = getExecutionDates(task.getFrequencyType(), task.getFrequencyDetail(), startDate, endDate);
-
- if (executionDates.isEmpty()) {
- return 0;
- }
-
- int count = 0;
- for (LocalDateTime executionDate : executionDates) {
- try {
- // 鎸夐『搴忕敓鎴愪繚鍏昏褰�
- DeviceMaintenance record = createMaintenanceRecord(task, executionDate);
- deviceMaintenanceService.save(record);
- count++;
- } catch (Exception e) {
- System.out.println(" 鉁� 鏃ユ湡: " + executionDate + " | 澶辫触: " + e.getMessage());
- }
- }
-
- // 鏇存柊浠诲姟鐨勪笅娆℃墽琛屾椂闂�
- if (count > 0) {
- // first_execution = 绗竴鏉¤褰曠殑鏃ユ湡 = last_execution_time
- LocalDateTime firstExecution = executionDates.get(0);
- // next_execution = 鏈�鍚庝竴鏉¤褰曠殑涓嬩竴娆℃墽琛屾棩鏈�
- LocalDateTime lastExecution = executionDates.get(executionDates.size() - 1);
- LocalDateTime nextExecution = calculateNextExecutionTime(task.getFrequencyType(), task.getFrequencyDetail(), lastExecution);
- updateTaskExecutionTime(task.getId(), firstExecution, nextExecution);
- }
-
- return count;
- }
-
- /**
- * 鑾峰彇鎸囧畾鏃ユ湡鑼冨洿鍐呮墍鏈夌鍚堥娆＄殑鎵ц鏃堕棿
- */
- private List<LocalDateTime> getExecutionDates(String frequencyType, String frequencyDetail, LocalDate startDate, LocalDate endDate) {
- List<LocalDateTime> dates = new ArrayList<>();
-
- switch (frequencyType) {
- case "DAILY":
- dates.addAll(getDailyDates(startDate, endDate, frequencyDetail));
- break;
- case "WEEKLY":
- dates.addAll(getWeeklyDates(startDate, endDate, frequencyDetail));
- break;
- case "MONTHLY":
- dates.addAll(getMonthlyDates(startDate, endDate, frequencyDetail));
- break;
- case "QUARTERLY":
- dates.addAll(getQuarterlyDates(startDate, endDate, frequencyDetail));
- break;
- }
-
- return dates;
- }
-
- private List<LocalDateTime> getDailyDates(LocalDate startDate, LocalDate endDate, String timeStr) {
- List<LocalDateTime> dates = new ArrayList<>();
- LocalTime time = LocalTime.parse(timeStr);
-
- LocalDate current = startDate;
- while (!current.isAfter(endDate)) {
- dates.add(LocalDateTime.of(current, time));
- current = current.plusDays(1);
- }
- return dates;
- }
-
- private List<LocalDateTime> getMonthlyDates(LocalDate startDate, LocalDate endDate, String detail) {
- List<LocalDateTime> dates = new ArrayList<>();
- String[] parts = detail.split(",");
- int dayOfMonth = Integer.parseInt(parts[0]);
- LocalTime time = LocalTime.parse(parts[1]);
-
- // 浠庣櫥璁版棩鏈熸墍鍦ㄦ湀鎵剧涓�涓鍚堟潯浠剁殑鏃ユ湡
- int actualDay = Math.min(dayOfMonth, startDate.lengthOfMonth());
- LocalDate current = startDate.withDayOfMonth(actualDay);
-
- // 濡傛灉杩欎釜鏃ユ湡鍦ㄧ櫥璁版棩鏈熶箣鍓嶏紝寰�鍚庢帹涓�涓湀
- if (current.isBefore(startDate)) {
- current = current.plusMonths(1);
- actualDay = Math.min(dayOfMonth, current.lengthOfMonth());
- current = current.withDayOfMonth(actualDay);
- }
-
- while (!current.isAfter(endDate)) {
- dates.add(LocalDateTime.of(current, time));
- current = current.plusMonths(1);
- actualDay = Math.min(dayOfMonth, current.lengthOfMonth());
- current = current.withDayOfMonth(actualDay);
- }
- return dates;
- }
-
- private List<LocalDateTime> getWeeklyDates(LocalDate startDate, LocalDate endDate, String detail) {
- List<LocalDateTime> dates = new ArrayList<>();
- String[] parts = detail.split(",");
- String dayOfWeekStr = parts[0];
- LocalTime time = LocalTime.parse(parts[1]);
-
- java.time.DayOfWeek targetDay = parseDayOfWeek(dayOfWeekStr);
-
- LocalDate current = startDate;
- while (!current.isAfter(endDate)) {
- if (current.getDayOfWeek() == targetDay) {
- dates.add(LocalDateTime.of(current, time));
- }
- current = current.plusDays(1);
- }
- return dates;
- }
-
- private List<LocalDateTime> getQuarterlyDates(LocalDate startDate, LocalDate endDate, String detail) {
- List<LocalDateTime> dates = new ArrayList<>();
- String[] parts = detail.split(",");
- int quarterMonth = Integer.parseInt(parts[0]);
- int dayOfMonth = Integer.parseInt(parts[1]);
- LocalTime time = LocalTime.parse(parts[2]);
-
- int currentMonth = startDate.getMonthValue();
- int targetMonth = ((currentMonth - 1) / 3) * 3 + quarterMonth;
- int yearAdjust = 0;
-
- if (targetMonth > 12) {
- targetMonth -= 12;
- yearAdjust = 1;
- }
-
- int actualDay = Math.min(dayOfMonth, YearMonth.of(startDate.getYear() + yearAdjust, targetMonth).lengthOfMonth());
- LocalDate current = startDate.withYear(startDate.getYear() + yearAdjust)
- .withMonth(targetMonth)
- .withDayOfMonth(actualDay);
-
- while (!current.isAfter(endDate)) {
- dates.add(LocalDateTime.of(current, time));
- current = current.plusMonths(3);
- actualDay = Math.min(dayOfMonth, current.lengthOfMonth());
- current = current.withDayOfMonth(actualDay);
- }
- return dates;
- }
-
- private java.time.DayOfWeek parseDayOfWeek(String dayOfWeekStr) {
- switch (dayOfWeekStr.toUpperCase()) {
- case "MON": return java.time.DayOfWeek.MONDAY;
- case "TUE": return java.time.DayOfWeek.TUESDAY;
- case "WED": return java.time.DayOfWeek.WEDNESDAY;
- case "THU": return java.time.DayOfWeek.THURSDAY;
- case "FRI": return java.time.DayOfWeek.FRIDAY;
- case "SAT": return java.time.DayOfWeek.SATURDAY;
- case "SUN": return java.time.DayOfWeek.SUNDAY;
- default: throw new IllegalArgumentException("鏃犳晥鐨勬槦鏈熷嚑: " + dayOfWeekStr);
- }
- }
-
- /**
- * 鍒涘缓淇濆吇璁板綍
- */
- private DeviceMaintenance createMaintenanceRecord(MaintenanceTask task, LocalDateTime executionDate) {
- DeviceMaintenance record = new DeviceMaintenance();
- record.setDeviceName(task.getTaskName());
- record.setMaintenanceTaskId(task.getId());
- record.setDeviceLedgerId(task.getTaskId());
- record.setMaintenancePlanTime(executionDate);
- record.setMaintenanceActuallyTime(executionDate);
- record.setMaintenanceActuallyName(task.getMaintenancePerson());
- record.setFrequencyType(task.getFrequencyType());
- record.setFrequencyDetail(task.getFrequencyDetail());
- record.setTenantId(task.getTenantId());
- record.setStatus(0); // 寰呬繚鍏�
- record.setDeviceModel(task.getDeviceModel());
- record.setMachineryCategory(task.getMachineryCategory());
- record.setCreateUser(Integer.parseInt(task.getRegistrantId().toString()));
- record.setUpdateTime(executionDate);
- record.setCreateTime(executionDate);
- record.setUpdateUser(Integer.parseInt(task.getRegistrantId().toString()));
- return record;
- }
-
- /**
- * 鏇存柊浠诲姟鐨勬墽琛屾椂闂�
- */
- private void updateTaskExecutionTime(Long taskId, LocalDateTime lastExecutionTime, LocalDateTime nextExecutionTime) {
- String sql = "UPDATE maintenance_task SET last_execution_time = ?, next_execution_time = ? WHERE id = ?";
- jdbcTemplate.update(sql, lastExecutionTime, nextExecutionTime, taskId);
- }
-
- /**
- * 璁＄畻涓嬫鎵ц鏃堕棿
- */
- private LocalDateTime calculateNextExecutionTime(String frequencyType, String frequencyDetail, LocalDateTime currentTime) {
- return switch (frequencyType) {
- case "DAILY" -> calculateDailyNextTime(frequencyDetail, currentTime);
- case "WEEKLY" -> calculateWeeklyNextTime(frequencyDetail, currentTime);
- case "MONTHLY" -> calculateMonthlyNextTime(frequencyDetail, currentTime);
- case "QUARTERLY" -> calculateQuarterlyNextTime(frequencyDetail, currentTime);
- default -> throw new IllegalArgumentException("涓嶆敮鎸佺殑棰戞绫诲瀷: " + frequencyType);
- };
- }
-
- private LocalDateTime calculateDailyNextTime(String timeStr, LocalDateTime current) {
- LocalTime executionTime = LocalTime.parse(timeStr);
- LocalDateTime nextTime = LocalDateTime.of(current.toLocalDate(), executionTime);
- return current.isBefore(nextTime) ? nextTime : nextTime.plusDays(1);
- }
-
- private LocalDateTime calculateMonthlyNextTime(String detail, LocalDateTime current) {
- String[] parts = detail.split(",");
- int dayOfMonth = Integer.parseInt(parts[0]);
- LocalTime time = LocalTime.parse(parts[1]);
- return current.plusMonths(1)
- .withDayOfMonth(Math.min(dayOfMonth, current.plusMonths(1).toLocalDate().lengthOfMonth()))
- .with(time);
- }
-
- private LocalDateTime calculateWeeklyNextTime(String detail, LocalDateTime current) {
- return current.plusWeeks(1);
- }
-
- private LocalDateTime calculateQuarterlyNextTime(String detail, LocalDateTime current) {
- String[] parts = detail.split(",");
- int quarterMonth = Integer.parseInt(parts[0]);
- int dayOfMonth = Integer.parseInt(parts[1]);
- LocalTime time = LocalTime.parse(parts[2]);
-
- int currentMonthInQuarter = (current.getMonthValue() - 1) % 3 + 1;
-
- YearMonth targetYearMonth;
- if (currentMonthInQuarter < quarterMonth) {
- targetYearMonth = YearMonth.from(current).plusMonths(quarterMonth - currentMonthInQuarter);
- } else {
- targetYearMonth = YearMonth.from(current).plusMonths(3 - currentMonthInQuarter + quarterMonth);
- }
-
- int adjustedDay = Math.min(dayOfMonth, targetYearMonth.lengthOfMonth());
- return LocalDateTime.of(targetYearMonth.getYear(), targetYearMonth.getMonthValue(), adjustedDay, time.getHour(), time.getMinute());
- }
-}
diff --git a/src/test/java/com/ruoyi/inspectiontask/service/impl/TimingTaskJobTest.java b/src/test/java/com/ruoyi/inspectiontask/service/impl/TimingTaskJobTest.java
deleted file mode 100644
index 426f332..0000000
--- a/src/test/java/com/ruoyi/inspectiontask/service/impl/TimingTaskJobTest.java
+++ /dev/null
@@ -1,339 +0,0 @@
-package com.ruoyi.inspectiontask.service.impl;
-
-import com.ruoyi.inspectiontask.mapper.InspectionTaskMapper;
-import com.ruoyi.inspectiontask.pojo.InspectionTask;
-import com.ruoyi.inspectiontask.pojo.TimingTask;
-import org.junit.jupiter.api.Test;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.boot.test.context.SpringBootTest;
-import org.springframework.jdbc.core.JdbcTemplate;
-
-import java.time.DayOfWeek;
-import java.time.LocalDateTime;
-import java.time.LocalTime;
-import java.time.YearMonth;
-import java.time.LocalDate;
-import java.util.HashSet;
-import java.util.List;
-import java.util.Set;
-
-/**
- * 璁惧宸℃瀹氭椂浠诲姟娴嬭瘯绫�
- *
- * 鏌ヨ鏁版嵁搴撲腑鐨勫贰妫�浠诲姟锛屾牴鎹櫥璁版棩鏈熴�侀娆°�佸紑濮嬫棩鏈熶笌鏃堕棿鏉ョ敓鎴愬贰妫�璁板綍
- * 骞舵洿鏂� timing_task 琛ㄧ殑鏈�鍚庢墽琛屾椂闂村拰涓嬫鎵ц鏃堕棿
- */
-@SpringBootTest
-public class TimingTaskJobTest {
-
- @Autowired
- private InspectionTaskMapper inspectionTaskMapper;
-
- @Autowired
- private JdbcTemplate jdbcTemplate;
-
- /**
- * 娴嬭瘯锛氭牴鎹墍鏈夊贰妫�浠诲姟鐢熸垚宸℃璁板綍
- * 浠庣櫥璁版棩鏈熷埌浠婂ぉ锛屾寜棰戞姣忓懆鐢熸垚涓�鏉¤褰�
- */
- @Test
- void testGenerateInspectionRecordForAllTasks() {
- // 鏌ヨ鎵�鏈夊惎鐢ㄧ殑宸℃浠诲姟
- String sql = "SELECT * FROM timing_task WHERE is_enabled = 1 AND deleted = 0";
- List<TimingTask> tasks = jdbcTemplate.query(sql,
- (rs, rowNum) -> {
- TimingTask task = new TimingTask();
- task.setId(rs.getLong("id"));
- task.setTaskName(rs.getString("task_name"));
- task.setInspectionProject(rs.getString("inspection_project"));
- task.setTaskId(rs.getInt("task_id"));
- task.setInspectorIds(rs.getString("inspector_ids"));
- task.setInspectionLocation(rs.getString("inspection_location"));
- task.setFrequencyType(rs.getString("frequency_type"));
- task.setFrequencyDetail(rs.getString("frequency_detail"));
- task.setRemarks(rs.getString("remarks"));
- task.setRegistrantId(rs.getLong("registrant_id"));
- task.setRegistrant(rs.getString("registrant"));
- task.setTenantId(rs.getLong("tenant_id"));
- // 鑾峰彇鐧昏鏃ユ湡
- java.sql.Date regDate = rs.getDate("registration_date");
- if (regDate != null) {
- task.setRegistrationDate(regDate.toLocalDate());
- }
- return task;
- });
-
- System.out.println("=== 鍏辨壘鍒� " + tasks.size() + " 涓贰妫�浠诲姟 ===\n");
-
- int totalRecords = 0;
- for (TimingTask task : tasks) {
- try {
- // 鐢熸垚鎵�鏈夌鍚堥娆＄殑璁板綍
- int recordCount = generateRecordsForTask(task);
- totalRecords += recordCount;
- System.out.println("鉁� 浠诲姟ID: " + task.getId() + " | " + task.getTaskName()
- + " | 鐧昏鏃ユ湡: " + task.getRegistrationDate()
- + " | 鐢熸垚璁板綍鏁�: " + recordCount);
- } catch (Exception e) {
- System.out.println("鉁� 浠诲姟ID: " + task.getId() + " | " + task.getTaskName() + " | 澶辫触: " + e.getMessage());
- }
- }
-
- System.out.println("\n=== 鎵ц瀹屾垚: 鍏辩敓鎴� " + totalRecords + " 鏉¤褰� ===");
- }
-
- /**
- * 涓哄崟涓换鍔＄敓鎴愭墍鏈夌鍚堥娆＄殑璁板綍
- */
- private int generateRecordsForTask(TimingTask task) {
- LocalDate startDate = task.getRegistrationDate();
- if (startDate == null) {
- startDate = LocalDate.now();
- }
- LocalDate endDate = LocalDate.now();
-
- // 鏍规嵁棰戞鑾峰彇鎵�鏈夐渶瑕佹墽琛岀殑鏃ユ湡
- List<LocalDateTime> executionDates = getExecutionDates(task.getFrequencyType(), task.getFrequencyDetail(), startDate, endDate);
-
- int count = 0;
- for (LocalDateTime executionDate : executionDates) {
- try {
- // 鐢熸垚宸℃璁板綍
- InspectionTask record = createInspectionRecord(task, executionDate);
- inspectionTaskMapper.insert(record);
-
- // 鏇存柊浠诲姟鐨勪笂娆℃墽琛屾椂闂�
- updateTaskLastExecutionTime(task.getId(), executionDate);
-
- count++;
- } catch (Exception e) {
- System.out.println(" 鉁� 鏃ユ湡: " + executionDate + " | 澶辫触: " + e.getMessage());
- }
- }
-
- // 鏇存柊浠诲姟鐨勪笅娆℃墽琛屾椂闂�
- if (count > 0) {
- LocalDateTime lastExecution = executionDates.get(executionDates.size() - 1);
- LocalDateTime nextExecution = calculateNextExecutionTime(task.getFrequencyType(), task.getFrequencyDetail(), lastExecution);
- updateTaskNextExecutionTime(task.getId(), nextExecution);
- }
-
- return count;
- }
-
- /**
- * 鑾峰彇鎸囧畾鏃ユ湡鑼冨洿鍐呮墍鏈夌鍚堥娆＄殑鎵ц鏃堕棿
- */
- private List<LocalDateTime> getExecutionDates(String frequencyType, String frequencyDetail, LocalDate startDate, LocalDate endDate) {
- List<LocalDateTime> dates = new java.util.ArrayList<>();
-
- switch (frequencyType) {
- case "DAILY":
- dates.addAll(getDailyDates(startDate, endDate, frequencyDetail));
- break;
- case "WEEKLY":
- dates.addAll(getWeeklyDates(startDate, endDate, frequencyDetail));
- break;
- case "MONTHLY":
- dates.addAll(getMonthlyDates(startDate, endDate, frequencyDetail));
- break;
- case "QUARTERLY":
- dates.addAll(getQuarterlyDates(startDate, endDate, frequencyDetail));
- break;
- }
-
- return dates;
- }
-
- private List<LocalDateTime> getDailyDates(LocalDate startDate, LocalDate endDate, String timeStr) {
- List<LocalDateTime> dates = new java.util.ArrayList<>();
- LocalTime time = LocalTime.parse(timeStr);
-
- LocalDate current = startDate;
- while (!current.isAfter(endDate)) {
- dates.add(LocalDateTime.of(current, time));
- current = current.plusDays(1);
- }
- return dates;
- }
-
- private List<LocalDateTime> getWeeklyDates(LocalDate startDate, LocalDate endDate, String detail) {
- List<LocalDateTime> dates = new java.util.ArrayList<>();
- String[] parts = detail.split(",");
- String dayOfWeekStr = parts[0];
- LocalTime time = LocalTime.parse(parts[1]);
-
- Set<DayOfWeek> targetDays = parseDayOfWeeks(dayOfWeekStr);
-
- LocalDate current = startDate;
- while (!current.isAfter(endDate)) {
- if (targetDays.contains(current.getDayOfWeek())) {
- dates.add(LocalDateTime.of(current, time));
- }
- current = current.plusDays(1);
- }
- return dates;
- }
-
- private List<LocalDateTime> getMonthlyDates(LocalDate startDate, LocalDate endDate, String detail) {
- List<LocalDateTime> dates = new java.util.ArrayList<>();
- String[] parts = detail.split(",");
- int dayOfMonth = Integer.parseInt(parts[0]);
- LocalTime time = LocalTime.parse(parts[1]);
-
- LocalDate current = startDate.plusMonths(0).withDayOfMonth(Math.min(dayOfMonth, startDate.lengthOfMonth()));
- while (!current.isAfter(endDate)) {
- dates.add(LocalDateTime.of(current, time));
- current = current.plusMonths(1).withDayOfMonth(Math.min(dayOfMonth, current.plusMonths(1).lengthOfMonth()));
- }
- return dates;
- }
-
- private List<LocalDateTime> getQuarterlyDates(LocalDate startDate, LocalDate endDate, String detail) {
- List<LocalDateTime> dates = new java.util.ArrayList<>();
- String[] parts = detail.split(",");
- int quarterMonth = Integer.parseInt(parts[0]);
- int dayOfMonth = Integer.parseInt(parts[1]);
- LocalTime time = LocalTime.parse(parts[2]);
-
- int currentMonth = startDate.getMonthValue();
- int targetMonth = ((currentMonth - 1) / 3) * 3 + quarterMonth;
- int yearAdjust = 0;
-
- if (targetMonth > 12) {
- targetMonth -= 12;
- yearAdjust = 1;
- }
-
- LocalDate current = startDate.withYear(startDate.getYear() + yearAdjust)
- .withMonth(targetMonth)
- .withDayOfMonth(Math.min(dayOfMonth, YearMonth.of(startDate.getYear() + yearAdjust, targetMonth).lengthOfMonth()));
-
- while (!current.isAfter(endDate)) {
- dates.add(LocalDateTime.of(current, time));
- current = current.plusMonths(3).withDayOfMonth(Math.min(dayOfMonth, current.plusMonths(3).lengthOfMonth()));
- }
- return dates;
- }
-
- private Set<DayOfWeek> parseDayOfWeeks(String dayOfWeekStr) {
- Set<DayOfWeek> days = new HashSet<>();
- String[] dayStrs = dayOfWeekStr.split("\\|");
-
- for (String dayStr : dayStrs) {
- switch (dayStr.toUpperCase()) {
- case "MON": days.add(DayOfWeek.MONDAY); break;
- case "TUE": days.add(DayOfWeek.TUESDAY); break;
- case "WED": days.add(DayOfWeek.WEDNESDAY); break;
- case "THU": days.add(DayOfWeek.THURSDAY); break;
- case "FRI": days.add(DayOfWeek.FRIDAY); break;
- case "SAT": days.add(DayOfWeek.SATURDAY); break;
- case "SUN": days.add(DayOfWeek.SUNDAY); break;
- }
- }
- return days;
- }
-
- /**
- * 璁＄畻涓嬫鎵ц鏃堕棿
- */
- private LocalDateTime calculateNextExecutionTime(String frequencyType, String frequencyDetail, LocalDateTime currentTime) {
- return switch (frequencyType) {
- case "DAILY" -> calculateDailyNextTime(frequencyDetail, currentTime);
- case "WEEKLY" -> calculateWeeklyNextTime(frequencyDetail, currentTime);
- case "MONTHLY" -> calculateMonthlyNextTime(frequencyDetail, currentTime);
- case "QUARTERLY" -> calculateQuarterlyNextTime(frequencyDetail, currentTime);
- default -> throw new IllegalArgumentException("涓嶆敮鎸佺殑棰戞绫诲瀷: " + frequencyType);
- };
- }
-
- private LocalDateTime calculateDailyNextTime(String timeStr, LocalDateTime current) {
- LocalTime executionTime = LocalTime.parse(timeStr);
- LocalDateTime nextTime = LocalDateTime.of(current.toLocalDate(), executionTime);
- return current.isBefore(nextTime) ? nextTime : nextTime.plusDays(1);
- }
-
- private LocalDateTime calculateMonthlyNextTime(String detail, LocalDateTime current) {
- String[] parts = detail.split(",");
- int dayOfMonth = Integer.parseInt(parts[0]);
- LocalTime time = LocalTime.parse(parts[1]);
- return current.plusMonths(1)
- .withDayOfMonth(Math.min(dayOfMonth, current.plusMonths(1).toLocalDate().lengthOfMonth()))
- .with(time);
- }
-
- private LocalDateTime calculateWeeklyNextTime(String detail, LocalDateTime current) {
- String[] parts = detail.split(",");
- String dayOfWeekStr = parts[0];
- LocalTime time = LocalTime.parse(parts[1]);
-
- Set<DayOfWeek> targetDays = parseDayOfWeeks(dayOfWeekStr);
-
- LocalDateTime nextTime = current;
- while (true) {
- nextTime = nextTime.plusDays(1);
- if (targetDays.contains(nextTime.getDayOfWeek())) {
- return LocalDateTime.of(nextTime.toLocalDate(), time);
- }
- if (nextTime.isAfter(current.plusYears(1))) {
- throw new RuntimeException("鏃犳硶鎵惧埌涓嬫鎵ц鏃堕棿");
- }
- }
- }
-
- private LocalDateTime calculateQuarterlyNextTime(String detail, LocalDateTime current) {
- String[] parts = detail.split(",");
- int quarterMonth = Integer.parseInt(parts[0]);
- int dayOfMonth = Integer.parseInt(parts[1]);
- LocalTime time = LocalTime.parse(parts[2]);
-
- int currentMonthInQuarter = (current.getMonthValue() - 1) % 3 + 1;
-
- YearMonth targetYearMonth;
- if (currentMonthInQuarter < quarterMonth) {
- targetYearMonth = YearMonth.from(current).plusMonths(quarterMonth - currentMonthInQuarter);
- } else {
- targetYearMonth = YearMonth.from(current).plusMonths(3 - currentMonthInQuarter + quarterMonth);
- }
-
- int adjustedDay = Math.min(dayOfMonth, targetYearMonth.lengthOfMonth());
- return LocalDateTime.of(targetYearMonth.getYear(), targetYearMonth.getMonthValue(), adjustedDay, time.getHour(), time.getMinute());
- }
-
- /**
- * 鍒涘缓宸℃璁板綍
- */
- private InspectionTask createInspectionRecord(TimingTask timingTask, LocalDateTime executionDate) {
- InspectionTask inspectionTask = new InspectionTask();
- inspectionTask.setTaskName(timingTask.getTaskName());
- inspectionTask.setInspectionProject(timingTask.getInspectionProject());
- inspectionTask.setTaskId(timingTask.getTaskId());
- inspectionTask.setInspectorId(timingTask.getInspectorIds());
- inspectionTask.setInspectionLocation(timingTask.getInspectionLocation());
- inspectionTask.setRemarks("鑷姩鐢熸垚鑷畾鏃朵换鍔D: " + timingTask.getId() + "锛�" + timingTask.getRemarks());
- inspectionTask.setRegistrantId(timingTask.getRegistrantId());
- inspectionTask.setFrequencyType(timingTask.getFrequencyType());
- inspectionTask.setFrequencyDetail(timingTask.getFrequencyDetail());
- inspectionTask.setTenantId(timingTask.getTenantId());
- // 璁剧疆鐧昏鏃ユ湡涓烘墽琛屾棩鏈�
- inspectionTask.setCreateTime(executionDate);
- inspectionTask.setUpdateTime(executionDate);
- return inspectionTask;
- }
-
- /**
- * 鏇存柊浠诲姟鐨勪笂娆℃墽琛屾椂闂�
- */
- private void updateTaskLastExecutionTime(Long taskId, LocalDateTime lastExecutionTime) {
- String updateSql = "UPDATE timing_task SET last_execution_time = ? WHERE id = ?";
- jdbcTemplate.update(updateSql, lastExecutionTime, taskId);
- }
-
- /**
- * 鏇存柊浠诲姟鐨勪笅娆℃墽琛屾椂闂�
- */
- private void updateTaskNextExecutionTime(Long taskId, LocalDateTime nextExecutionTime) {
- String updateSql = "UPDATE timing_task SET next_execution_time = ? WHERE id = ?";
- jdbcTemplate.update(updateSql, nextExecutionTime, taskId);
- }
-}
--
Gitblit v1.9.3