From 5c16e998fdcef5f5645eb71697a95c6f45157bbf Mon Sep 17 00:00:00 2001
From: k94314517 <8417338+k94314517@user.noreply.gitee.com>
Date: 星期四, 28 十二月 2023 14:18:26 +0800
Subject: [PATCH] ERP接口

---
 server/dmvisit_service/src/main/java/com/doumee/service/business/impl/CompanyServiceImpl.java |   89 +++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 83 insertions(+), 6 deletions(-)

diff --git a/server/dmvisit_service/src/main/java/com/doumee/service/business/impl/CompanyServiceImpl.java b/server/dmvisit_service/src/main/java/com/doumee/service/business/impl/CompanyServiceImpl.java
index 0a4dfc6..84656aa 100644
--- a/server/dmvisit_service/src/main/java/com/doumee/service/business/impl/CompanyServiceImpl.java
+++ b/server/dmvisit_service/src/main/java/com/doumee/service/business/impl/CompanyServiceImpl.java
@@ -19,18 +19,20 @@
 import com.doumee.core.model.PageWrap;
 import com.doumee.core.utils.Constants;
 import com.doumee.core.utils.Utils;
+import com.doumee.dao.admin.request.UpdateCompanySortDTO;
 import com.doumee.dao.admin.response.CompanyDTO;
 import com.doumee.dao.business.CompanyMapper;
 import com.doumee.dao.business.join.CompanyJoinMapper;
-import com.doumee.dao.business.model.Category;
-import com.doumee.dao.business.model.Company;
-import com.doumee.dao.business.model.Member;
+import com.doumee.dao.business.model.*;
+import com.doumee.dao.system.dto.UpdateSystemMenuSortDTO;
+import com.doumee.dao.system.model.SystemMenu;
 import com.doumee.dao.system.model.SystemUser;
 import com.doumee.service.business.CompanyService;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.doumee.service.common.tree.CompanyTree;
 import com.github.yulichang.wrapper.MPJLambdaWrapper;
 import io.swagger.annotations.ApiModelProperty;
 import org.apache.commons.collections.CollectionUtils;
@@ -81,11 +83,10 @@
             if(parent == null || Constants.equalsInteger(parent.getIsdeleted(),Constants.ONE)){
                 throw new BusinessException(ResponseStatus.DATA_EXISTS.getCode(), "瀵逛笉璧凤紝鐖剁骇缁勭粐淇℃伅涓嶅瓨鍦▇");
             }
-            if(StringUtils.isBlank(parent.getHkParentId())){
+            if(StringUtils.isBlank(parent.getHkId())){
                 throw new BusinessException(ResponseStatus.DATA_EXISTS.getCode(), "瀵逛笉璧凤紝鐖剁骇缁勭粐淇℃伅灏氭湭鍚屾涓嬪彂鎴愬姛~");
-
             }
-            company.setHkParentId(parent.getHkParentId());
+            company.setHkParentId(parent.getHkId());
         }
         company.setCreateDate(new Date());
         company.setCreator(user.getId());
@@ -198,6 +199,7 @@
         if(StringUtils.isBlank(model.getHkId())){
             throw new BusinessException(ResponseStatus.DATA_EMPTY.getCode(), "瀵逛笉璧凤紝缁勭粐淇℃伅灏氭湭鍚屾涓嬪彂鎴愬姛,鏆備笉鏀寔淇敼锛岃灏濊瘯鍒犻櫎鍚庨噸鏂版坊鍔爚");
         }
+
         Company company = new Company();
         company.setId(id);
         company.setEditDate(new Date());
@@ -494,4 +496,79 @@
             throw new BusinessException(ResponseStatus.DATA_EXISTS.getCode(),"鍏徃鍚嶇О宸插瓨鍦�");
         }
     }
+
+
+
+    @Override
+    public List<Company> companyTree(Integer type) {
+        MPJLambdaWrapper<Company> queryWrapper = new MPJLambdaWrapper<>();
+        queryWrapper.selectAll(Company.class);
+        queryWrapper.selectAs(Member::getName,Company::getHeadName);
+        queryWrapper.selectAs(Member::getPhone,Company::getHeadPhone);
+        queryWrapper.leftJoin(Member.class,Member::getId,Company::getHeadId);
+        queryWrapper.eq(Company::getIsdeleted,Constants.ZERO);
+        queryWrapper.eq(Company::getType,type);
+        List<Company> companyList = companyJoinMapper.selectJoinList(Company.class,queryWrapper);
+        //閰嶇疆缁勫憳浜哄憳绫诲瀷鏉ユ簮鏂瑰紡 0鑷缓 1ERP
+        String origin = systemDictDataBiz.queryByCode(Constants.SYSTEM,Constants.ORG_USER_ORIGIN).getCode();
+        CompanyTree treeBuild = new CompanyTree(companyList,Integer.valueOf(origin));
+        companyList = treeBuild.buildTree();
+        return companyList;
+    }
+
+
+    @Override
+    public void updateSort(UpdateCompanySortDTO dto) {
+        Company company = companyMapper.selectById(dto.getId());
+        if(Objects.isNull(company)){
+            throw new BusinessException(ResponseStatus.DATA_EMPTY);
+        }
+
+        List<Company> companyPool;
+        if (company.getParentId() == null) {
+            companyPool = companyMapper.selectList(new QueryWrapper<Company>().lambda()
+                    .eq(Company::getIsdeleted,Constants.ZERO)
+                    .isNull(Company::getParentId)
+                    .orderByAsc(Company::getSortnum)
+            );
+        } else {
+            companyPool = companyMapper.selectList(new QueryWrapper<Company>().lambda()
+                    .eq(Company::getIsdeleted,Constants.ZERO)
+                    .eq(Company::getParentId,company.getParentId())
+                    .orderByAsc(Company::getSortnum)
+            );
+        }
+        int currentMenuIndex = 0;
+        for (int i = 0; i < companyPool.size(); i++) {
+            if (companyPool.get(i).getId().equals(dto.getId())) {
+                currentMenuIndex = i;
+                break;
+            }
+        }
+        // 涓婄Щ
+        if ("top".equals(dto.getDirection())) {
+            if (currentMenuIndex - 1 < 0) {
+                return;
+            }
+            Company preMenu = companyPool.remove(currentMenuIndex - 1);
+            companyPool.add(currentMenuIndex, preMenu);
+        }
+        // 涓嬬Щ
+        else {
+            if (currentMenuIndex + 1 > companyPool.size() - 1) {
+                return;
+            }
+            Company nextMenu = companyPool.remove(currentMenuIndex + 1);
+            companyPool.add(currentMenuIndex, nextMenu);
+        }
+        for (int i = 0; i < companyPool.size(); i++) {
+            companyPool.get(i).setSortnum(i);
+        }
+
+        for (Company updCompany:companyPool) {
+            companyMapper.updateById(updCompany);
+        }
+
+    }
+
 }

--
Gitblit v1.9.3