导出mysql表结构生成grpc需要的proto文件工具

使用过grpc的同学都知道,写proto文件比较繁琐,尤其是写message,对应很多字段,为此写了一个简单的从mysql直接读取表结构,生成proto文件的工具。

工具的使用很简单,需要简单的配置,即可运行生成proto文件。
项目地址:https://github.com/guyan0319/...

准备

使用前需先安装依赖包go-sql-driver/mysql

$ go get -u github.com/go-sql-driver/mysql

使用说明:

func main() {
 //模板文件存放路径
 tpl := "d:/gopath/src/mysql-to-proto/template/proto.go.tpl"
 //生成proto文件路径
 file := "d:/gopath/src/mysql-to-proto/sso.proto"
 //数据库名,这里填你自己的数据库名
 dbName := "user"
 //配置连接数据库信息
 db, err := Connect("mysql", "root:123456@tcp(127.0.0.1:3306)/"+dbName+"?charset=utf8mb4&parseTime=true")
 //Table names to be excluded
 //需要排除表,这里的表不会生成对应的proto文件
 exclude := map[string]int{"user_log": 1}
 if err != nil {
 fmt.Println(err)
 }
 if IsFile(file) {
 fmt.Fprintf(os.Stderr, "Fatal error: ", "proto file already exist")
 return
 }
 t := Table{}
 //配置message,Cat 
 t.Message = map[string]Detail{
 "Filter": {
 Name: "Filter",
 Cat: "custom",
 Attr: []AttrDetail{{
 TypeName: "uint64", //类型
 AttrName: "id",//字段
 }},
 },
 "Request": {
 Name: "Request",
 Cat: "all",
 },
 "Response": {
 Name: "Response",
 Cat: "custom",
 Attr: []AttrDetail{
 {
 TypeName: "uint64",
 AttrName: "id",
 },
 {
 TypeName: "bool",
 AttrName: "success",
 },
 },
 },
 }
 //pachage名称
 t.PackageModels = "sso"
 //service名称
 t.ServiceName = "Sso"
 //配置services里面的rpc
 t.Method = map[string]MethodDetail{
 "Get": {Request: t.Message["Filter"], Response: t.Message["Request"]},
 "Create": {Request: t.Message["Request"], Response: t.Message["Response"]},
 "Update": {Request: t.Message["Request"], Response: t.Message["Response"]},
 }
 //处理数据库表字段属性
 t.TableColumn(db, dbName, exclude)
 //生成proto
 t.Generate(file, tpl)
}

links

• 目录

本文来自:Segmentfault

感谢作者:guyan0319

查看原文:导出mysql表结构生成grpc需要的proto文件工具

• 编辑
• 预览