feat: 更新系统

kongdeqiang

2026-03-23 0d3f0dc4931e2a8cd379c5b7e9a9c70276ebc1cf

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
package com.example.controller;
 
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.example.common.Result;
import com.example.entity.DataExcel;
import com.example.entity.DataTask;
import com.example.service.DataExcelService;
import com.example.service.DataTaskService;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
 
import java.util.List;
 
@RestController
@RequestMapping("/api/data-task")
public class DataTaskController {
 
    @Autowired
    private DataTaskService dataTaskService;
    @Autowired
    private DataExcelService dataExcelService;
 
    @GetMapping("/page")
    public Result<IPage<DataTask>> page(
            @RequestParam(defaultValue = "1") Long current,
            @RequestParam(defaultValue = "10") Long size,
            @RequestParam(required = false) String unitCode,
            @RequestParam(required = false) String status) {
        Page<DataTask> page = new Page<>(current, size);
        LambdaQueryWrapper<DataTask> wrapper = new LambdaQueryWrapper<>();
        if (unitCode != null && !unitCode.isEmpty()) {
            wrapper.eq(DataTask::getUnitCode, unitCode);
        }
        if (status != null && !status.isEmpty()) {
            wrapper.eq(DataTask::getStatus, status);
        }
        wrapper.orderByDesc(DataTask::getCreateTime);
        IPage<DataTask> dataTaskPage = dataTaskService.page(page, wrapper);
        return Result.success(dataTaskPage);
    }
 
    @GetMapping("/pageNew")
    public Result<IPage<DataTask>> pageNew(
            @RequestParam(defaultValue = "1") Long current,
            @RequestParam(defaultValue = "10") Long size,
            @RequestParam(required = false) String unitCode,
            @RequestParam(required = false) String status) {
        Page<DataTask> page = new Page<>(current, size);
        LambdaQueryWrapper<DataTask> wrapper = new LambdaQueryWrapper<>();
        if (unitCode != null && !unitCode.isEmpty()) {
            wrapper.eq(DataTask::getUnitCode, unitCode);
        }
        if (status != null && !status.isEmpty()) {
            wrapper.eq(DataTask::getStatus, status);
        }
        wrapper.orderByDesc(DataTask::getCreateTime);
        IPage<DataTask> dataTaskPage = dataTaskService.page(page, wrapper);
        for (DataTask record : dataTaskPage.getRecords()) {
            LambdaQueryWrapper<DataExcel> excelLambdaQueryWrapper = new LambdaQueryWrapper<>();
            excelLambdaQueryWrapper.eq(DataExcel::getUnitCode,record.getUnitCode());
            excelLambdaQueryWrapper.orderByDesc(DataExcel::getCreateTime);
            List<DataExcel> list = dataExcelService.list(excelLambdaQueryWrapper);
            if(list.isEmpty()){
                record.setStatus(0);
                record.setSuccessCount(0);
                dataTaskService.updateById(record);
            }else {
                record.setSuccessCount(list.size());
                if( list.size() >= record.getTaskCount()){
                    record.setStatus(1);
                }else {
                    record.setStatus(0);
                }
                dataTaskService.updateById(record);
            }
        }
 
        return Result.success(dataTaskPage);
    }
 
    @GetMapping("/{id}")
    public Result<DataTask> getById(@PathVariable Long id) {
        DataTask dataTask = dataTaskService.getById(id);
        return Result.success(dataTask);
    }
 
    @PostMapping
    public Result<Boolean> save(@RequestBody DataTask dataTask) {
        // 检查单位id是否已存在
        LambdaQueryWrapper<DataTask> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(DataTask::getUnitCode, dataTask.getUnitCode());
        if (dataTaskService.count(wrapper) > 0) {
            return Result.error("该单位任务已存在");
        }
 
        DataTask task = new DataTask();
        BeanUtils.copyProperties(dataTask,task);
        task.setStatus(0);
        return Result.success(dataTaskService.save(task));
    }
 
    @PutMapping
    public Result<Boolean> update(@RequestBody DataTask dataTask) {
        return Result.success(dataTaskService.updateById(dataTask));
    }
 
    @DeleteMapping("/{id}")
    public Result<Boolean> delete(@PathVariable Long id) {
        return Result.success(dataTaskService.removeById(id));
    }
}