Merge pull request #260 from yashsinghcodes/quick

added opensearch health status

Author: frikky

health.go

@@ -20,6 +20,7 @@ import (
 	"github.com/Masterminds/semver"
 	"github.com/frikky/kin-openapi/openapi3"
 	uuid "github.com/satori/go.uuid"
+	"github.com/shuffle/opensearch-go/v4/opensearchapi"
 )
 
 type appConfig struct {
@@ -723,6 +724,21 @@ func RunOpsHealthCheck(resp http.ResponseWriter, request *http.Request) {
 		errorChannel <- err
 	}()
 
+	if project.Environment != "cloud" {
+		opensearchHealthChannel := make(chan opensearchapi.ClusterHealthResp)
+		go func() {
+			opensearchHealth, err := RunOpensearchOps(ctx)
+			if err != nil {
+				log.Printf("[ERROR] Failed running opensearch health check: %s", err)
+			}
+
+			opensearchHealthChannel <- *opensearchHealth
+			errorChannel <- err
+		}()
+
+		platformHealth.OpensearchOps = <-opensearchHealthChannel
+	}
+
 	// TODO: More testing for onprem health checks
 	if project.Environment == "cloud" {
 		openapiAppHealthChannel := make(chan AppHealth)
@@ -3431,6 +3447,20 @@ func HandleRerunExecutions(resp http.ResponseWriter, request *http.Request) {
 	resp.Write([]byte(fmt.Sprintf(`{"success": true, "reason": "Successfully RERAN %d executions"}`, total)))
 }
 
+func RunOpensearchOps(ctx context.Context) (*opensearchapi.ClusterHealthResp, error) {
+	if project.Environment == "cloud" {
+		return nil, errors.New("Not running opensearch health check")
+	}
+	req := opensearchapi.ClusterHealthReq{}
+	resp, err := project.Es.Cluster.Health(ctx, &req)
+	if err != nil {
+		log.Printf("[ERROR] Failed to query cluster health: %s", err)
+		return nil, err
+	}
+
+	return resp, nil
+}
+
 // Send in deleteall=true to delete ALL executions for the environment ID
 func HandleStopExecutions(resp http.ResponseWriter, request *http.Request) {
 	cors := HandleCors(resp, request)

structs.go

@@ -4,6 +4,8 @@ import (
 	"encoding/xml"
 	"sync"
 	"time"
+
+	"github.com/shuffle/opensearch-go/v4/opensearchapi"
 )
 
 type AppContext struct {
@@ -4288,13 +4290,14 @@ type LiveExecutionStatus struct {
 }
 
 type HealthCheck struct {
-	Success    bool            `json:"success"`
-	Updated    int64           `json:"updated"`
-	Apps       AppHealth       `json:"apps"`
-	Workflows  WorkflowHealth  `json:"workflows"`
-	PythonApps AppHealth       `json:"python_apps"`
-	Datastore  DatastoreHealth `json:"datastore"`
-	FileOps    FileHealth      `json:"fileops"`
+	Success			bool            `json:"success"`
+	Updated			int64           `json:"updated"`
+	Apps			AppHealth       `json:"apps"`
+	Workflows		WorkflowHealth  `json:"workflows"`
+	PythonApps		AppHealth       `json:"python_apps"`
+	Datastore		DatastoreHealth `json:"datastore"`
+	FileOps			FileHealth      `json:"fileops"`
+	OpensearchOps	opensearchapi.ClusterHealthResp `json:"opensearch"`
 }
 
 type HealthCheckDB struct {