“邮客星球”论坛搭建之Gorm操作数据库

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

• 全功能 ORM (无限接近)
• 关联 (Has One, Has Many, Belongs To, Many To Many, 多态)
• 钩子 (在创建/保存/更新/删除/查找之前或之后)
• 预加载
• 事务
• 复合主键
• SQL 生成器
• 数据库自动迁移
• 自定义日志
• 可扩展性, 可基于 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 支持 sqlite，mysql，postgreSql，mssql，所以在连接数据库的时候需要指定你的数据库类型，下面我们以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{})