Merge pull request #140 from checkr/zz/add-health-handler

Add health handler
openflagr · Jul 6, 2018 · 3e8098c · 3e8098c
2 parents ba42137 + 044b8c6
commit 3e8098c
Show file tree

Hide file tree

Showing 10 changed files with 371 additions and 0 deletions.
diff --git a/docs/api_docs/bundle.yaml b/docs/api_docs/bundle.yaml
@@ -718,6 +718,17 @@ paths:
           description: generic error response
           schema:
             $ref: '#/definitions/error'
+  /health:
+    get:
+      tags:
+        - health
+      responses:
+        '200':
+          description: Flagr is healthy
+        default:
+          description: generic error response
+          schema:
+            $ref: '#/definitions/error'
 definitions:
   flag:
     type: object

diff --git a/pkg/handler/handler.go b/pkg/handler/handler.go
@@ -7,8 +7,10 @@ import (
 	"github.com/checkr/flagr/swagger_gen/restapi/operations/distribution"
 	"github.com/checkr/flagr/swagger_gen/restapi/operations/evaluation"
 	"github.com/checkr/flagr/swagger_gen/restapi/operations/flag"
+	"github.com/checkr/flagr/swagger_gen/restapi/operations/health"
 	"github.com/checkr/flagr/swagger_gen/restapi/operations/segment"
 	"github.com/checkr/flagr/swagger_gen/restapi/operations/variant"
+	"github.com/go-openapi/runtime/middleware"
 )
 
 var getDB = repo.GetDB
@@ -17,6 +19,7 @@ var getDB = repo.GetDB
 func Setup(api *operations.FlagrAPI) {
 	setupCRUD(api)
 	setupEvaluation(api)
+	setupHealth(api)
 }
 
 func setupCRUD(api *operations.FlagrAPI) {
@@ -62,3 +65,11 @@ func setupEvaluation(api *operations.FlagrAPI) {
 	api.EvaluationPostEvaluationHandler = evaluation.PostEvaluationHandlerFunc(e.PostEvaluation)
 	api.EvaluationPostEvaluationBatchHandler = evaluation.PostEvaluationBatchHandlerFunc(e.PostEvaluationBatch)
 }
+
+func setupHealth(api *operations.FlagrAPI) {
+	api.HealthGetHealthHandler = health.GetHealthHandlerFunc(
+		func(health.GetHealthParams) middleware.Responder {
+			return &health.GetHealthOK{}
+		},
+	)
+}
diff --git a/swagger/health.yaml b/swagger/health.yaml
@@ -0,0 +1,10 @@
+get:
+  tags:
+    - health
+  responses:
+    200:
+      description: Flagr is healthy
+    default:
+      description: generic error response
+      schema:
+        $ref: "#/definitions/error"
diff --git a/swagger/index.yaml b/swagger/index.yaml
@@ -64,6 +64,8 @@ paths:
     $ref: ./evaluation.yaml
   /evaluation/batch:
     $ref: ./evaluation_batch.yaml
+  /health:
+    $ref: ./health.yaml
 definitions:
   # Flag
   flag:

diff --git a/swagger_gen/restapi/embedded_spec.go b/swagger_gen/restapi/embedded_spec.go
diff --git a/swagger_gen/restapi/operations/flagr_api.go b/swagger_gen/restapi/operations/flagr_api.go
diff --git a/swagger_gen/restapi/operations/health/get_health.go b/swagger_gen/restapi/operations/health/get_health.go
diff --git a/swagger_gen/restapi/operations/health/get_health_parameters.go b/swagger_gen/restapi/operations/health/get_health_parameters.go
diff --git a/swagger_gen/restapi/operations/health/get_health_responses.go b/swagger_gen/restapi/operations/health/get_health_responses.go