grpc-native: Initial client setup

Signed-off-by: Johannes Zottele <[email protected]>

Author: Jozott00

cinterop-c/BUILD.bazel

@@ -1,13 +1,5 @@
 load("@rules_cc//cc:defs.bzl", "cc_library")
 
-cc_binary(
-    name = "testdemo",
-    srcs = ["src/main.cpp"],
-    deps = [
-        ":protowire",
-    ],
-)
-
 cc_static_library(
     name = "grpcpp_c_static",
     deps = [
@@ -24,6 +16,7 @@ cc_library(
     visibility = ["//visibility:public"],
     deps = [
         # TODO: Reduce the dependencies and only use required once. KRPC-185
+        "@com_github_grpc_grpc//:grpc",
         "@com_github_grpc_grpc//:channelz",
         "@com_github_grpc_grpc//:generic_stub",
         "@com_github_grpc_grpc//:grpc++",

cinterop-c/include/grpcpp_c.h

@@ -6,6 +6,7 @@
 #define GRPCPP_C_H
 
 #include <stdint.h>
+#include <grpc/grpc.h>
 #include <grpc/slice.h>
 #include <grpc/byte_buffer.h>
 
@@ -38,6 +39,14 @@ typedef enum StatusCode {
   GRPC_C_STATUS_DO_NOT_USE = -1
 } grpc_status_code_t;
 
+
+typedef struct {
+    grpc_completion_queue_functor functor;
+    void *user_data;
+} grpc_cb_tag;
+
+
+
 grpc_client_t *grpc_client_create_insecure(const char *target);
 void grpc_client_delete(const grpc_client_t *client);
 
@@ -59,6 +68,12 @@ uint32_t pb_decode_greeter_sayhello_response(grpc_slice response);
 
 grpc_status_code_t grpc_byte_buffer_dump_to_single_slice(grpc_byte_buffer *byte_buffer, grpc_slice *slice);
 
+
+/////// CHANNEL ///////
+
+typedef struct grpc_channel grpc_channel_t;
+typedef struct grpc_channel_credentials grpc_channel_credentials_t;
+
 #ifdef __cplusplus
     }
 #endif

cinterop-c/src/grpcpp_c.cpp

@@ -27,6 +27,10 @@ struct grpc_context {
     std::unique_ptr<grpc::ClientContext> context;
 };
 
+// struct grpc_channel {
+//     std::shared_ptr<grpc::Channel> channel;
+// };
+
 extern "C" {
 
     grpc_client_t *grpc_client_create_insecure(const char *target) {
@@ -196,6 +200,11 @@ extern "C" {
         return GRPC_C_STATUS_OK;
     }
 
+
+    //// CHANNEL ////
+
+
+
 }
 
 

grpc/grpc-core/src/nativeInterop/cinterop/libgrpcpp_c.def

@@ -1,5 +1,5 @@
-headers = grpcpp_c.h
-headerFilter= grpcpp_c.h grpc/slice.h grpc/byte_buffer.h
+headers = grpcpp_c.h grpc/grpc.h grpc/credentials.h
+headerFilter= grpcpp_c.h grpc/slice.h grpc/byte_buffer.h grpc/grpc.h grpc/impl/grpc_types.h grpc/credentials.h grpc/support/time.h
 
 noStringConversion = grpc_slice_from_copied_buffer my_grpc_slice_from_copied_buffer
 

grpc/grpc-core/src/nativeMain/kotlin/kotlinx/rpc/grpc/ManagedChannel.native.kt

@@ -3,10 +3,21 @@
  */
 
 @file:Suppress("EXPECT_ACTUAL_CLASSIFIERS_ARE_IN_BETA_WARNING")
+@file:OptIn(ExperimentalForeignApi::class, ExperimentalNativeApi::class)
 
 package kotlinx.rpc.grpc
 
+import cnames.structs.grpc_channel
+import kotlinx.cinterop.CPointer
+import kotlinx.cinterop.ExperimentalForeignApi
+import kotlinx.rpc.grpc.internal.ClientCall
+import kotlinx.rpc.grpc.internal.GrpcCallOptions
 import kotlinx.rpc.grpc.internal.GrpcChannel
+import kotlinx.rpc.grpc.internal.MethodDescriptor
+import libgrpcpp_c.*
+import kotlin.experimental.ExperimentalNativeApi
+import kotlin.native.ref.createCleaner
+import kotlin.time.Duration
 
 /**
  * Same as [ManagedChannel], but is platform-exposed.
@@ -23,7 +34,13 @@ public actual abstract class ManagedChannelBuilder<T : ManagedChannelBuilder<T>>
 }
 
 internal actual fun ManagedChannelBuilder<*>.buildChannel(): ManagedChannel {
-    error("Native target is not supported in gRPC")
+    return NativeManagedChannel(
+        target = "localhost:50051",
+        credentials = GrpcCredentials(
+            grpc_insecure_credentials_create()
+                ?: error("Failed to create credentials")
+        )
+    )
 }
 
 internal actual fun ManagedChannelBuilder(hostname: String, port: Int): ManagedChannelBuilder<*> {
@@ -33,3 +50,58 @@ internal actual fun ManagedChannelBuilder(hostname: String, port: Int): ManagedC
 internal actual fun ManagedChannelBuilder(target: String): ManagedChannelBuilder<*> {
     error("Native target is not supported in gRPC")
 }
+
+
+internal class NativeManagedChannel(
+    private val target: String,
+    // we must store them, otherwise the credentials are getting released
+    private val credentials: GrpcCredentials,
+) : ManagedChannel, ManagedChannelPlatform() {
+
+    internal val raw: CPointer<grpc_channel> = grpc_channel_create(target, credentials.raw, null)
+        ?: error("Failed to create channel")
+    private val rawCleaner = createCleaner(raw) {
+        grpc_channel_destroy(it)
+    }
+
+    override val platformApi: ManagedChannelPlatform = this
+
+    override val isShutdown: Boolean
+        get() = TODO("Not yet implemented")
+    override val isTerminated: Boolean
+        get() = TODO("Not yet implemented")
+
+    override suspend fun awaitTermination(duration: Duration): Boolean {
+        TODO("Not yet implemented")
+    }
+
+    override fun shutdown(): ManagedChannel {
+        TODO("Not yet implemented")
+    }
+
+    override fun shutdownNow(): ManagedChannel {
+        TODO("Not yet implemented")
+    }
+
+
+    override fun <RequestT, ResponseT> newCall(
+        methodDescriptor: MethodDescriptor<RequestT, ResponseT>,
+        callOptions: GrpcCallOptions,
+    ): ClientCall<RequestT, ResponseT> {
+        TODO("Not yet implemented")
+    }
+
+    override fun authority(): String {
+        TODO("Not yet implemented")
+    }
+
+}
+
+
+internal class GrpcCredentials(
+    internal val raw: CPointer<grpc_channel_credentials_t>,
+) {
+    val rawCleaner = createCleaner(raw) {
+        grpc_channel_credentials_release(it)
+    }
+}

grpc/grpc-core/src/nativeMain/kotlin/kotlinx/rpc/grpc/Status.native.kt

@@ -4,23 +4,21 @@
 
 package kotlinx.rpc.grpc
 
-public actual class Status {
-    public actual fun getDescription(): String? {
-        TODO("Not yet implemented")
-    }
+public actual class Status internal constructor(
+    private val description: String?,
+    internal val statusCode: StatusCode,
+    private val cause: Throwable?
+) {
+    public actual fun getDescription(): String? = description
 
-    public actual fun getCause(): Throwable? {
-        TODO("Not yet implemented")
-    }
+    public actual fun getCause(): Throwable? = cause
 }
 
 public actual val Status.code: StatusCode
-    get() = TODO("Not yet implemented")
+    get() = this.statusCode
 
 public actual fun Status(
     code: StatusCode,
     description: String?,
     cause: Throwable?,
-): Status {
-    TODO("Not yet implemented")
-}
+): Status = Status(description, code, cause)

grpc/grpc-core/src/nativeMain/kotlin/kotlinx/rpc/grpc/StatusException.native.kt

@@ -5,37 +5,39 @@
 package kotlinx.rpc.grpc
 
 public actual class StatusException : Exception {
-    public actual fun getStatus(): Status {
-        TODO("Not yet implemented")
-    }
+    private val status: Status
+    private val trailers: GrpcTrailers?
 
-    public actual fun getTrailers(): GrpcTrailers? {
-        TODO("Not yet implemented")
+    public actual constructor(status: Status) : super(status.getDescription()) {
+        this.status = status
+        this.trailers = null
     }
 
-    public actual constructor(status: Status) {
-        TODO("Not yet implemented")
+    public actual constructor(status: Status, trailers: GrpcTrailers?) : super(status.getDescription()) {
+        this.status = status
+        this.trailers = trailers
     }
 
-    public actual constructor(status: Status, trailers: GrpcTrailers?) {
-        TODO("Not yet implemented")
-    }
+    public actual fun getStatus(): Status = status
+
+    public actual fun getTrailers(): GrpcTrailers? = trailers
 }
 
 public actual class StatusRuntimeException : RuntimeException {
-    public actual fun getStatus(): Status {
-        TODO("Not yet implemented")
-    }
+    private val status: Status
+    private val trailers: GrpcTrailers?
 
-    public actual fun getTrailers(): GrpcTrailers? {
-        TODO("Not yet implemented")
+    public actual constructor(status: Status) : super(status.getDescription()) {
+        this.status = status
+        this.trailers = null
     }
 
-    public actual constructor(status: Status) {
-        TODO("Not yet implemented")
+    public actual constructor(status: Status, trailers: GrpcTrailers?) : super(status.getDescription()) {
+        this.status = status
+        this.trailers = trailers
     }
 
-    public actual constructor(status: Status, trailers: GrpcTrailers?) {
-        TODO("Not yet implemented")
-    }
+    public actual fun getStatus(): Status = status
+
+    public actual fun getTrailers(): GrpcTrailers? = trailers
 }

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

@@ -20,7 +20,8 @@ public actual abstract class ClientCall<Request, Response> {
     public actual abstract fun halfClose()
     public actual abstract fun sendMessage(message: Request)
     public actual open fun isReady(): Boolean {
-        TODO("Not yet implemented")
+        // Default implementation returns true - subclasses can override if they need flow control
+        return true
     }
 
     @InternalRpcApi
@@ -46,5 +47,21 @@ public actual fun <Message> clientCallListener(
     onClose: (status: Status, trailers: GrpcTrailers) -> Unit,
     onReady: () -> Unit,
 ): ClientCall.Listener<Message> {
-    TODO("Not yet implemented")
+    return object : ClientCall.Listener<Message>() {
+        override fun onHeaders(headers: GrpcTrailers) {
+            onHeaders(headers)
+        }
+
+        override fun onMessage(message: Message) {
+            onMessage(message)
+        }
+
+        override fun onClose(status: Status, trailers: GrpcTrailers) {
+            onClose(status, trailers)
+        }
+
+        override fun onReady() {
+            onReady()
+        }
+    }
 }