package com.zhitan.engine.controller; import com.zhitan.engine.scheduler.DataCleaningScheduler; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.format.annotation.DateTimeFormat; import org.springframework.http.ResponseEntity; import org.springframework.web.bind.annotation.*; import java.time.LocalDateTime; import java.util.HashMap; import java.util.Map; /** * 数据清洗控制器 * 提供手动触发数据清洗任务的API接口 */ @Slf4j @RestController @RequestMapping("/api/data-cleaning") public class DataCleaningController { @Autowired private DataCleaningScheduler dataCleaningScheduler; /** * 手动触发数据清洗任务 * * @param timeType 时间类型:HOUR, DAY, MONTH, YEAR * @param dateTime 统计时间点 * @return 处理结果 */ @PostMapping("/trigger") public ResponseEntity> triggerDataCleaning( @RequestParam("timeType") String timeType, @RequestParam("dateTime") @DateTimeFormat(pattern = "yyyy-MM-dd HH:mm:ss") LocalDateTime dateTime) { log.info("收到手动触发数据清洗请求,时间类型:{},时间点:{}", timeType, dateTime); Map result = new HashMap<>(); try { // 验证时间类型 if (!"HOUR".equals(timeType) && !"DAY".equals(timeType) && !"MONTH".equals(timeType) && !"YEAR".equals(timeType)) { result.put("success", false); result.put("message", "不支持的时间类型:" + timeType); return ResponseEntity.badRequest().body(result); } // 触发任务 dataCleaningScheduler.manualTrigger(timeType, dateTime); result.put("success", true); result.put("message", "任务已触发,请查看日志了解执行情况"); return ResponseEntity.ok(result); } catch (Exception e) { log.error("触发数据清洗任务失败:{}", e.getMessage(), e); result.put("success", false); result.put("message", "触发任务失败:" + e.getMessage()); return ResponseEntity.internalServerError().body(result); } } }