Go實現(xiàn)簡單的數(shù)據(jù)庫表轉結構體詳解
前言
碼上源碼
package main
import (
"database/sql"
"fmt"
_ "github.com/go-sql-driver/mysql"
"strings"
)
/**
自動將表生成 model結構,
通過創(chuàng)建數(shù)據(jù)庫連接,讀取數(shù)據(jù)庫的所有表并對所有的表元數(shù)據(jù)封裝轉化實體結構體
*/
type SchemaMeta struct {
Field string
Type string
Null string
Key string
Default interface{}
Extra string
}
func main() {
dbString := "{YOUR_MYSQL_CONNECTION}"
db, _ := sql.Open("mysql", dbString)
tables := getTables(db)
for _, table := range tables {
metas := getTableInfo(table, db)
result := changeMetas(table, metas)
fmt.Println(result)
}
}
func getTables(db *sql.DB) []string {
var tables []string
res, _ := db.Query("SHOW TABLES")
for res.Next() {
var table string
res.Scan(&table)
tables = append(tables, table)
}
return tables
}
func getTableInfo(tableName string, db *sql.DB) (metas []SchemaMeta) {
list, _ := db.Query(fmt.Sprintf("show columns from %s", tableName))
for list.Next() {
var data SchemaMeta
err := list.Scan(&data.Field, &data.Type, &data.Null, &data.Key, &data.Default, &data.Extra)
if err != nil {
fmt.Println(err.Error())
}
metas = append(metas, data)
}
return metas
}
func changeMetas(tableName string, metas []SchemaMeta) string {
var modelStr string
for _, val := range metas {
dataType := "interface{}"
if val.Type[:3] == "int" {
dataType = "int"
} else if val.Type[:7] == "varchar" {
dataType = "string"
} else if val.Type[:7] == "tinyint" {
dataType = "bool"
} else if val.Type == "datetime" {
dataType = "time.Time"
}
field := val.Field
field = strings.ToUpper(field[:1]) + field[1:]
modelStr += fmt.Sprintf("%s %s\n", field, dataType)
}
tableName = strings.ToUpper(tableName[:1]) + tableName[1:]
return fmt.Sprintf("type %s struct {\n %s }", tableName, modelStr)
}設計好了數(shù)據(jù)庫表之后最煩的就是又要在代碼中建一層實體層然后一個個創(chuàng)建對應表的結構體。關鍵那么多項目每次都是需要創(chuàng)建一份,所以就使用 go 實現(xiàn)一個簡單的將數(shù)據(jù)庫的表轉化為結構體。
基本的設計思路也是很簡單:
- 連接數(shù)據(jù)庫
- 獲取數(shù)據(jù)庫中所有的表
- 獲取表的所有的元數(shù)據(jù)
- 將元數(shù)據(jù)轉換為字符串,該字符串就是一個 struct的內容并輸出結果
設計過程
經常使用 MySQL 的時候,可以通過 SQL 語句查詢相關的信息,比如查看數(shù)據(jù)庫的狀態(tài)、數(shù)據(jù)庫的數(shù)據(jù)表、數(shù)據(jù)庫服務的狀態(tài)等等語句。
show databases 查看所有的數(shù)據(jù)庫 show tables [from dbName] 查看數(shù)據(jù)庫的所有表數(shù)據(jù) show columns from tableName [from dbName] 查看數(shù)據(jù)庫表的所有元信息 show status 查看數(shù)據(jù)服務的狀態(tài)
所以就可以使用以上的 SQL 語句進行相操作,比如獲取所有的表,獲取相關表的元數(shù)據(jù)
func getTableInfo(tableName string, db *sql.DB) (metas []SchemaMeta) {
list, _ := db.Query(fmt.Sprintf("show columns from %s", tableName))
for list.Next() {
var data SchemaMeta
err := list.Scan(&data.Field, &data.Type, &data.Null, &data.Key, &data.Default, &data.Extra)
if err != nil {
fmt.Println(err.Error())
}
metas = append(metas, data)
}
return metas
}
表的元數(shù)據(jù)獲取完之后就可以進行封裝成一個字符串。最主要的功能有兩個:
- db 數(shù)據(jù)類型轉 go 數(shù)據(jù)類型
- 首字母大小寫,
數(shù)據(jù)類型的轉換簡單處理的話可以使用枚舉的方式將 db 數(shù)據(jù)類型和 go 的數(shù)據(jù)類型建立一個map 進行一一對應,當然也可以使用模糊匹配的方式。我這里直接使用截取字符串值去匹配。
因要設置為對象類型,必須設置為公有的,所以結構體名稱和字段名稱都必須是大寫。直接使用截取第一位然后進行大寫轉換再重新拼接出去。比如 field = strings.ToUpper(field[:1]) + field[1:]。這部分的數(shù)據(jù)大小寫還需要根據(jù)自己的需求進行調整,比如有寫字段是這樣:“user_code”那么轉換成字段應該是“UserCode”。
轉換完之后就可以輸出,一個簡單的工具就完成了。
迭代升級
- 支持多種數(shù)據(jù)庫:現(xiàn)目前使用的是 MySQL 數(shù)據(jù)庫進行開發(fā),而市面上其實有很多種數(shù)據(jù),所以這部分可以繼續(xù)擴展支持多種數(shù)據(jù)庫。
- 數(shù)據(jù)類型轉換:現(xiàn)只支持自己所使用的相關數(shù)據(jù)類型
- 如果有前端搭配使用可以涉及按需生成
以上就是Go實現(xiàn)簡單的數(shù)據(jù)庫表轉結構體詳解的詳細內容,更多關于Go數(shù)據(jù)庫表轉結構體的資料請關注腳本之家其它相關文章!
相關文章
詳解Golang time包中的time.Duration類型
在日常開發(fā)過程中,會頻繁遇到對時間進行操作的場景,使用 Golang 中的 time 包可以很方便地實現(xiàn)對時間的相關操作,本文講解一下 time 包中的 time.Duration 類型,需要的朋友可以參考下2023-07-07
go浮點數(shù)轉字符串保留小數(shù)點后N位的完美解決方法
這篇文章主要介紹了go浮點數(shù)轉字符串保留小數(shù)點后N位解決辦法,本文給大家介紹的非常詳細,對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-05-05
golang實現(xiàn)http server提供文件下載功能
這篇文章主要介紹了golang實現(xiàn)http server提供文件下載功能,本文給大家簡單介紹了Golang的相關知識,非常不錯,具有一定的參考借鑒價值,需要的朋友可以參考下2020-02-02
Golang微服務框架Kratos實現(xiàn)分布式任務隊列Asynq的方法詳解
任務隊列(Task Queue) 一般用于跨線程或跨計算機分配工作的一種機制,在Golang語言里面,我們有像Asynq和Machinery這樣的類似于Celery的分布式任務隊列,本文就給大家詳細介紹一下Golang微服務框架Kratos實現(xiàn)分布式任務隊列Asynq的方法,需要的朋友可以參考下2023-09-09

