https://github.com/smallnest/gen

介紹

gen工具從給定的數(shù)據(jù)庫(kù)生成golang結(jié)構(gòu)，以便在.go文件中使用。它支持gorm標(biāo)簽并實(shí)現(xiàn)一些可用的方法。它還可以為這些結(jié)構(gòu)生成RESTful api。

通過(guò)從數(shù)據(jù)庫(kù)中讀取有關(guān)列結(jié)構(gòu)的詳細(xì)信息，gen生成具有所需列名，數(shù)據(jù)類(lèi)型和注釋的go兼容結(jié)構(gòu)類(lèi)型。

生成的數(shù)據(jù)類(lèi)型包括對(duì)可空列sql.NullX類(lèi)型或guregu null.X類(lèi)型 以及預(yù)期的基本內(nèi)置go類(lèi)型的支持。

gen基于Seth Shelnutt的db2struct的工作而受到啟發(fā)，而Db2Struct的基礎(chǔ)/靈感來(lái)自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ù)庫(kù)

目前支持

• MariaDB的
• MySQL的
• PostgreSQL的
• Microsoft SQL Server
• SQLite的

計(jì)劃支持

• 神諭

MariaDB的/ MySQL的

通過(guò)查詢INFORMATION_SCHEMA.Columns表，然后格式化類(lèi)型，列名和元數(shù)據(jù)來(lái)創(chuàng)建結(jié)構(gòu)，以創(chuàng)建可用的兼容結(jié)構(gòu)類(lèi)型。

支持的數(shù)據(jù)類(lèi)型

目前僅支持有限數(shù)量的數(shù)據(jù)類(lèi)型。初步支持包括：

• 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）
• 時(shí)間（null.Time）
• date（null.Time）
• 時(shí)間戳（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

問(wèn)題

• Postgres和SQLite驅(qū)動(dòng)程序支持sql.ColumnType.Nullable（）（＃3）
• 無(wú)法區(qū)分表的主鍵。僅將第一個(gè)字段設(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)
}