feat: recording cluster allocation explain to activity after cluster …

…hea… (#18) * feat: saving cluster allocation explain to activity after cluster health changed to `red` * chore: update release notes
infinilabs · Dec 19, 2024 · de16731 · de16731
1 parent a9b0736
commit de16731
Show file tree

Hide file tree

Showing 4 changed files with 41 additions and 0 deletions.
diff --git a/core/elastic/api.go b/core/elastic/api.go
@@ -137,6 +137,7 @@ type API interface {
 	DeleteILMPolicy(target string) error
 	GetRemoteInfo()([]byte, error)
 	Flush(indexName string) ([]byte, error)
+	ClusterAllocationExplain(ctx context.Context, body []byte, params url.Values)([]byte,error)
 }
 
 type TemplateAPI interface {

diff --git a/docs/content.en/docs/release-notes/_index.md b/docs/content.en/docs/release-notes/_index.md
@@ -15,6 +15,12 @@ Information about release notes of INFINI Framework is provided here.
 
 ### Improvements
 
+## v1.0.1
+### Bug fix
+### Features
+- Record cluster allocation explain to activity after cluster health status changed to `red`
+- Add elastic api method `ClusterAllocationExplain`
+
 ## v1.0.0
 
 ### 🚀 Features

diff --git a/modules/elastic/adapter/elasticsearch/v0.go b/modules/elastic/adapter/elasticsearch/v0.go
@@ -1906,4 +1906,23 @@ func (c *ESAPIV0) PutScript(scriptName string, script []byte)([]byte,error){
 
 func (c *ESAPIV0)SearchByTemplate(indexName,scriptName string,params map[string]interface{}) (*elastic.SearchResponse, error)  {
 	panic("not implemented")
+}
+
+func (c *ESAPIV0) ClusterAllocationExplain(ctx context.Context, body []byte, params url.Values)([]byte,error){
+	url := fmt.Sprintf("%s/_cluster/allocation/explain", c.GetEndpoint())
+	if len(params) > 0 {
+		url = fmt.Sprintf("%s?%s", url, params.Encode())
+	}
+	method := util.Verb_GET
+	if len(body) > 0 {
+		method = util.Verb_POST
+	}
+	resp, err := c.Request(ctx, method, url, body)
+	if err != nil {
+		return nil, err
+	}
+	if resp.StatusCode != 200 {
+		return nil, fmt.Errorf(string(resp.Body))
+	}
+	return resp.Body, nil
 }
diff --git a/modules/elastic/metadata.go b/modules/elastic/metadata.go
@@ -28,6 +28,7 @@
 package elastic
 
 import (
+	"context"
 	"errors"
 	"fmt"
 	"reflect"
@@ -155,6 +156,20 @@ func updateClusterHealthStatus(clusterID string, healthStatus string) {
 			},
 		},
 	}
+	if healthStatus == "red" {
+		targetClient := elastic.GetClient(clusterID)
+		ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+		defer cancel()
+		res, err := targetClient.ClusterAllocationExplain(ctx, nil, nil)
+		if err != nil {
+			log.Errorf("get cluster allocation explain of cluster [%s] error: %v", clusterID, err)
+		}
+		activityInfo.Fields = util.MapStr{
+			"cluster_health": util.MapStr{
+				"allocation_explain": string(res),
+			},
+		}
+	}
 	_, err = client.Index(orm.GetIndexName(activityInfo), "", activityInfo.ID, activityInfo, "")
 	if err != nil {
 		log.Error(err)