diff --git a/src/main/java/org/cmh/backend/authentication/controller/AuthenticationController.java b/src/main/java/org/cmh/backend/authentication/controller/AuthenticationController.java index 8d2fdb4..e58b5f1 100644 --- a/src/main/java/org/cmh/backend/authentication/controller/AuthenticationController.java +++ b/src/main/java/org/cmh/backend/authentication/controller/AuthenticationController.java @@ -59,8 +59,7 @@ class AuthenticationController { if (JwtUtil.isTokenValid(token)) { UserHS user = userService.getUserByUsername(JwtUtil.extractUsername(token)); if (user != null) { - - return new ResponseEntity<>(new UserProfileResponse( + UserProfileResponse response = new UserProfileResponse( user.getUsername(), user.getNickname(), user.getGender(), @@ -69,7 +68,12 @@ class AuthenticationController { user.getDepartment(), user.getRole(), user.getCreatedAt() - ), HttpStatus.OK); + ); + if (user.getSuperAdmin()) { + response.setDepartment("超级管理员"); + response.setRole("超级管理员"); + } + return new ResponseEntity<>(response, HttpStatus.OK); } } return new ResponseEntity<>(HttpStatus.NOT_FOUND); diff --git a/src/main/java/org/cmh/backend/authentication/model/UserHS.java b/src/main/java/org/cmh/backend/authentication/model/UserHS.java index 995186d..0b86a00 100644 --- a/src/main/java/org/cmh/backend/authentication/model/UserHS.java +++ b/src/main/java/org/cmh/backend/authentication/model/UserHS.java @@ -25,4 +25,5 @@ public class UserHS { private String department; private String role; private LocalDateTime createdAt; + private Boolean superAdmin = false; }