136 lines
2.7 KiB
Go
136 lines
2.7 KiB
Go
// Package json2go ...
|
|
//
|
|
// Description : json2go ...
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 2021-11-09 3:44 下午
|
|
package json2go
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"git.zhangdeman.cn/zhangdeman/gopkg/util"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/tidwall/gjson"
|
|
)
|
|
|
|
var (
|
|
// structName 结构体名称
|
|
structName = "Automatic"
|
|
)
|
|
|
|
// NewJSON2GO ...
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 6:13 下午 2021/11/9
|
|
func NewJSON2GO(name string) *JSON2GO {
|
|
if len(name) == 0 {
|
|
name = structName
|
|
}
|
|
return &JSON2GO{
|
|
structName: name,
|
|
result: "type " + name + " struct { \n",
|
|
}
|
|
}
|
|
|
|
// JSON2GO ...
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 6:12 下午 2021/11/9
|
|
type JSON2GO struct {
|
|
structName string
|
|
result string
|
|
}
|
|
|
|
// Parse ...
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 3:44 下午 2021/11/9
|
|
func (jg *JSON2GO) Parse(inputJSON string) (string, error) {
|
|
if !gjson.Valid(inputJSON) {
|
|
return "", errors.New("input json is invalid")
|
|
}
|
|
parseResult := gjson.Parse(inputJSON)
|
|
if parseResult.IsArray() {
|
|
jg.parseArray("", parseResult)
|
|
} else {
|
|
jg.parseObject("", parseResult)
|
|
}
|
|
jg.result += "}"
|
|
return jg.result, nil
|
|
}
|
|
|
|
// parseArray 解析array
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 4:37 下午 2021/11/9
|
|
func (jg *JSON2GO) parseArray(key string, parseResult gjson.Result) string {
|
|
for _, v := range parseResult.Array() {
|
|
if v.IsObject() {
|
|
jg.parseObject(key, v)
|
|
} else if v.IsArray() {
|
|
jg.parseArray("", v)
|
|
} else {
|
|
jg.result += util.SnakeCaseToCamel(key) + " " + v.Type.String() + " `json:\"" + key + "\"`"
|
|
}
|
|
}
|
|
return ""
|
|
}
|
|
|
|
// parseObject 解析object
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 4:37 下午 2021/11/9
|
|
func (jg *JSON2GO) parseObject(key string, parseResult gjson.Result) string {
|
|
if len(key) > 0 {
|
|
jg.result += util.SnakeCaseToCamel(key) + " struct { \n"
|
|
}
|
|
for k, v := range parseResult.Map() {
|
|
if v.IsObject() {
|
|
jg.parseObject(k, v)
|
|
} else if v.IsArray() {
|
|
jg.parseArray(k, v)
|
|
} else {
|
|
jg.result += util.SnakeCaseToCamel(k) + " " + jg.getDataType(v) + " `json:\"" + k + "\"`\n"
|
|
}
|
|
}
|
|
if len(key) > 0 {
|
|
jg.result += " } `json:\"" + key + "\"`\n"
|
|
}
|
|
return ""
|
|
}
|
|
|
|
// getDataType 获取数据类型
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 8:14 下午 2021/11/9
|
|
func (jg *JSON2GO) getDataType(val gjson.Result) string {
|
|
switch val.Type {
|
|
case gjson.True:
|
|
fallthrough
|
|
case gjson.False:
|
|
return "bool"
|
|
case gjson.Null:
|
|
return "map[string]interface{}"
|
|
case gjson.String:
|
|
return "string"
|
|
case gjson.Number:
|
|
tmpVal := fmt.Sprintf("%v", val.Num)
|
|
if strings.Contains(tmpVal, ".") {
|
|
return "float64"
|
|
}
|
|
return "int64"
|
|
default:
|
|
return "interface{}"
|
|
}
|
|
}
|