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

Go to download

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

Core modules of Suuchi

There is a newer version: 0.3.5

package in.ashwanthkumar.suuchi.rpc

import com.google.protobuf.ByteString
import in.ashwanthkumar.suuchi.rpc.generated.SuuchiPutGrpc
import in.ashwanthkumar.suuchi.rpc.generated.SuuchiRPC.{PutResponse, PutRequest, GetRequest, GetResponse}
import in.ashwanthkumar.suuchi.store.{WriteStore, ReadStore}
import io.grpc.stub.StreamObserver

class SuuchiPutService(store: WriteStore) extends SuuchiPutGrpc.SuuchiPutImplBase {
  override def put(request: PutRequest, responseObserver: StreamObserver[PutResponse]): Unit = {
    val key = request.getKey.toByteArray
    val value = request.getValue.toByteArray

    val status = store.put(key, value)
    responseObserver.onNext(PutResponse.newBuilder().setStatus(status).build())
    responseObserver.onCompleted()
  }
}