feat: support metricStore meteringMode (#363)

Author: crimson-gao

client_interface.go

@@ -160,6 +160,12 @@ type ClientInterface interface {
 	DeleteMetricStore(project, name string) error
 	// GetMetricStore return a metric store.
 	GetMetricStore(project, name string) (*LogStore, error)
+	// GetMetricStoreMeteringMode get the metering mode of metric store, eg. ChargeByFunction / ChargeByDataIngest
+	GetMetricStoreMeteringMode(project string, metricStore string) (*GetMeteringModeResponse, error)
+	// UpdateMetricStoreMeteringMode update the metering mode of metric store, eg. ChargeByFunction / ChargeByDataIngest
+	//
+	// Warning: this method may affect your billings, for more details ref: https://www.aliyun.com/price/detail/sls
+	UpdateMetricStoreMeteringMode(project string, metricStore string, meteringMode string) error
 
 	// #################### EventStore Operations #####################
 	// CreateEventStore creates a new event store in SLS.

client_metric_store.go

@@ -53,3 +53,16 @@ func (c *Client) DeleteMetricStore(project, name string) error {
 func (c *Client) GetMetricStore(project, name string) (*LogStore, error) {
 	return c.GetLogStore(project, name)
 }
+
+// GetMetricStoreMeteringMode get the metering mode of metric store, eg. ChargeByFunction / ChargeByDataIngest
+func (c *Client) GetMetricStoreMeteringMode(project string, metricStore string) (*GetMeteringModeResponse, error) {
+	ms := convertLogstore(c, project, metricStore)
+	return ms.GetMetricStoreMeteringMode()
+}
+
+// UpdateMetricStoreMeteringMode update the metering mode of metric store, eg. ChargeByFunction / ChargeByDataIngest
+// Warning: this method may affect your billings, for more details ref: https://www.aliyun.com/price/detail/sls
+func (c *Client) UpdateMetricStoreMeteringMode(project string, metricStore string, meteringMode string) error {
+	ms := convertLogstore(c, project, metricStore)
+	return ms.UpdateMetricStoreMeteringMode(meteringMode)
+}

log_store.go

@@ -1283,3 +1283,53 @@ func (s *LogStore) UpdateMeteringMode(meteringMode string) error {
 	defer r.Body.Close()
 	return nil
 }
+
+func (s *LogStore) GetMetricStoreMeteringMode() (*GetMeteringModeResponse, error) {
+	h := map[string]string{
+		"x-log-bodyrawsize": "0",
+		"Content-Type":      "application/json",
+	}
+	uri := fmt.Sprintf("/metricstores/%s/meteringmode", s.Name)
+	r, err := request(s.project, "GET", uri, h, nil)
+	if err != nil {
+		return nil, err
+	}
+	defer r.Body.Close()
+	data, err := ioutil.ReadAll(r.Body)
+	if err != nil {
+		return nil, NewBadResponseError("", r.Header, r.StatusCode)
+	}
+	res := GetMeteringModeResponse{}
+	err = json.Unmarshal(data, &res)
+	if err != nil {
+		return nil, NewBadResponseError(string(data), r.Header, r.StatusCode)
+
+	}
+	return &res, nil
+
+}
+
+func (s *LogStore) UpdateMetricStoreMeteringMode(meteringMode string) error {
+
+	body := map[string]string{
+		"meteringMode": meteringMode,
+	}
+	uri := fmt.Sprintf("/metricstores/%s/meteringmode", s.Name)
+	requestBody, err := json.Marshal(body)
+	if err != nil {
+		return fmt.Errorf("cant marshal body:%w", err)
+	}
+
+	h := map[string]string{
+		"Content-Type":      "application/json",
+		"x-log-bodyrawsize": strconv.Itoa(len(requestBody)),
+	}
+
+	r, err := request(s.project, "PUT", uri, h, requestBody)
+
+	if err != nil {
+		return err
+	}
+	defer r.Body.Close()
+	return nil
+}

token_auto_update_client.go

@@ -361,6 +361,26 @@ func (c *TokenAutoUpdateClient) UpdateLogStoreMeteringMode(project string, logst
 	return
 }
 
+func (c *TokenAutoUpdateClient) GetMetricStoreMeteringMode(project string, metricStore string) (res *GetMeteringModeResponse, err error) {
+	for i := 0; i < c.maxTryTimes; i++ {
+		res, err = c.logClient.GetMetricStoreMeteringMode(project, metricStore)
+		if !c.processError(err) {
+			return
+		}
+	}
+	return
+}
+
+func (c *TokenAutoUpdateClient) UpdateMetricStoreMeteringMode(project string, metricStore string, meteringMode string) (err error) {
+	for i := 0; i < c.maxTryTimes; i++ {
+		err = c.logClient.UpdateMetricStoreMeteringMode(project, metricStore, meteringMode)
+		if !c.processError(err) {
+			return
+		}
+	}
+	return
+}
+
 func (c *TokenAutoUpdateClient) ListMachines(project, machineGroupName string) (ms []*Machine, total int, err error) {
 	for i := 0; i < c.maxTryTimes; i++ {
 		ms, total, err = c.logClient.ListMachines(project, machineGroupName)