go模板-程式碼生成器

peng發表於2020-05-27

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

大概的思路:

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

組織後設資料

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

MySQL

MySQLinformation_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層的也實現了。

一定要用模板實現麼?當然不是,可以用你喜歡的方式去實現,只是正好閱讀了模板的官方文件,就用他做了實現。

Github地址: yggdrasill

相關文章