Code/operation_manager/src/main/java/com/huaxu/task/controller/PlanManageController.java

package com.huaxu.task.controller;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.huaxu.model.AjaxMessage;
import com.huaxu.model.LoginUser;
import com.huaxu.model.Pagination;
import com.huaxu.model.ResultStatus;
import com.huaxu.process.entity.ProcessDefinition;
import com.huaxu.process.service.WorkFlowService;
import com.huaxu.task.dto.PlanManageDto;
import com.huaxu.task.entity.PlanManage;
import com.huaxu.task.service.PlanManageService;
import com.huaxu.util.DatesUtil;
import com.huaxu.util.UserUtil;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import org.activiti.engine.task.Task;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.*;

import java.util.*;

@RestController
@RequestMapping("/workflow/task/")
@Api(tags = "计划管理")
public class PlanManageController {

    @Autowired
    PlanManageService planManageService;

    @Autowired
    WorkFlowService workFlowService;


    /**
     * 新增巡检计划
     * @param planName 任务名称
     * @param taskType 任务类型
     * @param planStartDate 计划开始时间
     * @param planEndDate 计划结束时间
     * @param userId 计划巡检人ID
     * @param taskContent 任务内容
     * @param taskAreaShape 任务区域坐标
     * @param taskAreaName 任务区域名称
     * @return
     */
    @GetMapping("/plan/add")
    @ApiOperation(value = "新增巡检计划")
    public AjaxMessage addPlan(@ApiParam(value = "任务名称", required = true) @RequestParam(required = true) String planName,
                               @ApiParam(value = "任务类型", required = true) @RequestParam(required = true) Integer taskType,
                               @ApiParam(value = "计划开始时间", required = true) @RequestParam(required = true) String planStartDate,
                               @ApiParam(value = "计划结束时间", required = true) @RequestParam(required = true) String planEndDate,
                               @ApiParam(value = "计划巡检人ID", required = true) @RequestParam(required = true) String userId,
                               @ApiParam(value = "任务内容", required = true) @RequestParam(required = true) String taskContent,
                               @ApiParam(value = "任务区域坐标", required = true) @RequestParam(required = true) String taskAreaShape,
                               @ApiParam(value = "任务区域名称", required = true) @RequestParam(required = true) String taskAreaName,
                               @ApiParam(value = "任务签到点", required = false) @RequestParam(required = false) String taskChecked) {
        LoginUser loginUser = UserUtil.getCurrentUser();
        String planId=DatesUtil.formatDate(new Date(),"yyyyMMddHHmmss")+String.valueOf((int) (Math.random()*(9999-1000)+1000));
        PlanManage record = new PlanManage();
        record.setUserId(userId);
        record.setPlanId(planId);
        record.setPlanName(planName);
        record.setTaskType(taskType);
        record.setTaskChecked(taskChecked);
        record.setPlanDate(new Date());
        record.setPlanStartDate(DatesUtil.parseDate(planStartDate,"yyyy-MM-dd"));
        record.setPlanEndDate(DatesUtil.parseDate(planEndDate,"yyyy-MM-dd"));
        record.setPlanStatus(0);
        record.setTaskContent(taskContent);
        record.setTaskAreaShape(taskAreaShape);
        record.setTaskAreaName(taskAreaName);
        record.setDateCreate(new Date());
        if(loginUser.getId() != null) {
            record.setPlanUserId(String.valueOf(loginUser.getId()));
            record.setCreateBy(String.valueOf(loginUser.getId()));
        }
        if(loginUser.getTenantId() != null) {
            record.setTenantId(loginUser.getTenantId());
        }
        int rows = planManageService.insertSelective(record);
        if(rows > 0) {
            return new AjaxMessage(ResultStatus.OK);
        }
        return new AjaxMessage(ResultStatus.ERROR);
    }

    @GetMapping("/plan/select")
    @ApiOperation(value = "查询巡检计划")
    public AjaxMessage<Pagination<PlanManageDto>> selectPlan(
            @ApiParam(value = "页数", required = true)@RequestParam Integer pageNum,
            @ApiParam(value = "条数", required = true)@RequestParam Integer pageSize,
            @ApiParam(value = "状态", required = false) @RequestParam(required = false) Integer planStatus,
            @ApiParam(value = "计划单号或计划名称", required = false) @RequestParam(required = false) String key,
            @ApiParam(value = "计划安排起始日期", required = false) @RequestParam(required = false) String startDate,
            @ApiParam(value = "计划安排截至日期", required = false) @RequestParam(required = false) String endDate){
        PlanManageDto planManageDto = new PlanManageDto();
        //根据用户编号，获取用户的权限
        LoginUser loginUser = UserUtil.getCurrentUser();
        planManageDto.setKey(key);
        if(!StringUtils.isEmpty(startDate)&& !StringUtils.isEmpty(endDate)) {
            planManageDto.setStartDate(startDate);
            planManageDto.setEndDate(endDate);
        }
        planManageDto.setPlanStatus(planStatus);
        planManageDto.setTenantId(loginUser.getTenantId());
        planManageDto.setProgramItems(loginUser.getProgramItemList());
        planManageDto.setUserType(loginUser.getType());
        //1是公司，2是公司及以下，3部门，4部门及以下，5自定义
        planManageDto.setPermissonType(loginUser.getPermissonType());
        IPage<PlanManageDto> iPage = new Page<>(pageNum, pageSize);
        iPage = planManageService.selectPage(iPage, planManageDto);
        Pagination<PlanManageDto> pages = new Pagination<>(iPage);
        for(PlanManage plan : pages.getList()){
            //plan.setCompanyOrgName(orgInfoUtil.getOrgName(plan.getCompanyOrgId()));
            //plan.setDeptOrgName(orgInfoUtil.getOrgName(plan.getDeptOrgId()));
        }
        return new AjaxMessage<>(ResultStatus.OK, pages);
    }
    /**
     * APP查询任务详情
     * @return Response对象
     */
    @RequestMapping(value = "selectPlanDetail", method = RequestMethod.GET)
    @ApiOperation(value = "APP查询任务详情")
    public AjaxMessage<PlanManageDto> selectPlanDetail(
            @ApiParam(value = "任务id", required = true) @RequestParam Integer flowId) {
        return new AjaxMessage<>(ResultStatus.OK,  planManageService.selectPlanDetail(flowId));
    }

    @GetMapping("/plan/submit")
    @ApiOperation(value = "提交巡检计划")
    public AjaxMessage submitPlan(
            @ApiParam(value = "计划单号", required = true) @RequestParam(required = false) Integer id){
        LoginUser loginUser = UserUtil.getCurrentUser();
        PlanManage planManage = planManageService.selectByPrimaryKey(id);
        //测试数据 test1 1 1
//        ProcessDefinition processDefinition = workFlowService.findProcessDefinition(
//                loginUser.getTenantId(),
//                loginUser.getCompanyId(),
//                planManage.getTaskType());
        ProcessDefinition processDefinition = workFlowService.findProcessDefinition(
                "test1",
                1,
                1);
        Map<String,Object> vars = new HashMap<>();
        vars.put("assineeFormUserId",planManage.getUserId());
        //巡检发送系统消息所需参数
        vars.put("任务类型",planManage.getTaskType());
        vars.put("任务编号",planManage.getPlanId());
        vars.put("templateId",4);
        vars.put("msgType",4);
        String startProcess = workFlowService.startProcess(processDefinition,vars);
        Task task = workFlowService.getProcessRuntimeTask(startProcess).get(0);
        Set<String> taskParticipator = workFlowService.getTaskParticipator(task.getId(),false,null);
        String currentUsers = null;
        for (String str : taskParticipator) {
            if(currentUsers == null){
                currentUsers = str;
            }
            else {
                currentUsers +=String.format(",%s",str);
            }
        }
        //更新计划表的流程相关字段值。
        planManage.setPlanStatus(1);//已执行
        if(planManage.getId() != null) {
            planManage.setUpdateBy(String.valueOf(loginUser.getId()));
        }
        planManage.setDateUpdate(new Date());
        planManage.setProcessInstanceId(task.getProcessInstanceId());
        planManage.setProcessDefId(task.getProcessDefinitionId());
        planManage.setCurrentTaskId(task.getId());
        planManage.setCurrentUsers(currentUsers);
        planManage.setCurrentTaskName(task.getName());
        int rows = planManageService.updateByPrimaryKeySelective(planManage);
        if(rows > 0) {
            return new AjaxMessage<>(ResultStatus.OK);
        }
        return new AjaxMessage<>(ResultStatus.ERROR);
    }

    @PostMapping("/plan/update")
    @ApiOperation(value = "修改巡检计划")
    public AjaxMessage updatePlan(@ApiParam(value = "修改巡检计划", required = true) @RequestBody PlanManage planManage) {
        PlanManage record = new PlanManage();
        record.setId(planManage.getId());
        record.setUserId(planManage.getUserId());
        record.setPlanName(planManage.getPlanName());
        record.setTaskType(planManage.getTaskType());
        record.setTaskChecked(planManage.getTaskChecked());
        record.setPlanStartDate(planManage.getPlanStartDate());
        record.setPlanEndDate(planManage.getPlanEndDate());
        record.setTaskContent(planManage.getTaskContent());
        record.setTaskAreaShape(planManage.getTaskAreaShape());
        record.setTaskAreaName(planManage.getTaskAreaName());
        LoginUser loginUser = UserUtil.getCurrentUser();
        if(loginUser.getId() != null) {
            record.setUpdateBy(String.valueOf(loginUser.getId()));
        }
        record.setDateUpdate(new Date());
        int rows = planManageService.updateByPrimaryKeySelective(record);
        if(rows > 0) {
            return new AjaxMessage(ResultStatus.OK);
        }
        return new AjaxMessage(ResultStatus.ERROR);
    }

    @RequestMapping(value="batchDelete" , method = RequestMethod.DELETE)
    @ApiOperation(value = "批量删除")
    public AjaxMessage<Integer> batchDelete( @ApiParam(value = "巡检计划ids") @RequestParam List<Integer> ids){
        return new AjaxMessage<>(ResultStatus.OK, planManageService.batchDelete(ids));
    }
}