|
@@ -6,13 +6,11 @@ import cn.iocoder.yudao.module.pms.controller.admin.iotdevicebom.vo.IotDeviceBom
|
|
|
import cn.iocoder.yudao.module.pms.controller.admin.iotdevicebom.vo.IotDeviceBomPageReqVO;
|
|
|
import cn.iocoder.yudao.module.pms.controller.admin.iotdevicebom.vo.IotDeviceBomSaveReqVO;
|
|
|
import cn.iocoder.yudao.module.pms.dal.dataobject.iotdevicebom.IotDeviceBomDO;
|
|
|
-import cn.iocoder.yudao.module.pms.dal.dataobject.iotmaterialgroup.IotMaterialGroupDO;
|
|
|
import cn.iocoder.yudao.module.pms.dal.mysql.iotdevicebom.IotDeviceBomMapper;
|
|
|
import org.springframework.stereotype.Service;
|
|
|
import org.springframework.validation.annotation.Validated;
|
|
|
|
|
|
import javax.annotation.Resource;
|
|
|
-
|
|
|
import java.util.Comparator;
|
|
|
import java.util.List;
|
|
|
|
|
@@ -36,6 +34,14 @@ public class IotDeviceBomServiceImpl implements IotDeviceBomService {
|
|
|
public Long createIotDeviceBom(IotDeviceBomSaveReqVO createReqVO) {
|
|
|
// 插入
|
|
|
IotDeviceBomDO iotDeviceBom = BeanUtils.toBean(createReqVO, IotDeviceBomDO.class);
|
|
|
+ // 设置当前节点为 叶子
|
|
|
+ iotDeviceBom.setLeafFlag(1);
|
|
|
+ // 设置父节点为 非叶子节点
|
|
|
+ if (!createReqVO.getParentId().equals(0L)) {
|
|
|
+ IotDeviceBomDO parentBomNode = iotDeviceBomMapper.selectParentNode(createReqVO);
|
|
|
+ parentBomNode.setLeafFlag(2);
|
|
|
+ iotDeviceBomMapper.updateById(parentBomNode);
|
|
|
+ }
|
|
|
iotDeviceBomMapper.insert(iotDeviceBom);
|
|
|
// 返回
|
|
|
return iotDeviceBom.getId();
|