easydo-zcgx/src/main/java/com/qdport/controller/PolicyStriveController.java

package com.qdport.controller;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.qdport.annotation.WebLog;
import com.qdport.entity.PolicyTodo;
import com.qdport.enums.QDPortEnum;
import com.qdport.modules.system.entity.SysDept;
import com.qdport.modules.system.entity.SysUser;
import com.qdport.modules.system.service.SysDeptService;
import com.qdport.modules.system.service.TSysUserService;
import com.qdport.service.PolicyFileService;
import com.qdport.service.PolicyLogService;
import com.qdport.service.PolicyTodoService;
import com.qdport.service.impl.PolicySystemService;
import com.qdport.util.StringUtil;
import com.qdport.vo.PolicyFileVO;
import com.qdport.entity.PolicyStrive;
import com.qdport.service.PolicyStriveService;
import com.qdport.query.PolicyStriveQuery;
import com.qdport.vo.PolicyLogVO;
import com.qdport.vo.PolicyStriveVO;
import com.qdport.wrapper.PolicyStriveWrapper;
import org.springframework.web.bind.annotation.*;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import com.qdport.core.tool.api.R;
import com.qdport.core.tool.utils.Func;
import com.qdport.core.boot.ctrl.QdportController;
import com.baomidou.mybatisplus.core.metadata.IPage;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.validation.Valid;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ExecutionException;

/**
 * 政策争取
 *
 * @author yuheng
 * @since 1.0.0 2024-10-14
 */
@RestController
@RequestMapping("qdport-zcgx/strive")
@Api(value = "政策争取", tags = "政策争取")
public class PolicyStriveController extends QdportController {
    @Resource
    private PolicyStriveService policyStriveService;
    @Resource
    private PolicyFileService policyFileService;
    @Resource
    private PolicyTodoService policyTodoService;
    @Resource
    private SysDeptService sysDeptService;

    /**
     * 分页
     */
    @GetMapping("page")
    @ApiOperation(value = "分页", notes = "分页")
    public R<IPage<PolicyStriveVO>> page(@Valid PolicyStriveQuery query) {
        IPage<PolicyStriveVO> page = policyStriveService.page(query);
        List<PolicyStriveVO> records = page.getRecords();
        for (PolicyStriveVO model : records) {

            SysDept dept = sysDeptService.getById(model.getDeptId());
            model.setDeptName(dept.getName());
            SysDept company = sysDeptService.getById(model.getCompanyId());
            model.setCompanyName(company.getName());

            List<PolicyFileVO> fileList = policyFileService.getFileList(model.getId(), QDPortEnum.POLICY_FILE_REFTYPE_POLICY_STRIVE.getValue());
            model.setFileList(fileList);
        }
        return R.data(page);
    }

    /**
     * 详情
     */
    @GetMapping("{id}")
    @ApiOperation(value = "详情", notes = "id")
    public R<PolicyStriveVO> get(@PathVariable("id") String id) {
        PolicyStrive entity = policyStriveService.getById(id);

        PolicyStriveVO vo = PolicyStriveWrapper.build().entityVO(entity);

        SysDept dept = sysDeptService.getById(vo.getDeptId());
        vo.setDeptName(dept.getName());
        SysDept company = sysDeptService.getById(vo.getCompanyId());
        vo.setCompanyName(company.getName());

        List<PolicyFileVO> fileList = policyFileService.getFileList(entity.getId().toString(), QDPortEnum.POLICY_FILE_REFTYPE_POLICY_STRIVE.getValue());

        vo.setFileList(fileList);
        return R.data(vo);
    }

    /**
     * 新增
     */
    @PostMapping("/save")
    @ApiOperation(value = "新增", notes = "传入PolicyStrive")
    @WebLog(operateName = "新增", operateType = "policy_strive")
    public R save(@RequestBody PolicyStriveVO vo) {
        vo.setStatus(QDPortEnum.POLICY_STRIVE_STATUS_ACTIVE.getValue());
        policyStriveService.save(vo);
        return R.success("保存成功");
    }

    @PostMapping("/saveDone")
    @ApiOperation(value = "新增", notes = "传入PolicyStrive")
    @WebLog(operateName = "管理员新增", operateType = "policy_strive")
    public R saveDone(@RequestBody PolicyStriveVO vo) {
        vo.setStatus(QDPortEnum.POLICY_STRIVE_STATUS_DONE.getValue());
        policyStriveService.save(vo);
        return R.success("保存成功");
    }

    @PostMapping("/saveApprove")
    @ApiOperation(value = "新增并且提交", notes = "传入PolicyStrive")
    @WebLog(operateName = "新增并且上报", operateType = "policy_strive")
    public R saveApprove(@RequestBody PolicyStriveVO vo) throws ExecutionException, InterruptedException {
        policyStriveService.saveApprove(vo);
        policyTodoService.save(vo);
        return R.success("新增并且提交成功");
    }

    /**
     * 修改
     */
    @PostMapping("/update")
    @ApiOperation(value = "编辑", notes = "传入PolicyStrive")
    @WebLog(operateName = "修改", operateType = "policy_strive")
    public R update(@RequestBody @Valid PolicyStriveVO vo) {
        policyStriveService.update(vo);
        return R.success("更新成功");
    }

    @PostMapping("/updateById")
    @ApiOperation(value = "月度维护、奖励分配", notes = "传入PolicyStrive")
    @WebLog(operateName = "月度维护/奖励分配", operateType = "policy_strive")
    public R updateById(@RequestBody @Valid PolicyStriveVO vo) {
        if (StringUtil.isNotEmpty(vo.getIsReward()) && vo.getIsReward() == 1) {
            String partPersonArr = vo.getPartPersonArr();
            JSONArray array = JSON.parseArray(partPersonArr);
            BigDecimal total = BigDecimal.ZERO;
            for (Object object : array) {
                JSONObject obj = (JSONObject) object;
                BigDecimal reward = obj.getBigDecimal("reward");
                if (StringUtil.isEmpty(reward)) {
                    continue;
                }
                total = total.add(reward);
            }
            if (vo.getRewardScore().compareTo(total) < 0) {
                return R.fail("分配总金额超出奖励标准");
            }
        }
        policyStriveService.updateById(vo);
        return R.success("更新成功");
    }

    @PostMapping("/withdraw")
    @ApiOperation(value = "撤回", notes = "传入PolicyStrive")
    @WebLog(operateName = "撤回", operateType = "policy_strive")
    public R withdraw(@RequestBody @Valid PolicyStriveVO vo) {
        policyStriveService.withdraw(vo);
        return R.success("撤回成功");
    }

    @PostMapping("/approve")
    @ApiOperation(value = "审批通过", notes = "传入PolicyStrive")
    @WebLog(operateName = "审批通过", operateType = "policy_strive")
    public R approve(@RequestBody @Valid PolicyStriveVO vo) throws ExecutionException, InterruptedException {
        vo.setIsWithdraw(0);
        policyStriveService.approve(vo);
        policyTodoService.save(vo);
        return R.success("更新成功");
    }

    @PostMapping("/refuse")
    @ApiOperation(value = "审批拒绝", notes = "传入PolicyStrive")
    @WebLog(operateName = "审批拒绝", operateType = "policy_strive")
    public R refuse(@RequestBody @Valid PolicyStriveVO vo) {
        vo.setIsWithdraw(1);
        policyStriveService.refuse(vo);
        return R.success("更新成功");
    }

    @PostMapping("/resubmit")
    @ApiOperation(value = "重新提交", notes = "传入PolicyStrive")
    @WebLog(operateName = "重新上报", operateType = "policy_strive")
    public R resubmit(@RequestBody @Valid PolicyStriveVO vo) throws ExecutionException, InterruptedException {
        vo.setStatus(QDPortEnum.POLICY_STRIVE_STATUS_APPROVE.getValue());
        policyStriveService.approve(vo);
        policyTodoService.save(vo);
        return R.success("重新发起成功");
    }

    /**
     * 删除
     */
    @PostMapping("/remove")
    @ApiOperation(value = "逻辑删除", notes = "传入ids")
    @WebLog(operateName = "删除", operateType = "policy_strive")
    public R delete(@ApiParam(value = "主键集合", required = true) @RequestParam(name = "ids") String ids) {
        List<String> idList = Func.toStrList(ids);
        policyStriveService.delete(idList);
        return R.success("删除成功");
    }

    @PostMapping("/calculate")
    @ApiOperation(value = "计算金额", notes = "传入PolicyStrive")
    public R calculate(@RequestBody @Valid PolicyStriveVO vo) {
        BigDecimal rewardRadix = vo.getRewardRadix();
        BigDecimal implementScore = vo.getImplementScore();
        BigDecimal promotionScore = vo.getPromotionScore();
        BigDecimal striveScore = vo.getStriveScore();

        BigDecimal number = rewardRadix.multiply(implementScore)
                .multiply(promotionScore).multiply(striveScore).setScale(2, RoundingMode.HALF_UP);
        return R.data(number);
    }
}