package main

import (
	"bytes"
	"dsBaseRpc/Utils/FileUtil"
	"encoding/json"
	"fmt"
	"github.com/iancoleman/strcase"
	"github.com/tidwall/gjson"
	"os"
	"reflect"
)

//根据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:\\t_base_organization.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")
	re.ForEach(func(k1, v1 gjson.Result) bool {
		fmt.Println(v1)
		return true // keep iterating
	})
	jsonStr := re.Str
	//转成map
	_map := make(map[string]interface{})
	err := json.Unmarshal([]byte(jsonStr), &_map)
	if err != nil {
		fmt.Println("error!")
	}

	str1 := "SET NAMES utf8mb4;\n"
	str1 += "ET FOREIGN_KEY_CHECKS = 0;\n"
	str1 = "CREATE TABLE `t_base_organization` (\n"

	for key, value := range _map {
		if reflect.TypeOf(value).Name() == "float64" {
			str1 += "`" + key + "` varchar(128) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci  NULL COMMENT '' ,\n"
		} else if reflect.TypeOf(value).Name() == "string" {
			//可以继续判断是不是日期格式//TODO
			str1 += "`" + key + "` float(10,0)  NULL,\n"
		}
	}
	str1 = str1[0 : len(str1)-2]
	str1 += ");\n"
	str1 += "SET FOREIGN_KEY_CHECKS = 1;\n"

	fmt.Println(str1)
}