快捷導(dǎo)航

golang 生成對應(yīng)的數(shù)據(jù)表struct定義操作

更新時間：2021年04月28日 15:14:31 作者：westhod

這篇文章主要介紹了golang 生成對應(yīng)的數(shù)據(jù)表struct定義操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

在開發(fā)過程中，常常需要將數(shù)據(jù)庫表對應(yīng)到golang的一個struct，特別是使用一些ORM工具，sqlx庫等，我是個懶人，即使數(shù)據(jù)表的字段不多，我也懶得去一個個對應(yīng)的敲入代碼，更別提數(shù)據(jù)表字段比較多的情況了，碼農(nóng)的時間，不能浪費在這啊，對吧？所以我在想，是不是有辦法可以自動生成。

我在工作時，用得最多的是mysql了，因此

本文針對mysql的數(shù)據(jù)表來自動生成golang 的struct定義

mysql有個自帶的數(shù)據(jù)庫information_schema，里面的信息量比較多，朋友們可以去百度下，我這里用到了表COLUMNS，它的字段包含數(shù)據(jù)庫名、表名、字段名、字段類型等，利用這個表的數(shù)據(jù)，把對應(yīng)的表的字段信息讀取出來，然后再根據(jù)golang的語法規(guī)則，生成文件就可以了。

大致思路確定了，開始動手。

我采用sqlx進行數(shù)據(jù)庫的訪問，首先定義一個struct來表示COLUMNS的數(shù)據(jù)，這里我只需要幾個字段，因此，沒有把表COLUMNS的所有字段都對應(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對應(yīng)的是哪個庫、哪個表，最終的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")

另外，我們項目習(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è)計數(shù)據(jù)庫時，是可空的，information_schema->COLUMNS中有個字段IS_NULLABLE專門表示，而golang的sql有幾個類型對應(yīng)：sql.NullString、sql.NullBool、sql.NullFloat64、sql.NullInt64，基本上是可以滿足使用要求的了。

有人可能會有疑問，假如字段類型為date、timestamp等，該對應(yīng)哪種呢？通常第三方的類庫會轉(zhuǎn)為string類型，那么就對應(yīng)sql.NullString好了。不過我這里沒有進行這方面的處理。

前期工作做好了，開始編碼：

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ù)")
 }
}

我把每個字段的tag，包括db和json的都加了了，在代碼最后，使用goimport工具添加需要import的package，它連format的工作都做了，實在不錯。

以下是我生成的一個用戶購物概要表的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"`
}

補充：Golang之方法（自定義類型，struct）

方法的使用，請看本天師的代碼

//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可以自動知道，接受者是指針，這里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()
}

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。

您可能感興趣的文章: