optimize owner member

java110

2023-03-31 4e12aa79fa3affb0d0dc3e5f9d368814702f8d2b

 service-common/src/main/java/com/java110/common/cmd/machine/SaveMachineCmd.java

@@ -15,22 +15,31 @@
 */
package com.java110.common.cmd.machine;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.java110.core.annotation.Java110Cmd;
import com.java110.core.annotation.Java110Transactional;
import com.java110.core.context.ICmdDataFlowContext;
import com.java110.core.event.cmd.AbstractServiceCmdListener;
import com.java110.core.event.cmd.Cmd;
import com.java110.core.event.cmd.CmdEvent;
import com.java110.core.factory.GenerateCodeFactory;
import com.java110.core.log.LoggerFactory;
import com.java110.dto.community.CommunityLocationDto;
import com.java110.dto.machine.MachineDto;
import com.java110.intf.common.IMachineAttrInnerServiceSMO;
import com.java110.intf.common.IMachineV1InnerServiceSMO;
import com.java110.intf.community.ICommunityLocationV1InnerServiceSMO;
import com.java110.po.machine.MachineAttrPo;
import com.java110.po.machine.MachinePo;
import com.java110.utils.exception.CmdException;
import com.java110.utils.util.Assert;
import com.java110.utils.util.BeanConvertUtil;
import com.java110.utils.util.DateUtil;
import com.java110.vo.ResultVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.slf4j.Logger;
import com.java110.core.log.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.List;

/**
 * 类表述：保存
@@ -43,40 +52,90 @@
 * // modify by 张三 at 2021-09-12 第10行在某种场景下存在某种bug 需要修复，注释10至20行 加入 20行至30行
 */
@Java110Cmd(serviceCode = "machine.saveMachine")
public class SaveMachineCmd extends AbstractServiceCmdListener {
public class SaveMachineCmd extends Cmd {

    private static Logger logger = LoggerFactory.getLogger( SaveMachineCmd.class );
    private static Logger logger = LoggerFactory.getLogger(SaveMachineCmd.class);

    public static final String CODE_PREFIX_ID = "10";

    @Autowired
    private IMachineV1InnerServiceSMO machineV1InnerServiceSMOImpl;

    @Autowired
    private ICommunityLocationV1InnerServiceSMO communityLocationV1InnerServiceSMOImpl;

    @Autowired
    private IMachineAttrInnerServiceSMO machineAttrInnerServiceSMOImpl;

    @Override
    public void validate(CmdEvent event, ICmdDataFlowContext cmdDataFlowContext, JSONObject reqJson) {
        Assert.hasKeyAndValue( reqJson, "machineCode", "请求报文中未包含machineCode" );
        Assert.hasKeyAndValue( reqJson, "machineVersion", "请求报文中未包含machineVersion" );
        Assert.hasKeyAndValue( reqJson, "machineTypeCd", "请求报文中未包含machineTypeCd" );
        Assert.hasKeyAndValue( reqJson, "communityId", "请求报文中未包含communityId" );
        Assert.hasKeyAndValue( reqJson, "machineName", "请求报文中未包含machineName" );
        Assert.hasKeyAndValue( reqJson, "authCode", "请求报文中未包含authCode" );
        Assert.hasKeyAndValue( reqJson, "direction", "请求报文中未包含direction" );
        Assert.hasKeyAndValue( reqJson, "typeId", "请求报文中未包含typeId" );

        Assert.hasKeyAndValue(reqJson, "machineCode", "必填，请填写设备编码");
        Assert.hasKeyAndValue(reqJson, "machineVersion", "必填，请填写设备版本号");
        Assert.hasKeyAndValue(reqJson, "machineName", "必填，请填写设备名称");
        Assert.hasKeyAndValue(reqJson, "machineTypeCd", "必填，请选择设备类型");
        Assert.hasKeyAndValue(reqJson, "direction", "必填，请选择设备方向");
        Assert.hasKeyAndValue(reqJson, "authCode", "必填，请填写鉴权编码");
        Assert.hasKeyAndValue(reqJson, "locationTypeCd", "必填，请选择位置类型");
        MachineDto machineDto = new MachineDto();
        machineDto.setMachineCode(reqJson.getString("machineCode"));
        int count = machineV1InnerServiceSMOImpl.queryMachinesCount(machineDto);
        if (count > 0) {
            throw new CmdException("设备已存在");
        }
        //属性校验
        Assert.judgeAttrValue(reqJson);
    }

    @Override
    @Java110Transactional
    public void doCmd(CmdEvent event, ICmdDataFlowContext cmdDataFlowContext, JSONObject reqJson) throws CmdException {

        MachinePo machinePo = BeanConvertUtil.covertBean( reqJson, MachinePo.class );
        machinePo.setMachineId( GenerateCodeFactory.getGeneratorId( CODE_PREFIX_ID ) );
        int flag = machineV1InnerServiceSMOImpl.saveMachine( machinePo );

        if (flag < 1) {
            throw new CmdException( "保存数据失败" );
        MachinePo machinePo = BeanConvertUtil.covertBean(reqJson, MachinePo.class);
        if (!MachineDto.MACHINE_TYPE_MONITOR.equals(reqJson.getString("machineTypeCd"))
            && !MachineDto.MACHINE_TYPE_ATTENDANCE.equals(reqJson.getString("machineTypeCd"))
        ) {
            CommunityLocationDto communityLocationDto = new CommunityLocationDto();
            communityLocationDto.setCommunityId(reqJson.getString("communityId"));
            communityLocationDto.setLocationId(reqJson.getString("locationTypeCd"));
            List<CommunityLocationDto> locationDtos = communityLocationV1InnerServiceSMOImpl.queryCommunityLocations(communityLocationDto);
            Assert.listOnlyOne(locationDtos, "位置不存在");
            machinePo.setLocationObjId(locationDtos.get(0).getLocationObjId());
        } else if(MachineDto.MACHINE_TYPE_ATTENDANCE.equals(reqJson.getString("machineTypeCd"))){
            machinePo.setLocationObjId(reqJson.getString("locationTypeCd"));
            machinePo.setLocationTypeCd(reqJson.getString("locationTypeCd"));
        }else {
            machinePo.setLocationObjId("-1");
            machinePo.setLocationTypeCd("-1");
        }
        machinePo.setMachineId(GenerateCodeFactory.getGeneratorId(CODE_PREFIX_ID));
        machinePo.setHeartbeatTime(DateUtil.getNow(DateUtil.DATE_FORMATE_STRING_A));
        machinePo.setState(MachineDto.MACHINE_STATE_ON);
        int flag = machineV1InnerServiceSMOImpl.saveMachine(machinePo);
        if (flag < 1) {
            throw new CmdException("保存数据失败");
        }
        dealMachineAttr(reqJson, machinePo);
        cmdDataFlowContext.setResponseEntity(ResultVo.success());
    }

        cmdDataFlowContext.setResponseEntity( ResultVo.success() );
    private void dealMachineAttr(JSONObject paramObj, MachinePo machinePo) {
        if (!paramObj.containsKey("attrs")) {
            return;
        }
        JSONArray attrs = paramObj.getJSONArray("attrs");
        if (attrs.size() < 1) {
            return;
        }
        MachineAttrPo attr = null;
        int flag = 0;
        for (int attrIndex = 0; attrIndex < attrs.size(); attrIndex++) {
            attr = BeanConvertUtil.covertBean(attrs.getJSONObject(attrIndex), MachineAttrPo.class);
            attr.setCommunityId(paramObj.getString("communityId"));
            attr.setMachineId(machinePo.getMachineId());
            attr.setAttrId(GenerateCodeFactory.getGeneratorId(CODE_PREFIX_ID));
            flag = machineAttrInnerServiceSMOImpl.saveMachineAttrs(attr);
            if (flag < 1) {
                throw new CmdException("保存数据失败");
            }
        }
    }
}