add CreateMetricStore

Author: chaolee50

client_interface.go

@@ -94,6 +94,16 @@ type ClientInterface interface {
 	// CheckLogstoreExist check logstore exist or not
 	CheckLogstoreExist(project string, logstore string) (bool, error)
 
+	// #################### MetricStore Operations #####################
+	// CreateMetricStore creates a new metric store in SLS.
+	CreateMetricStore(project string, metricStore LogStore) error
+	// UpdateMetricStore updates a metric store.
+	UpdateMetricStore(project string, metricStore LogStore) error
+	// DeleteMetricStore deletes a metric store.
+	DeleteMetricStore(project, name string) error
+	// GetMetricStore return a metric store.
+	GetMetricStore(project, name string) (*LogStore, error)
+
 	// #################### Logtail Operations #####################
 	// ListMachineGroup returns machine group name list and the total number of machine groups.
 	// The offset starts from 0 and the size is the max number of machine groups could be returned.

client_metric_store.go

@@ -3,16 +3,9 @@ package sls
 import "time"
 
 // CreateMetricStore .
-func (c *Client) CreateMetricStore(project, name string, ttl, shard int) error {
-	logStore := &LogStore{
-		Name:          name,
-		TTL:           ttl,
-		ShardCount:    shard,
-		TelemetryType: "Metrics",
-		AutoSplit:     true,
-		MaxSplitShard: 64,
-	}
-	err := c.CreateLogStoreV2(project, logStore)
+func (c *Client) CreateMetricStore(project string, metricStore LogStore) error {
+	metricStore.TelemetryType = "Metrics"
+	err := c.CreateLogStoreV2(project, &metricStore)
 	if err != nil {
 		return err
 	}
@@ -21,7 +14,7 @@ func (c *Client) CreateMetricStore(project, name string, ttl, shard int) error {
 	subStore.Name = "prom"
 	subStore.SortedKeyCount = 2
 	subStore.TimeIndex = 2
-	subStore.TTL = ttl
+	subStore.TTL = metricStore.TTL
 	subStore.Keys = append(subStore.Keys, SubStoreKey{
 		Name: "__name__",
 		Type: "text",
@@ -38,21 +31,17 @@ func (c *Client) CreateMetricStore(project, name string, ttl, shard int) error {
 	if !subStore.IsValid() {
 		panic("metric store invalid")
 	}
-	return c.CreateSubStore(project, name, subStore)
+	return c.CreateSubStore(project, metricStore.Name, subStore)
 }
 
 // UpdateMetricStore .
-func (c *Client) UpdateMetricStore(project, name string, ttl int) error {
-	metricStore := &LogStore{
-		Name:          name,
-		TelemetryType: "Metrics",
-		TTL:           ttl,
-	}
-	err := c.UpdateLogStoreV2(project, metricStore)
+func (c *Client) UpdateMetricStore(project string, metricStore LogStore) error {
+	metricStore.TelemetryType = "Metrics"
+	err := c.UpdateLogStoreV2(project, &metricStore)
 	if err != nil {
 		return err
 	}
-	return c.UpdateSubStoreTTL(project, name, ttl)
+	return c.UpdateSubStoreTTL(project, metricStore.Name, metricStore.TTL)
 }
 
 // DeleteMetricStore .

client_metric_store_test.go

@@ -49,14 +49,24 @@ func (m *MetricStoreTestSuite) TearDownSuite() {
 }
 
 func (m *MetricStoreTestSuite) TestClient_CreateAndDeleteMetricStore() {
-	ce := m.client.CreateMetricStore(m.projectName, m.metricStoreName, m.ttl, m.shardCnt)
+	metricStore := LogStore{
+		Name:       m.metricStoreName,
+		TTL:        m.ttl,
+		ShardCount: m.shardCnt,
+	}
+	ce := m.client.CreateMetricStore(m.projectName, metricStore)
 	m.Require().Nil(ce)
 	de := m.client.DeleteMetricStore(m.projectName, m.metricStoreName)
 	m.Require().Nil(de)
 }
 
 func (m *MetricStoreTestSuite) TestClient_UpdateAndGetMetricStore() {
-	ce := m.client.CreateMetricStore(m.projectName, m.metricStoreName, m.ttl, m.shardCnt)
+	metricStore1 := LogStore{
+		Name:       m.metricStoreName,
+		TTL:        m.ttl,
+		ShardCount: m.shardCnt,
+	}
+	ce := m.client.CreateMetricStore(m.projectName, metricStore1)
 	m.Require().Nil(ce)
 	metricStore, ge := m.client.GetMetricStore(m.projectName, m.metricStoreName)
 	m.Require().Nil(ge)
@@ -65,7 +75,8 @@ func (m *MetricStoreTestSuite) TestClient_UpdateAndGetMetricStore() {
 	m.Require().Equal(m.shardCnt, metricStore.ShardCount)
 	m.Require().Equal("Metrics", metricStore.TelemetryType)
 
-	ue := m.client.UpdateMetricStore(m.projectName, m.metricStoreName, 15)
+	metricStore1.TTL = 15
+	ue := m.client.UpdateMetricStore(m.projectName, metricStore1)
 	m.Require().Nil(ue)
 	metricStore2, ge2 := m.client.GetMetricStore(m.projectName, m.metricStoreName)
 	m.Require().Nil(ge2)

token_auto_update_client.go

@@ -1549,3 +1549,39 @@ func (c *TokenAutoUpdateClient) DeleteExport(project string, name string) (err e
 	}
 	return
 }
+func (c *TokenAutoUpdateClient) CreateMetricStore(project string, metricStore LogStore) (err error) {
+	for i := 0; i < c.maxTryTimes; i++ {
+		err = c.logClient.CreateMetricStore(project, metricStore)
+		if !c.processError(err) {
+			return
+		}
+	}
+	return
+}
+func (c *TokenAutoUpdateClient) UpdateMetricStore(project string, metricStore LogStore) (err error) {
+	for i := 0; i < c.maxTryTimes; i++ {
+		err = c.logClient.UpdateMetricStore(project, metricStore)
+		if !c.processError(err) {
+			return
+		}
+	}
+	return
+}
+func (c *TokenAutoUpdateClient) DeleteMetricStore(project, name string) (err error) {
+	for i := 0; i < c.maxTryTimes; i++ {
+		err = c.logClient.DeleteMetricStore(project, name)
+		if !c.processError(err) {
+			return
+		}
+	}
+	return
+}
+func (c *TokenAutoUpdateClient) GetMetricStore(project, name string) (metricStore *LogStore, err error) {
+	for i := 0; i < c.maxTryTimes; i++ {
+		metricStore, err = c.logClient.GetMetricStore(project, name)
+		if !c.processError(err) {
+			return
+		}
+	}
+	return
+}