tursom/TursomServer
0
0
Fork 0
mirror of https://github.com/tursom/TursomServer.git synced 2025-02-15 06:10:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

更新 RoutedHttpHandler 功能

Browse Source
This commit is contained in:
tursom 2019-11-27 11:32:47 +08:00
parent b05a554459
commit 76fce89ca0
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
web/src/main/kotlin/cn/tursom/web/router/RoutedHttpHandler.kt
View File

@ -48,15 +48,15 @@ open class RoutedHttpHandler<T : HttpContent, in E : ExceptionContent>(
} }
fun addRouter(route: String, handler: (T) -> Unit) { fun addRouter(route: String, handler: (T) -> Unit) {
router[route] = handler router[safeRoute(route)] = handler
} }
fun addRouter(method: String, route: String, handler: (T) -> Unit) { fun addRouter(method: String, route: String, handler: (T) -> Unit) {
getRouter(method)[route] = handler getRouter(method)[safeRoute(route)] = handler
} }
fun deleteRouter(route: String, method: String) { fun deleteRouter(route: String, method: String) {
getRouter(method).delRoute(route) getRouter(method).delRoute(safeRoute(route))
} }
private fun insertMapping(method: Method) { private fun insertMapping(method: Method) {
@ -108,7 +108,7 @@ open class RoutedHttpHandler<T : HttpContent, in E : ExceptionContent>(
} }
routes.forEach { route -> routes.forEach { route ->
log?.info("method {} mapped to route {}", method, route) log?.info("method {} mapped to route {}", method, route)
router[route] = { content -> method(this, content) } router[safeRoute(route)] = { content -> method(this, content) }
} }
} }
} }
@ -132,5 +132,7 @@ open class RoutedHttpHandler<T : HttpContent, in E : ExceptionContent>(
} catch (e: Exception) { } catch (e: Exception) {
null null
} }
private fun safeRoute(route: String) = if (route.first() == '/') route else "/$route"
} }
} }