in.ashwanthkumar.suuchi.rpc.SuuchiReadService.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of suuchi-examples Show documentation

Example modules using Suuchi

There is a newer version: 0.3.5

package in.ashwanthkumar.suuchi.rpc

import com.google.protobuf.ByteString
import in.ashwanthkumar.suuchi.examples.rpc.generated.{ReadGrpc, SuuchiRPC}
import in.ashwanthkumar.suuchi.examples.rpc.generated.SuuchiRPC.{GetRequest, GetResponse}
import in.ashwanthkumar.suuchi.store.ReadStore
import io.grpc.stub.StreamObserver

class SuuchiReadService(store: ReadStore) extends ReadGrpc.ReadImplBase {
  override def get(request: GetRequest, responseObserver: StreamObserver[GetResponse]): Unit = {
    val key = request.getKey.toByteArray
    store.get(key) match {
      case Some(value) =>
        responseObserver.onNext(
          SuuchiRPC.GetResponse
            .newBuilder()
            .setKey(ByteString.copyFrom(key))
            .setValue(ByteString.copyFrom(value))
            .build()
        )
        responseObserver.onCompleted()
      case None =>
        responseObserver.onNext(
          SuuchiRPC.GetResponse
            .newBuilder()
            .setKey(ByteString.copyFrom(key))
            .build()
        )
        responseObserver.onCompleted()
    }
  }
}