go模板-代码生成器

能用程序去做的事，就不要用手，编写自己的代码生成器就是用来解放你的双手，替你做一些重复性的工作。

上篇帖子写了模板的基础 go模板详说，有了基础就要做点什么东西，把所学到的东西应用起来才能更好的进步。于是用模板写了一个代码生成器，用于把数据库的表转换成go的struct，目前支持MySQL和PostgreSQL。这篇帖子写实现的思路和一些具体的代码实现。

Github地址: yggdrasill

看一下效果

大概的思路：

就是得到数据库下的所有的表数据。如果指定生成某几个表，只需查一下是否存在相应的表。
根据表名得到列信息，然后根据列的类型转换成go的类型，如果有自定的类型要做特殊处理。
利用模板把表的数据转换成结构体

组织元数据

这里的元数据也就是数据库的表信息，由于MySQL和PostgreSQL得到表和列信息的实现方式不同，但是最终的数据结构是一致的，就写了一个统一的接口来获取元数据信息。

MySQL

MySQL从information_schema.tables得到一个数据库下的所有表

代码语言：javascript复制

select table_name from information_schema.tables where table_schema = ? and table_type = 'base table';

如果只想查询指写的表，只需添加table_name in查询条件就行。

表的列信息从information_schema.columns获取

代码语言：javascript复制

select column_name, is_nullable, if(column_type = 'tinyint(1)', 'boolean', data_type), column_type like '%unsigned%' from information_schema.columns where table_schema = ? and  table_name = ? order by ordinal_position;

PostgreSQL

PostgreSQL也是从information_schema.tables获取表信息，不同的是，不需要指定数据库的名字，但MySql需要，感觉还是PostgreSQL更合理一些，因为在连接数据库的dsn里已经指定了要操作的db了

代码语言：javascript复制

select table_name from information_schema.tables where table_schema = 'public';

表的列信息从information_schema.columns获取

代码语言：javascript复制

select column_name, is_nullable, data_type, false
from information_schema.columns where table_schema = 'public' and table_name = $1 order by ordinal_position;