golang db2struct

db2struct

Converts a mysql table into a golang struc

地址

• https://github.com/Shelnutt2/db2struct

安装

• 首先需要安装go
• go get github.com/Shelnutt2/db2struct/cmd/db2struct

使用

• 首先看看工具的使用
  

• 为了更加的通用，我写量脚本，将数据库配置以及表名等参数独立了出来，方便使用

• 上述参数需要放到shell脚本同级的目录之下的.config文件中

• 执行脚本之后，会为每个表生产一个对应的*.go文件，文件中包含了转化之后的struct，struct名、文件名就是表名

• 脚本如下所示：

  • gen-model.sh

    1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35

    #!/usr/bin/env bash # check if db2struct exist check() { if !(hash ${convert_tool} 2>/dev/null); then echo "convert tool ${convert_tool} is absent, install..." go get github.com/Shelnutt2/db2struct/cmd/db2struct fi } # generate golang model gen() { for table in ${TABLES[@]}; do echo "del old model/${table}.go" && rm ../../model/${table}.go struct_name="$(echo ${table:0:1} | tr '[a-z]' '[A-Z]')${table:1}" db2struct -H ${HOST} -u ${USER} -p ${PASSWORD} \ --package ${GO_PACKAGE_NAME} -d ${DB} \ --table ${table} --struct ${struct_name} \ --target=${SAVE_PATH}/${table}.go \ --json -v echo "finish model/${table}.go\n" done } # import related config source .config convert_tool=db2struct check gen

• 配置文件实例如下：

  • .config

    1 2 3 4 5 6 7

    GO_PACKAGE_NAME=model SAVE_PATH=../../model HOST=127.0.0.1 USER=root PASSWORD=12345678 DB=gp_oj TABLES=("role" "tag" "algorithm")

  • .config中具体参数含义如下所示：

    参数名	意义
    GO_PACKAGE_NAME	model文件（*.go）的package name
    SAVE_PATH	model文件（*.go）保存路径
    HOST	MySQL host
    USER	MySQL username
    PASSWORD	MySQL password
    DB	MySQL db name
    TABLES	MySQL 需要转化成struct的表

demo

• 允许结果大致如下所示：
  • 我是通过makefile来运行的
    
• 然后在model文件夹就可以看到文件了