MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2025-12-13 15:52:01 +07:00
Code Issues Packages Projects Releases Wiki Activity

[fleet] move http ktor engine related code to rpc.server.ktor

Browse Source 
GitOrigin-RevId: 09aee497f8c5f532380918456c3a2a1a0f0e13cb
This commit is contained in:
Alexander Shparun
2024-09-23 04:00:12 +02:00
committed by intellij-monorepo-bot
parent 7968a25171
commit ca98770be4
3 changed files with 30 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
fleet/rpc.server/src/fleet/rpc/server/DirectRpcClient.kt
View File

@@ -8,40 +8,45 @@ import fleet.rpc.client.rpcClient
import fleet.rpc.core.Serialization import fleet.rpc.core.Serialization
import fleet.rpc.core.Transport import fleet.rpc.core.Transport
import fleet.rpc.core.TransportMessage import fleet.rpc.core.TransportMessage
import fleet.tracing.spannedScope
import fleet.util.UID import fleet.util.UID
import fleet.util.async.async import fleet.util.async.*
import fleet.util.async.resource
import fleet.util.async.use
import fleet.util.channels.channels import fleet.util.channels.channels
import kotlinx.coroutines.CoroutineScope import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.channels.Channel import kotlinx.coroutines.channels.Channel
import kotlinx.coroutines.launch import kotlinx.coroutines.launch
suspend fun RequestDispatcher.withDirectRpcClient(serialization: () -> Serialization, fun RequestDispatcher.directRpcClient(
interceptor: RpcInterceptor, serialization: () -> Serialization,
body: suspend CoroutineScope.(IRpcClient) -> Unit) { interceptor: RpcInterceptor,
spannedScope("withDirectRpcClient") { ): Resource<IRpcClient> =
resource { cc ->
val (dispatcherSend, clientReceive) = channels<TransportMessage>(Channel.BUFFERED) val (dispatcherSend, clientReceive) = channels<TransportMessage>(Channel.BUFFERED)
val (clientSend, dispatcherReceive) = channels<TransportMessage>(Channel.BUFFERED) val (clientSend, dispatcherReceive) = channels<TransportMessage>(Channel.BUFFERED)
val origin = UID.random() val origin = UID.random()
resource<IRpcClient> { continuation -> launch {
launch { handleConnection(route = origin,
handleConnection(route = origin, endpoint = EndpointKind.Client,
endpoint = EndpointKind.Client, send = dispatcherSend,
send = dispatcherSend, receive = dispatcherReceive,
receive = dispatcherReceive, presentableName = "directRpcClient")
presentableName = "directRpcClient") }.use {
}.use { rpcClient(transport = Transport(outgoing = clientSend, incoming = clientReceive),
rpcClient(transport = Transport(outgoing = clientSend, incoming = clientReceive), serialization = serialization,
serialization = serialization, origin = origin,
origin = origin, requestInterceptor = interceptor) { rpcClient ->
requestInterceptor = interceptor) { rpcClient -> cc(rpcClient)
continuation(rpcClient)
}
} }
}.async().use { rpcClientDeferred -> }
}.span("directRpcClient")
suspend fun RequestDispatcher.withDirectRpcClient(
serialization: () -> Serialization,
interceptor: RpcInterceptor,
body: suspend CoroutineScope.(IRpcClient) -> Unit,
) {
directRpcClient(serialization, interceptor)
.async()
.use { rpcClientDeferred ->
body(promisingRpcClient(rpcClientDeferred)) body(promisingRpcClient(rpcClientDeferred))
} }
}
} }

3
fleet/rpc.server/src/fleet/rpc/server/FleetService.kt
View File

@@ -16,7 +16,6 @@ class FleetService private constructor(val serviceId: UID,
job.join() job.join()
} }
//@fleet.kernel.plugins.InternalInPluginModules(where = ["fleet.app.fleet.tests"])
suspend fun terminate(cause: String) { suspend fun terminate(cause: String) {
terminate(CancellationException(cause)) terminate(CancellationException(cause))
} }
@@ -50,7 +49,7 @@ class FleetService private constructor(val serviceId: UID,
} }
} }
}.use { serviceJob -> }.use { serviceJob ->
body(FleetService(providerId, serviceJob)) body(FleetService(serviceId = providerId, job = serviceJob))
} }
} }
} }

1
fleet/rpc.server/src/fleet/rpc/server/RpcExecutor.kt
View File

@@ -28,7 +28,6 @@ import java.util.concurrent.ConcurrentHashMap
import kotlin.coroutines.EmptyCoroutineContext import kotlin.coroutines.EmptyCoroutineContext
import kotlin.coroutines.coroutineContext import kotlin.coroutines.coroutineContext
//@fleet.kernel.plugins.InternalInPluginModules(where = ["fleet.app.fleet.tests"])
class RpcExecutor private constructor(private val serialization: () -> Serialization, class RpcExecutor private constructor(private val serialization: () -> Serialization,
private val services: RpcServiceLocator, private val services: RpcServiceLocator,
private val route: UID, private val route: UID,