78 lines
1.8 KiB
Go
78 lines
1.8 KiB
Go
|
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)
|
|||
|
|
}
|