From 566c44cc5712ab9c997424ee5a5438e0a81a016e Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期三, 17 五月 2023 18:49:38 +0800
Subject: [PATCH] BUG修改

---
 management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TBranchOfficeController.java |   64 ++++++++++++++++++++++++++++++++
 1 files changed, 64 insertions(+), 0 deletions(-)

diff --git a/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TBranchOfficeController.java b/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TBranchOfficeController.java
index bf1f4eb..8e7f977 100644
--- a/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TBranchOfficeController.java
+++ b/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TBranchOfficeController.java
@@ -1,10 +1,13 @@
 package com.stylefeng.guns.modular.system.controller.general;
 
+import cn.hutool.crypto.SecureUtil;
 import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.mapper.EntityWrapper;
 import com.stylefeng.guns.core.base.controller.BaseController;
 import com.stylefeng.guns.core.base.tips.SuccessTip;
+import com.stylefeng.guns.core.common.constant.state.ManagerStatus;
 import com.stylefeng.guns.core.shiro.ShiroKit;
+import com.stylefeng.guns.core.util.SinataUtil;
 import com.stylefeng.guns.modular.system.controller.resp.TBranchOfficeResp;
 import com.stylefeng.guns.modular.system.controller.resp.TDriverResp;
 import com.stylefeng.guns.modular.system.enums.StatusEnum;
@@ -14,6 +17,8 @@
 import io.swagger.annotations.ApiOperation;
 import org.springframework.beans.BeanUtils;
 import org.springframework.stereotype.Controller;
+import org.springframework.transaction.annotation.Propagation;
+import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.CollectionUtils;
 import org.springframework.util.StringUtils;
 import org.springframework.web.bind.annotation.RequestMapping;
@@ -52,6 +57,8 @@
     private ITDriverService tDriverService;
     @Autowired
     private ITDriverWorkService tDriverWorkService;
+    @Autowired
+    private IUserService userService;
     @Autowired
     private RedisUtil redisUtil;
 
@@ -290,10 +297,15 @@
      */
     @RequestMapping(value = "/add")
     @ResponseBody
+    @Transactional(rollbackFor = Exception.class,propagation = Propagation.REQUIRES_NEW)
     public Object add(TBranchOffice tBranchOffice) {
         int count = tBranchOfficeService.selectCount(new EntityWrapper<TBranchOffice>().eq("branchOfficeName", tBranchOffice.getBranchOfficeName()));
         if(count>0){
             return new SuccessTip(500,"该分公司名称已存在!");
+        }
+        int count1 = userService.selectCount(new EntityWrapper<User>().eq("account", tBranchOffice.getAccount()));
+        if (count1 > 0){
+            return new SuccessTip(500,"该账号已存在!");
         }
 
         Object o = tBranchOfficeService.addOrUpdate(tBranchOffice);
@@ -304,6 +316,22 @@
         tBranchOffice.setStatus(StatusEnum.NORMAL.getCode());
 
         tBranchOfficeService.insert(tBranchOffice);
+
+        //添加User对象
+        User user = new User();
+        user.setAccount(tBranchOffice.getAccount());
+        user.setSalt(ShiroKit.getRandomSalt(5));
+        user.setPassword(ShiroKit.md5(tBranchOffice.getPassword(), user.getSalt()));
+        user.setStatus(ManagerStatus.OK.getCode());
+        user.setRoleid("3");
+        user.setDeptid(25);
+        user.setCreatetime(new Date());
+        user.setRoleType(2);
+        user.setObjectId(tBranchOffice.getId());
+        user.setName(tBranchOffice.getPrincipal());
+        user.setSex(1);
+        userService.insert(user);
+
         return SUCCESS_TIP;
     }
 
@@ -324,7 +352,43 @@
      */
     @RequestMapping(value = "/update")
     @ResponseBody
+    @Transactional(rollbackFor = Exception.class,propagation = Propagation.REQUIRES_NEW)
     public Object update(TBranchOffice tBranchOffice) {
+        TBranchOffice branchOffice = tBranchOfficeService.selectById(tBranchOffice.getId());
+        User user = userService.selectOne(new EntityWrapper<User>()
+                .eq("role_type", 2)
+                .eq("object_id", branchOffice.getId())
+                .last("LIMIT 1"));
+        //判断账号是否已存在
+        if (SinataUtil.isNotEmpty(tBranchOffice.getAccount()) && SinataUtil.isNotEmpty(tBranchOffice.getPassword())){
+            if(Objects.isNull(user)){
+                //添加User对象
+                user.setAccount(tBranchOffice.getAccount());
+                user.setSalt(ShiroKit.getRandomSalt(5));
+                user.setPassword(ShiroKit.md5(tBranchOffice.getPassword(), user.getSalt()));
+                user.setStatus(ManagerStatus.OK.getCode());
+                user.setRoleid("3");
+                user.setDeptid(25);
+                user.setCreatetime(new Date());
+                user.setRoleType(2);
+                user.setObjectId(tBranchOffice.getId());
+                user.setName(tBranchOffice.getPrincipal());
+                user.setSex(1);
+                userService.insert(user);
+            }else {
+                if (!tBranchOffice.getAccount().equals(user.getAccount())){
+                    //判断账号是否已存在
+                    int count = userService.selectCount(new EntityWrapper<User>().eq("account", tBranchOffice.getAccount()));
+                    if (count > 0){
+                        return new SuccessTip(500,"该账号已存在!");
+                    }
+                }
+                user.setAccount(tBranchOffice.getAccount());
+                user.setPassword(ShiroKit.md5(user.getPassword(), user.getSalt()));
+                userService.updateById(user);
+            }
+        }
+
         TBranchOffice office = tBranchOfficeService.selectOne(new EntityWrapper<TBranchOffice>().eq("branchOfficeName", tBranchOffice.getBranchOfficeName())
                 .last("LIMIT 1"));
         if(Objects.nonNull(office) && !tBranchOffice.getId().equals(office.getId())){

--
Gitblit v1.7.1