ray650128/KtorMongoTest
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

優化sendUnauthorized、sendNotFound

Browse Source
This commit is contained in:
Raymond Yang 2023-05-16 10:36:54 +08:00
parent 50dbcf60e1
commit 07597ad5f2
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/main/kotlin/com/ray650128/extension/ResponseExtension.kt
View File

@ -20,10 +20,10 @@ suspend fun <T : Any> ApplicationCall.sendSuccess(data: T?) {
)
}
suspend fun ApplicationCall.sendUnauthorized(errorResponse: ErrorResponse) {
suspend fun ApplicationCall.sendUnauthorized() {
this.respond(
status = HttpStatusCode.Unauthorized,
message = errorResponse
message = ErrorResponse.UNAUTHORIZED_RESPONSE
)
}
@ -34,9 +34,9 @@ suspend fun ApplicationCall.sendBadRequest(errorResponse: ErrorResponse) {
)
}
suspend fun ApplicationCall.sendNotFound(errorResponse: ErrorResponse) {
suspend fun ApplicationCall.sendNotFound() {
this.respond(
status = HttpStatusCode.NotFound,
message = errorResponse
message = ErrorResponse.NOT_FOUND_RESPONSE
)
}

8
src/main/kotlin/com/ray650128/plugins/UserRouting.kt
View File

@ -50,18 +50,18 @@ fun Application.configureUserRouting() {
}
call.sendSuccess(LoginResult(request.account, token!!))
} else {
call.sendUnauthorized(ErrorResponse("Account or Password wrong."))
call.sendBadRequest(ErrorResponse("Account or Password wrong."))
}
}
authenticate {
post("/logout") {
val account = call.authentication.principal<UserDto>()?.account ?: run {
call.sendUnauthorized(ErrorResponse("Unauthorized"))
call.sendUnauthorized()
return@post
}
val user = service.findByAccount(account) ?: run {
call.sendUnauthorized(ErrorResponse("Unauthorized"))
call.sendUnauthorized()
return@post
}
user.apply {
@ -83,7 +83,7 @@ fun Application.configureUserRouting() {
val id = call.parameters["id"].toString()
service.findById(id)
?.let { foundPerson -> call.respond(foundPerson.toDto()) }
?: call.sendNotFound(ErrorResponse.NOT_FOUND_RESPONSE)
?: call.sendNotFound()
}
get("/search") {