2
2
3
3
import java .util .List ;
4
4
5
+ import org .apache .catalina .connector .Response ;
5
6
import org .springframework .http .HttpStatus ;
6
7
import org .springframework .http .ResponseEntity ;
7
8
import org .springframework .security .access .prepost .PreAuthorize ;
@@ -65,7 +66,7 @@ public AuthDto login(@RequestBody UserRequest loginRequest) {
65
66
66
67
AuthDto authResponse = new AuthDto ();
67
68
authResponse .setAccessToken ("Bearer " + jwtToken );
68
- authResponse .setRefreshToken (refreshTokenService .createRefreshToken (user ));
69
+ authResponse .setRefreshToken (refreshTokenService .createRefreshToken (user ). getData () );
69
70
authResponse .setUserId (user .getId ());
70
71
authResponse .setMessage ("Successed." );
71
72
authResponse .setFirstName (user .getFirstName ());
@@ -108,7 +109,7 @@ public ResponseEntity<AuthDto> register(@RequestBody UserRequestForRegister regi
108
109
109
110
authResponse .setMessage ("User successfully registered." );
110
111
authResponse .setAccessToken ("Bearer " + jwtToken );
111
- authResponse .setRefreshToken (refreshTokenService .createRefreshToken (user ));
112
+ authResponse .setRefreshToken (refreshTokenService .createRefreshToken (user ). getData () );
112
113
authResponse .setUserId (user .getId ());
113
114
authResponse .setUserName (user .getUserName ());
114
115
authResponse .setFirstName (user .getFirstName ());
@@ -118,30 +119,33 @@ public ResponseEntity<AuthDto> register(@RequestBody UserRequestForRegister regi
118
119
}
119
120
120
121
@ PostMapping ("/refresh" )
121
- public ResponseEntity <AuthDto > refresh (@ RequestBody RefreshDto refreshRequest ) {
122
- AuthDto authResponse = new AuthDto ();
122
+ public ResponseEntity <? > refresh (@ RequestBody RefreshDto refreshRequest ) {
123
+ AuthDto authDto = new AuthDto ();
123
124
124
- RefreshToken token = refreshTokenService .getByUser (refreshRequest .getUserId ());
125
+ RefreshToken token = refreshTokenService .getByUser (refreshRequest .getUserId ()). getData () ;
125
126
126
127
if (token .getToken ().equals (refreshRequest .getRefreshToken ()) &&
127
- !refreshTokenService .isRefreshExpired (token )) {
128
+ !refreshTokenService .isRefreshExpired (token ). getData () ) {
128
129
129
130
User user = token .getUser ();
130
131
131
132
String jwtToken = jwtTokenProvider .generateJwtTokenByUserId (user .getId ());
132
133
133
- authResponse .setMessage ("Token has been refreshed successfully." );
134
- authResponse .setAccessToken ("Bearer " + jwtToken );
135
- authResponse .setUserId (user .getId ());
136
- authResponse .setFirstName (user .getFirstName ());
137
- authResponse .setLastName (user .getLastName ());
138
- authResponse .setUserName (user .getUserName ());
139
- authResponse .setRefreshToken (token .getToken ());
134
+ authDto .setMessage ("Token has been refreshed successfully." );
135
+ authDto .setAccessToken ("Bearer " + jwtToken );
136
+ authDto .setUserId (user .getId ());
137
+ authDto .setFirstName (user .getFirstName ());
138
+ authDto .setLastName (user .getLastName ());
139
+ authDto .setUserName (user .getUserName ());
140
+ authDto .setRefreshToken (token .getToken ());
141
+
142
+ return ResponseEntity
143
+ .status (HttpStatus .OK )
144
+ .body (authDto );
140
145
141
- return new ResponseEntity <>(authResponse , HttpStatus .OK );
142
146
} else {
143
- authResponse .setMessage ("refresh token is not valid." );
144
- return new ResponseEntity <>(authResponse , HttpStatus .UNAUTHORIZED );
147
+ authDto .setMessage ("Refresh token is not valid." );
148
+ return new ResponseEntity <>(authDto , HttpStatus .UNAUTHORIZED );
145
149
}
146
150
}
147
151
0 commit comments