set status of grpc health status manager

Author: fm3

src/main/scala/com/scalableminds/fossildb/FossilDBServer.scala

@@ -5,7 +5,7 @@ package com.scalableminds.fossildb
 
 import com.scalableminds.fossildb.db.StoreManager
 import com.scalableminds.fossildb.proto.fossildbapi.FossilDBGrpc
-import io.grpc.health.v1.HealthGrpc
+import io.grpc.health.v1.{HealthCheckResponse, HealthGrpc}
 import com.typesafe.scalalogging.LazyLogging
 import io.grpc.Server
 import io.grpc.netty.NettyServerBuilder
@@ -24,6 +24,7 @@ class FossilDBServer(storeManager: StoreManager, port: Int, executionContext: Ex
       .addService(FossilDBGrpc.bindService(new FossilDBGrpcImpl(storeManager), executionContext))
       .addService(healthStatusManager.getHealthService())
       .build.start
+    healthStatusManager.setStatus("", HealthCheckResponse.ServingStatus.SERVING)
     logger.info("Server started, listening on " + port)
     sys.addShutdownHook {
       logger.info("Shutting down gRPC server since JVM is shutting down")
@@ -36,6 +37,7 @@ class FossilDBServer(storeManager: StoreManager, port: Int, executionContext: Ex
     if (server != null) {
       server.shutdown()
       storeManager.close
+      healthStatusManager.setStatus("", HealthCheckResponse.ServingStatus.NOT_SERVING)
     }
   }
 

src/test/scala/com/scalableminds/fossildb/FossilDBSuite.scala

@@ -9,6 +9,7 @@ import java.nio.file.Paths
 import com.google.protobuf.ByteString
 import com.scalableminds.fossildb.db.StoreManager
 import com.scalableminds.fossildb.proto.fossildbapi._
+import io.grpc.health.v1._
 import io.grpc.netty.NettyChannelBuilder
 import org.scalatest.{BeforeAndAfterEach, FlatSpec}
 
@@ -21,7 +22,9 @@ class FossilDBSuite extends FlatSpec with BeforeAndAfterEach {
 
   val port = 21505
   var serverOpt: Option[FossilDBServer] = None
-  val client = FossilDBGrpc.blockingStub(NettyChannelBuilder.forAddress("127.0.0.1", port).maxInboundMessageSize(Int.MaxValue).usePlaintext(true).build)
+  val channel = NettyChannelBuilder.forAddress("127.0.0.1", port).maxInboundMessageSize(Int.MaxValue).usePlaintext().build
+  val client = FossilDBGrpc.blockingStub(channel)
+  val healthClient = HealthGrpc.newBlockingStub(channel)
 
   val collectionA = "collectionA"
   val collectionB = "collectionB"
@@ -65,6 +68,11 @@ class FossilDBSuite extends FlatSpec with BeforeAndAfterEach {
     assert(reply.success)
   }
 
+  "GRPC Standard Health Check" should "report SERVING" in {
+    val reply = healthClient.check(HealthCheckRequest.getDefaultInstance())
+    assert(reply.getStatus.toString == "SERVING")
+  }
+
   "Put" should "overwrite old value" in {
     client.put(PutRequest(collectionA, aKey, Some(0), testData1))
     client.put(PutRequest(collectionA, aKey, Some(0), testData2))