介紹
gen工具從給定的數(shù)據(jù)庫生成golang結(jié)構(gòu)，以便在.go文件中使用莉测。它支持gorm標(biāo)簽并實現(xiàn)一些可用的方法。它還可以為這些結(jié)構(gòu)生成RESTful api。
通過從數(shù)據(jù)庫中讀取有關(guān)列結(jié)構(gòu)的詳細(xì)信息，gen生成具有所需列名臀脏，數(shù)據(jù)類型和注釋的go兼容結(jié)構(gòu)類型。
生成的數(shù)據(jù)類型包括對可空列sql.NullX類型或guregu null.X類型 以及預(yù)期的基本內(nèi)置go類型的支持冀自。
gen基于Seth Shelnutt的db2struct的工作而受到啟發(fā)揉稚，而Db2Struct的基礎(chǔ)/靈感來自ChimeraCoder的gojson包gojson的工作。
獲取
go get github.com/smallnest/gen
使用
$ gen --connstr "root@tcp(127.0.0.1:3306)/employees?&parseTime=True" --database employees --json --gorm --guregu --rest
支持的數(shù)據(jù)庫
目前支持
- MariaDB的
- MySQL的
- PostgreSQL的
- Microsoft SQL Server
- SQLite的
計劃支持
- 神諭
MariaDB的/ MySQL的
通過查詢INFORMATION_SCHEMA.Columns表熬粗，然后格式化類型窃植，列名和元數(shù)據(jù)來創(chuàng)建結(jié)構(gòu)，以創(chuàng)建可用的兼容結(jié)構(gòu)類型荐糜。
支持的數(shù)據(jù)類型
目前僅支持有限數(shù)量的數(shù)據(jù)類型巷怜。初步支持包括：
- tinyint（sql.NullInt64或null.Int）
- int（sql.NullInt64或null.Int）
- smallint（sql.NullInt64或null.Int）
- mediumint（sql.NullInt64或null.Int）
- bigint（sql.NullInt64或null.Int）
- decimal（sql.NullFloat64或null.Float）
- float（sql.NullFloat64或null.Float）
- double（sql.NullFloat64或null.Float）
- datetime（null.Time）
- 時間（null.Time）
- date（null.Time）
- 時間戳（null.Time）
- var（sql.String或null.String）
- 枚舉（sql.String或null.String）
- varchar（sql.String或null.String）
- longtext（sql.String或null.String）
- mediumtext（sql.String或null.String）
- text（sql.String或null.String）
- tinytext（sql.String或null.String）
- 二進(jìn)制
- BLOB
- LONGBLOB
- MEDIUMBLOB
- VARBINARY
問題
- Postgres和SQLite驅(qū)動程序支持sql.ColumnType.Nullable（）（＃3）
- 無法區(qū)分表的主鍵。僅將第一個字段設(shè)置為主鍵暴氏。所以你需要在某些情況下改變它延塑。
gorm
package model
import (
"time"
"github.com/guregu/null"
"github.com/jinzhu/gorm"
)
type Title struct {
ToDate null.Time `gorm:"column:to_date" json:"to_date"`
EmpNo int `gorm:"column:emp_no" json:"emp_no"`
Title string `gorm:"column:title" json:"title"`
FromDate time.Time `gorm:"column:from_date" json:"from_date"`
}
// TableName sets the insert table name for this struct type
func (t *Title) TableName() string {
return "titles"
}
func (t *Title) CreateTitle(db *gorm.DB) error {
return db.Create(t).Error
}
RESTful api
package api
import (
"net/http"
"github.com/julienschmidt/httprouter"
"model"
)
func configTitlesRouter(router *httprouter.Router) {
router.GET("/titles", GetAllTitles)
router.POST("/titles", PostTitle)
router.GET("/titles/:id", GetTitle)
router.PUT("/titles/:id", PutTitle)
router.DELETE("/titles/:id", DeleteTitle)
}
func GetAllTitles(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
titles := []model.Title{}
DB.Find(&titles)
writeJSON(w, &titles)
}
func GetTitle(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
id := ps.ByName("id")
title := &model.Title{}
if DB.First(title, id).Error != nil {
http.NotFound(w, r)
return
}
writeJSON(w, title)
}
func PostTitle(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
title := &model.Title{}
if err := readJSON(r, title); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
if err := DB.Save(title).Error; err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
writeJSON(w, title)
}
func PutTitle(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
id := ps.ByName("id")
title := &model.Title{}
if DB.First(title, id).Error != nil {
http.NotFound(w, r)
return
}
updated := &model.Title{}
if err := readJSON(r, updated); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
// TODO: copy necessary fields from updated to title
if err := DB.Save(title).Error; err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
writeJSON(w, title)
}
func DeleteTitle(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
id := ps.ByName("id")
title := &model.Title{}
if DB.First(title, id).Error != nil {
http.NotFound(w, r)
return
}
if err := DB.Delete(title).Error; err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
w.WriteHeader(http.StatusOK)
}
