使用gin与gorm开发CRUD操作

gin启动

昨天学习到gin如何返回一个json数据之后，希望按照往常的开发经验，定义统一的返回报文体，于是定义了以下结构

type BaseResponse struct {
    code    string
    message string
}

然而使用下面的代码返回的数据一直是{}

func main() {
    r := gin.Default()
    r.GET("/hello", hello)
    r.Run() // listen and serve on 0.0.0.0:8080
}

func hello(c *gin.Context) {
    res := &BaseResponse{
        code:    "000000",
        message: "hello world",
    }
    c.JSON(200, res)
}

自己研究源码捣鼓半天也没弄出来个所以然，于是查阅资料才知道犯了一个很基础的错误。c.JSON方法底层使用了json.Marshal来序列化参数，这个方法只能序列化公有属性，也就是说要把BaseResponse中字段的首字母大写。首字母大写表示公有可导出的，小写是私有，这个知识点在GO中是比较基础的，很多地方都会用到，需要牢记。

gin结合gorm

接下来就开始使用gorm来进行crud操作了。我在service.go文件中提供了数据库连接的配置与初始化方法，在server.go文件中首先初始化数据库连接，再启动web容器接收请求。

server.go

type BaseResponse struct {
    Code    string
    Message string
}

func main() {
    ConfigDb("1", "2", "3", 4, "5")
    InitDb()
    defer Db.Close()

    r := gin.Default()
    r.POST("/author", addAuthor)
    r.Run("localhost:2046")
}

func addAuthor(c *gin.Context) {
    author := &Author{
        AuthorName: c.Query("AuthorName"),
        CreateTime: time.Now(),
    }
    NewAuthor(*author)
    message := fmt.Sprintf("Create author with name : %s", author.AuthorName)
    res := &BaseResponse{
        Code:    "000000",
        Message: message,
    }
    c.JSON(200, res)
}

service.go

type DBConfig struct {
    user, pw, host string
    port int
    dbName string
}

var config *DBConfig
var Db *gorm.DB

func ConfigDb(user, pw, host string, port int, dbName string) {
    config = &DBConfig{
        user:   user,
        pw:     pw,
        host:   host,
        port:   port,
        dbName: dbName,
    }
}

func InitDb() *gorm.DB {
    if config == nil {
        panic("Do configuration first!")
    }
    // connect to mysql
    connStr := fmt.Sprintf("%s:%s@(%s:%d)/%s?charset=utf8&parseTime=True&loc=Local", config.user, config.pw, config.host, config.port, config.dbName)
    var err error
    Db, err = gorm.Open("mysql", connStr)
    if err != nil {
        panic("Connect to database fail")
    }
    Db.SingularTable(true)
    Db.LogMode(true)
    return Db
}

// crud
func NewAuthor(author Author) {
    Db.Create(&author)
}

简单地做了一个逻辑处理，发送post请求localhost:2046/author?AuthorName=xxx，然后在数据库中新增一条author记录，返回结果成功。

{
"Code": "000000",
"Message": "Create author with name : xxx"
}