go模板-程式碼生成器

2020 年 5 月 27 日
筆記
Go, go sql builder, golang, go程式碼生成器, go模板, sql-builder, 程式碼生成器

能用程式去做的事，就不要用手，編寫自己的程式碼生成器就是用來解放你的雙手，替你做一些重複性的工作。
上篇帖子寫了模板的基礎 go模板詳說，有了基礎就要做點什麼東西，把所學到的東西應用起來才能更好的進步。於是用模板寫了一個程式碼生成器，用於把資料庫的錶轉換成go的struct，目前支援MySQL和PostgreSQL。這篇帖子寫實現的思路和一些具體的程式碼實現。
Github地址: yggdrasill
看一下效果

大概的思路：

就是得到資料庫下的所有的表數據。如果指定生成某幾個表，只需查一下是否存在相應的表。
根據表名得到列資訊，然後根據列的類型轉換成go的類型，如果有自定的類型要做特殊處理。
利用模板把表的數據轉換成結構體

組織元數據

這裡的元數據也就是資料庫的表資訊，由於MySQL和PostgreSQL得到表和列資訊的實現方式不同，但是最終的數據結構是一致的，就寫了一個統一的介面來獲取元數據資訊。

MySQL

MySQL從information_schema.tables得到一個資料庫下的所有表

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

如果只想查詢指寫的表，只需添加table_name in查詢條件就行。
表的列資訊從information_schema.columns獲取

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了

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

表的列資訊從information_schema.columns獲取

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