From 1e3cea15942a80932473bdc74a4c103ade713789 Mon Sep 17 00:00:00 2001 From: ye1023 <12588209+ye1023@user.noreply.gitee.com> Date: Wed, 27 May 2026 15:49:29 +0800 Subject: [PATCH] =?UTF-8?q?yxk-20260527-=E5=B7=A1=E8=A7=86=E6=95=B4?= =?UTF-8?q?=E6=94=B9=E6=8F=90=E5=8D=87=E5=8F=B0=E8=B4=A6=EF=BC=8C=E4=BB=A3?= =?UTF-8?q?=E7=A0=81=E7=94=9F=E6=88=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../DqInspectProblemController.java | 298 ++++++++++++++++++ .../entity/DqInspectProblem.java | 109 +++++++ .../mapper/DqInspectProblemMapper.java | 35 ++ .../mapper/xml/DqInspectProblemMapper.xml | 25 ++ .../service/IDqInspectProblemService.java | 74 +++++ .../impl/DqInspectProblemServiceImpl.java | 219 +++++++++++++ 6 files changed, 760 insertions(+) create mode 100644 jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/controller/DqInspectProblemController.java create mode 100644 jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/entity/DqInspectProblem.java create mode 100644 jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/mapper/DqInspectProblemMapper.java create mode 100644 jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/mapper/xml/DqInspectProblemMapper.xml create mode 100644 jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/service/IDqInspectProblemService.java create mode 100644 jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/service/impl/DqInspectProblemServiceImpl.java diff --git a/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/controller/DqInspectProblemController.java b/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/controller/DqInspectProblemController.java new file mode 100644 index 0000000..11b26c7 --- /dev/null +++ b/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/controller/DqInspectProblemController.java @@ -0,0 +1,298 @@ +package org.jeecg.modules.demo.dqinspectproblem.controller; + +import java.util.Arrays; +import java.util.List; +import java.util.Map; +import java.util.stream.Collectors; +import java.io.IOException; +import java.io.UnsupportedEncodingException; +import java.net.URLDecoder; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import org.jeecg.common.api.vo.Result; +import org.jeecg.common.system.query.QueryGenerator; +import org.jeecg.common.util.oConvertUtils; +import org.jeecg.common.system.vo.SelectTreeModel; +import org.jeecg.modules.demo.dqinspectproblem.entity.DqInspectProblem; +import org.jeecg.modules.demo.dqinspectproblem.service.IDqInspectProblemService; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import lombok.extern.slf4j.Slf4j; + +import org.jeecgframework.poi.excel.ExcelImportUtil; +import org.jeecgframework.poi.excel.def.NormalExcelConstants; +import org.jeecgframework.poi.excel.entity.ExportParams; +import org.jeecgframework.poi.excel.entity.ImportParams; +import org.jeecgframework.poi.excel.view.JeecgEntityExcelView; +import org.jeecg.common.system.base.controller.JeecgController; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.web.bind.annotation.*; +import org.springframework.web.multipart.MultipartFile; +import org.springframework.web.multipart.MultipartHttpServletRequest; +import org.springframework.web.servlet.ModelAndView; +import com.alibaba.fastjson.JSON; +import io.swagger.v3.oas.annotations.tags.Tag; +import io.swagger.v3.oas.annotations.Operation; +import org.jeecg.common.aspect.annotation.AutoLog; +import org.apache.shiro.authz.annotation.RequiresPermissions; + + /** + * @Description: dq_inspect_problem + * @Author: jeecg-boot + * @Date: 2026-05-27 + * @Version: V1.0 + */ +@Tag(name="dq_inspect_problem") +@RestController +@RequestMapping("/dqinspectproblem/dqInspectProblem") +@Slf4j +public class DqInspectProblemController extends JeecgController{ + @Autowired + private IDqInspectProblemService dqInspectProblemService; + + /** + * 分页列表查询 + * + * @param dqInspectProblem + * @param pageNo + * @param pageSize + * @param req + * @return + */ + //@AutoLog(value = "dq_inspect_problem-分页列表查询") + @Operation(summary="dq_inspect_problem-分页列表查询") + @GetMapping(value = "/rootList") + public Result> queryPageList(DqInspectProblem dqInspectProblem, + @RequestParam(name="pageNo", defaultValue="1") Integer pageNo, + @RequestParam(name="pageSize", defaultValue="10") Integer pageSize, + HttpServletRequest req) { + String hasQuery = req.getParameter("hasQuery"); + if(hasQuery != null && "true".equals(hasQuery)){ + QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(dqInspectProblem, req.getParameterMap()); + List list = dqInspectProblemService.queryTreeListNoPage(queryWrapper); + IPage pageList = new Page<>(1, 10, list.size()); + pageList.setRecords(list); + return Result.OK(pageList); + }else{ + String parentId = dqInspectProblem.getPid(); + if (oConvertUtils.isEmpty(parentId)) { + parentId = "0"; + } + dqInspectProblem.setPid(null); + QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(dqInspectProblem, req.getParameterMap()); + // 使用 eq 防止模糊查询 + queryWrapper.eq("pid", parentId); + Page page = new Page(pageNo, pageSize); + IPage pageList = dqInspectProblemService.page(page, queryWrapper); + return Result.OK(pageList); + } + } + + /** + * 【vue3专用】加载节点的子数据 + * + * @param pid + * @return + */ + @RequestMapping(value = "/loadTreeChildren", method = RequestMethod.GET) + public Result> loadTreeChildren(@RequestParam(name = "pid") String pid) { + Result> result = new Result<>(); + try { + List ls = dqInspectProblemService.queryListByPid(pid); + result.setResult(ls); + result.setSuccess(true); + } catch (Exception e) { + e.printStackTrace(); + result.setMessage(e.getMessage()); + result.setSuccess(false); + } + return result; + } + + /** + * 【vue3专用】加载一级节点/如果是同步 则所有数据 + * + * @param async + * @param pcode + * @return + */ + @RequestMapping(value = "/loadTreeRoot", method = RequestMethod.GET) + public Result> loadTreeRoot(@RequestParam(name = "async") Boolean async, @RequestParam(name = "pcode") String pcode) { + Result> result = new Result<>(); + try { + List ls = dqInspectProblemService.queryListByCode(pcode); + if (!async) { + loadAllChildren(ls); + } + result.setResult(ls); + result.setSuccess(true); + } catch (Exception e) { + e.printStackTrace(); + result.setMessage(e.getMessage()); + result.setSuccess(false); + } + return result; + } + + /** + * 【vue3专用】递归求子节点 同步加载用到 + * + * @param ls + */ + private void loadAllChildren(List ls) { + for (SelectTreeModel tsm : ls) { + List temp = dqInspectProblemService.queryListByPid(tsm.getKey()); + if (temp != null && temp.size() > 0) { + tsm.setChildren(temp); + loadAllChildren(temp); + } + } + } + + /** + * 获取子数据 + * @param dqInspectProblem + * @param req + * @return + */ + //@AutoLog(value = "dq_inspect_problem-获取子数据") + @Operation(summary="dq_inspect_problem-获取子数据") + @GetMapping(value = "/childList") + public Result> queryPageList(DqInspectProblem dqInspectProblem,HttpServletRequest req) { + QueryWrapper queryWrapper = QueryGenerator.initQueryWrapper(dqInspectProblem, req.getParameterMap()); + List list = dqInspectProblemService.list(queryWrapper); + IPage pageList = new Page<>(1, 10, list.size()); + pageList.setRecords(list); + return Result.OK(pageList); + } + + /** + * 批量查询子节点 + * @param parentIds 父ID（多个采用半角逗号分割） + * @return 返回 IPage + * @param parentIds + * @return + */ + //@AutoLog(value = "dq_inspect_problem-批量获取子数据") + @Operation(summary="dq_inspect_problem-批量获取子数据") + @GetMapping("/getChildListBatch") + public Result getChildListBatch(@RequestParam("parentIds") String parentIds) { + try { + QueryWrapper queryWrapper = new QueryWrapper<>(); + List parentIdList = Arrays.asList(parentIds.split(",")); + queryWrapper.in("pid", parentIdList); + List list = dqInspectProblemService.list(queryWrapper); + IPage pageList = new Page<>(1, 10, list.size()); + pageList.setRecords(list); + return Result.OK(pageList); + } catch (Exception e) { + log.error(e.getMessage(), e); + return Result.error("批量查询子节点失败：" + e.getMessage()); + } + } + + /** + * 添加 + * + * @param dqInspectProblem + * @return + */ + @AutoLog(value = "dq_inspect_problem-添加") + @Operation(summary="dq_inspect_problem-添加") + @RequiresPermissions("dqinspectproblem:dq_inspect_problem:add") + @PostMapping(value = "/add") + public Result add(@RequestBody DqInspectProblem dqInspectProblem) { + dqInspectProblemService.addDqInspectProblem(dqInspectProblem); + return Result.OK("添加成功！"); + } + + /** + * 编辑 + * + * @param dqInspectProblem + * @return + */ + @AutoLog(value = "dq_inspect_problem-编辑") + @Operation(summary="dq_inspect_problem-编辑") + @RequiresPermissions("dqinspectproblem:dq_inspect_problem:edit") + @RequestMapping(value = "/edit", method = {RequestMethod.PUT,RequestMethod.POST}) + public Result edit(@RequestBody DqInspectProblem dqInspectProblem) { + dqInspectProblemService.updateDqInspectProblem(dqInspectProblem); + return Result.OK("编辑成功!"); + } + + /** + * 通过id删除 + * + * @param id + * @return + */ + @AutoLog(value = "dq_inspect_problem-通过id删除") + @Operation(summary="dq_inspect_problem-通过id删除") + @RequiresPermissions("dqinspectproblem:dq_inspect_problem:delete") + @DeleteMapping(value = "/delete") + public Result delete(@RequestParam(name="id",required=true) String id) { + dqInspectProblemService.deleteDqInspectProblem(id); + return Result.OK("删除成功!"); + } + + /** + * 批量删除 + * + * @param ids + * @return + */ + @AutoLog(value = "dq_inspect_problem-批量删除") + @Operation(summary="dq_inspect_problem-批量删除") + @RequiresPermissions("dqinspectproblem:dq_inspect_problem:deleteBatch") + @DeleteMapping(value = "/deleteBatch") + public Result deleteBatch(@RequestParam(name="ids",required=true) String ids) { + this.dqInspectProblemService.removeByIds(Arrays.asList(ids.split(","))); + return Result.OK("批量删除成功！"); + } + + /** + * 通过id查询 + * + * @param id + * @return + */ + //@AutoLog(value = "dq_inspect_problem-通过id查询") + @Operation(summary="dq_inspect_problem-通过id查询") + @GetMapping(value = "/queryById") + public Result queryById(@RequestParam(name="id",required=true) String id) { + DqInspectProblem dqInspectProblem = dqInspectProblemService.getById(id); + if(dqInspectProblem==null) { + return Result.error("未找到对应数据"); + } + return Result.OK(dqInspectProblem); + } + + /** + * 导出excel + * + * @param request + * @param dqInspectProblem + */ + @RequiresPermissions("dqinspectproblem:dq_inspect_problem:exportXls") + @RequestMapping(value = "/exportXls") + public ModelAndView exportXls(HttpServletRequest request, DqInspectProblem dqInspectProblem) { + return super.exportXls(request, dqInspectProblem, DqInspectProblem.class, "dq_inspect_problem"); + } + + /** + * 通过excel导入数据 + * + * @param request + * @param response + * @return + */ + @RequiresPermissions("dqinspectproblem:dq_inspect_problem:importExcel") + @RequestMapping(value = "/importExcel", method = RequestMethod.POST) + public Result importExcel(HttpServletRequest request, HttpServletResponse response) { + return super.importExcel(request, response, DqInspectProblem.class); + } + +} diff --git a/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/entity/DqInspectProblem.java b/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/entity/DqInspectProblem.java new file mode 100644 index 0000000..b39f1c5 --- /dev/null +++ b/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/entity/DqInspectProblem.java @@ -0,0 +1,109 @@ +package org.jeecg.modules.demo.dqinspectproblem.entity; + +import java.io.Serializable; +import java.util.Date; +import java.math.BigDecimal; +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import com.baomidou.mybatisplus.annotation.TableLogic; +import org.jeecg.common.constant.ProvinceCityArea; +import org.jeecg.common.util.SpringContextUtils; +import lombok.Data; +import com.fasterxml.jackson.annotation.JsonFormat; +import org.springframework.format.annotation.DateTimeFormat; +import org.jeecgframework.poi.excel.annotation.Excel; +import org.jeecg.common.aspect.annotation.Dict; + +import io.swagger.v3.oas.annotations.media.Schema; +import java.io.UnsupportedEncodingException; + +/** + * @Description: dq_inspect_problem + * @Author: jeecg-boot + * @Date: 2026-05-27 + * @Version: V1.0 + */ +@Data +@TableName("dq_inspect_problem") +@Schema(description="dq_inspect_problem") +public class DqInspectProblem implements Serializable { + private static final long serialVersionUID = 1L; + + /**id*/ + @TableId(type = IdType.ASSIGN_ID) + @Schema(description = "id") + private java.lang.String id; + /**创建人*/ + @Schema(description = "创建人") + private java.lang.String createBy; + /**创建日期*/ + @JsonFormat(timezone = "GMT+8",pattern = "yyyy-MM-dd HH:mm:ss") + @DateTimeFormat(pattern="yyyy-MM-dd HH:mm:ss") + @Schema(description = "创建日期") + private java.util.Date createTime; + /**更新人*/ + @Schema(description = "更新人") + private java.lang.String updateBy; + /**更新日期*/ + @JsonFormat(timezone = "GMT+8",pattern = "yyyy-MM-dd HH:mm:ss") + @DateTimeFormat(pattern="yyyy-MM-dd HH:mm:ss") + @Schema(description = "更新日期") + private java.util.Date updateTime; + /**所属部门*/ + @Schema(description = "所属部门") + private java.lang.String sysOrgCode; + /**密级*/ + @Excel(name = "密级", width = 15, dicCode = "secret_level") + @Dict(dicCode = "secret_level") + @Schema(description = "密级") + private java.lang.String secretLevel; + /**年份*/ + @Excel(name = "年份", width = 15, dicCode = "super_year") + @Dict(dicCode = "super_year") + @Schema(description = "年份") + private java.lang.String year; + /**父节点ID*/ + @Excel(name = "父节点ID", width = 15) + @Schema(description = "父节点ID") + private java.lang.String pid; + /**是否有子节点*/ + @Excel(name = "是否有子节点", width = 15) + @Schema(description = "是否有子节点") + private java.lang.String hasChild; + /**节点类型：category/surface/specific*/ + @Excel(name = "节点类型：category/surface/specific", width = 15) + @Schema(description = "节点类型：category/surface/specific") + private java.lang.String nodeType; + /**问题名称*/ + @Excel(name = "问题名称", width = 15) + @Schema(description = "问题名称") + private java.lang.String problemName; + /**问题编码/分类字典值*/ + @Excel(name = "问题编码/分类字典值", width = 15) + @Schema(description = "问题编码/分类字典值") + private java.lang.String problemCode; + /**排序*/ + @Excel(name = "排序", width = 15) + @Schema(description = "排序") + private java.lang.Integer sortNo; + /**巡视建议*/ + @Excel(name = "巡视建议", width = 15) + @Schema(description = "巡视建议") + private java.lang.String inspectAdvice; + /**问题责任领导*/ + @Excel(name = "问题责任领导", width = 15, dictTable = "sys_depart", dicText = "depart_name", dicCode = "id") + @Dict(dictTable = "sys_depart", dicText = "depart_name", dicCode = "id") + @Schema(description = "问题责任领导") + private java.lang.String questionResLeader; + /**问题责任部门*/ + @Excel(name = "问题责任部门", width = 15, dictTable = "sys_user", dicText = "realname", dicCode = "username") + @Dict(dictTable = "sys_user", dicText = "realname", dicCode = "username") + @Schema(description = "问题责任部门") + private java.lang.String questionResDept; + /**删除标识*/ + @Excel(name = "删除标识", width = 15) + @Schema(description = "删除标识") + @TableLogic + private java.lang.String delFlag; +} diff --git a/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/mapper/DqInspectProblemMapper.java b/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/mapper/DqInspectProblemMapper.java new file mode 100644 index 0000000..00c805d --- /dev/null +++ b/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/mapper/DqInspectProblemMapper.java @@ -0,0 +1,35 @@ +package org.jeecg.modules.demo.dqinspectproblem.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import org.apache.ibatis.annotations.Param; +import org.jeecg.common.system.vo.SelectTreeModel; +import org.jeecg.modules.demo.dqinspectproblem.entity.DqInspectProblem; + +import java.util.List; +import java.util.Map; + +/** + * @Description: dq_inspect_problem + * @Author: jeecg-boot + * @Date: 2026-05-27 + * @Version: V1.0 + */ +public interface DqInspectProblemMapper extends BaseMapper { + + /** + * 编辑节点状态 + * @param id + * @param status + */ + void updateTreeNodeStatus(@Param("id") String id,@Param("status") String status); + + /** + * 【vue3专用】根据父级ID查询树节点数据 + * + * @param pid + * @param query + * @return + */ + List queryListByPid(@Param("pid") String pid, @Param("query") Map query); + +} diff --git a/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/mapper/xml/DqInspectProblemMapper.xml b/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/mapper/xml/DqInspectProblemMapper.xml new file mode 100644 index 0000000..ae65e21 --- /dev/null +++ b/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/mapper/xml/DqInspectProblemMapper.xml @@ -0,0 +1,25 @@ + + + + + + update dq_inspect_problem set has_child = #{status} where id = #{id} + + + + + + \ No newline at end of file diff --git a/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/service/IDqInspectProblemService.java b/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/service/IDqInspectProblemService.java new file mode 100644 index 0000000..d53b852 --- /dev/null +++ b/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/service/IDqInspectProblemService.java @@ -0,0 +1,74 @@ +package org.jeecg.modules.demo.dqinspectproblem.service; + +import org.jeecg.common.system.vo.SelectTreeModel; +import org.jeecg.modules.demo.dqinspectproblem.entity.DqInspectProblem; +import com.baomidou.mybatisplus.extension.service.IService; +import org.jeecg.common.exception.JeecgBootException; +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import java.util.List; + +/** + * @Description: dq_inspect_problem + * @Author: jeecg-boot + * @Date: 2026-05-27 + * @Version: V1.0 + */ +public interface IDqInspectProblemService extends IService { + + /**根节点父ID的值*/ + public static final String ROOT_PID_VALUE = "0"; + + /**树节点有子节点状态值*/ + public static final String HASCHILD = "1"; + + /**树节点无子节点状态值*/ + public static final String NOCHILD = "0"; + + /** + * 新增节点 + * + * @param dqInspectProblem + */ + void addDqInspectProblem(DqInspectProblem dqInspectProblem); + + /** + * 修改节点 + * + * @param dqInspectProblem + * @throws JeecgBootException + */ + void updateDqInspectProblem(DqInspectProblem dqInspectProblem) throws JeecgBootException; + + /** + * 删除节点 + * + * @param id + * @throws JeecgBootException + */ + void deleteDqInspectProblem(String id) throws JeecgBootException; + + /** + * 查询所有数据，无分页 + * + * @param queryWrapper + * @return List + */ + List queryTreeListNoPage(QueryWrapper queryWrapper); + + /** + * 【vue3专用】根据父级编码加载分类字典的数据 + * + * @param parentCode + * @return + */ + List queryListByCode(String parentCode); + + /** + * 【vue3专用】根据pid查询子节点集合 + * + * @param pid + * @return + */ + List queryListByPid(String pid); + +} diff --git a/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/service/impl/DqInspectProblemServiceImpl.java b/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/service/impl/DqInspectProblemServiceImpl.java new file mode 100644 index 0000000..e4a7236 --- /dev/null +++ b/jeecg-module-supervision/src/main/java/org/jeecg/modules/demo/dqinspectproblem/service/impl/DqInspectProblemServiceImpl.java @@ -0,0 +1,219 @@ +package org.jeecg.modules.demo.dqinspectproblem.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import org.jeecg.common.exception.JeecgBootException; +import org.jeecg.common.util.oConvertUtils; +import org.jeecg.common.system.vo.SelectTreeModel; +import org.jeecg.modules.demo.dqinspectproblem.entity.DqInspectProblem; +import org.jeecg.modules.demo.dqinspectproblem.mapper.DqInspectProblemMapper; +import org.jeecg.modules.demo.dqinspectproblem.service.IDqInspectProblemService; +import org.springframework.stereotype.Service; +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import org.springframework.transaction.annotation.Transactional; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.List; + +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; + +/** + * @Description: dq_inspect_problem + * @Author: jeecg-boot + * @Date: 2026-05-27 + * @Version: V1.0 + */ +@Service +public class DqInspectProblemServiceImpl extends ServiceImpl implements IDqInspectProblemService { + + @Override + public void addDqInspectProblem(DqInspectProblem dqInspectProblem) { + //新增时设置hasChild为0 + dqInspectProblem.setHasChild(IDqInspectProblemService.NOCHILD); + if(oConvertUtils.isEmpty(dqInspectProblem.getPid())){ + dqInspectProblem.setPid(IDqInspectProblemService.ROOT_PID_VALUE); + }else{ + //如果当前节点父ID不为空 则设置父节点的hasChildren 为1 + DqInspectProblem parent = baseMapper.selectById(dqInspectProblem.getPid()); + if(parent!=null && !"1".equals(parent.getHasChild())){ + parent.setHasChild("1"); + baseMapper.updateById(parent); + } + } + baseMapper.insert(dqInspectProblem); + } + + @Override + public void updateDqInspectProblem(DqInspectProblem dqInspectProblem) { + DqInspectProblem entity = this.getById(dqInspectProblem.getId()); + if(entity==null) { + throw new JeecgBootException("未找到对应实体"); + } + String old_pid = entity.getPid(); + String new_pid = dqInspectProblem.getPid(); + if(!old_pid.equals(new_pid)) { + updateOldParentNode(old_pid); + if(oConvertUtils.isEmpty(new_pid)){ + dqInspectProblem.setPid(IDqInspectProblemService.ROOT_PID_VALUE); + } + if(!IDqInspectProblemService.ROOT_PID_VALUE.equals(dqInspectProblem.getPid())) { + baseMapper.updateTreeNodeStatus(dqInspectProblem.getPid(), IDqInspectProblemService.HASCHILD); + } + } + baseMapper.updateById(dqInspectProblem); + } + + @Override + @Transactional(rollbackFor = Exception.class) + public void deleteDqInspectProblem(String id) throws JeecgBootException { + //查询选中节点下所有子节点一并删除 + id = this.queryTreeChildIds(id); + if(id.indexOf(",")>0) { + StringBuffer sb = new StringBuffer(); + String[] idArr = id.split(","); + for (String idVal : idArr) { + if(idVal != null){ + DqInspectProblem dqInspectProblem = this.getById(idVal); + String pidVal = dqInspectProblem.getPid(); + //查询此节点上一级是否还有其他子节点 + List dataList = baseMapper.selectList(new QueryWrapper().eq("pid", pidVal).notIn("id",Arrays.asList(idArr))); + boolean flag = (dataList == null || dataList.size() == 0) && !Arrays.asList(idArr).contains(pidVal) && !sb.toString().contains(pidVal); + if(flag){ + //如果当前节点原本有子节点 现在木有了，更新状态 + sb.append(pidVal).append(","); + } + } + } + //批量删除节点 + baseMapper.deleteBatchIds(Arrays.asList(idArr)); + //修改已无子节点的标识 + String[] pidArr = sb.toString().split(","); + for(String pid : pidArr){ + this.updateOldParentNode(pid); + } + }else{ + DqInspectProblem dqInspectProblem = this.getById(id); + if(dqInspectProblem==null) { + throw new JeecgBootException("未找到对应实体"); + } + updateOldParentNode(dqInspectProblem.getPid()); + baseMapper.deleteById(id); + } + } + + @Override + public List queryTreeListNoPage(QueryWrapper queryWrapper) { + List dataList = baseMapper.selectList(queryWrapper); + List mapList = new ArrayList<>(); + for(DqInspectProblem data : dataList){ + String pidVal = data.getPid(); + //递归查询子节点的根节点 + if(pidVal != null && !IDqInspectProblemService.NOCHILD.equals(pidVal)){ + DqInspectProblem rootVal = this.getTreeRoot(pidVal); + if(rootVal != null && !mapList.contains(rootVal)){ + mapList.add(rootVal); + } + }else{ + if(!mapList.contains(data)){ + mapList.add(data); + } + } + } + return mapList; + } + + @Override + public List queryListByCode(String parentCode) { + String pid = ROOT_PID_VALUE; + if (oConvertUtils.isNotEmpty(parentCode)) { + LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>(); + queryWrapper.eq(DqInspectProblem::getPid, parentCode); + List list = baseMapper.selectList(queryWrapper); + if (list == null || list.size() == 0) { + throw new JeecgBootException("该编码【" + parentCode + "】不存在，请核实!"); + } + if (list.size() > 1) { + throw new JeecgBootException("该编码【" + parentCode + "】存在多个，请核实!"); + } + pid = list.get(0).getId(); + } + return baseMapper.queryListByPid(pid, null); + } + + @Override + public List queryListByPid(String pid) { + if (oConvertUtils.isEmpty(pid)) { + pid = ROOT_PID_VALUE; + } + return baseMapper.queryListByPid(pid, null); + } + + /** + * 根据所传pid查询旧的父级节点的子节点并修改相应状态值 + * @param pid + */ + private void updateOldParentNode(String pid) { + if(!IDqInspectProblemService.ROOT_PID_VALUE.equals(pid)) { + Long count = baseMapper.selectCount(new QueryWrapper().eq("pid", pid)); + if(count==null || count<=1) { + baseMapper.updateTreeNodeStatus(pid, IDqInspectProblemService.NOCHILD); + } + } + } + + /** + * 递归查询节点的根节点 + * @param pidVal + * @return + */ + private DqInspectProblem getTreeRoot(String pidVal){ + DqInspectProblem data = baseMapper.selectById(pidVal); + if(data != null && !IDqInspectProblemService.ROOT_PID_VALUE.equals(data.getPid())){ + return this.getTreeRoot(data.getPid()); + }else{ + return data; + } + } + + /** + * 根据id查询所有子节点id + * @param ids + * @return + */ + private String queryTreeChildIds(String ids) { + //获取id数组 + String[] idArr = ids.split(","); + StringBuffer sb = new StringBuffer(); + for (String pidVal : idArr) { + if(pidVal != null){ + if(!sb.toString().contains(pidVal)){ + if(sb.toString().length() > 0){ + sb.append(","); + } + sb.append(pidVal); + this.getTreeChildIds(pidVal,sb); + } + } + } + return sb.toString(); + } + + /** + * 递归查询所有子节点 + * @param pidVal + * @param sb + * @return + */ + private StringBuffer getTreeChildIds(String pidVal,StringBuffer sb){ + List dataList = baseMapper.selectList(new QueryWrapper().eq("pid", pidVal)); + if(dataList != null && dataList.size()>0){ + for(DqInspectProblem tree : dataList) { + if(!sb.toString().contains(tree.getId())){ + sb.append(",").append(tree.getId()); + } + this.getTreeChildIds(tree.getId(),sb); + } + } + return sb; + } + +}