Golang 将数据库转换为gorm结构和RESTful api

栏目: Go · 发布时间: 5年前

内容简介:gen工具从给定的数据库生成golang结构,以便在.go文件中使用。它支持通过从数据库中读取有关列结构的详细信息,gen生成具有所需列名,数据类型和注释的go兼容结构类型。

https://github.com/smallnest/gen

介绍

gen工具从给定的数据库生成golang结构,以便在.go文件中使用。它支持 gorm 标签并实现一些可用的方法。它还可以为这些结构生成RESTful api。

通过从数据库中读取有关列结构的详细信息,gen生成具有所需列名,数据类型和注释的 go 兼容结构类型。

生成的数据类型包括对可空列 sql.NullX类型guregu null.X类型 以及预期的基本内置go类型的支持。

gen基于Seth Shelnutt的 db2struct 的工作而受到启发,而Db2Struct的基础/灵感来自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

支持的数据库

目前支持

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

计划支持

  • 神谕

MariaDB的/ MySQL的

通过查询INFORMATION_SCHEMA.Columns表,然后格式化类型,列名和元数据来创建结构,以创建可用的兼容结构类型。

支持的数据类型

目前仅支持有限数量的数据类型。初步支持包括:

  • 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)
  • 二进制
  • BLOB
  • LONGBLOB
  • MEDIUMBLOB
  • VARBINARY

问题

  • Postgres和 SQLite 驱动程序支持sql.ColumnType.Nullable()( #3
  • 无法区分表的主键。仅将第一个字段设置为主键。所以你需要在某些情况下改变它。

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

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持 码农网

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

大型网站技术架构演进与性能优化

大型网站技术架构演进与性能优化

许令波 / 电子工业出版社 / 2018-6 / 79

《大型网站技术架构演进与性能优化》从一名亲历者的角度,阐述了一个网站在业务量飞速发展的过程中所遇到的技术转型等各种问题及解决思路。从技术发展上看,网站经历了Web应用系统从分布式、无线多端、中台到国际化的改造;在解决大流量问题的方向上,涉及了从端的优化到管道到服务端甚至到基础环境优化的各个层面。 《大型网站技术架构演进与性能优化》总结的宝贵经验教训可以帮助读者了解当网站遇到类似问题时,应如何......一起来看看 《大型网站技术架构演进与性能优化》 这本书的介绍吧!

RGB转16进制工具
RGB转16进制工具

RGB HEX 互转工具

HTML 编码/解码
HTML 编码/解码

HTML 编码/解码

SHA 加密
SHA 加密

SHA 加密工具