pms树

yudao-module-pms/yudao-module-pms-api/src/main/java/cn/iocoder/yudao/module/pms/enums/ErrorCodeConstant.java

@@ -17,4 +17,5 @@ public interface ErrorCodeConstant{
     ErrorCode IOT_DEVICE_MATERIAL_NOT_EXISTS = new ErrorCode(108, "设备BOM物料关联不存在");
     ErrorCode IOT_DEVICE_BOM_NOT_EXISTS = new ErrorCode(109, "设备BOM 关系不存在");
     ErrorCode IOT_INFO_NOT_EXISTS = new ErrorCode( 122, "资料数据不存在");
+    ErrorCode IOT_TREE_NOT_EXISTS = new ErrorCode(122, "pms树不存在");
 }

yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/IotTreeController.java

@@ -0,0 +1,97 @@
+package cn.iocoder.yudao.module.pms.controller.admin;
+
+import cn.iocoder.yudao.module.pms.controller.admin.vo.IotTreePageReqVO;
+import cn.iocoder.yudao.module.pms.controller.admin.vo.IotTreeRespVO;
+import cn.iocoder.yudao.module.pms.controller.admin.vo.IotTreeSaveReqVO;
+import cn.iocoder.yudao.module.pms.dal.dataobject.IotTreeDO;
+import cn.iocoder.yudao.module.pms.service.IotTreeService;
+import org.springframework.web.bind.annotation.*;
+import javax.annotation.Resource;
+import org.springframework.validation.annotation.Validated;
+import org.springframework.security.access.prepost.PreAuthorize;
+import io.swagger.v3.oas.annotations.tags.Tag;
+import io.swagger.v3.oas.annotations.Parameter;
+import io.swagger.v3.oas.annotations.Operation;
+
+import javax.validation.constraints.*;
+import javax.validation.*;
+import javax.servlet.http.*;
+import java.util.*;
+import java.io.IOException;
+
+import cn.iocoder.yudao.framework.common.pojo.PageParam;
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.common.pojo.CommonResult;
+import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
+import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
+
+import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
+
+import cn.iocoder.yudao.framework.apilog.core.annotation.ApiAccessLog;
+import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.*;
+
+
+@Tag(name = "管理后台 - pms树")
+@RestController
+@RequestMapping("/rq/iot-tree")
+@Validated
+public class IotTreeController {
+
+    @Resource
+    private IotTreeService iotTreeService;
+
+    @PostMapping("/create")
+    @Operation(summary = "创建pms树")
+    @PreAuthorize("@ss.hasPermission('rq:iot-tree:create')")
+    public CommonResult<Long> createIotTree(@Valid @RequestBody IotTreeSaveReqVO createReqVO) {
+        return success(iotTreeService.createIotTree(createReqVO));
+    }
+
+    @PutMapping("/update")
+    @Operation(summary = "更新pms树")
+    @PreAuthorize("@ss.hasPermission('rq:iot-tree:update')")
+    public CommonResult<Boolean> updateIotTree(@Valid @RequestBody IotTreeSaveReqVO updateReqVO) {
+        iotTreeService.updateIotTree(updateReqVO);
+        return success(true);
+    }
+
+    @DeleteMapping("/delete")
+    @Operation(summary = "删除pms树")
+    @Parameter(name = "id", description = "编号", required = true)
+    @PreAuthorize("@ss.hasPermission('rq:iot-tree:delete')")
+    public CommonResult<Boolean> deleteIotTree(@RequestParam("id") Long id) {
+        iotTreeService.deleteIotTree(id);
+        return success(true);
+    }
+
+    @GetMapping("/get")
+    @Operation(summary = "获得pms树")
+    @Parameter(name = "id", description = "编号", required = true, example = "1024")
+    @PreAuthorize("@ss.hasPermission('rq:iot-tree:query')")
+    public CommonResult<IotTreeRespVO> getIotTree(@RequestParam("id") Long id) {
+        IotTreeDO iotTree = iotTreeService.getIotTree(id);
+        return success(BeanUtils.toBean(iotTree, IotTreeRespVO.class));
+    }
+
+    @GetMapping("/page")
+    @Operation(summary = "获得pms树分页")
+    @PreAuthorize("@ss.hasPermission('rq:iot-tree:query')")
+    public CommonResult<PageResult<IotTreeRespVO>> getIotTreePage(@Valid IotTreePageReqVO pageReqVO) {
+        PageResult<IotTreeDO> pageResult = iotTreeService.getIotTreePage(pageReqVO);
+        return success(BeanUtils.toBean(pageResult, IotTreeRespVO.class));
+    }
+
+    @GetMapping("/export-excel")
+    @Operation(summary = "导出pms树 Excel")
+    @PreAuthorize("@ss.hasPermission('rq:iot-tree:export')")
+    @ApiAccessLog(operateType = EXPORT)
+    public void exportIotTreeExcel(@Valid IotTreePageReqVO pageReqVO,
+              HttpServletResponse response) throws IOException {
+        pageReqVO.setPageSize(PageParam.PAGE_SIZE_NONE);
+        List<IotTreeDO> list = iotTreeService.getIotTreePage(pageReqVO).getList();
+        // 导出 Excel
+        ExcelUtils.write(response, "pms树.xls", "数据", IotTreeRespVO.class,
+                        BeanUtils.toBean(list, IotTreeRespVO.class));
+    }
+
+}

yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/vo/IotTreePageReqVO.java

@@ -0,0 +1,43 @@
+package cn.iocoder.yudao.module.pms.controller.admin.vo;
+
+import lombok.*;
+import java.util.*;
+import io.swagger.v3.oas.annotations.media.Schema;
+import cn.iocoder.yudao.framework.common.pojo.PageParam;
+import org.springframework.format.annotation.DateTimeFormat;
+import java.time.LocalDateTime;
+
+import static cn.iocoder.yudao.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND;
+
+@Schema(description = "管理后台 - pms树分页 Request VO")
+@Data
+@EqualsAndHashCode(callSuper = true)
+@ToString(callSuper = true)
+public class IotTreePageReqVO extends PageParam {
+
+    @Schema(description = "原id", example = "19643")
+    private Long originId;
+
+    @Schema(description = "父分类id", example = "3195")
+    private Long parentId;
+
+    @Schema(description = "分类名称", example = "赵六")
+    private String name;
+
+    @Schema(description = "类型", example = "2")
+    private String type;
+
+    @Schema(description = "分类排序")
+    private Integer sort;
+
+    @Schema(description = "开启状态", example = "1")
+    private Integer status;
+
+    @Schema(description = "创建时间")
+    @DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND)
+    private LocalDateTime[] createTime;
+
+    @Schema(description = "备注", example = "随便")
+    private String remark;
+
+}

yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/vo/IotTreeRespVO.java

@@ -0,0 +1,51 @@
+package cn.iocoder.yudao.module.pms.controller.admin.vo;
+
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.*;
+import java.util.*;
+import org.springframework.format.annotation.DateTimeFormat;
+import java.time.LocalDateTime;
+import com.alibaba.excel.annotation.*;
+
+@Schema(description = "管理后台 - pms树 Response VO")
+@Data
+@ExcelIgnoreUnannotated
+public class IotTreeRespVO {
+
+    @Schema(description = "主键id", requiredMode = Schema.RequiredMode.REQUIRED, example = "17585")
+    @ExcelProperty("主键id")
+    private Long id;
+
+    @Schema(description = "原id", requiredMode = Schema.RequiredMode.REQUIRED, example = "19643")
+    @ExcelProperty("原id")
+    private Long originId;
+
+    @Schema(description = "父分类id", requiredMode = Schema.RequiredMode.REQUIRED, example = "3195")
+    @ExcelProperty("父分类id")
+    private Long parentId;
+
+    @Schema(description = "分类名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "赵六")
+    @ExcelProperty("分类名称")
+    private String name;
+
+    @Schema(description = "类型", requiredMode = Schema.RequiredMode.REQUIRED, example = "2")
+    @ExcelProperty("类型")
+    private String type;
+
+    @Schema(description = "分类排序")
+    @ExcelProperty("分类排序")
+    private Integer sort;
+
+    @Schema(description = "开启状态", example = "1")
+    @ExcelProperty("开启状态")
+    private Integer status;
+
+    @Schema(description = "创建时间")
+    @ExcelProperty("创建时间")
+    private LocalDateTime createTime;
+
+    @Schema(description = "备注", example = "随便")
+    @ExcelProperty("备注")
+    private String remark;
+
+}

yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/vo/IotTreeSaveReqVO.java

@@ -0,0 +1,40 @@
+package cn.iocoder.yudao.module.pms.controller.admin.vo;
+
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.*;
+import java.util.*;
+import javax.validation.constraints.*;
+
+@Schema(description = "管理后台 - pms树新增/修改 Request VO")
+@Data
+public class IotTreeSaveReqVO {
+
+    @Schema(description = "主键id", requiredMode = Schema.RequiredMode.REQUIRED, example = "17585")
+    private Long id;
+
+    @Schema(description = "原id", requiredMode = Schema.RequiredMode.REQUIRED, example = "19643")
+    @NotNull(message = "原id不能为空")
+    private Long originId;
+
+    @Schema(description = "父分类id", requiredMode = Schema.RequiredMode.REQUIRED, example = "3195")
+    @NotNull(message = "父分类id不能为空")
+    private Long parentId;
+
+    @Schema(description = "分类名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "赵六")
+    @NotEmpty(message = "分类名称不能为空")
+    private String name;
+
+    @Schema(description = "类型", requiredMode = Schema.RequiredMode.REQUIRED, example = "2")
+    @NotEmpty(message = "类型不能为空")
+    private String type;
+
+    @Schema(description = "分类排序")
+    private Integer sort;
+
+    @Schema(description = "开启状态", example = "1")
+    private Integer status;
+
+    @Schema(description = "备注", example = "随便")
+    private String remark;
+
+}

yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/dal/dataobject/IotTreeDO.java

@@ -0,0 +1,59 @@
+package cn.iocoder.yudao.module.pms.dal.dataobject;
+
+import lombok.*;
+import java.util.*;
+import java.time.LocalDateTime;
+import java.time.LocalDateTime;
+import com.baomidou.mybatisplus.annotation.*;
+import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
+
+/**
+ * pms树 DO
+ *
+ * @author 芋道源码
+ */
+@TableName("rq_iot_tree")
+@KeySequence("rq_iot_tree_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
+@Data
+@EqualsAndHashCode(callSuper = true)
+@ToString(callSuper = true)
+@Builder
+@NoArgsConstructor
+@AllArgsConstructor
+public class IotTreeDO extends BaseDO {
+
+    /**
+     * 主键id
+     */
+    @TableId
+    private Long id;
+    /**
+     * 原id
+     */
+    private Long originId;
+    /**
+     * 父分类id
+     */
+    private Long parentId;
+    /**
+     * 分类名称
+     */
+    private String name;
+    /**
+     * 类型
+     */
+    private String type;
+    /**
+     * 分类排序
+     */
+    private Integer sort;
+    /**
+     * 开启状态
+     */
+    private Integer status;
+    /**
+     * 备注
+     */
+    private String remark;
+
+}

yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/dal/mysql/IotTreeMapper.java

@@ -0,0 +1,33 @@
+package cn.iocoder.yudao.module.pms.dal.mysql;
+
+import java.util.*;
+
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
+import cn.iocoder.yudao.module.pms.controller.admin.vo.IotTreePageReqVO;
+import cn.iocoder.yudao.module.pms.dal.dataobject.IotTreeDO;
+import org.apache.ibatis.annotations.Mapper;
+
+/**
+ * pms树 Mapper
+ *
+ * @author 芋道源码
+ */
+@Mapper
+public interface IotTreeMapper extends BaseMapperX<IotTreeDO> {
+
+    default PageResult<IotTreeDO> selectPage(IotTreePageReqVO reqVO) {
+        return selectPage(reqVO, new LambdaQueryWrapperX<IotTreeDO>()
+                .eqIfPresent(IotTreeDO::getOriginId, reqVO.getOriginId())
+                .eqIfPresent(IotTreeDO::getParentId, reqVO.getParentId())
+                .likeIfPresent(IotTreeDO::getName, reqVO.getName())
+                .eqIfPresent(IotTreeDO::getType, reqVO.getType())
+                .eqIfPresent(IotTreeDO::getSort, reqVO.getSort())
+                .eqIfPresent(IotTreeDO::getStatus, reqVO.getStatus())
+                .betweenIfPresent(IotTreeDO::getCreateTime, reqVO.getCreateTime())
+                .eqIfPresent(IotTreeDO::getRemark, reqVO.getRemark())
+                .orderByDesc(IotTreeDO::getId));
+    }
+
+}

yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/service/IotTreeService.java

@@ -0,0 +1,56 @@
+package cn.iocoder.yudao.module.pms.service;
+
+import java.util.*;
+import javax.validation.*;
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.common.pojo.PageParam;
+import cn.iocoder.yudao.module.pms.controller.admin.vo.IotTreePageReqVO;
+import cn.iocoder.yudao.module.pms.controller.admin.vo.IotTreeSaveReqVO;
+import cn.iocoder.yudao.module.pms.dal.dataobject.IotTreeDO;
+
+/**
+ * pms树 Service 接口
+ *
+ * @author 芋道源码
+ */
+public interface IotTreeService {
+
+    /**
+     * 创建pms树
+     *
+     * @param createReqVO 创建信息
+     * @return 编号
+     */
+    Long createIotTree(@Valid IotTreeSaveReqVO createReqVO);
+
+    /**
+     * 更新pms树
+     *
+     * @param updateReqVO 更新信息
+     */
+    void updateIotTree(@Valid IotTreeSaveReqVO updateReqVO);
+
+    /**
+     * 删除pms树
+     *
+     * @param id 编号
+     */
+    void deleteIotTree(Long id);
+
+    /**
+     * 获得pms树
+     *
+     * @param id 编号
+     * @return pms树
+     */
+    IotTreeDO getIotTree(Long id);
+
+    /**
+     * 获得pms树分页
+     *
+     * @param pageReqVO 分页查询
+     * @return pms树分页
+     */
+    PageResult<IotTreeDO> getIotTreePage(IotTreePageReqVO pageReqVO);
+
+}

yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/service/IotTreeServiceImpl.java

@@ -0,0 +1,75 @@
+package cn.iocoder.yudao.module.pms.service;
+
+import cn.iocoder.yudao.module.pms.controller.admin.vo.IotTreePageReqVO;
+import cn.iocoder.yudao.module.pms.controller.admin.vo.IotTreeSaveReqVO;
+import cn.iocoder.yudao.module.pms.dal.dataobject.IotTreeDO;
+import cn.iocoder.yudao.module.pms.dal.mysql.IotTreeMapper;
+import org.springframework.stereotype.Service;
+import javax.annotation.Resource;
+import org.springframework.validation.annotation.Validated;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.util.*;
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.common.pojo.PageParam;
+import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
+
+
+import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
+import static cn.iocoder.yudao.module.pms.enums.ErrorCodeConstant.IOT_TREE_NOT_EXISTS;
+
+/**
+ * pms树 Service 实现类
+ *
+ * @author 芋道源码
+ */
+@Service
+@Validated
+public class IotTreeServiceImpl implements IotTreeService {
+
+    @Resource
+    private IotTreeMapper iotTreeMapper;
+
+    @Override
+    public Long createIotTree(IotTreeSaveReqVO createReqVO) {
+        // 插入
+        IotTreeDO iotTree = BeanUtils.toBean(createReqVO, IotTreeDO.class);
+        iotTreeMapper.insert(iotTree);
+        // 返回
+        return iotTree.getId();
+    }
+
+    @Override
+    public void updateIotTree(IotTreeSaveReqVO updateReqVO) {
+        // 校验存在
+        validateIotTreeExists(updateReqVO.getId());
+        // 更新
+        IotTreeDO updateObj = BeanUtils.toBean(updateReqVO, IotTreeDO.class);
+        iotTreeMapper.updateById(updateObj);
+    }
+
+    @Override
+    public void deleteIotTree(Long id) {
+        // 校验存在
+        validateIotTreeExists(id);
+        // 删除
+        iotTreeMapper.deleteById(id);
+    }
+
+    private void validateIotTreeExists(Long id) {
+        if (iotTreeMapper.selectById(id) == null) {
+            throw exception(IOT_TREE_NOT_EXISTS);
+        }
+    }
+
+    @Override
+    public IotTreeDO getIotTree(Long id) {
+        return iotTreeMapper.selectById(id);
+    }
+
+    @Override
+    public PageResult<IotTreeDO> getIotTreePage(IotTreePageReqVO pageReqVO) {
+        return iotTreeMapper.selectPage(pageReqVO);
+    }
+
+}