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

guyan0319 · · 631 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

使用过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文件工具

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

631 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传