Refactor ilm collectors

Signed-off-by: Шабакаев Илья Исмаилович <[email protected]>
Signed-off-by: Шабакаев Илья Исмаилович <[email protected]>

Author: Шабакаев Илья Исмаилович

collector/ilm_indices.go

@@ -1,4 +1,4 @@
-// Copyright 2021 The Prometheus Authors
+// Copyright 2023 The Prometheus Authors
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
@@ -16,6 +16,7 @@ package collector
 import (
 	"encoding/json"
 	"fmt"
+	"io/ioutil"
 	"net/http"
 	"net/url"
 	"path"
@@ -45,6 +46,19 @@ type IlmIndiciesCollector struct {
 	ilmMetric ilmMetric
 }
 
+type IlmResponse struct {
+	Indices map[string]IlmIndexResponse `json:"indices"`
+}
+
+type IlmIndexResponse struct {
+	Index          string  `json:"index"`
+	Managed        bool    `json:"managed"`
+	Phase          string  `json:"phase"`
+	Action         string  `json:"action"`
+	Step           string  `json:"step"`
+	StepTimeMillis float64 `json:"step_time_millis"`
+}
+
 var (
 	defaultIlmIndicesMappingsLabels = []string{"index", "phase", "action", "step"}
 )
@@ -117,7 +131,13 @@ func (i *IlmIndiciesCollector) fetchAndDecodeIlm() (IlmResponse, error) {
 		return ir, fmt.Errorf("HTTP Request failed with code %d", res.StatusCode)
 	}
 
-	if err := json.NewDecoder(res.Body).Decode(&ir); err != nil {
+	bts, err := ioutil.ReadAll(res.Body)
+	if err != nil {
+		i.jsonParseFailures.Inc()
+		return ir, err
+	}
+
+	if err := json.Unmarshal(bts, &ir); err != nil {
 		i.jsonParseFailures.Inc()
 		return ir, err
 	}

collector/ilm_indices_response.go

@@ -1,4 +1,4 @@
-// Copyright 2021 The Prometheus Authors
+// Copyright 2023 The Prometheus Authors
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at

collector/ilm_indices_test.go

@@ -1,4 +1,4 @@
-// Copyright 2021 The Prometheus Authors
+// Copyright 2023 The Prometheus Authors
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
@@ -49,6 +49,10 @@ type IlmStatusCollector struct {
 	metric ilmStatusMetric
 }
 
+type IlmStatusResponse struct {
+	OperationMode string `json:"operation_mode"`
+}
+
 // NewIlmStatus defines Indices IndexIlms Prometheus metrics
 func NewIlmStatus(logger log.Logger, client *http.Client, url *url.URL) *IlmStatusCollector {
 	subsystem := "ilm"
@@ -95,7 +99,10 @@ func (im *IlmStatusCollector) Describe(ch chan<- *prometheus.Desc) {
 	ch <- im.jsonParseFailures.Desc()
 }
 
-func (im *IlmStatusCollector) getAndParseURL(u *url.URL) (*IlmStatusResponse, error) {
+func (im *IlmStatusCollector) fetchAndDecodeIlm() (*IlmStatusResponse, error) {
+	u := *im.url
+	u.Path = path.Join(im.url.Path, "/_ilm/status")
+
 	res, err := im.client.Get(u.String())
 	if err != nil {
 		return nil, fmt.Errorf("failed to get from %s://%s:%s%s: %s",
@@ -127,12 +134,6 @@ func (im *IlmStatusCollector) getAndParseURL(u *url.URL) (*IlmStatusResponse, er
 	return &imr, nil
 }
 
-func (im *IlmStatusCollector) fetchAndDecodeIlm() (*IlmStatusResponse, error) {
-	u := *im.url
-	u.Path = path.Join(u.Path, "/_ilm/status")
-	return im.getAndParseURL(&u)
-}
-
 // Collect gets all indices Ilms metric values
 func (im *IlmStatusCollector) Collect(ch chan<- prometheus.Metric) {
 

collector/ilm_status.go

@@ -1,4 +1,4 @@
-// Copyright 2021 The Prometheus Authors
+// Copyright 2023 The Prometheus Authors
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at