| | |
| | | import org.springframework.web.bind.annotation.RequestMapping; |
| | | import org.springframework.web.bind.annotation.RestController; |
| | | |
| | | import java.util.HashMap; |
| | | import java.util.Map; |
| | | |
| | | @RestController |
| | | @RequestMapping("/cleanup") |
| | | public class DataCleanupController { |
| | |
| | | return "清理失败: " + e.getMessage(); |
| | | } |
| | | } |
| | | |
| | | @PostMapping("/clear-all-test-data") |
| | | public Map<String, Object> clearAllTestData() { |
| | | Map<String, Object> result = new HashMap<>(); |
| | | try { |
| | | // 禁用外键检查 |
| | | jdbcTemplate.execute("SET FOREIGN_KEY_CHECKS = 0"); |
| | | |
| | | // 按照外键依赖关系的顺序删除 |
| | | int deletedRatingItems = jdbcTemplate.update("DELETE FROM t_activity_rating_item"); |
| | | int deletedRatings = jdbcTemplate.update("DELETE FROM t_activity_rating"); |
| | | int deletedActivityPlayers = jdbcTemplate.update("DELETE FROM t_activity_player"); |
| | | int deletedActivityJudges = jdbcTemplate.update("DELETE FROM t_activity_judge"); |
| | | int deletedPlayers = jdbcTemplate.update("DELETE FROM t_player"); |
| | | int deletedActivities = jdbcTemplate.update("DELETE FROM t_activity"); |
| | | |
| | | // 重新启用外键检查 |
| | | jdbcTemplate.execute("SET FOREIGN_KEY_CHECKS = 1"); |
| | | |
| | | result.put("success", true); |
| | | result.put("message", "所有测试数据已清空"); |
| | | result.put("deletedCounts", Map.of( |
| | | "t_activity_rating_item", deletedRatingItems, |
| | | "t_activity_rating", deletedRatings, |
| | | "t_activity_player", deletedActivityPlayers, |
| | | "t_activity_judge", deletedActivityJudges, |
| | | "t_player", deletedPlayers, |
| | | "t_activity", deletedActivities |
| | | )); |
| | | |
| | | } catch (Exception e) { |
| | | result.put("success", false); |
| | | result.put("message", "清理失败: " + e.getMessage()); |
| | | } |
| | | |
| | | return result; |
| | | } |
| | | } |