mitao
3 天以前 dfe379f10f5d16a3b34b33f260d3a7ca97c7fdc6
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
115
116
117
118
119
120
121
122
123
124
125
126
package com.ruoyi.web.controller.api;
 
 
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.basic.PageInfo;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.common.enums.BusinessType;
import com.ruoyi.framework.web.service.TokenService;
import com.ruoyi.system.dto.YearTaskAuditDTO;
import com.ruoyi.system.model.ProjectInvestment;
import com.ruoyi.system.model.ProjectMain;
import com.ruoyi.system.model.YearTask;
import com.ruoyi.system.model.YearTaskDetail;
import com.ruoyi.system.query.ProjectMainListQuery;
import com.ruoyi.system.query.YearTaskListQuery;
import com.ruoyi.system.service.*;
import com.ruoyi.system.vo.ProjectMainListVO;
import com.ruoyi.system.vo.YearTaskListVO;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.*;
 
import javax.annotation.Resource;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
 
/**
 * <p>
 * 年度任务主表 前端控制器
 * </p>
 *
 * @author WuGuanFengYue
 * @since 2025-10-16
 */
@RestController
@Api(tags = "年度任务")
@RequestMapping("/year-task")
public class YearTaskController {
    @Resource
    private YearTaskService yearTaskService;
    @Resource
    private ISysUserService sysUserService;
    @Resource
    private TokenService tokenService;
    @Resource
    private YearTaskDetailService yearTaskDetailService;
    @Resource
    private ProjectMainService projectMainService;
 
    @ApiOperation(value = "项目不分页列表")
    @PostMapping(value = "/listProject")
    public R<List<ProjectMain>> listProject() {
        return R.ok(projectMainService.list());
    }
    @ApiOperation(value = "年度任务分页列表")
    @PostMapping(value = "/pageList")
    public R<PageInfo<YearTaskListVO>> pageList(@RequestBody YearTaskListQuery query) {
        return R.ok(yearTaskService.pageList(query));
    }
    @ApiOperation(value = "添加")
    @Transactional
    @Log(title = "年度任务-添加", businessType = BusinessType.INSERT)
    @PostMapping(value = "/add")
    public R<Boolean> save(@RequestBody YearTask entity) {
        yearTaskService.save(entity);
        List<YearTaskDetail> yearTaskDetails = entity.getYearTaskDetails();
        for (YearTaskDetail yearTaskDetail : yearTaskDetails) {
            yearTaskDetail.setYearTaskId(entity.getId());
        }
        yearTaskDetailService.saveBatch(yearTaskDetails);
        return R.ok();
    }
 
    @ApiOperation(value = "修改")
    @Transactional
    @Log(title = "年度任务-修改", businessType = BusinessType.UPDATE)
    @PostMapping(value = "/edit")
    public R<Boolean> edit(@RequestBody ProjectMain entity) {
        projectMainService.updateById(entity);
        return R.ok();
    }
    @ApiOperation(value = "年度任务投资目标-审核")
    @Transactional
    @Log(title = "年度任务投资目标-审核", businessType = BusinessType.UPDATE)
    @PostMapping(value = "/audit")
    public R<Boolean> audit(@RequestBody YearTaskAuditDTO  entity) {
        YearTaskDetail yearTaskDetail = yearTaskDetailService.getById(entity.getId());
        yearTaskDetail.setIsAudit( true);
        yearTaskDetail.setAuditUserId(tokenService.getLoginUser().getUser().getUserId().intValue());
        yearTaskDetailService.updateById(yearTaskDetail);
        return R.ok();
    }
    @ApiOperation(value = "年度任务投资目标-修改")
    @Transactional
    @Log(title = "年度任务投资目标-修改", businessType = BusinessType.UPDATE)
    @PostMapping(value = "/editYearTaskDetail")
    public R<Boolean> edit(@RequestBody YearTaskDetail  entity) {
        YearTaskDetail yearTaskDetail = yearTaskDetailService.getById(entity.getId());
        yearTaskDetail.setIsAudit( true);
        yearTaskDetail.setCompeleteUserId(tokenService.getLoginUser().getUser().getUserId().intValue());
        yearTaskDetailService.updateById(yearTaskDetail);
        return R.ok();
    }
    @ApiOperation(value = "年度任务详情")
    @GetMapping(value = "/detail")
    public R<YearTask> detail(Integer id ) {
        YearTask yearTask = yearTaskService.getById(id);
        List<YearTaskDetail> list = yearTaskDetailService.lambdaQuery()
                .eq(YearTaskDetail::getYearTaskId, id).list();
        Map<Long, String> userMaps = sysUserService.selectAllList().stream()
                .collect(Collectors.toMap(SysUser::getUserId, SysUser::getNickName));
        for (YearTaskDetail yearTaskDetail : list) {
            if (yearTaskDetail.getAuditUserId()!=null){
                yearTaskDetail.setAuditUserName(userMaps.get(Long.valueOf(yearTaskDetail.getAuditUserId())));
            }
            if (yearTaskDetail.getCompeleteUserId()!=null){
                yearTaskDetail.setCompeleteUserName(userMaps.get(Long.valueOf(yearTaskDetail.getCompeleteUserId())));
            }
        }
        yearTask.setYearTaskDetails(list);
        return R.ok(yearTask);
    }
}