optimize owner member

java110

2023-03-31 4e12aa79fa3affb0d0dc3e5f9d368814702f8d2b

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

@@ -24,16 +24,22 @@
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.slf4j.Logger;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.List;

/**
 * 类表述：保存
@@ -48,12 +54,15 @@
@Java110Cmd(serviceCode = "machine.saveMachine")
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;
@@ -67,53 +76,66 @@
        Assert.hasKeyAndValue(reqJson, "direction", "必填，请选择设备方向");
        Assert.hasKeyAndValue(reqJson, "authCode", "必填，请填写鉴权编码");
        Assert.hasKeyAndValue(reqJson, "locationTypeCd", "必填，请选择位置类型");
        Assert.hasKeyAndValue(reqJson, "locationObjId", "必填，请填写位置对象ID");

        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");
        }

        dealMachineAttr(reqJson);

        cmdDataFlowContext.setResponseEntity( ResultVo.success() );
        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());
    }

    private void dealMachineAttr(JSONObject paramObj) {

    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(paramObj.getString("machineId"));
            attr.setAttrId(GenerateCodeFactory.getGeneratorId(CODE_PREFIX_ID ));
            attr.setMachineId(machinePo.getMachineId());
            attr.setAttrId(GenerateCodeFactory.getGeneratorId(CODE_PREFIX_ID));
            flag = machineAttrInnerServiceSMOImpl.saveMachineAttrs(attr);
            if (flag < 1) {
                throw new CmdException( "保存数据失败" );
                throw new CmdException("保存数据失败");
            }
        }

    }
}