一行命令将mysql表转为golang struct

Github地址

中文文档| English

okcli 可以将指定数据表生成为对应的model文件,目前支持将列名、数据类型、默认值、注释带入指定model中,
就像如下文件:

import "time"

type (
    Student struct {
        Id         int32     `db:"id"`
        Name       string    `db:"name"`        // the student's name
        Gender     int8      `db:"gender"`      // the student's gender,0-male,1-female,DEFAULT:0
        Age        int8      `db:"age"`         // the student's age,DEFAULT:0
        Class      string    `db:"class"`       // the student's class
        CreateTime time.Time `db:"create_time"` // the column create time,DEFAULT:CURRENT_TIMESTAMP
        UpdateTime time.Time `db:"update_time"` // the column last update time,DEFAULT:CURRENT_TIMESTAMP
    }
)

准备工作

根据自己的操作系统平台下载不同的二进制文件

或者clone源码进行编译。

其次,将下载或编译后的二进制文件设置到环境变量中,以mac OS为例子:
假设二进制文件okcli存放在/usr/local/okcli文件目录下

$ sudo vi ~/.bash_profile

在文件末尾添加:

$ export OKCLI=/usr/local/okcli
$ export PATH=$PATH:$OKCLI

最后执行以下source ~/.bash_profile

使用说明

数据库准备

这里以本地数据库举例,在demo库下新建student、course、student_score表,sql如下:

CREATE TABLE `student` (
  `id` int(20) NOT NULL AUTO_INCREMENT,
  `name` varchar(255) COLLATE utf8mb4_general_ci NOT NULL DEFAULT '' COMMENT 'the student''s name',
  `gender` tinyint(1) NOT NULL DEFAULT '0' COMMENT 'the student''s gender,0-male,1-female',
  `age` tinyint(3) unsigned NOT NULL DEFAULT '0' COMMENT 'the student''s age',
  `class` varchar(255) COLLATE utf8mb4_general_ci NOT NULL DEFAULT '' COMMENT 'the student''s class',
  `create_time` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT 'the column create time',
  `update_time` timestamp NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT 'the column last update time',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB  DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='Student info';

CREATE TABLE `course` (
  `id` int(20) NOT NULL AUTO_INCREMENT,
  `name` varchar(255) COLLATE utf8mb4_general_ci NOT NULL DEFAULT '' COMMENT 'course name',
  `teacher` varchar(255) COLLATE utf8mb4_general_ci NOT NULL DEFAULT '' COMMENT 'teach person',
  `create_time` timestamp NULL DEFAULT CURRENT_TIMESTAMP,
  `update_time` timestamp NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci;


CREATE TABLE `student_score` (
  `id` int(20) NOT NULL,
  `student_id` int(20) NOT NULL DEFAULT '0',
  `course_id` int(20) NOT NULL DEFAULT '0',
  `score` tinyint(3) unsigned NOT NULL,
  `create_time` timestamp NULL DEFAULT CURRENT_TIMESTAMP,
  `update_time` timestamp NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
  PRIMARY KEY (`id`),
  KEY `student_course_unique` (`student_id`,`course_id`),
  KEY `course_idx` (`course_id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci;

okcli命令准备

命令行参数

$ okcli

NAME:
   okcli - a cli tool to generate model

USAGE:
   okcli [global options] command [command options] [arguments...]

VERSION:
   0.0.1

COMMANDS:
   init     generate the json configuration template
   gen      generated from a configuration file
   help, h  Shows a list of commands or help for one command

GLOBAL OPTIONS:
   --help, -h     show help
   --version, -v  print the version

生成model

1、配置文件生成

$ okcli init

执行命令后将会在当前目录生成一个config.json文件,内容模板如下:

// TODO: FILLED IN YOUR REAL VALUE AND DELETE THIS LINE
{
    "username": "root",
    "password": "123456",
    "host": "localhost:3306",
    "database":{
        "name":"demo",
        "tables":["student","course","student_score"]
    },
    "tagPrefix":"db"
}
  • username:
    数据库连接名
  • password:
    数据库连接密码
  • host:
    数据库连接地址
  • database:
    数据库名称和数据表名列表
  • tagPrefix:
    待格式化model中tag的前缀,默认为"db"

将config.json配置文件修改为你需要连接的数据库地址,填入待生成model的数据库名和表名。

2、执行okcli gen 命令

$ okcli gen -c ./config.json -o ./model

successful,the path: /Users/anqiansong/go/src/okgo/model/student.go
successful,the path: /Users/anqiansong/go/src/okgo/model/course.go
successful,the path: /Users/anqiansong/go/src/okgo/model/studentscore.go

  • c 指定配置文件路径
  • o 指定model文件存放目录,默认为当前目录

执行以上命令将按照配置文件生成对应model文件,并存放在model文件夹中,如果指定model文件已经存在,则会提示你

WARNING:the file [student.go] would be override,Y/N?

3、生成model文件

  • Student
package model

import "time"

type (
    Student struct {
        Id         int32     `gorm:"id"`
        Name       string    `gorm:"name"`        // the student's name
        Gender     int8      `gorm:"gender"`      // the student's gender,0-male,1-female,DEFAULT:0
        Age        int8      `gorm:"age"`         // the student's age,DEFAULT:0
        Class      string    `gorm:"class"`       // the student's class
        CreateTime time.Time `gorm:"create_time"` // the column create time,DEFAULT:CURRENT_TIMESTAMP
        UpdateTime time.Time `gorm:"update_time"` // the column last update time,DEFAULT:CURRENT_TIMESTAMP
    }
)
  • Course
package model

import "time"

type (
    Course struct {
        Id         int32     `gorm:"id"`
        Name       string    `gorm:"name"`    // course name
        Teacher    string    `gorm:"teacher"` // teach person
        CreateTime time.Time `gorm:"create_time"`
        UpdateTime time.Time `gorm:"update_time"`
    }
)
  • StudentScore
package model

import "time"

type (
    StudentScore struct {
        Id         int32     `gorm:"id"`
        StudentId  int32     `gorm:"student_id"`
        CourseId   int32     `gorm:"course_id"`
        Score      int8      `gorm:"score"`
        CreateTime time.Time `gorm:"create_time"`
        UpdateTime time.Time `gorm:"update_time"`
    }
)

注意事项

model文件生成后目前可支持代码格式化,但是暂不支持包的自动引入,因此,在生成文件后你需要对model文件的
包进行检查导入。

结尾

如有对你有帮助,请帮忙点击一下在github给一下你的小星星作为我前进的动力。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,544评论 6 501
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,430评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,764评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,193评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,216评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,182评论 1 299
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,063评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,917评论 0 274
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,329评论 1 310
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,543评论 2 332
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,722评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,425评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,019评论 3 326
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,671评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,825评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,729评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,614评论 2 353