add CreateMetricStore

Author: chaolee50

client_metric_store.go

@@ -0,0 +1,66 @@
+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)
+	if err != nil {
+		return err
+	}
+	time.Sleep(time.Second * 3)
+	subStore := &SubStore{}
+	subStore.Name = "prom"
+	subStore.SortedKeyCount = 2
+	subStore.TimeIndex = 2
+	subStore.TTL = ttl
+	subStore.Keys = append(subStore.Keys, SubStoreKey{
+		Name: "__name__",
+		Type: "text",
+	}, SubStoreKey{
+		Name: "__labels__",
+		Type: "text",
+	}, SubStoreKey{
+		Name: "__time_nano__",
+		Type: "long",
+	}, SubStoreKey{
+		Name: "__value__",
+		Type: "double",
+	})
+	if !subStore.IsValid() {
+		panic("metric store invalid")
+	}
+	return c.CreateSubStore(project, 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)
+	if err != nil {
+		return err
+	}
+	return c.UpdateSubStoreTTL(project, name, ttl)
+}
+
+// DeleteMetricStore .
+func (c *Client) DeleteMetricStore(project, name string) error {
+	return c.DeleteLogStore(project, name)
+}
+
+// GetMetricStore .
+func (c *Client) GetMetricStore(project, name string) (*LogStore, error) {
+	return c.GetLogStore(project, name)
+}

client_metric_store_test.go

@@ -0,0 +1,77 @@
+package sls
+
+import (
+	"fmt"
+	"github.com/stretchr/testify/suite"
+	"os"
+	"testing"
+	"time"
+)
+
+func TestMetricStore(t *testing.T) {
+	suite.Run(t, new(MetricStoreTestSuite))
+}
+
+type MetricStoreTestSuite struct {
+	suite.Suite
+	endpoint        string
+	projectName     string
+	metricStoreName string
+	accessKeyID     string
+	accessKeySecret string
+	ttl             int
+	shardCnt        int
+	client          *Client
+}
+
+func (m *MetricStoreTestSuite) SetupSuite() {
+	m.endpoint = os.Getenv("LOG_TEST_ENDPOINT")
+	m.accessKeyID = os.Getenv("LOG_TEST_ACCESS_KEY_ID")
+	m.accessKeySecret = os.Getenv("LOG_TEST_ACCESS_KEY_SECRET")
+	suffix := time.Now().Unix()
+	m.projectName = fmt.Sprintf("test-metric-store-%d", suffix)
+	m.metricStoreName = "test"
+	m.ttl = 30
+	m.shardCnt = 2
+	m.client = &Client{
+		Endpoint:        m.endpoint,
+		AccessKeyID:     m.accessKeyID,
+		AccessKeySecret: m.accessKeySecret,
+	}
+	_, err := m.client.CreateProject(m.projectName, "test metric store")
+	m.Require().Nil(err)
+	time.Sleep(time.Minute)
+}
+
+func (m *MetricStoreTestSuite) TearDownSuite() {
+	err := m.client.DeleteProject(m.projectName)
+	m.Require().Nil(err)
+}
+
+func (m *MetricStoreTestSuite) TestClient_CreateAndDeleteMetricStore() {
+	ce := m.client.CreateMetricStore(m.projectName, m.metricStoreName, m.ttl, m.shardCnt)
+	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)
+	m.Require().Nil(ce)
+	metricStore, ge := m.client.GetMetricStore(m.projectName, m.metricStoreName)
+	m.Require().Nil(ge)
+	m.Require().Equal(m.metricStoreName, metricStore.Name)
+	m.Require().Equal(m.ttl, metricStore.TTL)
+	m.Require().Equal(m.shardCnt, metricStore.ShardCount)
+	m.Require().Equal("Metrics", metricStore.TelemetryType)
+
+	ue := m.client.UpdateMetricStore(m.projectName, m.metricStoreName, 15)
+	m.Require().Nil(ue)
+	metricStore2, ge2 := m.client.GetMetricStore(m.projectName, m.metricStoreName)
+	m.Require().Nil(ge2)
+	m.Require().Equal(m.metricStoreName, metricStore2.Name)
+	m.Require().Equal(15, metricStore2.TTL)
+	m.Require().Equal("Metrics", metricStore.TelemetryType)
+	de := m.client.DeleteMetricStore(m.projectName, m.metricStoreName)
+	m.Require().Nil(de)
+}