From 1c940b2c9b9cb8f8591045e5048eac8de5923a14 Mon Sep 17 00:00:00 2001
From: mitao <2763622819@qq.com>
Date: 星期三, 04 十二月 2024 18:52:21 +0800
Subject: [PATCH] 系统管理:区域管理接口

---
 medicalWaste-admin/src/main/java/com/sinata/web/controller/backend/system/SysLoginController.java |   24 ++++++++++++++++++++++--
 1 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/medicalWaste-admin/src/main/java/com/sinata/web/controller/backend/system/SysLoginController.java b/medicalWaste-admin/src/main/java/com/sinata/web/controller/backend/system/SysLoginController.java
index 21944f6..ba386c5 100644
--- a/medicalWaste-admin/src/main/java/com/sinata/web/controller/backend/system/SysLoginController.java
+++ b/medicalWaste-admin/src/main/java/com/sinata/web/controller/backend/system/SysLoginController.java
@@ -3,15 +3,20 @@
 import com.sinata.common.constant.Constants;
 import com.sinata.common.core.domain.AjaxResult;
 import com.sinata.common.core.domain.entity.SysMenu;
+import com.sinata.common.core.domain.entity.SysRole;
 import com.sinata.common.core.domain.entity.SysUser;
 import com.sinata.common.core.domain.model.LoginBody;
+import com.sinata.common.core.domain.model.LoginUser;
 import com.sinata.common.utils.SecurityUtils;
 import com.sinata.framework.web.service.SysLoginService;
 import com.sinata.framework.web.service.SysPermissionService;
+import com.sinata.framework.web.service.TokenService;
 import com.sinata.system.service.ISysMenuService;
+import com.sinata.system.service.ISysRoleService;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.util.CollectionUtils;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
@@ -40,6 +45,12 @@
     @Autowired
     private SysPermissionService permissionService;
 
+    @Autowired
+    private TokenService tokenService;
+
+    @Autowired
+    private ISysRoleService roleService;
+
     /**
      * 登录方法
      * 
@@ -52,9 +63,18 @@
     {
         AjaxResult ajax = AjaxResult.success();
         // 生成令牌
-        String token = loginService.login(loginBody.getUsername(), loginBody.getPassword(), loginBody.getCode(),
+        LoginUser loginUser = loginService.login(loginBody.getUsername(), loginBody.getPassword(), loginBody.getCode(),
                 loginBody.getUuid());
-        ajax.put(Constants.TOKEN, token);
+        ajax.put(Constants.TOKEN, tokenService.createToken(loginUser));
+        List<SysRole> roles = loginUser.getUser().getRoles();
+        if (CollectionUtils.isEmpty(roles)) {
+            return AjaxResult.error("请关联角色!");
+        }
+
+        List<SysMenu> menus = roleService.roleInfoFromUserId(loginUser.getUserId());
+
+        ajax.put("menus", menus);
+        ajax.put("userInfo", loginUser);
         return ajax;
     }
 

--
Gitblit v1.7.1