添加StringArray

Author: yumaojun03

go.mod

@@ -37,5 +37,6 @@ require (
 	golang.org/x/sys v0.0.0-20191220220014-0732a990476f
 	golang.org/x/time v0.0.0-20191024005414-555d28b269f0 // indirect
 	google.golang.org/grpc v1.26.0 // indirect
+	gopkg.in/go-playground/assert.v1 v1.2.1
 	sigs.k8s.io/yaml v1.1.0 // indirect
 )

go.sum

@@ -139,6 +139,7 @@ github.com/sirupsen/logrus v1.4.2/go.mod h1:tLMulIdttU9McNUspp0xgXVQah82FyeX6Mwd
 github.com/soheilhy/cmux v0.1.4 h1:0HKaf1o97UwFjHH9o5XsHUOF+tqmdA7KEzXLpiyaw0E=
 github.com/soheilhy/cmux v0.1.4/go.mod h1:IM3LyeVVIOuxMH7sFAkER9+bJ4dT7Ms6E4xg4kGIyLM=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/objx v0.1.1 h1:2vfRuCMp5sSVIDSqO8oNnWJq7mPa6KVP3iPIwFBuy8A=
 github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
 github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
 github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
@@ -240,6 +241,8 @@ gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8
 gopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI=
 gopkg.in/fsnotify.v1 v1.4.7 h1:xOHLXZwVvI9hhs+cLKq5+I5onOuwQLhQwiu63xxlHs4=
 gopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys=
+gopkg.in/go-playground/assert.v1 v1.2.1 h1:xoYuJVE7KT85PYWrN730RguIQO0ePzVRfFMXadIrXTM=
+gopkg.in/go-playground/assert.v1 v1.2.1/go.mod h1:9RXL0bg/zibRAgZUYszZSwO/z8Y/a8bDuhia5mkpMnE=
 gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ=
 gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=
 gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=

types/array/string.go

@@ -0,0 +1,88 @@
+package array
+
+import (
+	"database/sql/driver"
+	"encoding/json"
+	"fmt"
+	"strings"
+)
+
+var (
+	// Separator 序列化时的分隔符
+	Separator = ";"
+	// StartEndMark 开始和结尾标志
+	StartEndMark = ";"
+)
+
+// NewStringArray todo
+func NewStringArray(items []string) *StringArray {
+	return &StringArray{
+		items: items,
+	}
+}
+
+// StringArray todo
+type StringArray struct {
+	items []string
+}
+
+func (t *StringArray) add(item string) {
+	if item != "" {
+		t.items = append(t.items, item)
+	}
+}
+
+// Items 列表
+func (t StringArray) Items() []string {
+	return t.items
+}
+
+// Length 长度
+func (t StringArray) Length() int {
+	return len(t.items)
+}
+
+func (t StringArray) String() string {
+	if t.Length() == 0 {
+		return ""
+	}
+
+	return StartEndMark + strings.Join(t.items, Separator) + StartEndMark
+}
+
+// Scan 实现sql的反序列化
+func (t *StringArray) Scan(value interface{}) error {
+	if value == nil {
+		return nil
+	}
+
+	switch data := value.(type) {
+	case []byte:
+		raw := string(data)
+		if raw == "" {
+			t.items = []string{}
+			return nil
+		}
+		for _, item := range strings.Split(raw, ";") {
+			t.add(item)
+		}
+		return nil
+	default:
+		return fmt.Errorf("unsupport type: %s", data)
+	}
+}
+
+// Value 实现sql的序列化
+func (t StringArray) Value() (driver.Value, error) {
+	return t.String(), nil
+}
+
+// MarshalJSON todo
+func (t StringArray) MarshalJSON() ([]byte, error) {
+	return json.Marshal(t.items)
+}
+
+// UnmarshalJSON todo
+func (t *StringArray) UnmarshalJSON(b []byte) error {
+	return json.Unmarshal(b, &t.items)
+}

types/array/string_test.go

@@ -0,0 +1,34 @@
+package array_test
+
+import (
+	"testing"
+
+	"github.com/infraboard/mcube/types/array"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestStringArrayString(t *testing.T) {
+	should := assert.New(t)
+	data := array.NewStringArray([]string{"1", "2", "3"})
+	should.Equal(data.String(), ";1;2;3;")
+}
+
+func TestStringArrayValue(t *testing.T) {
+	should := assert.New(t)
+	data := array.NewStringArray([]string{"1", "2", "3"})
+	v, err := data.Value()
+	if should.NoError(err) {
+		strv, ok := v.(string)
+		if should.True(ok) {
+			should.Equal(strv, ";1;2;3;")
+		}
+	}
+}
+
+func TestStringArrayScan(t *testing.T) {
+	should := assert.New(t)
+	data := array.NewStringArray([]string{})
+	if should.NoError(data.Scan([]byte(";1;2;3;"))) {
+		should.Equal(data.Items(), []string{"1", "2", "3"})
+	}
+}