1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
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)
}
}
|
