support metricsconfig's management

Author: 陈辉

log_project.go

@@ -5,7 +5,9 @@ import (
 	"encoding/json"
 	"fmt"
 	"io/ioutil"
+	"log"
 	"net/http"
+	"reflect"
 	"regexp"
 	"strings"
 	"time"
@@ -540,6 +542,135 @@ func (p *LogProject) DeleteMachineGroup(name string) (err error) {
 	return nil
 }
 
+func (p *LogProject) CreateMetricConfig(metricStore string, metricConfig *MetricsConfig) error {
+	body, err := json.Marshal(metricConfig)
+	if err != nil {
+		return NewClientError(err)
+	}
+	jsonBody := map[string]interface{}{
+		"metricStore":         metricStore,
+		"metricsConfigDetail": string(body),
+	}
+	body, err = json.Marshal(jsonBody)
+	if err != nil {
+		return NewClientError(err)
+	}
+
+	h := map[string]string{
+		"x-log-bodyrawsize": fmt.Sprintf("%v", len(body)),
+		"Content-Type":      "application/json",
+		"Accept-Encoding":   "deflate", // TODO: support lz4
+	}
+	r, err := request(p, "POST", "/metricsconfigs", h, body)
+	if err != nil {
+		return NewClientError(err)
+	}
+	defer r.Body.Close()
+	body, err = ioutil.ReadAll(r.Body)
+	if r.StatusCode != http.StatusOK {
+		err := new(Error)
+		json.Unmarshal(body, err)
+		return err
+	}
+	return nil
+}
+
+func (p *LogProject) DeleteMetricConfig(metricStore string) (err error) {
+	h := map[string]string{
+		"x-log-bodyrawsize": "0",
+	}
+	r, err := request(p, "DELETE", "/metricsconfigs/"+metricStore, h, nil)
+	if err != nil {
+		return NewClientError(err)
+	}
+	defer r.Body.Close()
+	body, _ := ioutil.ReadAll(r.Body)
+	if r.StatusCode != http.StatusOK {
+		err := new(Error)
+		json.Unmarshal(body, err)
+		return err
+	}
+	return nil
+}
+
+func (p *LogProject) GetMetricConfig(metricStore string) (*MetricsConfig, error) {
+	h := map[string]string{
+		"x-log-bodyrawsize": "0",
+	}
+	r, err := request(p, "GET", "/metricsconfigs/"+metricStore, h, nil)
+	if err != nil {
+		return nil, NewClientError(err)
+	}
+	defer r.Body.Close()
+	buf, _ := ioutil.ReadAll(r.Body)
+	if r.StatusCode != http.StatusOK {
+		err := new(Error)
+		json.Unmarshal(buf, err)
+		return nil, err
+	}
+	type OuterJSON struct {
+		MetricStore         string `json:"metricStore"`
+		MetricsConfigDetail string `json:"metricsConfigDetail"`
+	}
+
+	var outerData OuterJSON
+	if err := json.Unmarshal(buf, &outerData); err != nil {
+		log.Fatalf("Error parsing outer JSON: %v", err)
+	}
+
+	m := &MetricsConfig{}
+	err = json.Unmarshal([]byte(outerData.MetricsConfigDetail), m)
+	if err != nil {
+		return nil, err
+	}
+	if IsDebugLevelMatched(4) {
+		level.Info(Logger).Log("msg", "Get MetricConfig config, result", *m)
+	}
+
+	if reflect.DeepEqual(m, MetricsConfig{}) {
+		fmt.Println("MetricsConfig is empty")
+	}
+
+	return m, err
+}
+
+func (p *LogProject) UpdateMetricConfig(metricStore string, metricConfig *MetricsConfig) (err error) {
+	body, err := json.Marshal(metricConfig)
+	if err != nil {
+		return NewClientError(err)
+	}
+	jsonBody := map[string]interface{}{
+		"metricStore":         metricStore,
+		"metricsConfigDetail": string(body),
+	}
+	body, err = json.Marshal(jsonBody)
+	if err != nil {
+		return NewClientError(err)
+	}
+	body, err = json.Marshal(jsonBody)
+	if err != nil {
+		return NewClientError(err)
+	}
+
+	h := map[string]string{
+		"x-log-bodyrawsize": fmt.Sprintf("%v", len(body)),
+		"Content-Type":      "application/json",
+		"Accept-Encoding":   "deflate", // TODO: support lz4
+	}
+	r, err := request(p, "PUT", "/metricsconfigs/"+metricStore, h, body)
+	if err != nil {
+		return NewClientError(err)
+	}
+	defer r.Body.Close()
+	body, _ = ioutil.ReadAll(r.Body)
+	if r.StatusCode != http.StatusOK {
+		err := new(Error)
+		json.Unmarshal(body, err)
+		return err
+	}
+	return nil
+}
+
 // ListConfig returns config names list and the total number of configs.
 // The offset starts from 0 and the size is the max number of configs could be returned.
 func (p *LogProject) ListConfig(offset, size int) (cfgNames []string, total int, err error) {