golang 生成對(duì)應(yīng)的數(shù)據(jù)表struct定義操作
在開發(fā)過程中，常常需要將數(shù)據(jù)庫表對(duì)應(yīng)到golang的一個(gè)struct，特別是使用一些ORM工具，sqlx庫等，我是個(gè)懶人，即使數(shù)據(jù)表的字段不多，我也懶得去一個(gè)個(gè)對(duì)應(yīng)的敲入代碼，更別提數(shù)據(jù)表字段比較多的情況了，碼農(nóng)的時(shí)間，不能浪費(fèi)在這啊，對(duì)吧？所以我在想，是不是有辦法可以自動(dòng)生成。
我在工作時(shí)，用得最多的是mysql了，因此
本文針對(duì)mysql的數(shù)據(jù)表來自動(dòng)生成golang 的struct定義
mysql有個(gè)自帶的數(shù)據(jù)庫information_schema，里面的信息量比較多，朋友們可以去百度下，我這里用到了表COLUMNS，它的字段包含數(shù)據(jù)庫名、表名、字段名、字段類型等，利用這個(gè)表的數(shù)據(jù)，把對(duì)應(yīng)的表的字段信息讀取出來，然后再根據(jù)golang的語法規(guī)則，生成文件就可以了。
大致思路確定了，開始動(dòng)手。
我采用sqlx進(jìn)行數(shù)據(jù)庫的訪問，首先定義一個(gè)struct來表示COLUMNS的數(shù)據(jù)，這里我只需要幾個(gè)字段，因此，沒有把表COLUMNS的所有字段都對(duì)應(yīng)到struct：
type FieldInfo struct {
ColName string `db:"COLUMN_NAME"`
DataType string `db:"DATA_TYPE"`
ColComment string `db:"COLUMN_COMMENT"`
IsNullable string `db:"IS_NULLABLE"`
}
需要指定生成的struct對(duì)應(yīng)的是哪個(gè)庫、哪個(gè)表，最終的golang文件保存地址
這里利用命令行參數(shù)來輸入：
var dbname= flag.String("db", "", "the database name")
var tblname = flag.String("tbl", "", "the table name to export")
var savepath = flag.String("path", "./", "the path to save file")
另外，我們項(xiàng)目習(xí)慣使用下劃線“_”來分割單詞，比如info_user，表示user表，而生成的struct名稱為InfoUser，字段名也是類似規(guī)則
因此定義了如下函數(shù)來處理這種情況：
func fmtFieldDefine(src string) string {
temp := strings.Split(src, "_") // 有下劃線的，需要拆分
var str string
for i := 0; i < len(temp); i++ {
b := []rune(temp[i])
for j := 0; j < len(b); j++ {
if j == 0 {
// 首字母大寫轉(zhuǎn)換
b[j] -= 32
str += string(b[j])
} else {
str += string(b[j])
}
}
}
return str
}
即把下劃線去掉，且將單詞的首字母改為大寫。
有些字段，在設(shè)計(jì)數(shù)據(jù)庫時(shí)，是可空的，information_schema->COLUMNS中有個(gè)字段IS_NULLABLE專門表示，而golang的sql有幾個(gè)類型對(duì)應(yīng)：sql.NullString、sql.NullBool、sql.NullFloat64、sql.NullInt64，基本上是可以滿足使用要求的了。
有人可能會(huì)有疑問，假如字段類型為date、timestamp等，該對(duì)應(yīng)哪種呢？通常第三方的類庫會(huì)轉(zhuǎn)為string類型，那么就對(duì)應(yīng)sql.NullString好了。不過我這里沒有進(jìn)行這方面的處理。
前期工作做好了，開始編碼：
func main() {
flag.Parse()
fmt.Println("table name -->", *tblname)
dns := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8", dbuser, dbpwd, dbhost, "information_schema")
db := sqlx.MustConnect("mysql", dns)
var fs []FieldInfo
err := db.Select(&fs, "SELECT COLUMN_NAME, DATA_TYPE, COLUMN_COMMENT, IS_NULLABLE FROM COLUMNS WHERE TABLE_NAME=? and table_schema=?", *tblname, *dbname)
if err != nil {
fmt.Println(err)
panic(err)
}
if len(fs) > 0 {
var buffer bytes.Buffer
buffer.WriteString("package models\n")
buffer.WriteString("type " + fmtFieldDefine(*tblname) + " struct {\n")
for _, v := range fs {
buffer.WriteString("" + fmtFieldDefine(v.ColName) + " ")
switch v.DataType {
case "int", "tinyint", "smallint":
if v.IsNullable == "YES" {
buffer.WriteString("sql.NullInt64 ")
} else {
buffer.WriteString("int ")
}
case "bigint":
if v.IsNullable == "YES" {
buffer.WriteString("sql.NullInt64 ")
} else {
buffer.WriteString("int64 ")
}
case "char", "varchar", "longtext", "text", "tinytext":
if v.IsNullable == "YES" {
buffer.WriteString("sql.NullString ")
} else {
buffer.WriteString("string ")
}
case "date", "datetime", "timestamp":
buffer.WriteString("time.Time ")
case "double", "float":
if v.IsNullable == "YES" {
buffer.WriteString("sql.NullFloat64 ")
} else {
buffer.WriteString("float64 ")
}
default:
// 其他類型當(dāng)成string處理
if v.IsNullable == "YES" {
buffer.WriteString("sql.NullString ")
} else {
buffer.WriteString("string ")
}
}
buffer.WriteString(fmt.Sprintf("`db:\"%s\" json:\"%s\"`\n", v.ColName, v.ColName))
}
buffer.WriteString(`}`)
fmt.Println(buffer.String())
filename := *savepath + "\\" + *tblname + ".go"
f, _ := os.Create(filename)
f.Write([]byte(buffer.String()))
f.Close()
cmd := exec.Command("goimports", "-w", filename)
cmd.Run()
} else {
fmt.Println("查詢不到數(shù)據(jù)")
}
}
我把每個(gè)字段的tag，包括db和json的都加了了，在代碼最后，使用goimport工具添加需要import的package，它連format的工作都做了，實(shí)在不錯(cuò)。
以下是我生成的一個(gè)用戶購物概要表的struct定義：
package models
import (
"database/sql"
"time"
)
type InfoUserShoppingSummary struct {
Id int `db:"id" json:"id"`
TransactionId sql.NullString `db:"transaction_id" json:"transaction_id"`
OutTradeNo sql.NullString `db:"out_trade_no" json:"out_trade_no"`
WuId int `db:"wu_id" json:"wu_id"`
WdId int `db:"wd_id" json:"wd_id"`
TotalFee float64 `db:"total_fee" json:"total_fee"`
PayStaus sql.NullInt64 `db:"pay_staus" json:"pay_staus"`
CreateTime time.Time `db:"create_time" json:"create_time"`
UpdateTime time.Time `db:"update_time" json:"update_time"`
Address sql.NullString `db:"address" json:"address"`
}
補(bǔ)充：Golang之方法（自定義類型，struct）
方法的使用，請(qǐng)看本天師的代碼
//Golang的方法定義
//Golang中的方法是作用在特定類型的變量上，因此自定義類型，都可以有方法，不僅僅是struct
//定義：func (recevier type) methodName(參數(shù)列表)(返回值列表){}
//方法和函數(shù)的區(qū)別
/*
1，函數(shù)調(diào)用：function(variable，參數(shù)列表)
2， 方法，variable.function(參數(shù)列表)
方法的控制，通過大小寫空格控制
*/
package main
//Golang的方法定義
//Golang中的方法是作用在特定類型的變量上，因此自定義類型，都可以有方法，不僅僅是struct
//定義：func (recevier type) methodName(參數(shù)列表)(返回值列表){}
import "fmt"
type integer int
func (p integer) print() {
fmt.Println("p is:", p)
}
//這里傳遞的是副本，想改變p的值，需要傳遞指針
func (p *integer) set(b integer) {
*p = b
}
type Student struct {
Name string
Age int
Score int
sex int
}
//這里需要接受指針 *Student（接收者），否則修改不了值
func (p *Student) init(name string, age int, score int) {
p.Name = name
p.Age = age
p.Score = score
fmt.Println(p)
}
func (p Student) get() Student {
return p
}
func main() {
var stu Student
//修改地址的寫法(&stu).init
//但是go可以自動(dòng)知道，接受者是指針，這里stu就傳遞地址
stu.init("stu", 18, 99)
stu1 := stu.get()
fmt.Println(stu1)
//type integer方法
var a integer
a = 100
a.print()
a.set(1000)
a.print()
}
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教。
相關(guān)文章
關(guān)于go語言載入json可能遇到的一個(gè)坑
Go 語言從新手到大神，每個(gè)人多少都會(huì)踩一些坑，那么下面這篇文章主要給大家介紹了關(guān)于go語言載入json可能遇到的一個(gè)坑，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面來一起看看吧。2017-07-07
Golang實(shí)現(xiàn)四種負(fù)載均衡的算法(隨機(jī),輪詢等)
本文介紹了示例介紹了Golang 負(fù)載均衡的四種實(shí)現(xiàn)，主要包括了隨機(jī)，輪詢，加權(quán)輪詢負(fù)載，一致性hash，感興趣的小伙伴們可以參考一下2021-06-06
位運(yùn)算可能在平常的編程中使用的并不多，但涉及到底層優(yōu)化，一些算法及源碼可能會(huì)經(jīng)常遇見。下面這篇文章主要給大家介紹了關(guān)于Go語言中你所不知道的位操作用法的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，需要的朋友可以參考下。2017-12-12
掌握GoLang Fiber路由和中間件技術(shù)進(jìn)行高效Web開發(fā)
這篇文章主要為大家介紹了GoLang Fiber路由和中間件進(jìn)行高效Web開發(fā),本文將深入探討 Fiber 中的路由細(xì)節(jié),學(xué)習(xí)如何創(chuàng)建和處理路由,深入了解使用路由參數(shù)的動(dòng)態(tài)路由,并掌握在 Fiber 應(yīng)用程序中實(shí)現(xiàn)中間件的藝術(shù)2024-01-01
這篇文章主要介紹了Go語言使用組合的方式實(shí)現(xiàn)多繼承的方法,實(shí)例分析了多繼承的原理與使用組合方式來實(shí)現(xiàn)多繼承的技巧,需要的朋友可以參考下，希望可以幫助到你2021-06-06
源碼剖析Golang中map擴(kuò)容底層的實(shí)現(xiàn)
之前的文章詳細(xì)介紹過Go切片和map的基本使用，以及切片的擴(kuò)容機(jī)制。本文針對(duì)map的擴(kuò)容，會(huì)從源碼的角度全面的剖析一下map擴(kuò)容的底層實(shí)現(xiàn)，需要的可以參考一下2023-03-03
詳解如何用Golang處理每分鐘100萬個(gè)請(qǐng)求
在項(xiàng)目開發(fā)中，我們常常會(huì)遇到處理來自數(shù)百萬個(gè)端點(diǎn)的大量POST請(qǐng)求，本文主要介紹了Golang實(shí)現(xiàn)處理每分鐘100萬個(gè)請(qǐng)求的方法，希望對(duì)大家有所幫助2023-04-04
golang不到30行代碼實(shí)現(xiàn)依賴注入的方法
這篇文章主要介紹了golang不到30行代碼實(shí)現(xiàn)依賴注入的方法,小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧2018-07-07
這篇文章主要介紹了ubuntu下搭建Go語言（golang）環(huán)境,需要的朋友可以參考下2015-01-01
Golang Map value不可尋址使用指針類型代替示例詳解
這篇文章主要為大家介紹了Golang Map value不可尋址使用指針類型代替示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-11-11