From 665c0a06d0bca5a22fb42406c465d736f876b04e Mon Sep 17 00:00:00 2001
From: zhaoxiaohao <913652501@qq.com>
Date: Fri, 10 Apr 2020 17:53:16 +0800
Subject: [PATCH] 添加公司管理的功能

---
 kidgrow-business/kidgrow-usercenter/kidgrow-usercenter-biz/src/main/java/com/kidgrow/usercenter/service/impl/SysCompanyServiceImpl.java |   69 +++++++++++++++++++++++++++++++++-
 1 files changed, 67 insertions(+), 2 deletions(-)

diff --git a/kidgrow-business/kidgrow-usercenter/kidgrow-usercenter-biz/src/main/java/com/kidgrow/usercenter/service/impl/SysCompanyServiceImpl.java b/kidgrow-business/kidgrow-usercenter/kidgrow-usercenter-biz/src/main/java/com/kidgrow/usercenter/service/impl/SysCompanyServiceImpl.java
index 15b101e..6cf4234 100644
--- a/kidgrow-business/kidgrow-usercenter/kidgrow-usercenter-biz/src/main/java/com/kidgrow/usercenter/service/impl/SysCompanyServiceImpl.java
+++ b/kidgrow-business/kidgrow-usercenter/kidgrow-usercenter-biz/src/main/java/com/kidgrow/usercenter/service/impl/SysCompanyServiceImpl.java
@@ -1,10 +1,17 @@
 package com.kidgrow.usercenter.service.impl;
 
+import com.kidgrow.common.model.ResultBody;
+import com.kidgrow.usercenter.model.SysArea;
+import com.kidgrow.usercenter.service.ISysAreaService;
+import com.kidgrow.usercenter.vo.SysCompanyVo;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.cglib.beans.BeanCopier;
 import org.springframework.stereotype.Service;
 import com.kidgrow.common.model.PageResult;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.kidgrow.common.service.impl.SuperServiceImpl;
 
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
 import org.apache.commons.collections4.MapUtils;
@@ -25,16 +32,58 @@
 @Slf4j
 @Service
 public class SysCompanyServiceImpl extends SuperServiceImpl<SysCompanyMapper, SysCompany> implements ISysCompanyService {
+    @Autowired
+    private ISysAreaService sysAreaService;
+    private final  Integer SHENG=1;
+    private final  Integer SHI=2;
+    private final  Integer XIAN=3;
     /**
      * 列表
      * @param params
      * @return
      */
     @Override
-    public PageResult<SysCompany> findList(Map<String, Object> params){
+    public PageResult<SysCompanyVo> findList(Map<String, Object> params){
         Page<SysCompany> page = new Page<>(MapUtils.getInteger(params, "page"), MapUtils.getInteger(params, "limit"));
         List<SysCompany> list  =  baseMapper.findList(page, params);
-        return PageResult.<SysCompany>builder().data(list).code(0).count(page.getTotal()).build();
+        List<SysCompanyVo> listVo=new ArrayList<>();
+        list.forEach(e -> {
+            SysCompanyVo vo=new SysCompanyVo();
+            BeanCopier beanCopier = BeanCopier.create(SysCompany.class, SysCompanyVo.class, false);
+            beanCopier.copy(e,vo,null);
+            if(vo.getCompanyArea()!=null){
+                SysArea byId = sysAreaService.getById(vo.getCompanyArea());
+                if(byId.getAreaLeveltype()==XIAN){
+                    vo.setXianId(byId.getId());
+                    vo.setXian(byId.getAreaName());
+                    SysArea shi = sysAreaService.getById(byId.getAreaParentId());
+                    if(shi!=null){
+                        vo.setShi(shi.getAreaName());
+                        vo.setShiId(shi.getId());
+                        SysArea sheng = sysAreaService.getById(shi.getAreaParentId());
+                        if (sheng != null) {
+                            vo.setSheng(sheng.getAreaName());
+                            vo.setShengId(sheng.getId());
+                        }
+                    }
+                }
+                if(byId.getAreaLeveltype()==SHI){
+                    vo.setShi(byId.getAreaName());
+                    vo.setShiId(byId.getId());
+                    SysArea sheng = sysAreaService.getById(byId.getAreaParentId());
+                    if (sheng != null) {
+                        vo.setSheng(sheng.getAreaName());
+                        vo.setShengId(sheng.getId());
+                    }
+                }
+                if(byId.getAreaLeveltype()==SHENG){
+                    vo.setSheng(byId.getAreaName());
+                    vo.setShengId(byId.getId());
+                }
+            }
+            listVo.add(vo);
+        });
+        return PageResult.<SysCompanyVo>builder().data(listVo).code(0).count(page.getTotal()).build();
     }
 
     /**
@@ -46,4 +95,20 @@
     public SysCompany findByObject(SysCompany sysCompany){
         return baseMapper.findByObject(sysCompany);
     }
+
+    @Override
+    public ResultBody updateEnabled(Map<String, Object> params) {
+        Long id = MapUtils.getLong(params, "id");
+        if (id != null) {
+            SysCompany sysCompany = baseMapper.selectById(id);
+            if (sysCompany != null) {
+                Boolean enabled = MapUtils.getBoolean(params, "enabled");
+                sysCompany.setEnabled(enabled);
+                baseMapper.updateById(sysCompany);
+            }
+        }else {
+            return ResultBody.failed("请选择一条数据");
+        }
+        return ResultBody.ok();
+    }
 }

--
Gitblit v1.8.0