From ec6d43aa07ee0e8faf34498057ebcfbb446aa015 Mon Sep 17 00:00:00 2001
From: hjl <1657978663@qq.com>
Date: 星期二, 16 七月 2024 09:21:51 +0800
Subject: [PATCH] feat: 代码重构

---
 ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/FranchiseeController.java |  115 ++++++++++++++++++++++++++++++++++++++++++++++++++-------
 1 files changed, 100 insertions(+), 15 deletions(-)

diff --git a/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/FranchiseeController.java b/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/FranchiseeController.java
index 85e8c4f..9d42088 100644
--- a/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/FranchiseeController.java
+++ b/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/controller/FranchiseeController.java
@@ -5,7 +5,11 @@
 import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.ruoyi.admin.entity.Franchisee;
+import com.ruoyi.admin.entity.SysUser;
+import com.ruoyi.admin.entity.UserRole;
 import com.ruoyi.admin.service.FranchiseeService;
+import com.ruoyi.admin.service.SysUserService;
+import com.ruoyi.admin.service.UserRoleService;
 import com.ruoyi.common.core.constant.Constants;
 import com.ruoyi.common.core.domain.R;
 import com.ruoyi.common.core.exception.GlobalException;
@@ -39,6 +43,10 @@
 
     @Resource
     private FranchiseeService franchiseeService;
+    @Resource
+    private SysUserService sysUserService;
+    @Resource
+    private UserRoleService userRoleService;
 
     /**
      * 加盟商信息分页列表
@@ -80,8 +88,23 @@
     @ApiImplicitParams({
             @ApiImplicitParam(value = "加盟商信息id", name = "id", dataType = "Integer", required = true)
     })
-    public R<Franchisee> detail(@RequestParam Integer id) {
-        return R.ok(franchiseeService.getById(id));
+    public R<Franchisee> detail(@RequestParam("id") Integer id) {
+        return R.ok(franchiseeService.lambdaQuery()
+                .eq(Franchisee::getId, id).eq(Franchisee::getIsDelete, 0).one());
+    }
+
+    /**
+     * 加盟商管辖城市详情
+     * -- 远程调用
+     *
+     * @param id 加盟商信息id
+     */
+    @ApiOperation(value = "加盟商管辖城市详情", tags = {"后台-加盟商管理"})
+    @GetMapping(value = "/cityDetail")
+    public R<List<String>> cityDetail(@RequestParam("id") Integer id) {
+        Franchisee franchisee = franchiseeService.lambdaQuery()
+                .eq(Franchisee::getId, id).eq(Franchisee::getIsDelete, 0).one();
+        return R.ok(Arrays.stream(franchisee.getCity().split(",")).collect(Collectors.toList()));
     }
 
     /**
@@ -93,12 +116,27 @@
     @ApiOperation(value = "新增加盟商信息", tags = {"后台-加盟商管理"})
     @PostMapping(value = "/save")
     public R<String> save(@RequestBody @Validated Franchisee franchisee) {
-        checkFranchisee(franchisee);
         if (null == franchisee.getAdminPassword() || StringUtils.isBlank(franchisee.getAdminPassword())) {
             throw new GlobalException("请输入管理员初始密码!");
         }
+        String md5Password = checkFranchisee(franchisee);
+        boolean save = franchiseeService.save(franchisee);
+        // 生成sysUser账号
+        SysUser sysUser = new SysUser();
+        sysUser.setFranchiseeId(franchisee.getId());
+        sysUser.setNickName(franchisee.getName());
+        sysUser.setAccount(franchisee.getAdminAccount());
+        sysUser.setPassword(md5Password);
+        sysUser.setIsEnable(Constants.ONE);
+        sysUser.setIsDelete(Constants.ZERO);
+        save = save && sysUserService.save(sysUser);
+        // 添加账号与角色关联
+        UserRole userRole = new UserRole();
+        userRole.setUserId(sysUser.getUserId());
+        userRole.setRoleId(franchisee.getRoleId().longValue());
+        save = save && userRoleService.save(userRole);
         //franchisee.setCityStr(String.valueOf(franchisee.getCityArr()));
-        return franchiseeService.save(franchisee) ? R.ok() : R.fail();
+        return save ? R.ok() : R.fail();
     }
 
     /**
@@ -110,7 +148,27 @@
     @ApiOperation(value = "修改加盟商信息", tags = {"后台-加盟商管理"})
     @PostMapping(value = "/update")
     public R<String> update(@RequestBody @Validated Franchisee franchisee) {
-        checkFranchisee(franchisee);
+        String md5Password = checkFranchisee(franchisee);
+        SysUser sysUser = sysUserService.lambdaQuery()
+                .eq(SysUser::getFranchiseeId, franchisee.getId())
+                .eq(SysUser::getIsDelete, Constants.ZERO).one();
+        if (null == sysUser) {
+            sysUser = new SysUser();
+            sysUser.setNickName(franchisee.getName());
+            sysUser.setAccount(franchisee.getAdminAccount());
+            sysUser.setPassword(md5Password);
+            sysUser.setIsEnable(franchisee.getIsEnable());
+            sysUser.setIsDelete(Constants.ZERO);
+            sysUserService.save(sysUser);
+        } else {
+            // 生成sysUser账号
+            sysUser.setNickName(franchisee.getName());
+            sysUser.setAccount(franchisee.getAdminAccount());
+            sysUser.setPassword(md5Password);
+            sysUser.setIsEnable(franchisee.getIsEnable());
+            sysUserService.updateById(sysUser);
+        }
+
         return franchiseeService.updateById(franchisee) ? R.ok() : R.fail();
     }
 
@@ -130,6 +188,10 @@
     public R<String> enable(@RequestParam Integer id, @RequestParam Integer enable) {
         boolean update = franchiseeService.lambdaUpdate().set(Franchisee::getIsEnable, enable)
                 .eq(Franchisee::getId, id).update();
+        // 启动/关闭后台账号
+        sysUserService.lambdaUpdate()
+                .eq(SysUser::getFranchiseeId, id)
+                .set(SysUser::getIsEnable, enable).update();
         return update ? R.ok() : R.fail();
     }
 
@@ -138,20 +200,39 @@
      *
      * @param franchisee 加盟商信息
      */
-    private void checkFranchisee(Franchisee franchisee) {
-        String city = franchisee.getCity();
-        List<String> cityList = Arrays.stream(city.split(",")).collect(Collectors.toList());
-        for (String c : cityList) {
-            Franchisee one = franchiseeService.lambdaQuery().like(Franchisee::getCity, c)
-                    .eq(Franchisee::getIsDelete, 0).one();
-            if (null != one) {
-                throw new GlobalException("当前所选城市中 " + c + " 已有加盟商所管辖!");
-            }
+    private String checkFranchisee(Franchisee franchisee) {
+        LambdaQueryChainWrapper<Franchisee> wrapper = franchiseeService.lambdaQuery()
+                .eq(Franchisee::getAdminAccount, franchisee.getAdminAccount())
+                .eq(Franchisee::getIsDelete, 0);
+        // 校验账号是否唯一
+        Franchisee only;
+        if (null == franchisee.getId()) {
+            only = wrapper.one();
+        } else {
+            only = wrapper.ne(Franchisee::getId, franchisee.getId()).one();
         }
+        // 校验后台账号是否存在当前账号
+        SysUser user = sysUserService.lambdaQuery()
+                .eq(SysUser::getAccount, franchisee.getAdminAccount())
+                .eq(SysUser::getIsDelete, Constants.ZERO)
+                .eq(SysUser::getFranchiseeId, null).one();
+        if (null != only || null != user) {
+            throw new GlobalException("该账号已存在!");
+        }
+//        String city = franchisee.getCity();
+//        List<String> cityList = Arrays.stream(city.split(",")).collect(Collectors.toList());
+//        for (String c : cityList) {
+//            Franchisee one = franchiseeService.lambdaQuery().like(Franchisee::getCity, c)
+//                    .eq(Franchisee::getIsDelete, 0).one();
+//            if (null != one) {
+//                throw new GlobalException("当前所选城市中 " + c + " 已有加盟商所管辖!");
+//            }
+//        }
         // MD5加密登录密码(新)
         BCryptPasswordEncoder passwordEncoder = new BCryptPasswordEncoder();
-        String md5Password = passwordEncoder.encode(Constants.DEFAULT_PASSWORD);
+        String md5Password = passwordEncoder.encode(franchisee.getAdminPassword());
         franchisee.setAdminPassword(md5Password);
+        return md5Password;
     }
 
     /**
@@ -169,6 +250,10 @@
         List<String> idList = Arrays.stream(ids.split(",")).collect(Collectors.toList());
         List<Franchisee> list = franchiseeService.lambdaQuery().in(Franchisee::getId, idList).list();
         list.forEach(data -> data.setIsDelete(1));
+        // 删除对应sysUser账号
+        sysUserService.lambdaUpdate()
+                .in(SysUser::getFranchiseeId, idList)
+                .set(SysUser::getIsDelete, Constants.ONE).update();
         return franchiseeService.updateBatchById(list) ? R.ok() : R.fail();
     }
 

--
Gitblit v1.7.1