fjhx-cloud/hx-common/service-flow/src/main/java/com/fjhx/feign/FlowClient.java

package com.fjhx.feign;

import cn.hutool.core.util.ObjectUtil;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.fjhx.base.BaseEntity;
import com.fjhx.constants.FlowConstant;
import com.fjhx.constants.FlowExplainConstant;
import com.fjhx.constants.StatusConstant;
import com.fjhx.entity.example.ExampleDetails;
import com.fjhx.entity.example.ExampleInfo;
import com.fjhx.entity.process.ProcessInfo;
import com.fjhx.entity.process.ProcessNode;
import com.fjhx.entity.process.ProcessNodeButton;
import com.fjhx.entity.process.ProcessTenant;
import com.fjhx.enums.ButtonNameEnum;
import com.fjhx.enums.HandleResultEnum;
import com.fjhx.enums.ProcessNodeHandleObjectTypeEnum;
import com.fjhx.params.ExampleResult;
import com.fjhx.params.JumpVo;
import com.fjhx.service.example.ExampleDetailsService;
import com.fjhx.service.example.ExampleInfoService;
import com.fjhx.service.process.ProcessInfoService;
import com.fjhx.service.process.ProcessNodeButtonService;
import com.fjhx.service.process.ProcessNodeService;
import com.fjhx.service.process.ProcessTenantService;
import com.fjhx.utils.Assert;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.core.tool.api.R;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

@RestController
public class FlowClient implements IFlowClient {

    @Autowired
    private ExampleInfoService exampleInfoService;

    @Autowired
    private ProcessInfoService processInfoService;

    @Autowired
    private ProcessTenantService processTenantService;

    @Autowired
    private ProcessNodeService processNodeService;

    @Autowired
    private ProcessNodeButtonService processNodeButtonService;

    @Autowired
    private ExampleDetailsService exampleDetailsService;

    @Transactional(rollbackFor = Exception.class)
    @GetMapping(CREATE)
    @Override
    public R<ExampleResult> create(Long flowLinkNo, String code, String nodeCode, String title, String remarks, String cacheData) {

        ExampleResult exampleResult = new ExampleResult();

        Long count = exampleInfoService.count(Wrappers.<ExampleInfo>lambdaQuery()
                .eq(ExampleInfo::getFlowLinkNo, flowLinkNo)
                .eq(ExampleInfo::getComplete, StatusConstant.NO));

        Assert.eqZero(count, FlowExplainConstant.EXPLAIN_NOT_EXIST);

        // 查找通用流程
        ProcessInfo processInfo = processInfoService.getOne(ProcessInfo::getCode, code);
        Assert.notEmpty(processInfo, String.format(FlowExplainConstant.PROCESS_NOT_EXIST, code));

        // 获取租户流程
        ProcessTenant processTenant = processTenantService.getOne(Wrappers.<ProcessTenant>lambdaQuery()
                .eq(ProcessTenant::getProcessInfoId, processInfo.getId())
                .eq(ProcessTenant::getBindingTenantId, AuthUtil.getTenantId())
                .eq(ProcessTenant::getCurrentVersion, StatusConstant.YES));

        // 创建流程实例
        ExampleInfo exampleInfo = new ExampleInfo();
        exampleInfo.setProcessInfoId(processInfo.getId());
        exampleInfo.setTitle(title);
        exampleInfo.setFlowLinkNo(flowLinkNo);
        exampleInfo.setCacheData(cacheData);

        // 走默认流程，直接完成
        if (processTenant == null) {
            exampleInfo.setComplete(StatusConstant.YES);
            exampleInfo.setProcessNodeId(FlowConstant.OVER_PROCESS_FLAG);
            exampleInfo.setProcessNodeCode(FlowConstant.END_CODE);
            exampleInfo.setFlowType(FlowConstant.FLOW_TYPE_DEFAULT);
            exampleInfo.setHandleResult(HandleResultEnum.SUCCESS.getType());
            exampleInfoService.save(exampleInfo);

            // 开始流程明细
            ExampleDetails startDetails = new ExampleDetails();
            startDetails.setExampleInfoId(exampleInfo.getId());
            startDetails.setProcessInfoId(processInfo.getId());
            startDetails.setNameType(ButtonNameEnum.START.getType());
            startDetails.setHandleUserId(AuthUtil.getUserId());
            startDetails.setRemarks(remarks);

            // 结束流程明细
            ExampleDetails endDetails = new ExampleDetails();
            endDetails.setExampleInfoId(exampleInfo.getId());
            endDetails.setProcessInfoId(processInfo.getId());
            endDetails.setNameType(ButtonNameEnum.END.getType());

            // 封装返回值
            exampleResult.setOldExampleInfo(exampleInfo);
            exampleResult.setExampleInfo(exampleInfo);
            exampleResult.setExampleDetailsIdList(Arrays.asList(startDetails.getId(), endDetails.getId()));
            exampleResult.setHandleType(ButtonNameEnum.END.getType());
            exampleResult.setCacheDataStr(cacheData);
            return R.data(exampleResult);
        }

        // 查询节点
        ProcessNodeButton processNodeButton = processNodeButtonService.getOne(Wrappers.<ProcessNodeButton>lambdaQuery()
                .eq(ProcessNodeButton::getProcessTenantId, processTenant.getId())
                .eq(ProcessNodeButton::getProcessNodeCode, FlowConstant.START_CODE));
        Assert.notEmpty(processNodeButton, FlowExplainConstant.SPECIFY_BRANCH_EMPTY);

        exampleInfo.setFlowType(FlowConstant.FLOW_TYPE_TENANT);
        exampleInfo.setProcessTenantId(processTenant.getId());
        exampleInfo.setProcessNodeId(processNodeButton.getProcessNodeId());
        exampleInfo.setProcessNodeCode(processNodeButton.getProcessNodeCode());
        exampleInfo.setComplete(StatusConstant.NO);
        exampleInfo.setHandleResult(HandleResultEnum.IN_PROGRESS.getType());
        exampleInfoService.save(exampleInfo);

        JumpVo jumpVo = new JumpVo();
        setJumpNode(jumpVo, processNodeButton, nodeCode);
        exampleResult.setJumpVo(jumpVo);

        return R.data(exampleResult);
    }

    @GetMapping(GET_JUMP_VO)
    @Override
    public R<JumpVo> getJumpVo(Long buttonId, String nodeCode) {
        JumpVo jumpVo = new JumpVo();

        ProcessNodeButton processNodeButton = processNodeButtonService.getById(buttonId);
        Assert.notEmpty(processNodeButton, "nodeButton is null");

        setJumpNode(jumpVo, processNodeButton, nodeCode);

        return R.data(jumpVo);
    }


    @Transactional(rollbackFor = Exception.class)
    @PostMapping(JUMP)
    @Override
    public R<ExampleResult> jump(JumpVo jumpVo) {

        // 返回值
        ExampleResult exampleResult = new ExampleResult();

        // 查询流程实例
        ExampleInfo oleExampleInfo = exampleInfoService.getOne(Wrappers.<ExampleInfo>lambdaQuery()
                .eq(ExampleInfo::getFlowLinkNo, jumpVo.getFlowLinkNo())
                .eq(ExampleInfo::getComplete, StatusConstant.NO));
        Assert.notEmpty(oleExampleInfo, FlowExplainConstant.EXAMPLE_INFO_NULL);

        // 复制除一个对象做修改
        ExampleInfo exampleInfo = ObjectUtil.clone(oleExampleInfo);

        String cacheDataStr = jumpVo.getCacheDataStr();
        if (ObjectUtil.isNotEmpty(cacheDataStr)) {
            exampleInfo.setCacheData(cacheDataStr);
        }

        // 明细
        ExampleDetails exampleDetails = createExampleDetails(exampleInfo, jumpVo.getNameType());
        exampleDetails.setProcessNodeButtonId(jumpVo.getButtonId());
        exampleDetails.setHandleUserId(AuthUtil.getUserId());
        exampleDetails.setRemarks(jumpVo.getRemarks());

        // 赋值按钮处理方法
        exampleResult.setHandleType(jumpVo.getNameType());

        // 驳回
        if (ButtonNameEnum.REJECT.getType().equals(jumpVo.getNameType())) {
            exampleInfo.setHandleResult(HandleResultEnum.FAIL.getType());
        }
        // 通过
        else if (FlowConstant.END_CODE.equals(jumpVo.getJumpNodeCode())) {
            exampleInfo.setHandleResult(HandleResultEnum.SUCCESS.getType());
            // 赋值结束处理方法
            exampleResult.setHandleType(ButtonNameEnum.END.getType());
        }

        exampleInfo.setProcessNodeId(jumpVo.getJumpNodeId());
        exampleInfo.setProcessNodeCode(jumpVo.getJumpNodeCode());
        exampleInfo.setHandleObjectType(jumpVo.getHandleObjectType());
        exampleInfo.setHandleObjectIdSet(jumpVo.getHandleObjectIdSet());

        List<Long> exampleDetailsIdList = new ArrayList<>();

        // 流程结束
        if (FlowConstant.END_CODE.equals(jumpVo.getJumpNodeCode())) {
            // 赋值流程完成
            exampleInfo.setComplete(StatusConstant.YES);

            // 流程明细
            ExampleDetails endExampleDetails = createExampleDetails(exampleInfo, ButtonNameEnum.END.getType());
            exampleDetailsService.saveBatch(Arrays.asList(exampleDetails, endExampleDetails));
            exampleDetailsIdList.add(exampleDetails.getId());
            exampleDetailsIdList.add(endExampleDetails.getId());
        }
        // 正常跳转
        else {
            // 流程明细
            exampleDetailsService.save(exampleDetails);
            exampleDetailsIdList.add(exampleDetails.getId());

            // 如果跳转节点到发起（一直回退到发起人），处理人只能是发起人
            if (FlowConstant.START_CODE.equals(jumpVo.getJumpNodeCode())) {
                exampleInfo.setHandleObjectType(ProcessNodeHandleObjectTypeEnum.USER.getType());
                exampleInfo.setHandleObjectIdSet(exampleInfo.getCreateUser().toString());
            }
        }

        exampleInfoService.updateById(exampleInfo);

        exampleResult.setOldExampleInfo(oleExampleInfo);
        exampleResult.setExampleInfo(exampleInfo);
        exampleResult.setCacheDataStr(exampleInfo.getCacheData());
        exampleResult.setExampleDetailsIdList(exampleDetailsIdList);

        return R.data(exampleResult);
    }

    @PostMapping(ROLL_BACK)
    @Override
    public void rollBack(ExampleResult result) {
        ExampleInfo oldExampleInfo = result.getOldExampleInfo();

        if (result.isDeleteExample()) {
            exampleInfoService.update(Wrappers.<ExampleInfo>lambdaUpdate()
                    .eq(BaseEntity::getId, oldExampleInfo.getId())
                    .set(ExampleInfo::getDelFlag, 1)
                    .set(ExampleInfo::getRemarks, "业务异常,本次流程实例已删除")
            );
        } else {
            exampleInfoService.updateById(oldExampleInfo);
        }

        exampleDetailsService.update(Wrappers.<ExampleDetails>lambdaUpdate()
                .in(BaseEntity::getId, result.getExampleDetailsIdList())
                .set(ExampleDetails::getDelFlag, 1)
                .set(ExampleDetails::getRemarks, "业务异常,本次流程明细已删除")
        );
    }


    private void setJumpNode(JumpVo jumpVo, ProcessNodeButton processNodeButton, String nodeCode) {
        // 赋值按钮名称类型
        jumpVo.setNameType(processNodeButton.getNameType());
        jumpVo.setButtonId(processNodeButton.getId());

        ProcessNode processNode;

        // 非分支跳转
        if (ObjectUtil.isNotEmpty(processNodeButton.getJumpNodeId())) {
            processNode = processNodeService.getById(processNodeButton.getJumpNodeId());
        }
        // 分支跳转
        else {
            // 跳转节点编码不为空
            Assert.notEmpty(nodeCode, FlowExplainConstant.SPECIFY_BRANCH_EMPTY);

            // 查询跳转到哪个分支节点
            processNode = processNodeService.getOne(Wrappers.<ProcessNode>lambdaQuery()
                    .eq(ProcessNode::getParentId, processNodeButton.getProcessNodeId())
                    .eq(ProcessNode::getCode, nodeCode));

            // 分支节点不为空
            Assert.notEmpty(processNode, FlowExplainConstant.SPECIFY_BRANCH_EMPTY);
        }

        jumpVo.setJumpNodeId(processNode.getId());
        jumpVo.setJumpNodeCode(processNode.getCode());
        jumpVo.setHandleObjectType(processNode.getHandleObjectType());
        jumpVo.setHandleObjectIdSet(processNode.getHandleObjectIdSet());

    }

    /**
     * 创建流程实例明细
     */
    private ExampleDetails createExampleDetails(ExampleInfo exampleInfo, Integer nameType) {
        ExampleDetails exampleDetails = new ExampleDetails();
        exampleDetails.setExampleInfoId(exampleInfo.getId());
        exampleDetails.setProcessInfoId(exampleInfo.getProcessInfoId());
        exampleDetails.setProcessTenantId(exampleInfo.getProcessTenantId());
        exampleDetails.setProcessNodeId(exampleInfo.getProcessNodeId());
        exampleDetails.setNameType(nameType);
        return exampleDetails;
    }

}