-
Notifications
You must be signed in to change notification settings - Fork 36
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
11 changed files
with
577 additions
and
2 deletions.
There are no files selected for viewing
41 changes: 41 additions & 0 deletions
41
rsocket-transport-ktor-websocket-client/api/rsocket-transport-ktor-websocket-client.api
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
244 changes: 244 additions & 0 deletions
244
.../kotlin/io/rsocket/kotlin/transport/ktor/websocket/client/KtorWebSocketClientTransport.kt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,244 @@ | ||
| /* | ||
| * Copyright 2015-2023 the original author or authors. | ||
| * | ||
| * Licensed under the Apache License, Version 2.0 (the "License"); | ||
| * you may not use this file except in compliance with the License. | ||
| * You may obtain a copy of the License at | ||
| * | ||
| * http://www.apache.org/licenses/LICENSE-2.0 | ||
| * | ||
| * Unless required by applicable law or agreed to in writing, software | ||
| * distributed under the License is distributed on an "AS IS" BASIS, | ||
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| * See the License for the specific language governing permissions and | ||
| * limitations under the License. | ||
| */ | ||
|
|
||
| package io.rsocket.kotlin.transport.ktor.websocket.client | ||
|
|
||
| import io.ktor.client.* | ||
| import io.ktor.client.engine.* | ||
| import io.ktor.client.plugins.websocket.* | ||
| import io.ktor.client.request.* | ||
| import io.ktor.http.* | ||
| import io.rsocket.kotlin.internal.io.* | ||
| import io.rsocket.kotlin.transport.* | ||
| import io.rsocket.kotlin.transport.ktor.websocket.* | ||
| import kotlinx.coroutines.* | ||
| import kotlin.coroutines.* | ||
|
|
||
| public sealed interface KtorWebSocketClientTransport : RSocketClientTransport { | ||
| public val url: Url | ||
| public val headers: Headers | ||
|
|
||
| public companion object Factory : RSocketTransportFactory< | ||
| HttpRequestBuilder.() -> Unit, | ||
| KtorWebSocketClientTransport, | ||
| KtorWebSocketClientTransportBuilder>(::KtorWebSocketClientTransportBuilderImpl) { | ||
|
|
||
| public operator fun invoke( | ||
| context: CoroutineContext, | ||
| urlString: String, | ||
| request: HttpRequestBuilder.() -> Unit = {}, | ||
| block: KtorWebSocketClientTransportBuilder.() -> Unit = {}, | ||
| ): KtorWebSocketClientTransport = invoke( | ||
| context = context, | ||
| method = HttpMethod.Get, host = null, port = null, path = null, | ||
| request = { | ||
| url.protocol = URLProtocol.WS | ||
| url.port = port | ||
|
|
||
| url.takeFrom(urlString) | ||
| request() | ||
| }, | ||
| block = block | ||
| ) | ||
|
|
||
| public operator fun invoke( | ||
| context: CoroutineContext, | ||
| method: HttpMethod = HttpMethod.Get, | ||
| host: String? = null, | ||
| port: Int? = null, | ||
| path: String? = null, | ||
| request: HttpRequestBuilder.() -> Unit = {}, | ||
| block: KtorWebSocketClientTransportBuilder.() -> Unit = {}, | ||
| ): KtorWebSocketClientTransport = invoke(context, { | ||
| this.method = method | ||
| url("ws", host, port, path) | ||
| request() | ||
| }, block) | ||
| } | ||
| } | ||
|
|
||
| public sealed interface KtorWebSocketClientTransportEngine : | ||
| RSocketTransportEngine<HttpRequestBuilder.() -> Unit, KtorWebSocketClientTransport> { | ||
|
|
||
| public fun createTransport( | ||
| urlString: String, | ||
| request: HttpRequestBuilder.() -> Unit = {}, | ||
| ): KtorWebSocketClientTransport = createTransport( | ||
| method = HttpMethod.Get, host = null, port = null, path = null, | ||
| ) { | ||
| url.protocol = URLProtocol.WS | ||
| url.port = port | ||
| url.takeFrom(urlString) | ||
| request() | ||
| } | ||
|
|
||
| public fun createTransport( | ||
| method: HttpMethod = HttpMethod.Get, | ||
| host: String? = null, | ||
| port: Int? = null, | ||
| path: String? = null, | ||
| request: HttpRequestBuilder.() -> Unit = {}, | ||
| ): KtorWebSocketClientTransport = createTransport { | ||
| this.method = method | ||
| url("ws", host, port, path) | ||
| request() | ||
| } | ||
|
|
||
| public companion object Factory : RSocketTransportEngineFactory< | ||
| HttpRequestBuilder.() -> Unit, | ||
| KtorWebSocketClientTransport, | ||
| KtorWebSocketClientTransportEngine, | ||
| KtorWebSocketClientTransportBuilder>(::KtorWebSocketClientTransportBuilderImpl) | ||
| } | ||
|
|
||
| public sealed interface KtorWebSocketClientTransportBuilder : | ||
| RSocketTransportBuilder<HttpRequestBuilder.() -> Unit, KtorWebSocketClientTransport>, | ||
| RSocketTransportEngineBuilder<HttpRequestBuilder.() -> Unit, KtorWebSocketClientTransport, KtorWebSocketClientTransportEngine> { | ||
|
|
||
| public fun httpEngine(configure: HttpClientEngineConfig.() -> Unit) | ||
| public fun httpEngine(engine: HttpClientEngine, configure: HttpClientEngineConfig.() -> Unit = {}) | ||
| public fun <T : HttpClientEngineConfig> httpEngine(factory: HttpClientEngineFactory<T>, configure: T.() -> Unit = {}) | ||
|
|
||
| public fun webSocketsConfig(block: WebSockets.Config.() -> Unit) | ||
| } | ||
|
|
||
| private class KtorWebSocketClientTransportBuilderImpl : KtorWebSocketClientTransportBuilder { | ||
| private var httpClientFactory: HttpClientFactory = HttpClientFactory.Default | ||
| private var webSocketsConfig: WebSockets.Config.() -> Unit = {} | ||
|
|
||
| override fun httpEngine(configure: HttpClientEngineConfig.() -> Unit) { | ||
| this.httpClientFactory = HttpClientFactory.FromConfiguration(configure) | ||
| } | ||
|
|
||
| override fun httpEngine(engine: HttpClientEngine, configure: HttpClientEngineConfig.() -> Unit) { | ||
| this.httpClientFactory = HttpClientFactory.FromEngine(engine, configure) | ||
| } | ||
|
|
||
| override fun <T : HttpClientEngineConfig> httpEngine(factory: HttpClientEngineFactory<T>, configure: T.() -> Unit) { | ||
| this.httpClientFactory = HttpClientFactory.FromFactory(factory, configure) | ||
| } | ||
|
|
||
| override fun webSocketsConfig(block: WebSockets.Config.() -> Unit) { | ||
| this.webSocketsConfig = block | ||
| } | ||
|
|
||
| private fun buildHttpClient(context: CoroutineContext): Pair<CoroutineContext, HttpClient> { | ||
| val httpClient = httpClientFactory.createHttpClient { | ||
| install(WebSockets, webSocketsConfig) | ||
| } | ||
| val newContext = httpClient.coroutineContext + context.supervisorContext() | ||
| val newJob = newContext.job | ||
| val httpClientJob = httpClient.coroutineContext.job | ||
|
|
||
| httpClientJob.invokeOnCompletion { newJob.cancel("HttpClient closed", it) } | ||
| newJob.invokeOnCompletion { httpClientJob.cancel("Transport closed", it) } | ||
|
|
||
| return newContext to httpClient | ||
| } | ||
|
|
||
| @RSocketTransportApi | ||
| override fun buildTransport(context: CoroutineContext, target: HttpRequestBuilder.() -> Unit): KtorWebSocketClientTransport { | ||
| val (newContext, httpClient) = buildHttpClient(context) | ||
| return KtorWebSocketClientTransportImpl( | ||
| coroutineContext = newContext, | ||
| requestBlock = target, | ||
| httpClient = httpClient, | ||
| ) | ||
| } | ||
|
|
||
| @RSocketTransportApi | ||
| override fun buildEngine(context: CoroutineContext): KtorWebSocketClientTransportEngine { | ||
| val (newContext, httpClient) = buildHttpClient(context) | ||
| return KtorWebSocketClientTransportEngineImpl( | ||
| coroutineContext = newContext, | ||
| httpClient = httpClient, | ||
| ) | ||
| } | ||
| } | ||
|
|
||
| private class KtorWebSocketClientTransportEngineImpl( | ||
| override val coroutineContext: CoroutineContext, | ||
| private val httpClient: HttpClient, | ||
| ) : KtorWebSocketClientTransportEngine { | ||
| override fun createTransport(target: HttpRequestBuilder.() -> Unit): KtorWebSocketClientTransport { | ||
| return KtorWebSocketClientTransportImpl( | ||
| coroutineContext = coroutineContext.supervisorContext(), | ||
| requestBlock = target, | ||
| httpClient = httpClient | ||
| ) | ||
| } | ||
| } | ||
|
|
||
| private class KtorWebSocketClientTransportImpl( | ||
| override val coroutineContext: CoroutineContext, | ||
| private val requestBlock: HttpRequestBuilder.() -> Unit, | ||
| private val httpClient: HttpClient, | ||
| ) : KtorWebSocketClientTransport { | ||
| private val requestData: HttpRequestData by lazy { | ||
| HttpRequestBuilder().apply { | ||
| url { | ||
| protocol = URLProtocol.WS | ||
| port = protocol.defaultPort | ||
| } | ||
| }.apply(requestBlock).build() | ||
| } | ||
| override val url: Url get() = requestData.url | ||
| override val headers: Headers get() = requestData.headers | ||
|
|
||
| @RSocketTransportApi | ||
| override suspend fun createSession(): RSocketTransportSession { | ||
| ensureActive() | ||
|
|
||
| return KtorWebSocketSession(httpClient.webSocketSession(requestBlock)) | ||
| } | ||
| } | ||
|
|
||
| private sealed class HttpClientFactory { | ||
| abstract fun createHttpClient(block: HttpClientConfig<*>.() -> Unit): HttpClient | ||
|
|
||
| object Default : HttpClientFactory() { | ||
| override fun createHttpClient(block: HttpClientConfig<*>.() -> Unit): HttpClient = HttpClient(block) | ||
| } | ||
|
|
||
| class FromConfiguration( | ||
| private val configure: HttpClientEngineConfig.() -> Unit, | ||
| ) : HttpClientFactory() { | ||
| override fun createHttpClient(block: HttpClientConfig<*>.() -> Unit): HttpClient = HttpClient { | ||
| engine(configure) | ||
| block() | ||
| } | ||
| } | ||
|
|
||
| class FromEngine( | ||
| private val engine: HttpClientEngine, | ||
| private val configure: HttpClientEngineConfig.() -> Unit, | ||
| ) : HttpClientFactory() { | ||
| override fun createHttpClient(block: HttpClientConfig<*>.() -> Unit): HttpClient = HttpClient(engine) { | ||
| engine(configure) | ||
| block() | ||
| } | ||
| } | ||
|
|
||
| class FromFactory<T : HttpClientEngineConfig>( | ||
| private val factory: HttpClientEngineFactory<T>, | ||
| private val configure: T.() -> Unit, | ||
| ) : HttpClientFactory() { | ||
| override fun createHttpClient(block: HttpClientConfig<*>.() -> Unit): HttpClient = HttpClient(factory) { | ||
| engine(configure) | ||
| block() | ||
| } | ||
| } | ||
| } |
31 changes: 31 additions & 0 deletions
31
rsocket-transport-ktor-websocket-server/api/rsocket-transport-ktor-websocket-server.api
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.