From 698ae84adaf1b8d0e8dd61d7279863fe17c1e81d Mon Sep 17 00:00:00 2001
From: luodangjia <luodangjia>
Date: 星期四, 23 一月 2025 14:20:39 +0800
Subject: [PATCH] 1

---
 ruoyi-modules/ruoyi-company/src/main/java/com/ruoyi/company/service/impl/CompanyServiceImpl.java |  220 ++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 209 insertions(+), 11 deletions(-)

diff --git a/ruoyi-modules/ruoyi-company/src/main/java/com/ruoyi/company/service/impl/CompanyServiceImpl.java b/ruoyi-modules/ruoyi-company/src/main/java/com/ruoyi/company/service/impl/CompanyServiceImpl.java
index 917df53..6b4da84 100644
--- a/ruoyi-modules/ruoyi-company/src/main/java/com/ruoyi/company/service/impl/CompanyServiceImpl.java
+++ b/ruoyi-modules/ruoyi-company/src/main/java/com/ruoyi/company/service/impl/CompanyServiceImpl.java
@@ -1,55 +1,89 @@
 package com.ruoyi.company.service.impl;
 
+import cn.idev.excel.FastExcel;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.ruoyi.common.core.constant.CacheConstants;
+import com.ruoyi.common.core.exception.ServiceException;
 import com.ruoyi.common.core.page.BeanUtils;
 import com.ruoyi.common.core.page.PageDTO;
 import com.ruoyi.common.core.utils.StringUtils;
+import com.ruoyi.common.redis.service.RedisService;
+import com.ruoyi.common.security.utils.SecurityUtils;
 import com.ruoyi.company.api.domain.Company;
 import com.ruoyi.company.api.domain.User;
+import com.ruoyi.company.api.domain.dto.MgtCompanyDTO;
+import com.ruoyi.company.api.domain.excel.MgtCompanyExcel;
 import com.ruoyi.company.api.domain.query.MgtCompanyQuery;
-import com.ruoyi.company.api.domain.vo.MgtCompanyVO;
+import com.ruoyi.company.domain.vo.MgtCompanyDetailVO;
+import com.ruoyi.company.domain.vo.MgtCompanyVO;
 import com.ruoyi.company.mapper.CompanyMapper;
 import com.ruoyi.company.service.CompanyService;
 import com.ruoyi.company.service.UserService;
+import com.ruoyi.company.utils.AliyunCloudAuthUtil;
+import com.ruoyi.system.api.model.AppUser;
 import lombok.RequiredArgsConstructor;
+import org.springframework.context.annotation.Lazy;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.net.URLEncoder;
+import java.util.Collection;
+import java.util.List;
 import java.util.Objects;
 
 @Service
-@RequiredArgsConstructor
+@RequiredArgsConstructor(onConstructor_ = {@Lazy})
 public class CompanyServiceImpl extends ServiceImpl<CompanyMapper, Company> implements CompanyService {
     private final UserService userService;
+    private final HttpServletResponse response;
+
+    private final AliyunCloudAuthUtil aliyunCloudAuthUtil;
+
+    private final RedisService redisService;
     /**
      * 获取企业列表
+     *
      * @param query
      * @return
      */
     @Override
     public PageDTO<MgtCompanyVO> queryPage(MgtCompanyQuery query) {
         Page<Company> page = this.lambdaQuery()
-                .like(StringUtils.isNotBlank(query.getCompanyInfo()), Company::getCompanyName, query.getCompanyInfo())
-                .like(StringUtils.isNotBlank(query.getCompanyInfo()), Company::getSocialCode, query.getCompanyInfo())
-                .like(StringUtils.isNotBlank(query.getLegalPersonInfo()), Company::getLegalPersonName, query.getLegalPersonInfo())
-                .like(StringUtils.isNotBlank(query.getLegalPersonInfo()), Company::getIdCardNumber, query.getLegalPersonInfo())
-                .like(StringUtils.isNotBlank(query.getContactInfo()), Company::getContactName, query.getContactInfo())
-                .like(StringUtils.isNotBlank(query.getContactInfo()), Company::getContactPhone, query.getContactInfo())
+                .and(StringUtils.isNotBlank(query.getCompanyInfo()),
+                        queryWrapper -> queryWrapper
+                                .like(Company::getCompanyName, query.getCompanyInfo())
+                                .or()
+                                .like(Company::getSocialCode, query.getCompanyInfo()))
+                .and(StringUtils.isNotBlank(query.getLegalPersonInfo()),
+                        queryWrapper -> queryWrapper
+                                .like(Company::getLegalPersonName, query.getLegalPersonInfo())
+                                .or()
+                                .like(Company::getIdCardNumber, query.getLegalPersonInfo()))
+                .and(StringUtils.isNotBlank(query.getContactInfo()),
+                        queryWrapper -> queryWrapper
+                                .like(Company::getContactName, query.getContactInfo())
+                                .or()
+                                .like(Company::getContactPhone, query.getContactInfo()))
+                .orderByDesc(Company::getCreateTime)
                 .page(new Page<>(query.getPageCurr(), query.getPageSize()));
         return PageDTO.of(page,MgtCompanyVO.class);
     }
 
     /**
      * 获取企业详情
+     *
      * @param id
      * @return
      */
     @Override
-    public MgtCompanyVO queryCompanyDetail(Long id) {
+    public MgtCompanyDetailVO queryCompanyDetail(Long id) {
         Company company = this.getById(id);
-        MgtCompanyVO mgtCompanyVO = BeanUtils.copyBean(company, MgtCompanyVO.class);
+        MgtCompanyDetailVO mgtCompanyVO = BeanUtils.copyBean(company, MgtCompanyDetailVO.class);
         if (Objects.isNull(mgtCompanyVO)) {
-            mgtCompanyVO = new MgtCompanyVO();
+            mgtCompanyVO = new MgtCompanyDetailVO();
             return mgtCompanyVO;
         }
         User user = userService.getById(company.getUserId());
@@ -59,4 +93,168 @@
         }
         return mgtCompanyVO;
     }
+
+    /**
+     * 新增企业
+     *
+     * @param dto
+     * @return
+     */
+    @Transactional(rollbackFor = Exception.class)
+    @Override
+    public void saveCompany(MgtCompanyDTO dto) {
+        Long accountCount = userService.lambdaQuery().eq(User::getAccountName, dto.getAccountName()).count();
+        if (accountCount > 0) {
+            throw new ServiceException("账户名重复");
+        }
+        //添加用户信息
+        User user = BeanUtils.copyBean(dto, User.class);
+        user.setPassword(SecurityUtils.encryptPassword(dto.getPassword()));
+        user.setCreateBy(SecurityUtils.getUserId());
+        userService.save(user);
+
+        Company company = BeanUtils.copyBean(dto, Company.class);
+        company.setUserId(user.getUserId());
+        //根据公司名称查询数据库
+        Long count = this.lambdaQuery().eq(Company::getCompanyName, company.getCompanyName()).eq(Company::getSocialCode, dto.getSocialCode()).count();
+        if (count > 0) {
+            throw new ServiceException("该企业已存在");
+        }
+        //身份证二要素校验
+        identityVerification(dto, user);
+        this.save(company);
+    }
+
+    /**
+     * 编辑企业
+     *
+     * @param dto
+     */
+    @Override
+    public void editCompany(MgtCompanyDTO dto) {
+        if (Objects.isNull(dto.getId())) {
+            throw new ServiceException("企业id不能为空");
+        }
+        //查询企业
+        Company company = this.getById(dto.getId());
+        if (Objects.isNull(company)) {
+            throw new ServiceException("该企业不存在");
+        }
+        //查询企业账号
+        User user = userService.getById(company.getUserId());
+        if (Objects.isNull(user)) {
+            throw new ServiceException("该企业账号不存在");
+        }
+        Long accountCount = userService.lambdaQuery()
+                .ne(User::getUserId, user.getUserId())
+                .eq(User::getAccountName, dto.getAccountName())
+                .count();
+        if (accountCount > 0) {
+            throw new ServiceException("账户名重复");
+        }
+        //用户信息
+        User userUpd = BeanUtils.copyBean(dto, User.class);
+        if (StringUtils.isNotBlank(dto.getPassword())) {
+            userUpd.setPassword(SecurityUtils.encryptPassword(dto.getPassword()));
+        }
+        userUpd.setUserId(user.getUserId());
+        userUpd.setUpdateBy(SecurityUtils.getUserId());
+        userService.updateById(userUpd);
+        //根据公司名称查询数据库
+        if (StringUtils.isNotBlank(company.getCompanyName())){
+            Long count = this.lambdaQuery()
+                    .ne(Company::getId, dto.getId())
+                    .and(i -> {
+                    i.eq(Company::getSocialCode, dto.getSocialCode());
+                    i.or();
+                    i.eq(Company::getCompanyName, dto.getCompanyName());
+                    })
+                    .count();
+            if (count > 0) {
+                throw new ServiceException("该企业已存在");
+            }
+        }
+        identityVerification(dto, user);
+        Company companyUpd = BeanUtils.copyBean(dto, Company.class);
+        companyUpd.setId(company.getId());
+        this.updateById(companyUpd);
+    }
+
+    private void identityVerification(MgtCompanyDTO dto, User user) {
+        //身份证二要素校验
+        Boolean res = aliyunCloudAuthUtil.verifyIdCard(dto.getLegalPersonName(), dto.getIdCardNumber());
+        if (!res) {
+            throw new ServiceException("身份证信息不匹配");
+        }
+        //营业执照二要素校验
+        res = aliyunCloudAuthUtil.verifyBusinessLicense(dto.getSocialCode(), dto.getCompanyName(), user.getUserId());
+        if (!res) {
+            throw new ServiceException("营业执照信息不匹配");
+        }
+    }
+
+    /**
+     * 删除企业
+     *
+     * @param id
+     */
+    @Override
+    public void deleteCompany(Long id) {
+        //查询企业
+        Company company = this.getById(id);
+        if (Objects.isNull(company)) {
+            throw new ServiceException("删除失败,该企业不存在");
+        }
+        //删除企业账号
+        userService.removeById(company.getUserId());
+
+        //删除企业信息
+        this.removeById(id);
+
+        //强退用户
+        forceLogout(company.getUserId());
+
+    }
+
+    public void forceLogout(Long userId) {
+        Collection<String> keys = redisService.keys(CacheConstants.LOGIN_TOKEN_KEY + "*");
+        for (String key : keys) {
+            Object user = redisService.getCacheObject(key);
+            if (user instanceof AppUser) {
+                AppUser appUser = (AppUser) user;
+                if (appUser.getUserId().equals(userId)) {
+                    redisService.deleteObject(key);
+                }
+            }
+        }
+    }
+
+    @Override
+    public void export(MgtCompanyQuery query) throws IOException {
+        List<Company> list = this.lambdaQuery()
+                .and(StringUtils.isNotBlank(query.getCompanyInfo()),
+                        queryWrapper -> queryWrapper
+                                .like(Company::getCompanyName, query.getCompanyInfo())
+                                .or()
+                                .like(Company::getSocialCode, query.getCompanyInfo()))
+                .and(StringUtils.isNotBlank(query.getLegalPersonInfo()),
+                        queryWrapper -> queryWrapper
+                                .like(Company::getLegalPersonName, query.getLegalPersonInfo())
+                                .or()
+                                .like(Company::getIdCardNumber, query.getLegalPersonInfo()))
+                .and(StringUtils.isNotBlank(query.getContactInfo()),
+                        queryWrapper -> queryWrapper
+                                .like(Company::getContactName, query.getContactInfo())
+                                .or()
+                                .like(Company::getContactPhone, query.getContactInfo()))
+                .orderByDesc(Company::getCreateTime).list();
+        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
+        response.setCharacterEncoding("utf-8");
+        String fileName = URLEncoder.encode("企业信息导出数据", "UTF-8").replaceAll("\\+", "%20");
+        response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
+
+        FastExcel.write(response.getOutputStream(), MgtCompanyExcel.class)
+                .sheet("企业信息导出数据")
+                .doWrite(BeanUtils.copyList(list, MgtCompanyExcel.class));
+    }
 }

--
Gitblit v1.7.1