能用程式去做的事,就不要用手,編寫自己的程式碼生成器就是用來解放你的雙手,替你做一些重複性的工作。
上篇帖子寫了模板的基礎 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;
型別轉換
從資料庫裡獲取完表和列的資訊後,就要把資料庫的型別轉換成go
相關的型別。
需要考慮的是是否有些型別要特殊處理,或自定義,特殊處理後的型別是否需要在
import
里加上具體的匯入。
方法getGoType
把得到的資料庫型別進行對比,返回go的型別。
模板
先看一下模板的程式碼,裡面的知識點上篇帖子都有說過
主要就是包名稱、是否有import、結構體資訊
import
應該是會用到的,有用到時間型別就需要import time
,還有前面說的,注意自定義的型別的匯入。
還有一點就是這個Tag
,由於模板的功能還不夠豐富,就完全用方法去實現了
看一下生成的效果
也可以根據你的業務對這個庫進行擴充套件,由於專案不讓用orm
完全手寫sql
,我就根據自己的業務進行了擴充套件,把dao
層的也實現了。
一定要用模板實現麼?當然不是,可以用你喜歡的方式去實現,只是正好閱讀了模板的官方文件,就用他做了實現。