一、初识 gorm

Gorm 是一款强大且优秀的 Go 语言 orm 框架。主要功能你如下:

  • 全功能 ORM (无限接近)
  • 关联 (Has One, Has Many, Belongs To, Many To Many, 多态)
  • 钩子 (在创建/保存/更新/删除/查找之前或之后)
  • 预加载
  • 事务
  • 复合主键
  • SQL 生成器
  • 数据库自动迁移
  • 自定义日志
  • 可扩展性, 可基于 GORM 回调编写插件
  • 所有功能都被测试覆盖
  • 开发者友好

二、Gorm入门

添加 gorm 依赖

go get -u github.com/jinzhu/gorm

模型定义

type User struct {
    Id    int64  `gorm:"PRIMARY_KEY;AUTO_INCREMENT"` // 编号,主键、自增长
    Name  string `gorm:"size:32;not null"`           // 用户名,长度32,不可空
    Email string `gorm:"size:128;not null"`          // 邮箱,长度128,不可空
}

模型的每个字段都可以添加标记(Tag),通过 Tag 可以配置字段的属性,gorm 支持以下标记:

二、连接数据库

gorm 支持 sqlitemysqlpostgreSqlmssql,所以在连接数据库的时候需要指定你的数据库类型,下面我们以mysql为例进行讲解。

首先我们连上 mysql 创建数据库:test_db,创建数据库命令如下:

CREATE DATABASE `test_db` DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;

接下来编写 Go 代码,使用gorm连接数据库:

package main

import (
    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/mysql"
)

func main() {
    // 连接数据库
    db, err := gorm.Open("mysql", "root@tcp(localhost:3306)/test_db?charset=utf8mb4&parseTime=True&loc=Local")
    if err != nil {
        panic("failed to connect database")
    }
    defer db.Close()
}

首先我们需要将 gorm 和 mysql 支持引入进来,即:

import (
    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/mysql"
)

那么链接数据库就变得很简单了, 只需要调用 gorm.Open 方法即可,该方法有两个参数,第一个参数是指定数据库类型,第二个参数是数据库的链接。这里我们将数据库类型设定为 mysql,mysql 的链接格式为:

username:password@tcp(数据库地址:端口)/数据库名称

gorm.Open有两个返回值,第一个返回值就是数据库链接,第二个返回值是错误信息,当错误信息为空时表示数据库连接成功,连接成功之后我们即可正常操作数据库了。

另外,gorm 还提供了AutoMigrate功能,他会根据我们结构体中的定义来自动为我们创建表,开启只需要调用AutoMigrate方法,并且将我们的结构体传入进去即可,代码如下:

// 自动根据结构体创建
db.AutoMigrate(&User{})

三、数据库增删改查

连接好数据库后,我们来试试基础的增删改查功能

插入一条数据

db.Create(&User{
    Name:  "小明",
    Email: "xiaoming@gmail.com",
})

更新数据

var user User
db.Where("name = ?", "小明").First(&user)
fmt.Println("小明的邮箱为:" + user.Email)

// 重新赋值,然后更新
user.Email = "xiaoming2@gmail.com"
db.Save(user)

查询数据

//查询单条数据:
var user User
db.Where("name = ?", "小明").First(&user)
fmt.Println("小明的邮箱为:" + user.Email)
//查询列表数据:
var users []User
db.Find(&users)
fmt.Println("总用户数:", len(users))

删除数据

db.Where("name = ?", "小明").Delete(&User{})