Golang数据库编程之GORM库入门

原创作者，公众号【程序员读书】，欢迎关注公众号，转载文章请注明出处哦。

在上一篇文章中我们讲解了使用Go语言的标准库sql/database包操作数据库的过程，虽然使用sql/database包操作数据也是挺方便的，但是需要自己写每一条SQL语句，因此我们可能会自己再度进行封装，以便更好地使用，而使用现有Go语言开源ORM框架则是代替自己封装的一个更好的方式。

ORM，即对象关系映射(Object Relational Mapping)，可以简单理解为将关系型数据库中的数据表映射为编程语言中的具体的数据类型(如struct)，而GORM库就是一个使用Go语言实现的且功能非常完善易使用的ORM框架。

下面一起来探索一下如何使用GORM框架吧！

特性

• 关联 (Has One, Has Many, Belongs To, Many To Many, 多态)
• 钩子 (在创建/保存/更新/删除/查找之前或之后)
• 预加载
• 事务
• 复合主键
• SQL 生成器
• 数据库自动迁移
• 自定义日志
• 可扩展性, 可基于 GORM 回调编写插件

如何安装

安装GORM非常简单，使用go get -u就可以在GOPATH目录下安装最新GROM框架。

go get -u github.com/jinzhu/gorm

安装之后，便可以使用import关键字导入GORM库，开始使用啦！

import "github.com/jinzhu/gorm"

支持的数据库

GORM框架支持MySQL,SQL Server,Sqlite3,PostgreSQL四种数据库驱动，如果我们要连接这些数据库，则需要导入不同的驱动包及定义不同格式的DSN(Data Source Name)。

MySQL

1. 导入

import _ "github.com/jinzhu/gorm/dialects/mysql"
//或者//import _ "github.com/go-sql-driver/mysql"

2. DSN

//user指用户名，password指密码,dbname指数据库名
"user:password@/dbname?charset=utf8&parseTime=True&loc=Local"

SQL Server

1. 导入

import _ "github.com/jinzhu/gorm/dialects/mssql"

2. DSN

//username指用户名，password指密码,host指主机地址，port指端口号，database指数据库名
"sqlserver://username:password@host:port?database=dbname"

Sqlite3

1. 导入

import _ "github.com/jinzhu/gorm/dialects/sqlite"

2. DSN

连接Sqlite3数据库的DSN只需要指定Sqlite3的数据库文件的路径即可，如：

//数据库路径
/tmp/gorm.db

PostgreSQL

1. 导入

import _ "github.com/jinzhu/gorm/dialects/postgres"

2. DSN

//host指主机地址,port指端口号,user指用户名,dbname指数据库名,password指密码
host=myhost port=myport user=gorm dbname=gorm password=mypassword

连接数据库

上面我们定义了连接不同的数据库的DSN，下面演示如果连接数据库，使用gorm.Open()方法可以初始化并返回一个gorm.DB结构体，这个结构体封装了GORM框架所有的数据库操作方法，下面是gorm.Open()方法的定义：

func Open(dialect string, args ...interface{}) (db *DB, err error)

示例代码：

package main

import "github.com/jinzhu/gorm"
import _ "github.com/jinzhu/gorm/dialects/mysql"//导入连接MySQL数据库的驱动包

//DSN
const DSN = "root:123456@tcp(localhost:3306)/test?charset=utf8&parseTime=True&loc=Local"
//指定驱动
const DRIVER = "mysql"

var db *gorm.DB

func init() {
    var err error
    db,err = gorm.Open(DRIVER,DSN)
    if err != nil{
        panic(err)
    }
}

func main(){
    defer db.Close()//退出前执行关闭
    //调用db执行具体的逻辑
}

在上面的例子中，我们在init方法中初始化gorm.DB结构体，这样在下面的例子可以直接使用变量db直接进行数据库操作。

基本操作

使用gorm.Open()函数返回一个gorm.DB结构体后，我们可以使用gorm.DB结构体提供的方法操作数据库，下面我们演示如何使用gorm.DB进行创建、查询、更新、删除等最基本的操作。

其实gorm.DB是在Go语言的database/sql库中的sql.DB结构体上再封装，因为gorm.DB提供许多和sql.DB一样的方法，如下所示：

func (s *DB) Exec(sql string, values ...interface{}) *DB
func (s *DB) Row() *sql.Row
func (s *DB) Rows() (*sql.Rows, error)
func (s *DB) Scan(dest interface{}) *DB

另外，使用gorm.DB结构体中的DB()方法，可以返回一个sql.DB对象，如下：

func (s *DB) DB() *sql.DB

下面演示的是使用gorm.DB结构体中一些更简便的方法进行数据库基本操作，不过，在演示之前，我们需要先定义一个模型，如下:

type User struct {
    Id       int   //对应数据表的自增id
    Username string
    Password string
    Email    string
    Phone    string
}

我们定义了一个名称为User的结构体，GROM支持将结构体按规则映射为某个数据表的一行，结构体的每个字段表示数据表的列，结构体的字段首字母必须是大写的。

创建

使用gorm.DB中的Create()方法，GORM会根据传给Create()方法的模型，向数据表插入一行。

func (s *DB) Create(value interface{}) *DB  //创建一行
func (s *DB) NewRecord(value interface{}) bool //根据自增id判断主键是否存在

示例

func main() {
    defer db.Close()
    //具体的逻辑
    u := &User{Username: "test_one", Password: "testOne123456", Email: "test_one@163.com", Phone: "13711112222"}
    db.Create(u)
    if db.NewRecord(u){
        fmt.Println("写入失败")
    }else{
    	fmt.Println("写入成功")
    }
}

查询

GROM框架在sql/database包的原生基础上封装了简便的方法，可以直接调用便将数据映射到对应的结构体模型中，用起来非常简单，如下面这几个方法：

//返回第一条
func (s *DB) First(out interface{}, where ...interface{}) *DB
//返回最后一条
func (s *DB) Last(out interface{}, where ...interface{}) *DB
//返回符合条件的内容
func (s *DB) Find(out interface{}, where ...interface{}) *DB
//返回Count(*)结果
func (s *DB) Count(value interface{}) *DB

示例代码

//Find方法示例
func find() {
    var users = make([]*User, 0)
    db.Model(&User2{}).Find(&users)
    fmt.Println(users)
}

//First方法示例
func first()  {
    var user1,user2 User
    db.First(&user1)
    fmt.Println(user1)
    db.First(&user2,"id = ?",20)
    fmt.Println(user2)
}

//Last方法示例
func last()  {
    var user1,user2 User
    db.Last(&user1)
    fmt.Println(user1)
    db.First(&user2,"id = ?",19)
    fmt.Println(user2)
}

//Count方法示例
func count()  {
    var count int
    db.Model(&User{}).Count(&count)
    fmt.Println(count)
}

更新

更新数据可以使用gorm.DB的Save()或Update(),UpdateColumn(),UpdateColumns(),Updates()等方法，后面这四个方法需要与Model()方法一起使用。

func (s *DB) Save(value interface{}) *DB

func (s *DB) Model(value interface{}) *DB
//下面的方法需要与Model方法一起使用，通过Model方法指定更新数据的条件
func (s *DB) Update(attrs ...interface{}) *DB
func (s *DB) UpdateColumn(attrs ...interface{}) *DB
func (s *DB) UpdateColumns(values interface{}) *DB
func (s *DB) Updates(values interface{}, ignoreProtectedAttrs ...bool) *DB

代码示例

//Save()方法示例
func save(){
    u := &User{}
    db.First(u)
    u.Email = "test@163.com"
    db.Save(u)
    fmt.Println(u)
}

//Update方法示例
func update() {
    u := &User{}
    db.First(u)
    db.Model(u).Update("username", "hello")
}

//Updates方法示例
func updates() {
    u := &User{}
    db.First(u)
    db.Model(&u).Updates(map[string]interface{}{"username": "hello2"})
}

删除

使用gorm.DB的Delete()方法可以很简单地删除满足条件的记录，下面是Delete()方法的定义：

//value如果有主键id，则包含在判断条件内，通过where可以指定其他条件
func (s *DB) Delete(value interface{}, where ...interface{}) *DB

示例代码

func delete(){
    defer db.Close()
    u := &User{Id: 16}
    db.Delete(u)//根据id
    db.Delete(&User{},"username = ? ","test_one")//根据额外条件删除
}

小结

在这篇文章中我们只是讲解使用GROM框架如何连接和简单操作数据库而已，其实GROM框架还有许多更加高级功能，可以让我们的开发变得更加简洁，在之后的文章中，我们再进行详细讲解吧。

---

你的关注，是我写作路上最大的鼓励！