gRPC/Common - Client/Server + Jvm Impl

Author: Mr3zee

grpc/grpc-core/src/commonMain/kotlin/kotlinx/rpc/grpc/GrpcTrailers.kt

@@ -0,0 +1,10 @@
+/*
+ * Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
+ */
+
+package kotlinx.rpc.grpc
+
+@Suppress("RedundantConstructorKeyword")
+public expect class GrpcTrailers constructor() {
+    public fun merge(trailers: GrpcTrailers)
+}

grpc/grpc-core/src/commonMain/kotlin/kotlinx/rpc/grpc/ManagedChannel.kt

@@ -6,12 +6,13 @@
 
 package kotlinx.rpc.grpc
 
+import kotlinx.rpc.grpc.internal.GrpcChannel
 import kotlin.time.Duration
 
 /**
  * Same as [ManagedChannel], but is platform-exposed.
  */
-public expect abstract class ManagedChannelPlatform
+public expect abstract class ManagedChannelPlatform : GrpcChannel
 
 /**
  * A virtual connection to a conceptual endpoint, to perform RPCs.
@@ -66,7 +67,9 @@ public interface ManagedChannel {
 /**
  * Builder class for [ManagedChannel].
  */
-public expect abstract class ManagedChannelBuilder<T : ManagedChannelBuilder<T>>
+public expect abstract class ManagedChannelBuilder<T : ManagedChannelBuilder<T>> {
+    public fun usePlaintext(): T
+}
 
 internal expect fun ManagedChannelBuilder(hostname: String, port: Int): ManagedChannelBuilder<*>
 internal expect fun ManagedChannelBuilder(target: String): ManagedChannelBuilder<*>

grpc/grpc-core/src/commonMain/kotlin/kotlinx/rpc/grpc/ServerServiceDefinition.kt

@@ -6,7 +6,20 @@
 
 package kotlinx.rpc.grpc
 
+import kotlinx.rpc.grpc.internal.ServerMethodDefinition
+import kotlinx.rpc.grpc.internal.ServiceDescriptor
+import kotlinx.rpc.internal.utils.InternalRpcApi
+
 /**
  * Definition of a service to be exposed via a Server.
  */
-public expect class ServerServiceDefinition
+public expect class ServerServiceDefinition {
+    public fun getServiceDescriptor(): ServiceDescriptor
+    public fun getMethods(): Collection<ServerMethodDefinition<*, *>>
+}
+
+@InternalRpcApi
+public expect fun serverServiceDefinition(
+    serviceDescriptor: ServiceDescriptor,
+    methods: Collection<ServerMethodDefinition<*, *>>
+): ServerServiceDefinition

grpc/grpc-core/src/commonMain/kotlin/kotlinx/rpc/grpc/Status.kt

@@ -7,15 +7,15 @@
 package kotlinx.rpc.grpc
 
 /**
- * Defines the status of an operation by providing a standard [Code] in conjunction with an
+ * Defines the status of an operation by providing a standard [StatusCode] in conjunction with an
  * optional descriptive message.
  *
  * For clients, every remote call will return a status on completion.
  * In the case of errors this
  * status may be propagated to blocking stubs as a [RuntimeException] or to a listener as an
  * explicit parameter.
  *
- * Similarly, servers can report a status by throwing [StatusRuntimeException]
+ * Similarly, servers can report a status by throwing [StatusException]
  * or by passing the status to a callback.
  *
  * Utility functions are provided to convert a status to an exception and to extract them
@@ -25,30 +25,33 @@ package kotlinx.rpc.grpc
  * can be found at
  * [doc/statuscodes.md](https://github.com/grpc/grpc/blob/master/doc/statuscodes.md)
  */
-public interface Status {
-    public val code: Code
-    public val description: String?
-    public val cause: Throwable?
+public expect class Status {
+    public fun getDescription(): String?
+    public fun getCause(): Throwable?
+}
+
+public expect fun Status(code: StatusCode, description: String? = null, cause: Throwable? = null): Status
+
+public expect val Status.code: StatusCode
 
-    public enum class Code(public val value: Int) {
-        OK(0),
-        CANCELLED(1),
-        UNKNOWN(2),
-        INVALID_ARGUMENT(3),
-        DEADLINE_EXCEEDED(4),
-        NOT_FOUND(5),
-        ALREADY_EXISTS(6),
-        PERMISSION_DENIED(7),
-        RESOURCE_EXHAUSTED(8),
-        FAILED_PRECONDITION(9),
-        ABORTED(10),
-        OUT_OF_RANGE(11),
-        UNIMPLEMENTED(12),
-        INTERNAL(13),
-        UNAVAILABLE(14),
-        DATA_LOSS(15),
-        UNAUTHENTICATED(16);
+public enum class StatusCode(public val value: Int) {
+    OK(0),
+    CANCELLED(1),
+    UNKNOWN(2),
+    INVALID_ARGUMENT(3),
+    DEADLINE_EXCEEDED(4),
+    NOT_FOUND(5),
+    ALREADY_EXISTS(6),
+    PERMISSION_DENIED(7),
+    RESOURCE_EXHAUSTED(8),
+    FAILED_PRECONDITION(9),
+    ABORTED(10),
+    OUT_OF_RANGE(11),
+    UNIMPLEMENTED(12),
+    INTERNAL(13),
+    UNAVAILABLE(14),
+    DATA_LOSS(15),
+    UNAUTHENTICATED(16);
 
-        public val valueAscii: ByteArray = value.toString().encodeToByteArray()
-    }
+    public val valueAscii: ByteArray = value.toString().encodeToByteArray()
 }

grpc/grpc-core/src/commonMain/kotlin/kotlinx/rpc/grpc/StatusException.kt

@@ -0,0 +1,24 @@
+/*
+ * Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
+ */
+
+package kotlinx.rpc.grpc
+
+/**
+ * [Status] in Exception form, for propagating Status information via exceptions.
+ */
+public expect class StatusException : Exception {
+    public constructor(status: Status)
+    public constructor(status: Status, trailers: GrpcTrailers?)
+
+    public fun getStatus(): Status
+    public fun getTrailers(): GrpcTrailers?
+}
+
+public expect class StatusRuntimeException : RuntimeException {
+    public constructor(status: Status)
+    public constructor(status: Status, trailers: GrpcTrailers?)
+
+    public fun getStatus(): Status
+    public fun getTrailers(): GrpcTrailers?
+}

grpc/grpc-core/src/commonMain/kotlin/kotlinx/rpc/grpc/StatusRuntimeException.kt

@@ -0,0 +1,35 @@
+/*
+ * Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
+ */
+
+package kotlinx.rpc.grpc.internal
+
+import kotlinx.rpc.grpc.GrpcTrailers
+import kotlinx.rpc.grpc.Status
+import kotlinx.rpc.internal.utils.InternalRpcApi
+
+@InternalRpcApi
+public expect abstract class ClientCall<Request, Response> {
+    @InternalRpcApi
+    public abstract class Listener<Message> {
+        public open fun onHeaders(headers: GrpcTrailers)
+        public open fun onMessage(message: Message)
+        public open fun onClose(status: Status, trailers: GrpcTrailers)
+        public open fun onReady()
+    }
+
+    public abstract fun start(responseListener: Listener<Response>, headers: GrpcTrailers)
+    public abstract fun request(numMessages: Int)
+    public abstract fun cancel(message: String?, cause: Throwable?)
+    public abstract fun halfClose()
+    public abstract fun sendMessage(message: Request)
+    public open fun isReady(): Boolean
+}
+
+@InternalRpcApi
+public expect fun <Message> clientCallListener(
+    onHeaders: (headers: GrpcTrailers) -> Unit,
+    onMessage: (message: Message) -> Unit,
+    onClose: (status: Status, trailers: GrpcTrailers) -> Unit,
+    onReady: () -> Unit,
+): ClientCall.Listener<Message>

grpc/grpc-core/src/commonMain/kotlin/kotlinx/rpc/grpc/internal/ClientCall.kt

@@ -0,0 +1,13 @@
+/*
+ * Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
+ */
+
+package kotlinx.rpc.grpc.internal
+
+import kotlinx.rpc.internal.utils.InternalRpcApi
+
+@InternalRpcApi
+public expect class GrpcCallOptions
+
+@InternalRpcApi
+public expect val GrpcDefaultCallOptions: GrpcCallOptions

grpc/grpc-core/src/commonMain/kotlin/kotlinx/rpc/grpc/internal/GrpcCallOptions.kt

@@ -0,0 +1,17 @@
+/*
+ * Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
+ */
+
+package kotlinx.rpc.grpc.internal
+
+import kotlinx.rpc.internal.utils.InternalRpcApi
+
+@InternalRpcApi
+public expect abstract class GrpcChannel {
+    public abstract fun <RequestT, ResponseT> newCall(
+        methodDescriptor: MethodDescriptor<RequestT, ResponseT>,
+        callOptions: GrpcCallOptions,
+    ): ClientCall<RequestT, ResponseT>
+
+    public abstract fun authority(): String
+}

grpc/grpc-core/src/commonMain/kotlin/kotlinx/rpc/grpc/internal/GrpcChannel.kt

@@ -0,0 +1,18 @@
+/*
+ * Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
+ */
+
+package kotlinx.rpc.grpc.internal
+
+import kotlin.coroutines.CoroutineContext
+
+internal expect class GrpcContext
+internal expect val CurrentGrpcContext: GrpcContext
+
+internal expect class GrpcContextElement : CoroutineContext.Element {
+    companion object Key : CoroutineContext.Key<GrpcContextElement> {
+        fun current(): GrpcContextElement
+    }
+
+    override val key: CoroutineContext.Key<GrpcContextElement>
+}