bbit/AILab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修复AccessToken、RefreshToken均可更新Token的问题

Browse Source
This commit is contained in:
BBIT-Kai
2025-12-29 16:28:10 +08:00
parent e6a9a09492
commit e1fcc64236
1 changed files with 15 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ktor/src/main/kotlin/ink/snowflake/server/controller/User.kt
+15 -18
View File
@@ -76,7 +76,7 @@ fun Application.User(config: AppConfig) {
} else { } else {
// 账号密码不匹配 // 账号密码不匹配
BaseResponse(status = false, message = "账号密码不匹配，请重新登录", data = null) BaseResponse(status = false, message = "账号密码不匹配，请重新登录", data = null)
} }
} else { } else {
BaseResponse(status = false, message = "账号已被禁用，请联系管理员", data = null) BaseResponse(status = false, message = "账号已被禁用，请联系管理员", data = null)
} }
@@ -149,15 +149,16 @@ fun Application.User(config: AppConfig) {
message = "拿什么乱七八糟的东西跟我换Access Token呢，？？？", message = "拿什么乱七八糟的东西跟我换Access Token呢，？？？",
data = null data = null
) )
} } else {
val userId = UUID.fromString(decodedJWT.getClaim("user_id").asString()) val userId = UUID.fromString(decodedJWT.getClaim("user_id").asString())
// 生成新的access token和refresh token // 生成新的access token和refresh token
BaseResponse( BaseResponse(
status = true, data = RefreshTokenResponse( status = true, data = RefreshTokenResponse(
generateAccessToken(config, userId), generateAccessToken(config, userId),
generateRefreshToken(config, userId) generateRefreshToken(config, userId)
)
) )
) }
} catch (ex: Exception) { } catch (ex: Exception) {
BaseResponse(status = false, message = "token解析错误", data = null) BaseResponse(status = false, message = "token解析错误", data = null)
} }
@@ -166,16 +167,12 @@ fun Application.User(config: AppConfig) {
authenticate { authenticate {
get("/getUserInfo") { get("/getUserInfo") {
val userId = getUserIdByToken(call) val userId = getUserIdByToken(call)
if (userId != null) { val userInfo = UserDAO.getUserInfoByUserId(userId)
val userInfo = UserDAO.getUserInfoByUserId(userId) if (userInfo != null) {
if (userInfo != null) { val response = BaseResponse(data = userInfo)
val response = BaseResponse(data = userInfo) call.respond(response)
call.respond(response)
} else {
call.respond(BaseResponse(data = "查无此人"))
}
} else { } else {
call.respond(BaseResponse(data = "Token出错")) call.respond(BaseResponse(data = "查无此人"))
} }
} }
} }