dev: 增加格式化方法

Author: zhangxu19830126

format/string.go

@@ -30,6 +30,31 @@ func MustParseStrUInt64(data string) uint64 {
 	return value
 }
 
+// ParseStrInt str -> int
+func ParseStrInt(data string) (int, error) {
+	v, err := strconv.ParseInt(data, 10, 32)
+	if err != nil {
+		return 0, err
+	}
+
+	return int(v), nil
+}
+
+// MustParseStrInt str -> int
+func MustParseStrInt(data string) int {
+	value, err := ParseStrInt(data)
+	if err != nil {
+		buf := make([]byte, 4096)
+		runtime.Stack(buf, true)
+		log.Fatalf("parse to int failed, data=<%s> errors:\n %+v \n %s",
+			data,
+			err,
+			buf)
+	}
+
+	return value
+}
+
 // ParseStrInt64 str -> int64
 func ParseStrInt64(data string) (int64, error) {
 	return strconv.ParseInt(data, 10, 64)
@@ -69,3 +94,51 @@ func MustParseStrFloat64(data string) float64 {
 
 	return value
 }
+
+// ParseStrIntSlice parse []string -> []int
+func ParseStrIntSlice(data []string) ([]int, error) {
+	var target []int
+
+	for _, str := range data {
+		id, err := ParseStrInt(str)
+		if err != nil {
+			return nil, err
+		}
+
+		target = append(target, id)
+	}
+
+	return target, nil
+}
+
+// ParseStrInt64Slice parse []string -> []int64
+func ParseStrInt64Slice(data []string) ([]int64, error) {
+	var target []int64
+
+	for _, str := range data {
+		id, err := ParseStrInt64(str)
+		if err != nil {
+			return nil, err
+		}
+
+		target = append(target, id)
+	}
+
+	return target, nil
+}
+
+// ParseStrUInt64Slice parse []string -> []uint64
+func ParseStrUInt64Slice(data []string) ([]uint64, error) {
+	var target []uint64
+
+	for _, str := range data {
+		id, err := ParseStrUInt64(str)
+		if err != nil {
+			return nil, err
+		}
+
+		target = append(target, id)
+	}
+
+	return target, nil
+}

format/string_test.go

@@ -5,7 +5,7 @@ import (
 )
 
 func TestParseStrInt64(t *testing.T) {
-	value := []byte("10")
+	value := "10"
 	got, err := ParseStrInt64(value)
 	if err != nil {
 		t.Errorf("failed, errors:%+v", err)
@@ -17,7 +17,7 @@ func TestParseStrInt64(t *testing.T) {
 }
 
 func TestParseStrFloat64(t *testing.T) {
-	value := []byte("10.10")
+	value := "10.10"
 	got, err := ParseStrFloat64(value)
 	if err != nil {
 		t.Errorf("failed, errors:%+v", err)
@@ -27,3 +27,99 @@ func TestParseStrFloat64(t *testing.T) {
 		t.Errorf("failed, got=<%+v>, expect=<%+v>", got, 10.10)
 	}
 }
+
+func TestParseStrInt(t *testing.T) {
+	value := "10"
+	got, err := ParseStrInt(value)
+	if err != nil {
+		t.Errorf("failed, errors:%+v", err)
+	}
+
+	if got != 10 {
+		t.Errorf("failed, got=<%+v>, expect=<%+v>", got, 10)
+	}
+}
+
+func TestParseStrIntSlice(t *testing.T) {
+	value := []string{"10", "11", "12"}
+	got, err := ParseStrIntSlice(value)
+	if err != nil {
+		t.Errorf("failed, errors:%+v", err)
+	}
+
+	if len(got) != len(value) {
+		t.Errorf("failed, got=<%d>, expect=<%d>", len(got), len(value))
+		return
+	}
+
+	if got[0] != 10 {
+		t.Errorf("failed, got=<%+v>, expect=<%+v>", got[0], 10)
+		return
+	}
+
+	if got[1] != 11 {
+		t.Errorf("failed, got=<%+v>, expect=<%+v>", got[1], 11)
+		return
+	}
+
+	if got[2] != 12 {
+		t.Errorf("failed, got=<%+v>, expect=<%+v>", got[1], 11)
+		return
+	}
+}
+
+func TestParseStrInt64Slice(t *testing.T) {
+	value := []string{"10", "11", "12"}
+	got, err := ParseStrInt64Slice(value)
+	if err != nil {
+		t.Errorf("failed, errors:%+v", err)
+	}
+
+	if len(got) != len(value) {
+		t.Errorf("failed, got=<%d>, expect=<%d>", len(got), len(value))
+		return
+	}
+
+	if got[0] != 10 {
+		t.Errorf("failed, got=<%+v>, expect=<%+v>", got[0], 10)
+		return
+	}
+
+	if got[1] != 11 {
+		t.Errorf("failed, got=<%+v>, expect=<%+v>", got[1], 11)
+		return
+	}
+
+	if got[2] != 12 {
+		t.Errorf("failed, got=<%+v>, expect=<%+v>", got[1], 11)
+		return
+	}
+}
+
+func TestParseStrUInt64Slice(t *testing.T) {
+	value := []string{"10", "11", "12"}
+	got, err := ParseStrUInt64Slice(value)
+	if err != nil {
+		t.Errorf("failed, errors:%+v", err)
+	}
+
+	if len(got) != len(value) {
+		t.Errorf("failed, got=<%d>, expect=<%d>", len(got), len(value))
+		return
+	}
+
+	if got[0] != 10 {
+		t.Errorf("failed, got=<%+v>, expect=<%+v>", got[0], 10)
+		return
+	}
+
+	if got[1] != 11 {
+		t.Errorf("failed, got=<%+v>, expect=<%+v>", got[1], 11)
+		return
+	}
+
+	if got[2] != 12 {
+		t.Errorf("failed, got=<%+v>, expect=<%+v>", got[1], 11)
+		return
+	}
+}