kv_storage.KvStorageGrpcKt.kt Maven / Gradle / Ivy

Go to download
package kv_storage

import com.google.bytestream.ByteStreamProto
import io.grpc.CallOptions
import io.grpc.CallOptions.DEFAULT
import io.grpc.Channel
import io.grpc.Metadata
import io.grpc.MethodDescriptor
import io.grpc.ServerServiceDefinition
import io.grpc.ServerServiceDefinition.builder
import io.grpc.ServiceDescriptor
import io.grpc.Status.UNIMPLEMENTED
import io.grpc.StatusException
import io.grpc.kotlin.AbstractCoroutineServerImpl
import io.grpc.kotlin.AbstractCoroutineStub
import io.grpc.kotlin.ClientCalls.clientStreamingRpc
import io.grpc.kotlin.ClientCalls.serverStreamingRpc
import io.grpc.kotlin.ServerCalls.clientStreamingServerMethodDefinition
import io.grpc.kotlin.ServerCalls.serverStreamingServerMethodDefinition
import io.grpc.kotlin.StubFor
import kotlin.String
import kotlin.coroutines.CoroutineContext
import kotlin.coroutines.EmptyCoroutineContext
import kotlin.jvm.JvmOverloads
import kotlin.jvm.JvmStatic
import kotlinx.coroutines.flow.Flow
import kv_storage.KVStorageGrpc.getServiceDescriptor

/**
 * Holder for Kotlin coroutine-based client and server APIs for kv_storage.KVStorage.
 */
public object KVStorageGrpcKt {
  public const val SERVICE_NAME: String = KVStorageGrpc.SERVICE_NAME

  @JvmStatic
  public val serviceDescriptor: ServiceDescriptor
    get() = getServiceDescriptor()

  public val getMethod: MethodDescriptor
    @JvmStatic
    get() = KVStorageGrpc.getGetMethod()

  public val putMethod:
      MethodDescriptor
    @JvmStatic
    get() = KVStorageGrpc.getPutMethod()

  /**
   * A stub for issuing RPCs to a(n) kv_storage.KVStorage service as suspending coroutines.
   */
  @StubFor(KVStorageGrpc::class)
  public class KVStorageCoroutineStub @JvmOverloads constructor(
    channel: Channel,
    callOptions: CallOptions = DEFAULT,
  ) : AbstractCoroutineStub(channel, callOptions) {
    override fun build(channel: Channel, callOptions: CallOptions): KVStorageCoroutineStub =
        KVStorageCoroutineStub(channel, callOptions)

    /**
     * Returns a [Flow] that, when collected, executes this RPC and emits responses from the
     * server as they arrive.  That flow finishes normally if the server closes its response with
     * [`Status.OK`][io.grpc.Status], and fails by throwing a [StatusException] otherwise.  If
     * collecting the flow downstream fails exceptionally (including via cancellation), the RPC
     * is cancelled with that exception as a cause.
     *
     * @param request The request message to send to the server.
     *
     * @param headers Metadata to attach to the request.  Most users will not need this.
     *
     * @return A flow that, when collected, emits the responses from the server.
     */
    public fun `get`(request: ByteStreamProto.ReadRequest, headers: Metadata = Metadata()):
        Flow = serverStreamingRpc(
      channel,
      KVStorageGrpc.getGetMethod(),
      request,
      callOptions,
      headers
    )

    /**
     * Executes this RPC and returns the response message, suspending until the RPC completes
     * with [`Status.OK`][io.grpc.Status].  If the RPC completes with another status, a
     * corresponding
     * [StatusException] is thrown.  If this coroutine is cancelled, the RPC is also cancelled
     * with the corresponding exception as a cause.
     *
     * This function collects the [Flow] of requests.  If the server terminates the RPC
     * for any reason before collection of requests is complete, the collection of requests
     * will be cancelled.  If the collection of requests completes exceptionally for any other
     * reason, the RPC will be cancelled for that reason and this method will throw that
     * exception.
     *
     * @param requests A [Flow] of request messages.
     *
     * @param headers Metadata to attach to the request.  Most users will not need this.
     *
     * @return The single response from the server.
     */
    public suspend fun put(requests: Flow, headers: Metadata =
        Metadata()): ByteStreamProto.WriteResponse = clientStreamingRpc(
      channel,
      KVStorageGrpc.getPutMethod(),
      requests,
      callOptions,
      headers
    )
  }

  /**
   * Skeletal implementation of the kv_storage.KVStorage service based on Kotlin coroutines.
   */
  public abstract class KVStorageCoroutineImplBase(
    coroutineContext: CoroutineContext = EmptyCoroutineContext,
  ) : AbstractCoroutineServerImpl(coroutineContext) {
    /**
     * Returns a [Flow] of responses to an RPC for kv_storage.KVStorage.Get.
     *
     * If creating or collecting the returned flow fails with a [StatusException], the RPC
     * will fail with the corresponding [io.grpc.Status].  If it fails with a
     * [java.util.concurrent.CancellationException], the RPC will fail with status
     * `Status.CANCELLED`.  If creating
     * or collecting the returned flow fails for any other reason, the RPC will fail with
     * `Status.UNKNOWN` with the exception as a cause.
     *
     * @param request The request from the client.
     */
    public open fun `get`(request: ByteStreamProto.ReadRequest): Flow
        = throw
        StatusException(UNIMPLEMENTED.withDescription("Method kv_storage.KVStorage.Get is unimplemented"))

    /**
     * Returns the response to an RPC for kv_storage.KVStorage.Put.
     *
     * If this method fails with a [StatusException], the RPC will fail with the corresponding
     * [io.grpc.Status].  If this method fails with a [java.util.concurrent.CancellationException],
     * the RPC will fail
     * with status `Status.CANCELLED`.  If this method fails for any other reason, the RPC will
     * fail with `Status.UNKNOWN` with the exception as a cause.
     *
     * @param requests A [Flow] of requests from the client.  This flow can be
     *        collected only once and throws [java.lang.IllegalStateException] on attempts to
     * collect
     *        it more than once.
     */
    public open suspend fun put(requests: Flow):
        ByteStreamProto.WriteResponse = throw
        StatusException(UNIMPLEMENTED.withDescription("Method kv_storage.KVStorage.Put is unimplemented"))

    final override fun bindService(): ServerServiceDefinition = builder(getServiceDescriptor())
      .addMethod(serverStreamingServerMethodDefinition(
      context = this.context,
      descriptor = KVStorageGrpc.getGetMethod(),
      implementation = ::`get`
    ))
      .addMethod(clientStreamingServerMethodDefinition(
      context = this.context,
      descriptor = KVStorageGrpc.getPutMethod(),
      implementation = ::put
    )).build()
  }
}