add HealthStatusManager

Author: fm3

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

@@ -6,7 +6,6 @@ package com.scalableminds.fossildb
 import java.io.{PrintWriter, StringWriter}
 
 import com.google.protobuf.ByteString
-import io.grpc.health.v1.HealthGrpc
 import com.scalableminds.fossildb.db.StoreManager
 import com.scalableminds.fossildb.proto.fossildbapi._
 import com.trueaccord.scalapb.GeneratedMessage
@@ -16,8 +15,7 @@ import scala.concurrent.Future
 
 class FossilDBGrpcImpl(storeManager: StoreManager)
   extends FossilDBGrpc.FossilDB
-  with LazyLogging
-  with HealthGrpc.HealthImplBase {
+  with LazyLogging {
 
   override def health(req: HealthRequest) = withExceptionHandler(req) {
     HealthReply(true)

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

@@ -5,18 +5,25 @@ package com.scalableminds.fossildb
 
 import com.scalableminds.fossildb.db.StoreManager
 import com.scalableminds.fossildb.proto.fossildbapi.FossilDBGrpc
+import io.grpc.health.v1.HealthGrpc
 import com.typesafe.scalalogging.LazyLogging
 import io.grpc.Server
 import io.grpc.netty.NettyServerBuilder
+import io.grpc.services.HealthStatusManager
 
 import scala.concurrent.ExecutionContext
 
 class FossilDBServer(storeManager: StoreManager, port: Int, executionContext: ExecutionContext) extends LazyLogging
 { self =>
   private[this] var server: Server = null
+  private[this] var healthStatusManager: HealthStatusManager = null
 
   def start(): Unit = {
-    server = NettyServerBuilder.forPort(port).maxMessageSize(Int.MaxValue).addService(FossilDBGrpc.bindService(new FossilDBGrpcImpl(storeManager), executionContext)).build.start
+    healthStatusManager = new HealthStatusManager()
+    server = NettyServerBuilder.forPort(port).maxMessageSize(Int.MaxValue)
+      .addService(FossilDBGrpc.bindService(new FossilDBGrpcImpl(storeManager), executionContext))
+      .addService(healthStatusManager.getHealthService())
+      .build.start
     logger.info("Server started, listening on " + port)
     sys.addShutdownHook {
       logger.info("Shutting down gRPC server since JVM is shutting down")