|
|
package main
|
|
|
|
|
|
import (
|
|
|
"bytes"
|
|
|
"dsBaseRpc/Utils/CommonUtil"
|
|
|
"dsBaseRpc/Utils/FileUtil"
|
|
|
"encoding/json"
|
|
|
"fmt"
|
|
|
"github.com/iancoleman/strcase"
|
|
|
"github.com/tidwall/gjson"
|
|
|
"os"
|
|
|
"path/filepath"
|
|
|
"reflect"
|
|
|
"strings"
|
|
|
)
|
|
|
|
|
|
//根据json对象字符串生成结构体
|
|
|
func createStruct(jsonStr string, structName string) string {
|
|
|
m := make(map[string]interface{})
|
|
|
err := json.Unmarshal([]byte(jsonStr), &m)
|
|
|
if err != nil {
|
|
|
fmt.Println("转化错误:", err)
|
|
|
}
|
|
|
var buffer bytes.Buffer
|
|
|
buffer.WriteString("type ")
|
|
|
buffer.WriteString(structName)
|
|
|
buffer.WriteString(" struct {\n")
|
|
|
for k, v := range m {
|
|
|
runes := []rune(k)
|
|
|
buffer.WriteString(strcase.ToCamel(string(runes))) //驼峰命名法
|
|
|
buffer.WriteString(" ")
|
|
|
buffer.WriteString((reflect.TypeOf(v)).String())
|
|
|
buffer.WriteString(" `json:\"")
|
|
|
buffer.WriteString(k)
|
|
|
buffer.WriteString("\"`")
|
|
|
buffer.WriteString("\n")
|
|
|
}
|
|
|
buffer.WriteString("}")
|
|
|
return buffer.String()
|
|
|
}
|
|
|
|
|
|
func main() {
|
|
|
//文件名称
|
|
|
fileName := `e:\\TBaseOrganization.json`
|
|
|
//1、读取json文件
|
|
|
j := FileUtil.ReadFileContent(fileName)
|
|
|
result := gjson.Get(j, "datas")
|
|
|
|
|
|
//2、生成JSON对应的struct
|
|
|
if len(result.Array()) == 0 {
|
|
|
fmt.Println("未找到数据，无法生成Struct~")
|
|
|
os.Exit(200)
|
|
|
}
|
|
|
r1 := result.Array()[0]
|
|
|
fmt.Println(r1.Get("data_id"))
|
|
|
fmt.Println(r1.Get("del_flag"))
|
|
|
re := r1.Get("data")
|
|
|
jsonStr := re.Str
|
|
|
c := strings.Replace(filepath.Base(fileName), ".json", "", -1)
|
|
|
content := "package Struct\n" + createStruct(jsonStr, c)
|
|
|
parentPath, _ := os.Getwd()
|
|
|
parentPath = strings.Replace(parentPath, "\\", "/", -1)
|
|
|
//保存文件
|
|
|
p1 := parentPath + "/Test/TestFenXiJson/Struct/" + c + ".go"
|
|
|
FileUtil.WriteContent(p1, content)
|
|
|
//格式化文件
|
|
|
CommonUtil.Exec("gofmt", "-l", "-w", p1)
|
|
|
}
|
