mirror of
https://github.com/charlienet/go-mixed.git
synced 2025-07-17 16:12:42 +08:00
139 lines
2.7 KiB
Go
139 lines
2.7 KiB
Go
package json
|
||
|
||
import (
|
||
"bytes"
|
||
"log"
|
||
"regexp"
|
||
"strconv"
|
||
"strings"
|
||
"unicode"
|
||
|
||
"github.com/charlienet/go-mixed/bytesconv"
|
||
)
|
||
|
||
// 下划线
|
||
type SnakeCase struct {
|
||
Value any
|
||
}
|
||
|
||
func (c SnakeCase) MarshalJSON() ([]byte, error) {
|
||
var keyMatchRegex = regexp.MustCompile(`\"(\w+)\":`)
|
||
var wordBarrierRegex = regexp.MustCompile(`(\w)([A-Z])`)
|
||
|
||
marshalled, err := Marshal(c.Value)
|
||
converted := keyMatchRegex.ReplaceAllFunc(
|
||
marshalled,
|
||
func(match []byte) []byte {
|
||
return bytes.ToLower(wordBarrierRegex.ReplaceAll(
|
||
match,
|
||
bytesconv.StringToBytes(`${1}_${2}`),
|
||
))
|
||
},
|
||
)
|
||
return converted, err
|
||
}
|
||
|
||
// 驼峰
|
||
type CamelCase struct {
|
||
Value any
|
||
}
|
||
|
||
func (c CamelCase) MarshalJSON() ([]byte, error) {
|
||
var keyMatchRegex = regexp.MustCompile(`\"(\w+)\":`)
|
||
|
||
marshalled, err := Marshal(c.Value)
|
||
converted := keyMatchRegex.ReplaceAllFunc(
|
||
marshalled,
|
||
func(match []byte) []byte {
|
||
matchStr := bytesconv.BytesToString(match)
|
||
key := matchStr[1 : len(matchStr)-2]
|
||
resKey := Lcfirst(Case2Camel(key))
|
||
return bytesconv.StringToBytes(`"` + resKey + `":`)
|
||
},
|
||
)
|
||
return converted, err
|
||
}
|
||
|
||
// 驼峰式写法转为下划线写法
|
||
func Camel2Case(name string) string {
|
||
buffer := NewBuffer()
|
||
for i, r := range name {
|
||
if unicode.IsUpper(r) {
|
||
if i != 0 {
|
||
buffer.Append('_')
|
||
}
|
||
buffer.Append(unicode.ToLower(r))
|
||
} else {
|
||
buffer.Append(r)
|
||
}
|
||
}
|
||
return buffer.String()
|
||
}
|
||
|
||
// 下划线写法转为驼峰写法
|
||
func Case2Camel(name string) string {
|
||
name = strings.Replace(name, "_", " ", -1)
|
||
// name = strings.Title(name)
|
||
return strings.Replace(name, " ", "", -1)
|
||
}
|
||
|
||
func Case2CamelBytes(name []byte) []byte {
|
||
ret := Case2Camel(string(name))
|
||
return bytesconv.StringToBytes(ret)
|
||
}
|
||
|
||
// 首字母大写
|
||
func Ucfirst(str string) string {
|
||
for i, v := range str {
|
||
return string(unicode.ToUpper(v)) + str[i+1:]
|
||
}
|
||
return ""
|
||
}
|
||
|
||
// 首字母小写
|
||
func Lcfirst(str string) string {
|
||
for i, v := range str {
|
||
return string(unicode.ToLower(v)) + str[i+1:]
|
||
}
|
||
return ""
|
||
}
|
||
|
||
// 内嵌bytes.Buffer，支持连写
|
||
type Buffer struct {
|
||
*bytes.Buffer
|
||
}
|
||
|
||
func NewBuffer() *Buffer {
|
||
return &Buffer{Buffer: new(bytes.Buffer)}
|
||
}
|
||
|
||
func (b *Buffer) Append(i any) *Buffer {
|
||
switch val := i.(type) {
|
||
case int:
|
||
b.append(strconv.Itoa(val))
|
||
case int64:
|
||
b.append(strconv.FormatInt(val, 10))
|
||
case uint:
|
||
b.append(strconv.FormatUint(uint64(val), 10))
|
||
case uint64:
|
||
b.append(strconv.FormatUint(val, 10))
|
||
case string:
|
||
b.append(val)
|
||
case []byte:
|
||
b.Write(val)
|
||
case rune:
|
||
b.WriteRune(val)
|
||
}
|
||
return b
|
||
}
|
||
|
||
func (b *Buffer) append(s string) *Buffer {
|
||
defer func() {
|
||
if err := recover(); err != nil {
|
||
log.Println("*****内存不够了!******")
|
||
}
|
||
}()
|
||
b.WriteString(s)
|
||
return b
|
||
}
|