lbqms.git - Gitblit

			@@ -8,11 +8,16 @@
			import com.ruoyi.common.core.domain.entity.SysUser;
			import com.ruoyi.common.core.domain.model.LoginBody;
			import com.ruoyi.common.utils.SecurityUtils;
			import com.ruoyi.system.domain.vo.RouterVo;
			import com.ruoyi.system.service.ISysMenuService;
			import com.ruoyi.system.service.ISysUserService;
			import com.ruoyi.system.service.SysLoginService;
			import com.ruoyi.system.service.SysPermissionService;
			import io.swagger.annotations.Api;
			import io.swagger.annotations.ApiOperation;
			import lombok.RequiredArgsConstructor;
			import org.springframework.beans.factory.annotation.Autowired;
			import org.springframework.validation.annotation.Validated;
			import org.springframework.web.bind.annotation.GetMapping;
			import org.springframework.web.bind.annotation.PostMapping;
			import org.springframework.web.bind.annotation.RequestBody;
			@@ -26,22 +31,18 @@
			/**
			* 登录验证
			*
			* @author ruoyi
			* @author Lion Li
			*/
			@Validated
			@Api(value = "数据字典信息控制器", tags = {"数据字典信息管理"})
			@RequiredArgsConstructor(onConstructor_ = @Autowired)
			@RestController
			public class SysLoginController
			{
			@Autowired
			private SysLoginService loginService;
			public class SysLoginController {

			@Autowired
			private ISysMenuService menuService;

			@Autowired
			private SysPermissionService permissionService;

			@Autowired
			private ISysUserService iSysUserService;
			private final SysLoginService loginService;
			private final ISysMenuService menuService;
			private final SysPermissionService permissionService;
			private final ISysUserService iSysUserService;

			/**
			* 登录方法
			@@ -49,10 +50,10 @@
			* @param loginBody 登录信息
			* @return 结果
			*/
			@ApiOperation("登录方法")
			@PostMapping("/login")
			public AjaxResult login(@RequestBody LoginBody loginBody)
			{
			Map<String,Object> ajax = new HashMap<>();
			public AjaxResult<Map<String, Object>> login(@RequestBody LoginBody loginBody) {
			Map<String, Object> ajax = new HashMap<>();
			// 生成令牌
			String token = loginService.login(loginBody.getUsername(), loginBody.getPassword(), loginBody.getCode(),
			loginBody.getUuid());
			@@ -60,6 +61,7 @@
			return AjaxResult.success(ajax);
			}

			@ApiOperation("登出方法")
			@PostMapping("/logout")
			public AjaxResult logout(){
			try {
			@@ -74,15 +76,15 @@
			*
			* @return 用户信息
			*/
			@ApiOperation("获取用户信息")
			@GetMapping("getInfo")
			public AjaxResult getInfo()
			{
			public AjaxResult<Map<String, Object>> getInfo() {
			SysUser user = SecurityUtils.getUser();
			// 角色集合
			Set<String> roles = permissionService.getRolePermission(user);
			// 权限集合
			Set<String> permissions = permissionService.getMenuPermission(user);
			Map<String,Object> ajax = new HashMap<>();
			Map<String, Object> ajax = new HashMap<>();
			ajax.put("user", user);
			ajax.put("roles", roles);
			ajax.put("permissions", permissions);
			@@ -94,10 +96,11 @@
			*
			* @return 路由信息
			*/
			@ApiOperation("获取路由信息")
			@GetMapping("getRouters")
			public AjaxResult getRouters()
			{
			List<SysMenu> menus = menuService.selectMenuTreeByUserId(SecurityUtils.getUserId());
			public AjaxResult<List<RouterVo>> getRouters() {
			Long userId = SecurityUtils.getUserId();
			List<SysMenu> menus = menuService.selectMenuTreeByUserId(userId);
			return AjaxResult.success(menuService.buildMenus(menus));
			}
			}