diff options
author | Robert Jaros <rjaros@finn.pl> | 2020-04-18 16:08:26 +0200 |
---|---|---|
committer | Robert Jaros <rjaros@finn.pl> | 2020-04-18 16:08:26 +0200 |
commit | b3a0c96cce241a3f141dec2ffe773b24957d4040 (patch) | |
tree | 1b0b8d9aa29fce8be159256d3ddd63f7d42115ec /kvision-modules/kvision-server-vertx/src/jvmMain | |
parent | 68ebd29099f0bcc97d528d8829b1aba32b188839 (diff) | |
download | kvision-b3a0c96cce241a3f141dec2ffe773b24957d4040.tar.gz kvision-b3a0c96cce241a3f141dec2ffe773b24957d4040.tar.bz2 kvision-b3a0c96cce241a3f141dec2ffe773b24957d4040.zip |
Vert.x support (#141)
Diffstat (limited to 'kvision-modules/kvision-server-vertx/src/jvmMain')
3 files changed, 726 insertions, 0 deletions
diff --git a/kvision-modules/kvision-server-vertx/src/jvmMain/kotlin/pl/treksoft/kvision/remote/KVModules.kt b/kvision-modules/kvision-server-vertx/src/jvmMain/kotlin/pl/treksoft/kvision/remote/KVModules.kt new file mode 100644 index 00000000..a68c7693 --- /dev/null +++ b/kvision-modules/kvision-server-vertx/src/jvmMain/kotlin/pl/treksoft/kvision/remote/KVModules.kt @@ -0,0 +1,94 @@ +/* + * Copyright (c) 2017-present Robert Jaros + * + * Permission is hereby granted, free of charge, to any person obtaining a copy + * of this software and associated documentation files (the "Software"), to deal + * in the Software without restriction, including without limitation the rights + * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell + * copies of the Software, and to permit persons to whom the Software is + * furnished to do so, subject to the following conditions: + * + * The above copyright notice and this permission notice shall be included in all + * copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, + * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE + * SOFTWARE. + */ +package pl.treksoft.kvision.remote + +import com.google.inject.AbstractModule +import com.google.inject.Guice +import com.google.inject.Module +import io.vertx.core.Vertx +import io.vertx.core.http.HttpServer +import io.vertx.ext.web.Router +import io.vertx.ext.web.RoutingContext +import io.vertx.ext.web.handler.BodyHandler +import io.vertx.ext.web.handler.StaticHandler + +const val KV_INJECTOR_KEY = "pl.treksoft.kvision.injector.key" + +/** + * Initialization function for Vert.x server. + */ +fun Vertx.kvisionInit( + router: Router, + vararg modules: Module +) { + router.route("/*").last().handler(StaticHandler.create()) + router.route("/kv/*").handler(BodyHandler.create(false)) + @Suppress("SpreadOperator") + val injector = Guice.createInjector(MainModule(this), *modules) + + router.route("/kv/*").handler { rctx -> + rctx.put(KV_INJECTOR_KEY, injector.createChildInjector(RoutingContextModule(rctx))) + rctx.next() + } +} + +/** + * Initialization function for Vert.x server with support for WebSockets. + */ +fun Vertx.kvisionInit( + router: Router, + server: HttpServer, + wsServiceManagers: List<KVServiceManager<*>> = emptyList(), + vararg modules: Module +) { + router.route("/*").last().handler(StaticHandler.create()) + router.route("/kv/*").handler(BodyHandler.create(false)) + + @Suppress("SpreadOperator") + val injector = Guice.createInjector(MainModule(this), *modules) + + router.route("/kv/*").handler { rctx -> + rctx.put(KV_INJECTOR_KEY, injector.createChildInjector(RoutingContextModule(rctx))) + rctx.next() + } + wsServiceManagers.forEach { serviceManager -> + if (serviceManager.webSocketRequests.isNotEmpty()) { + server.webSocketHandler { webSocket -> + serviceManager.webSocketRequests[webSocket.path()]?.let { + it(injector, webSocket) + } ?: webSocket.reject() + } + } + } +} + +internal class MainModule(private val vertx: Vertx) : AbstractModule() { + override fun configure() { + bind(Vertx::class.java).toInstance(vertx) + } +} + +internal class RoutingContextModule(private val rctx: RoutingContext) : AbstractModule() { + override fun configure() { + bind(RoutingContext::class.java).toInstance(rctx) + } +} diff --git a/kvision-modules/kvision-server-vertx/src/jvmMain/kotlin/pl/treksoft/kvision/remote/KVServiceManager.kt b/kvision-modules/kvision-server-vertx/src/jvmMain/kotlin/pl/treksoft/kvision/remote/KVServiceManager.kt new file mode 100644 index 00000000..c64689b1 --- /dev/null +++ b/kvision-modules/kvision-server-vertx/src/jvmMain/kotlin/pl/treksoft/kvision/remote/KVServiceManager.kt @@ -0,0 +1,586 @@ +/* + * Copyright (c) 2017-present Robert Jaros + * + * Permission is hereby granted, free of charge, to any person obtaining a copy + * of this software and associated documentation files (the "Software"), to deal + * in the Software without restriction, including without limitation the rights + * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell + * copies of the Software, and to permit persons to whom the Software is + * furnished to do so, subject to the following conditions: + * + * The above copyright notice and this permission notice shall be included in all + * copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, + * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE + * SOFTWARE. + */ +package pl.treksoft.kvision.remote + +import com.fasterxml.jackson.databind.module.SimpleModule +import com.fasterxml.jackson.module.kotlin.jacksonObjectMapper +import com.fasterxml.jackson.module.kotlin.readValue +import com.google.inject.Injector +import io.vertx.core.Vertx +import io.vertx.core.http.ServerWebSocket +import io.vertx.core.json.Json +import io.vertx.core.logging.Logger +import io.vertx.core.logging.LoggerFactory +import io.vertx.ext.web.Router +import io.vertx.ext.web.RoutingContext +import io.vertx.kotlin.coroutines.dispatcher +import kotlinx.coroutines.ExperimentalCoroutinesApi +import kotlinx.coroutines.GlobalScope +import kotlinx.coroutines.channels.Channel +import kotlinx.coroutines.channels.ReceiveChannel +import kotlinx.coroutines.channels.SendChannel +import kotlinx.coroutines.coroutineScope +import kotlinx.coroutines.launch +import pl.treksoft.kvision.types.* +import java.math.BigDecimal +import java.time.LocalDate +import java.time.LocalDateTime +import java.time.LocalTime +import java.time.OffsetDateTime +import java.time.OffsetTime +import kotlin.reflect.KClass + +/** + * Multiplatform service manager for Vert.x. + */ +@Suppress("LargeClass", "TooManyFunctions", "BlockingMethodInNonBlockingContext") +actual open class KVServiceManager<T : Any> actual constructor(val serviceClass: KClass<T>) { + + companion object { + val LOG: Logger = LoggerFactory.getLogger(KVServiceManager::class.java.name) + const val KV_WS_INCOMING_KEY = "pl.treksoft.kvision.ws.incoming.key" + const val KV_WS_OUTGOING_KEY = "pl.treksoft.kvision.ws.outgoing.key" + } + + val getRequests: MutableMap<String, (RoutingContext) -> Unit> = mutableMapOf() + val postRequests: MutableMap<String, (RoutingContext) -> Unit> = mutableMapOf() + val putRequests: MutableMap<String, (RoutingContext) -> Unit> = mutableMapOf() + val deleteRequests: MutableMap<String, (RoutingContext) -> Unit> = + mutableMapOf() + val optionsRequests: MutableMap<String, (RoutingContext) -> Unit> = + mutableMapOf() + val webSocketRequests: MutableMap<String, (Injector, ServerWebSocket) -> Unit> = + mutableMapOf() + + val mapper = jacksonObjectMapper().apply { + val module = SimpleModule() + module.addSerializer(LocalDateTime::class.java, LocalDateTimeSerializer()) + module.addSerializer(LocalDate::class.java, LocalDateSerializer()) + module.addSerializer(LocalTime::class.java, LocalTimeSerializer()) + module.addSerializer(OffsetDateTime::class.java, OffsetDateTimeSerializer()) + module.addSerializer(OffsetTime::class.java, OffsetTimeSerializer()) + module.addSerializer(BigDecimal::class.java, BigDecimalSerializer()) + module.addDeserializer(LocalDateTime::class.java, LocalDateTimeDeserializer()) + module.addDeserializer(LocalDate::class.java, LocalDateDeserializer()) + module.addDeserializer(LocalTime::class.java, LocalTimeDeserializer()) + module.addDeserializer(OffsetDateTime::class.java, OffsetDateTimeDeserializer()) + module.addDeserializer(OffsetTime::class.java, OffsetTimeDeserializer()) + module.addDeserializer(BigDecimal::class.java, BigDecimalDeserializer()) + this.registerModule(module) + } + var counter: Int = 0 + + /** + * Binds a given route with a function of the receiver. + * @param function a function of the receiver + * @param method a HTTP method + * @param route a route + */ + @Suppress("TooGenericExceptionCaught") + protected actual inline fun <reified RET> bind( + noinline function: suspend T.() -> RET, + method: HttpMethod, route: String? + ) { + val routeDef = route ?: "route${this::class.simpleName}${counter++}" + addRoute(method, "/kv/$routeDef") { ctx -> + val jsonRpcRequest = if (method == HttpMethod.GET) { + JsonRpcRequest(ctx.request().getParam("id").toInt(), "", listOf()) + } else { + ctx.bodyAsJson.mapTo(JsonRpcRequest::class.java) + } + val injector = ctx.get<Injector>(KV_INJECTOR_KEY) + val service = injector.getInstance(serviceClass.java) + GlobalScope.launch(ctx.vertx().dispatcher()) { + val response = try { + val result = function.invoke(service) + JsonRpcResponse( + id = jsonRpcRequest.id, + result = mapper.writeValueAsString(result) + ) + } catch (e: Exception) { + if (e !is ServiceException) LOG.error(e.message, e) + JsonRpcResponse( + id = jsonRpcRequest.id, error = e.message ?: "Error", + exceptionType = e.javaClass.canonicalName + ) + } + ctx.response().putHeader("Content-Type", "application/json").end(Json.encode(response)) + } + } + } + + /** + * Binds a given route with a function of the receiver. + * @param function a function of the receiver + * @param method a HTTP method + * @param route a route + */ + @Suppress("TooGenericExceptionCaught") + protected actual inline fun <reified PAR, reified RET> bind( + noinline function: suspend T.(PAR) -> RET, + method: HttpMethod, route: String? + ) { + if (method == HttpMethod.GET) + throw UnsupportedOperationException("GET method is only supported for methods without parameters") + val routeDef = route ?: "route${this::class.simpleName}${counter++}" + addRoute(method, "/kv/$routeDef") { ctx -> + val jsonRpcRequest = ctx.bodyAsJson.mapTo(JsonRpcRequest::class.java) + if (jsonRpcRequest.params.size == 1) { + val param = getParameter<PAR>(jsonRpcRequest.params[0]) + val injector = ctx.get<Injector>(KV_INJECTOR_KEY) + val service = injector.getInstance(serviceClass.java) + GlobalScope.launch(ctx.vertx().dispatcher()) { + val response = try { + val result = function.invoke(service, param) + JsonRpcResponse( + id = jsonRpcRequest.id, + result = mapper.writeValueAsString(result) + ) + } catch (e: Exception) { + if (e !is ServiceException) LOG.error(e.message, e) + JsonRpcResponse( + id = jsonRpcRequest.id, error = e.message ?: "Error", + exceptionType = e.javaClass.canonicalName + ) + } + ctx.response().putHeader("Content-Type", "application/json").end(Json.encode(response)) + } + } else { + ctx.response().putHeader("Content-Type", "application/json") + .end(Json.encode(JsonRpcResponse(id = jsonRpcRequest.id, error = "Invalid parameters"))) + } + } + } + + /** + * Binds a given route with a function of the receiver. + * @param function a function of the receiver + * @param method a HTTP method + * @param route a route + */ + @Suppress("TooGenericExceptionCaught") + protected actual inline fun <reified PAR1, reified PAR2, reified RET> bind( + noinline function: suspend T.(PAR1, PAR2) -> RET, + method: HttpMethod, route: String? + ) { + if (method == HttpMethod.GET) + throw UnsupportedOperationException("GET method is only supported for methods without parameters") + val routeDef = route ?: "route${this::class.simpleName}${counter++}" + addRoute(method, "/kv/$routeDef") { ctx -> + val jsonRpcRequest = ctx.bodyAsJson.mapTo(JsonRpcRequest::class.java) + if (jsonRpcRequest.params.size == 2) { + val param1 = getParameter<PAR1>(jsonRpcRequest.params[0]) + val param2 = getParameter<PAR2>(jsonRpcRequest.params[1]) + val injector = ctx.get<Injector>(KV_INJECTOR_KEY) + val service = injector.getInstance(serviceClass.java) + GlobalScope.launch(ctx.vertx().dispatcher()) { + val response = try { + val result = function.invoke(service, param1, param2) + JsonRpcResponse( + id = jsonRpcRequest.id, + result = mapper.writeValueAsString(result) + ) + } catch (e: Exception) { + if (e !is ServiceException) LOG.error(e.message, e) + JsonRpcResponse( + id = jsonRpcRequest.id, error = e.message ?: "Error", + exceptionType = e.javaClass.canonicalName + ) + } + ctx.response().putHeader("Content-Type", "application/json").end(Json.encode(response)) + } + } else { + ctx.response().putHeader("Content-Type", "application/json") + .end(Json.encode(JsonRpcResponse(id = jsonRpcRequest.id, error = "Invalid parameters"))) + } + } + } + + /** + * Binds a given route with a function of the receiver. + * @param function a function of the receiver + * @param method a HTTP method + * @param route a route + */ + @Suppress("TooGenericExceptionCaught") + protected actual inline fun <reified PAR1, reified PAR2, reified PAR3, reified RET> bind( + noinline function: suspend T.(PAR1, PAR2, PAR3) -> RET, + method: HttpMethod, route: String? + ) { + if (method == HttpMethod.GET) + throw UnsupportedOperationException("GET method is only supported for methods without parameters") + val routeDef = route ?: "route${this::class.simpleName}${counter++}" + addRoute(method, "/kv/$routeDef") { ctx -> + val jsonRpcRequest = ctx.bodyAsJson.mapTo(JsonRpcRequest::class.java) + @Suppress("MagicNumber") + if (jsonRpcRequest.params.size == 3) { + val param1 = getParameter<PAR1>(jsonRpcRequest.params[0]) + val param2 = getParameter<PAR2>(jsonRpcRequest.params[1]) + val param3 = getParameter<PAR3>(jsonRpcRequest.params[2]) + val injector = ctx.get<Injector>(KV_INJECTOR_KEY) + val service = injector.getInstance(serviceClass.java) + GlobalScope.launch(ctx.vertx().dispatcher()) { + val response = try { + val result = function.invoke(service, param1, param2, param3) + JsonRpcResponse( + id = jsonRpcRequest.id, + result = mapper.writeValueAsString(result) + ) + } catch (e: Exception) { + if (e !is ServiceException) LOG.error(e.message, e) + JsonRpcResponse( + id = jsonRpcRequest.id, error = e.message ?: "Error", + exceptionType = e.javaClass.canonicalName + ) + } + ctx.response().putHeader("Content-Type", "application/json").end(Json.encode(response)) + } + } else { + ctx.response().putHeader("Content-Type", "application/json") + .end(Json.encode(JsonRpcResponse(id = jsonRpcRequest.id, error = "Invalid parameters"))) + } + } + } + + /** + * Binds a given route with a function of the receiver. + * @param function a function of the receiver + * @param method a HTTP method + * @param route a route + */ + @Suppress("TooGenericExceptionCaught") + protected actual inline fun <reified PAR1, reified PAR2, reified PAR3, reified PAR4, reified RET> bind( + noinline function: suspend T.(PAR1, PAR2, PAR3, PAR4) -> RET, + method: HttpMethod, route: String? + ) { + if (method == HttpMethod.GET) + throw UnsupportedOperationException("GET method is only supported for methods without parameters") + val routeDef = route ?: "route${this::class.simpleName}${counter++}" + addRoute(method, "/kv/$routeDef") { ctx -> + val jsonRpcRequest = ctx.bodyAsJson.mapTo(JsonRpcRequest::class.java) + @Suppress("MagicNumber") + if (jsonRpcRequest.params.size == 4) { + val param1 = getParameter<PAR1>(jsonRpcRequest.params[0]) + val param2 = getParameter<PAR2>(jsonRpcRequest.params[1]) + val param3 = getParameter<PAR3>(jsonRpcRequest.params[2]) + val param4 = getParameter<PAR4>(jsonRpcRequest.params[3]) + val injector = ctx.get<Injector>(KV_INJECTOR_KEY) + val service = injector.getInstance(serviceClass.java) + GlobalScope.launch(ctx.vertx().dispatcher()) { + val response = try { + val result = function.invoke(service, param1, param2, param3, param4) + JsonRpcResponse( + id = jsonRpcRequest.id, + result = mapper.writeValueAsString(result) + ) + } catch (e: Exception) { + if (e !is ServiceException) LOG.error(e.message, e) + JsonRpcResponse( + id = jsonRpcRequest.id, error = e.message ?: "Error", + exceptionType = e.javaClass.canonicalName + ) + } + ctx.response().putHeader("Content-Type", "application/json").end(Json.encode(response)) + } + } else { + ctx.response().putHeader("Content-Type", "application/json") + .end(Json.encode(JsonRpcResponse(id = jsonRpcRequest.id, error = "Invalid parameters"))) + } + } + } + + /** + * Binds a given route with a function of the receiver. + * @param function a function of the receiver + * @param method a HTTP method + * @param route a route + */ + @Suppress("TooGenericExceptionCaught") + protected actual inline fun <reified PAR1, reified PAR2, reified PAR3, + reified PAR4, reified PAR5, reified RET> bind( + noinline function: suspend T.(PAR1, PAR2, PAR3, PAR4, PAR5) -> RET, + method: HttpMethod, route: String? + ) { + if (method == HttpMethod.GET) + throw UnsupportedOperationException("GET method is only supported for methods without parameters") + val routeDef = route ?: "route${this::class.simpleName}${counter++}" + addRoute(method, "/kv/$routeDef") { ctx -> + val jsonRpcRequest = ctx.bodyAsJson.mapTo(JsonRpcRequest::class.java) + @Suppress("MagicNumber") + if (jsonRpcRequest.params.size == 5) { + val param1 = getParameter<PAR1>(jsonRpcRequest.params[0]) + val param2 = getParameter<PAR2>(jsonRpcRequest.params[1]) + val param3 = getParameter<PAR3>(jsonRpcRequest.params[2]) + val param4 = getParameter<PAR4>(jsonRpcRequest.params[3]) + val param5 = getParameter<PAR5>(jsonRpcRequest.params[4]) + val injector = ctx.get<Injector>(KV_INJECTOR_KEY) + val service = injector.getInstance(serviceClass.java) + GlobalScope.launch(ctx.vertx().dispatcher()) { + val response = try { + val result = function.invoke(service, param1, param2, param3, param4, param5) + JsonRpcResponse( + id = jsonRpcRequest.id, + result = mapper.writeValueAsString(result) + ) + } catch (e: Exception) { + if (e !is ServiceException) LOG.error(e.message, e) + JsonRpcResponse( + id = jsonRpcRequest.id, error = e.message ?: "Error", + exceptionType = e.javaClass.canonicalName + ) + } + ctx.response().putHeader("Content-Type", "application/json").end(Json.encode(response)) + } + } else { + ctx.response().putHeader("Content-Type", "application/json") + .end(Json.encode(JsonRpcResponse(id = jsonRpcRequest.id, error = "Invalid parameters"))) + } + } + } + + /** + * Binds a given route with a function of the receiver. + * @param function a function of the receiver + * @param method a HTTP method + * @param route a route + */ + @Suppress("TooGenericExceptionCaught") + protected actual inline fun <reified PAR1, reified PAR2, reified PAR3, + reified PAR4, reified PAR5, reified PAR6, reified RET> bind( + noinline function: suspend T.(PAR1, PAR2, PAR3, PAR4, PAR5, PAR6) -> RET, + method: HttpMethod, route: String? + ) { + if (method == HttpMethod.GET) + throw UnsupportedOperationException("GET method is only supported for methods without parameters") + val routeDef = route ?: "route${this::class.simpleName}${counter++}" + addRoute(method, "/kv/$routeDef") { ctx -> + val jsonRpcRequest = ctx.bodyAsJson.mapTo(JsonRpcRequest::class.java) + @Suppress("MagicNumber") + if (jsonRpcRequest.params.size == 6) { + val param1 = getParameter<PAR1>(jsonRpcRequest.params[0]) + val param2 = getParameter<PAR2>(jsonRpcRequest.params[1]) + val param3 = getParameter<PAR3>(jsonRpcRequest.params[2]) + val param4 = getParameter<PAR4>(jsonRpcRequest.params[3]) + val param5 = getParameter<PAR5>(jsonRpcRequest.params[4]) + val param6 = getParameter<PAR6>(jsonRpcRequest.params[5]) + val injector = ctx.get<Injector>(KV_INJECTOR_KEY) + val service = injector.getInstance(serviceClass.java) + GlobalScope.launch(ctx.vertx().dispatcher()) { + val response = try { + val result = function.invoke(service, param1, param2, param3, param4, param5, param6) + JsonRpcResponse( + id = jsonRpcRequest.id, + result = mapper.writeValueAsString(result) + ) + } catch (e: Exception) { + if (e !is ServiceException) LOG.error(e.message, e) + JsonRpcResponse( + id = jsonRpcRequest.id, error = e.message ?: "Error", + exceptionType = e.javaClass.canonicalName + ) + } + ctx.response().putHeader("Content-Type", "application/json").end(Json.encode(response)) + } + } else { + ctx.response().putHeader("Content-Type", "application/json") + .end(Json.encode(JsonRpcResponse(id = jsonRpcRequest.id, error = "Invalid parameters"))) + } + } + } + + /** + * Binds a given web socket connection with a function of the receiver. + * @param function a function of the receiver + * @param route a route + */ + @OptIn(ExperimentalCoroutinesApi::class) + protected actual inline fun <reified PAR1 : Any, reified PAR2 : Any> bind( + noinline function: suspend T.(ReceiveChannel<PAR1>, SendChannel<PAR2>) -> Unit, + route: String? + ) { + val routeDef = "route${this::class.simpleName}${counter++}" + webSocketRequests["/kvws/$routeDef"] = { injector, ws -> + val incoming = Channel<String>() + val outgoing = Channel<String>() + val service = injector.getInstance(serviceClass.java) + val vertx = injector.getInstance(Vertx::class.java) + ws.textMessageHandler { message -> + GlobalScope.launch(vertx.dispatcher()) { + incoming.send(message) + } + } + ws.closeHandler { + GlobalScope.launch(vertx.dispatcher()) { + outgoing.close() + incoming.close() + } + } + ws.exceptionHandler { + GlobalScope.launch(vertx.dispatcher()) { + outgoing.close() + incoming.close() + } + } + GlobalScope.launch(vertx.dispatcher()) { + coroutineScope { + launch { + for (text in outgoing) { + ws.writeTextMessage(text) + } + if (!ws.isClosed) ws.close() + } + launch { + val requestChannel = Channel<PAR1>() + val responseChannel = Channel<PAR2>() + coroutineScope { + launch { + for (p in incoming) { + val jsonRpcRequest = getParameter<JsonRpcRequest>(p) + if (jsonRpcRequest.params.size == 1) { + val par = getParameter<PAR1>(jsonRpcRequest.params[0]) + requestChannel.send(par) + } + } + requestChannel.close() + } + launch { + for (p in responseChannel) { + val text = mapper.writeValueAsString( + JsonRpcResponse( + id = 0, + result = mapper.writeValueAsString(p) + ) + ) + outgoing.send(text) + } + } + launch { + function.invoke(service, requestChannel, responseChannel) + if (!responseChannel.isClosedForReceive) responseChannel.close() + } + } + if (!outgoing.isClosedForReceive) outgoing.close() + } + } + } + } + } + + /** + * Binds a given function of the receiver as a tabulator component source + * @param function a function of the receiver + */ + @Suppress("TooGenericExceptionCaught") + protected actual inline fun <reified RET> bindTabulatorRemote( + noinline function: suspend T.(Int?, Int?, List<RemoteFilter>?, List<RemoteSorter>?, String?) -> RemoteData<RET> + ) { + val routeDef = "route${this::class.simpleName}${counter++}" + addRoute(HttpMethod.POST, "/kv/$routeDef") { ctx -> + val jsonRpcRequest = ctx.bodyAsJson.mapTo(JsonRpcRequest::class.java) + @Suppress("MagicNumber") + if (jsonRpcRequest.params.size == 5) { + val param1 = getParameter<Int?>(jsonRpcRequest.params[0]) + val param2 = getParameter<Int?>(jsonRpcRequest.params[1]) + val param3 = getParameter<List<RemoteFilter>?>(jsonRpcRequest.params[2]) + + @Suppress("MagicNumber") + val param4 = getParameter<List<RemoteSorter>?>(jsonRpcRequest.params[3]) + + @Suppress("MagicNumber") + val param5 = getParameter<String?>(jsonRpcRequest.params[4]) + val injector = ctx.get<Injector>(KV_INJECTOR_KEY) + val service = injector.getInstance(serviceClass.java) + GlobalScope.launch(ctx.vertx().dispatcher()) { + val response = try { + val result = function.invoke(service, param1, param2, param3, param4, param5) + JsonRpcResponse( + id = jsonRpcRequest.id, + result = mapper.writeValueAsString(result) + ) + } catch (e: Exception) { + if (e !is ServiceException) LOG.error(e.message, e) + JsonRpcResponse( + id = jsonRpcRequest.id, error = e.message ?: "Error", + exceptionType = e.javaClass.canonicalName + ) + } + ctx.response().putHeader("Content-Type", "application/json").end(Json.encode(response)) + } + } else { + ctx.response().putHeader("Content-Type", "application/json") + .end(Json.encode(JsonRpcResponse(id = jsonRpcRequest.id, error = "Invalid parameters"))) + } + } + } + + /** + * @suppress Internal method + */ + fun addRoute( + method: HttpMethod, + path: String, + handler: (RoutingContext) -> Unit + ) { + when (method) { + HttpMethod.GET -> getRequests[path] = handler + HttpMethod.POST -> postRequests[path] = handler + HttpMethod.PUT -> putRequests[path] = handler + HttpMethod.DELETE -> deleteRequests[path] = handler + HttpMethod.OPTIONS -> optionsRequests[path] = handler + } + } + + /** + * @suppress Internal method + */ + protected inline fun <reified T> getParameter(str: String?): T { + return str?.let { + if (T::class == String::class) { + str as T + } else { + mapper.readValue(str) + } + } ?: null as T + } + +} + +/** + * A function to generate routes based on definitions from the service manager. + */ +fun <T : Any> Vertx.applyRoutes(router: Router, serviceManager: KVServiceManager<T>) { + serviceManager.getRequests.forEach { (path, handler) -> + router.route(io.vertx.core.http.HttpMethod.GET, path).handler(handler) + } + serviceManager.postRequests.forEach { (path, handler) -> + router.route(io.vertx.core.http.HttpMethod.POST, path).handler(handler) + } + serviceManager.putRequests.forEach { (path, handler) -> + router.route(io.vertx.core.http.HttpMethod.PUT, path).handler(handler) + } + serviceManager.deleteRequests.forEach { (path, handler) -> + router.route(io.vertx.core.http.HttpMethod.DELETE, path).handler(handler) + } + serviceManager.optionsRequests.forEach { (path, handler) -> + router.route(io.vertx.core.http.HttpMethod.OPTIONS, path).handler(handler) + } +} diff --git a/kvision-modules/kvision-server-vertx/src/jvmMain/kotlin/pl/treksoft/kvision/remote/Security.kt b/kvision-modules/kvision-server-vertx/src/jvmMain/kotlin/pl/treksoft/kvision/remote/Security.kt new file mode 100644 index 00000000..188c1f5f --- /dev/null +++ b/kvision-modules/kvision-server-vertx/src/jvmMain/kotlin/pl/treksoft/kvision/remote/Security.kt @@ -0,0 +1,46 @@ +/* + * Copyright (c) 2017-present Robert Jaros + * + * Permission is hereby granted, free of charge, to any person obtaining a copy + * of this software and associated documentation files (the "Software"), to deal + * in the Software without restriction, including without limitation the rights + * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell + * copies of the Software, and to permit persons to whom the Software is + * furnished to do so, subject to the following conditions: + * + * The above copyright notice and this permission notice shall be included in all + * copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, + * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE + * SOFTWARE. + */ + +package pl.treksoft.kvision.remote + +import io.vertx.core.Handler +import io.vertx.core.http.HttpMethod +import io.vertx.ext.web.Router +import io.vertx.ext.web.RoutingContext + +fun Router.serviceRoute(service: KVServiceManager<*>, handler: Handler<RoutingContext>) { + service.getRequests.keys.forEach { + this.route(HttpMethod.GET, it).handler(handler) + } + service.postRequests.keys.forEach { + this.route(HttpMethod.POST, it).handler(handler) + } + service.putRequests.keys.forEach { + this.route(HttpMethod.PUT, it).handler(handler) + } + service.deleteRequests.keys.forEach { + this.route(HttpMethod.DELETE, it).handler(handler) + } + service.optionsRequests.keys.forEach { + this.route(HttpMethod.OPTIONS, it).handler(handler) + } +} |