feat: 登录接口返回新增用户角色

This commit is contained in:
BenjaminNH 2025-06-21 23:23:12 +08:00
parent 98d938976b
commit 89335ea55b
2 changed files with 9 additions and 1 deletions

View File

@ -1,6 +1,7 @@
package github.benjamin.equipreservebackend.controller;
import github.benjamin.equipreservebackend.dto.LoginRequest;
import github.benjamin.equipreservebackend.entity.Role;
import github.benjamin.equipreservebackend.vo.LoginResponse;
import github.benjamin.equipreservebackend.response.ResponseResult;
import github.benjamin.equipreservebackend.entity.User;
@ -12,6 +13,8 @@ import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;
import java.util.List;
@RestController
@RequiredArgsConstructor
public class UserController {
@ -25,6 +28,7 @@ public class UserController {
User user = userService.login(request.getUsername(), request.getPassword());
SecurityUser securityUser = userService.loadSecurityUserById(user.getId());
String token = jwtUtil.generateToken(securityUser);
return ResponseResult.success(new LoginResponse(user.getId(), user.getName(), token));
List<String> roles = securityUser.getRoles().stream().map(Role::getCode).toList();
return ResponseResult.success(new LoginResponse(user.getId(), user.getName(), roles, token));
}
}

View File

@ -1,14 +1,18 @@
package github.benjamin.equipreservebackend.vo;
import github.benjamin.equipreservebackend.entity.Role;
import lombok.AllArgsConstructor;
import lombok.Data;
import java.util.List;
@Data
@AllArgsConstructor
public class LoginResponse {
private Long userId;
private String name;
private List<String> roles;
private String token;
}