first commit

util.go

@@ -0,0 +1,77 @@
+package db
+
+import (
+	"reflect"
+	"regexp"
+	"strings"
+)
+
+var matchFirstCap = regexp.MustCompile("(.)([A-Z][a-z]+)")
+var matchAllCap = regexp.MustCompile("([a-z0-9])([A-Z])")
+
+func ToSnakeCase(str string) string {
+	snake := matchFirstCap.ReplaceAllString(str, "${1}_${2}")
+	snake = matchAllCap.ReplaceAllString(snake, "${1}_${2}")
+	return strings.ToLower(snake)
+}
+func ExtractStruct(target interface{}) map[string]interface{} {
+	tv := reflect.Indirect(reflect.ValueOf(target))
+	tt := tv.Type()
+	result := make(map[string]interface{}, tv.NumField())
+
+	for i := 0; i < tv.NumField(); i++ {
+		key := ToSnakeCase(tt.Field(i).Name)
+		result[key] = tv.Field(i).Interface()
+	}
+
+	return result
+}
+func InterfaceToSlice(param interface{}) []interface{} {
+	if p, ok := param.([]interface{}); ok {
+		return p
+	}
+	tv := reflect.Indirect(reflect.ValueOf(param))
+	var res []interface{}
+	if tv.Type().Kind() == reflect.Slice {
+		for i := 0; i < tv.Len(); i++ {
+			res = append(res, tv.Index(i).Interface())
+		}
+	} else {
+		panic("not slice")
+	}
+	return res
+}
+
+// addslashes() 函数返回在预定义字符之前添加反斜杠的字符串。
+// 预定义字符是：
+// 单引号（'）
+// 双引号（"）
+// 反斜杠（\）
+func Addslashes(str string) string {
+	tmpRune := []rune{}
+	strRune := []rune(str)
+	for _, ch := range strRune {
+		switch ch {
+		case []rune{'\\'}[0], []rune{'"'}[0], []rune{'\''}[0]:
+			tmpRune = append(tmpRune, []rune{'\\'}[0])
+			tmpRune = append(tmpRune, ch)
+		default:
+			tmpRune = append(tmpRune, ch)
+		}
+	}
+	return string(tmpRune)
+}
+
+// stripslashes() 函数删除由 addslashes() 函数添加的反斜杠。
+func Stripslashes(str string) string {
+	dstRune := []rune{}
+	strRune := []rune(str)
+	strLenth := len(strRune)
+	for i := 0; i < strLenth; i++ {
+		if strRune[i] == []rune{'\\'}[0] {
+			i++
+		}
+		dstRune = append(dstRune, strRune[i])
+	}
+	return string(dstRune)
+}