From 867126130946a9d3c3da7ff8e64b59b14da13599 Mon Sep 17 00:00:00 2001
From: Stan Hu <stanhu@gmail.com>
Date: Sun, 21 Jan 2018 15:34:09 -0800
Subject: Add a gRPC health check to ensure Gitaly is up

This will enable Geo to skip shards that not operational.

Relates to gitlab-org/gitlab-ee#4329
---
 lib/gitlab/gitaly_client/health_check_service.rb | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 lib/gitlab/gitaly_client/health_check_service.rb

(limited to 'lib/gitlab/gitaly_client/health_check_service.rb')

diff --git a/lib/gitlab/gitaly_client/health_check_service.rb b/lib/gitlab/gitaly_client/health_check_service.rb
new file mode 100644
index 00000000000..6c1213f5e20
--- /dev/null
+++ b/lib/gitlab/gitaly_client/health_check_service.rb
@@ -0,0 +1,19 @@
+module Gitlab
+  module GitalyClient
+    class HealthCheckService
+      def initialize(storage)
+        @storage = storage
+      end
+
+      # Sends a gRPC health ping to the Gitaly server for the storage shard.
+      def check
+        request = Grpc::Health::V1::HealthCheckRequest.new
+        response = GitalyClient.call(@storage, :health_check, :check, request, timeout: GitalyClient.fast_timeout)
+
+        { success: response&.status == :SERVING }
+      rescue GRPC::BadStatus => e
+        { success: false, message: e.to_s }
+      end
+    end
+  end
+end
-- 
cgit v1.2.3