ZLGoLib/db.go

package mokuai

import (
	"fmt"
	"github.com/gohouse/converter"
)

//转换数据库schema到go struct
//dsn 数据库格式："user:password@/dbname 如 user:mima@tcp(127.0.0.1:3306)/dbname"
//path 生成的结构体保存路径
//pacName 生成struct的包名(默认为空的话, 则取名为: package model)
func GormT(dsn string, path string, pacName string) bool {
	// 初始化
	t2t := converter.NewTable2Struct()
	// 个性化配置
	t2t.Config(&converter.T2tConfig{
		// 如果字段首字母本来就是大写, 就不添加tag, 默认false添加, true不添加
		RmTagIfUcFirsted: false,
		// tag的字段名字是否转换为小写, 如果本身有大写字母的话, 默认false不转
		TagToLower: false,
		// 字段首字母大写的同时, 是否要把其他字母转换为小写,默认false不转换
		UcFirstOnly: false,
		//// 每个struct放入单独的文件,默认false,放入同一个文件(暂未提供)
	})
	// 开始迁移转换
	err := t2t.
		// 是否添加json tag
		EnableJsonTag(true).
		// 生成struct的包名(默认为空的话, 则取名为: package model)
		PackageName(pacName).
		// tag字段的key值,默认是orm
		TagKey("gorm").
		// 生成的结构体保存路径
		SavePath(path).
		// 数据库dsn,这里可以使用 t2t.DB() 代替,参数为 *sql.DB 对象
		Dsn(dsn).
		// 执行
		Run()

	if err != nil {
		fmt.Println(err)
		return false
	}
	return true
}