From e2f8fb1fbe26ba6d92f3e5dfcaeb0c69abe76b25 Mon Sep 17 00:00:00 2001
From: jiangping <jp@doumee.com>
Date: 星期四, 14 十二月 2023 14:28:57 +0800
Subject: [PATCH] 海康接口对接开发

---
 server/dmvisit_service/src/main/java/com/doumee/service/business/impl/CompanyServiceImpl.java |   66 ++++++++++++++++++++++++++++++---
 1 files changed, 60 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 3161c88..68aca87 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
@@ -1,5 +1,7 @@
 package com.doumee.service.business.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.doumee.core.annotation.excel.ExcelColumn;
 import com.doumee.core.constants.ResponseStatus;
 import com.doumee.core.exception.BusinessException;
 import com.doumee.core.model.LoginUserInfo;
@@ -7,10 +9,12 @@
 import com.doumee.core.model.PageWrap;
 import com.doumee.core.utils.Constants;
 import com.doumee.core.utils.Utils;
+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.system.model.SystemUser;
 import com.doumee.service.business.CompanyService;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
@@ -18,12 +22,18 @@
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.github.yulichang.wrapper.MPJLambdaWrapper;
+import io.swagger.annotations.ApiModelProperty;
 import org.apache.commons.collections.CollectionUtils;
 import org.apache.shiro.SecurityUtils;
+import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+
+import java.util.ArrayList;
 import java.util.Date;
 import java.util.List;
+import java.util.Objects;
+import java.util.stream.Collectors;
 
 /**
  * 浼佷笟淇℃伅琛⊿ervice瀹炵幇
@@ -50,7 +60,6 @@
         LoginUserInfo loginUserInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
 
         checkUnique(company);
-
         Company insert = new Company();
         insert.setCreator(loginUserInfo.getId());
         insert.setCreateDate(new Date());
@@ -135,6 +144,7 @@
         IPage<Company> page = new Page<>(pageWrap.getPage(), pageWrap.getCapacity());
         MPJLambdaWrapper<Company> queryWrapper = new MPJLambdaWrapper<>();
         Utils.MP.blankToNull(pageWrap.getModel());
+        pageWrap.getModel().setIsdeleted(Constants.ZERO);
         if (pageWrap.getModel().getId() != null) {
             queryWrapper.eq(Company::getId, pageWrap.getModel().getId());
         }
@@ -230,10 +240,12 @@
         }
         queryWrapper.orderByDesc(Company::getCreateDate);
         queryWrapper.leftJoin(Category.class,Category::getId,Company::getCategoryId)
-                    .leftJoin(SystemUser.class,SystemUser::getId,Company::getCreator);
+                    .leftJoin(Member.class, Member::getId,Company::getHeadId);
         queryWrapper.selectAll(Company.class)
                     .selectAs(Category::getName,Company::getCategoryName)
-                    .selectAs(SystemUser::getRealname,Company::getCreatorName);
+                    .selectAs(Member::getName,Company::getHeadName)
+                    .selectAs(Member::getPhone,Company::getHeadPhone)
+                    .select("count(t1.id) as countNum");
         IPage<Company> companyIPage = companyJoinMapper.selectJoinPage(page, Company.class, queryWrapper);
         return PageData.from(companyIPage);
     }
@@ -243,6 +255,41 @@
         QueryWrapper<Company> wrapper = new QueryWrapper<>(company);
         return companyMapper.selectCount(wrapper);
     }
+
+
+    @Override
+    public List<CompanyDTO> findCompanyTreePage(Integer type) {
+
+        LambdaQueryWrapper<Company> wrapper = new LambdaQueryWrapper<>();
+
+        wrapper.eq(Company::getIsdeleted,Constants.ZERO)
+                .eq(Constants.equalsInteger(type,Constants.ONE),Company::getType,type);
+        List<Company> companies = companyMapper.selectList(wrapper);
+        return getCompanyTree( companies, null);
+    }
+
+
+    private List<CompanyDTO> getCompanyTree(List<Company> companies, Integer parentId){
+        if (CollectionUtils.isEmpty(companies)){
+            return null;
+        }
+        List<Company> collect = companies.stream()
+                                        .filter(s -> Constants.equalsInteger(s.getParentId(), parentId))
+                                        .collect(Collectors.toList());
+        if (CollectionUtils.isEmpty(companies)){
+            return null;
+        }
+        List<CompanyDTO> companyDTOList = new ArrayList<>();
+        for (int i = 0; i < collect.size(); i++) {
+            Company company = collect.get(i);
+            CompanyDTO companyDTO = new CompanyDTO();
+            BeanUtils.copyProperties(company,companyDTO);
+            companyDTO.setCompanyDTOList(getCompanyTree(companies, company.getId()));
+            companyDTOList.add(companyDTO);
+        }
+        return companyDTOList;
+    }
+
 
     /**
      * 鍔冲姟鍏徃鍞竴鎬ф牎楠�
@@ -255,9 +302,16 @@
                 .eq(Company::getType,Constants.ZERO)
                 .eq(Company::getName,company.getName());
         List<Company> companies = companyMapper.selectList(wrapper);
-
-        if (CollectionUtils.isNotEmpty(companies)){
-            throw new BusinessException(ResponseStatus.DATA_EXISTS.getCode(),"浼佷笟淇℃伅宸插瓨鍦�");
+        if (CollectionUtils.isEmpty(companies)){
+            return;
+        }
+        //鏂板鐨勬儏鍐靛瓨鍦ㄨ褰�
+        if (Objects.isNull(company.getId())){
+            throw new BusinessException(ResponseStatus.DATA_EXISTS.getCode(),"鍏徃鍚嶇О宸插瓨鍦�");
+        }
+        //淇敼鐨勬儏鍐靛瓨鍦ㄨ褰� 鍖归厤涓嶅埌 own
+        if (!companies.stream().map(s -> s.getId()).anyMatch(s -> s.equals(company.getId()))){
+            throw new BusinessException(ResponseStatus.DATA_EXISTS.getCode(),"鍏徃鍚嶇О宸插瓨鍦�");
         }
     }
 }

--
Gitblit v1.9.3