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

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

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

使用说明：

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

• 目录