golang 生成對(duì)應(yīng)的數(shù)據(jù)表struct定義操作
在開(kāi)發(fā)過(guò)程中,常常需要將數(shù)據(jù)庫(kù)表對(duì)應(yīng)到golang的一個(gè)struct,特別是使用一些ORM工具,sqlx庫(kù)等,我是個(gè)懶人,即使數(shù)據(jù)表的字段不多,我也懶得去一個(gè)個(gè)對(duì)應(yīng)的敲入代碼,更別提數(shù)據(jù)表字段比較多的情況了,碼農(nóng)的時(shí)間,不能浪費(fèi)在這啊,對(duì)吧?所以我在想,是不是有辦法可以自動(dòng)生成。
我在工作時(shí),用得最多的是mysql了,因此
本文針對(duì)mysql的數(shù)據(jù)表來(lái)自動(dòng)生成golang 的struct定義
mysql有個(gè)自帶的數(shù)據(jù)庫(kù)information_schema,里面的信息量比較多,朋友們可以去百度下,我這里用到了表COLUMNS,它的字段包含數(shù)據(jù)庫(kù)名、表名、字段名、字段類(lèi)型等,利用這個(gè)表的數(shù)據(jù),把對(duì)應(yīng)的表的字段信息讀取出來(lái),然后再根據(jù)golang的語(yǔ)法規(guī)則,生成文件就可以了。
大致思路確定了,開(kāi)始動(dòng)手。
我采用sqlx進(jìn)行數(shù)據(jù)庫(kù)的訪(fǎng)問(wèn),首先定義一個(gè)struct來(lái)表示COLUMNS的數(shù)據(jù),這里我只需要幾個(gè)字段,因此,沒(méi)有把表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è)庫(kù)、哪個(gè)表,最終的golang文件保存地址
這里利用命令行參數(shù)來(lái)輸入:
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í)慣使用下劃線(xiàn)“_”來(lái)分割單詞,比如info_user,表示user表,而生成的struct名稱(chēng)為InfoUser,字段名也是類(lèi)似規(guī)則
因此定義了如下函數(shù)來(lái)處理這種情況:
func fmtFieldDefine(src string) string { temp := strings.Split(src, "_") // 有下劃線(xiàn)的,需要拆分 var str string for i := 0; i < len(temp); i++ { b := []rune(temp[i]) for j := 0; j < len(b); j++ { if j == 0 { // 首字母大寫(xiě)轉(zhuǎn)換 b[j] -= 32 str += string(b[j]) } else { str += string(b[j]) } } } return str }
即把下劃線(xiàn)去掉,且將單詞的首字母改為大寫(xiě)。
有些字段,在設(shè)計(jì)數(shù)據(jù)庫(kù)時(shí),是可空的,information_schema->COLUMNS中有個(gè)字段IS_NULLABLE專(zhuān)門(mén)表示,而golang的sql有幾個(gè)類(lèi)型對(duì)應(yīng):sql.NullString、sql.NullBool、sql.NullFloat64、sql.NullInt64,基本上是可以滿(mǎn)足使用要求的了。
有人可能會(huì)有疑問(wèn),假如字段類(lèi)型為date、timestamp等,該對(duì)應(yīng)哪種呢?通常第三方的類(lèi)庫(kù)會(huì)轉(zhuǎn)為string類(lèi)型,那么就對(duì)應(yīng)sql.NullString好了。不過(guò)我這里沒(méi)有進(jìn)行這方面的處理。
前期工作做好了,開(kāi)始編碼:
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: // 其他類(lèi)型當(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("查詢(xún)不到數(shù)據(jù)") } }
我把每個(gè)字段的tag,包括db和json的都加了了,在代碼最后,使用goimport工具添加需要import的package,它連format的工作都做了,實(shí)在不錯(cuò)。
以下是我生成的一個(gè)用戶(hù)購(gòu)物概要表的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之方法(自定義類(lèi)型,struct)
方法的使用,請(qǐng)看本天師的代碼
//Golang的方法定義 //Golang中的方法是作用在特定類(lèi)型的變量上,因此自定義類(lèi)型,都可以有方法,不僅僅是struct //定義:func (recevier type) methodName(參數(shù)列表)(返回值列表){} //方法和函數(shù)的區(qū)別 /* 1,函數(shù)調(diào)用:function(variable,參數(shù)列表) 2, 方法,variable.function(參數(shù)列表) 方法的控制,通過(guò)大小寫(xiě)空格控制 */
package main //Golang的方法定義 //Golang中的方法是作用在特定類(lèi)型的變量上,因此自定義類(lèi)型,都可以有方法,不僅僅是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 //修改地址的寫(xiě)法(&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語(yǔ)言載入json可能遇到的一個(gè)坑
Go 語(yǔ)言從新手到大神,每個(gè)人多少都會(huì)踩一些坑,那么下面這篇文章主要給大家介紹了關(guān)于go語(yǔ)言載入json可能遇到的一個(gè)坑,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面來(lái)一起看看吧。2017-07-07Golang實(shí)現(xiàn)四種負(fù)載均衡的算法(隨機(jī),輪詢(xún)等)
本文介紹了示例介紹了Golang 負(fù)載均衡的四種實(shí)現(xiàn),主要包括了隨機(jī),輪詢(xún),加權(quán)輪詢(xún)負(fù)載,一致性hash,感興趣的小伙伴們可以參考一下2021-06-06掌握GoLang Fiber路由和中間件技術(shù)進(jìn)行高效Web開(kāi)發(fā)
這篇文章主要為大家介紹了GoLang Fiber路由和中間件進(jìn)行高效Web開(kāi)發(fā),本文將深入探討 Fiber 中的路由細(xì)節(jié),學(xué)習(xí)如何創(chuàng)建和處理路由,深入了解使用路由參數(shù)的動(dòng)態(tài)路由,并掌握在 Fiber 應(yīng)用程序中實(shí)現(xiàn)中間件的藝術(shù)2024-01-01源碼剖析Golang中map擴(kuò)容底層的實(shí)現(xiàn)
之前的文章詳細(xì)介紹過(guò)Go切片和map的基本使用,以及切片的擴(kuò)容機(jī)制。本文針對(duì)map的擴(kuò)容,會(huì)從源碼的角度全面的剖析一下map擴(kuò)容的底層實(shí)現(xiàn),需要的可以參考一下2023-03-03詳解如何用Golang處理每分鐘100萬(wàn)個(gè)請(qǐng)求
在項(xiàng)目開(kāi)發(fā)中,我們常常會(huì)遇到處理來(lái)自數(shù)百萬(wàn)個(gè)端點(diǎn)的大量POST請(qǐng)求,本文主要介紹了Golang實(shí)現(xiàn)處理每分鐘100萬(wàn)個(gè)請(qǐng)求的方法,希望對(duì)大家有所幫助2023-04-04golang不到30行代碼實(shí)現(xiàn)依賴(lài)注入的方法
這篇文章主要介紹了golang不到30行代碼實(shí)現(xiàn)依賴(lài)注入的方法,小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2018-07-07ubuntu下搭建Go語(yǔ)言(golang)環(huán)境
這篇文章主要介紹了ubuntu下搭建Go語(yǔ)言(golang)環(huán)境,需要的朋友可以參考下2015-01-01Golang Map value不可尋址使用指針類(lèi)型代替示例詳解
這篇文章主要為大家介紹了Golang Map value不可尋址使用指針類(lèi)型代替示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-11-11