最近在想给这个小站增长点赞和评论功能，第三方系统又有各类限制，就想本身弄个后端，实现相似的功能，对于我的来讲，数据量不是很大，单机的 mysql 足够存下全部数据，mysql 做为底层存储是个不错的选择php

github.com/go-sql-driver/mysqlgithub.com/jinzhu/gorm

ORM（Object Relation Mapping），对象关系映射，实际上就是对数据库的操做进行封装，对上层开发人员屏蔽数据操做的细节，开发人员看到的就是一个个对象，大大简化了开发工做，提升了生产效率mysql

好了，下面我以这个点赞评论系统为例，介绍一下 gorm 的简单用法，如下使用的完整代码：https://github.com/hatlonely/microservices/blob/master/internal/comment_like/comment_like.golaravel

gorm 用法介绍

库安装

go get -u github.com/jinzhu/gorm

数据库链接

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

var db *gorm.DB

func init() {
    var err error
    db, err = gorm.Open("mysql", "<user>:<password>/<database>?charset=utf8&parseTime=True&loc=Local")
    if err != nil {
        panic(err)
    }
}
gorm.Open
github.com/jinzhu/gorm/dialects/mysqlgithub.com/go-sql-driver/mysql

这里我用的 mysql，实际上支持基本上全部主流的关系数据库，链接方式上略有不一样golang

db.DB().SetMaxIdleConns(10)
db.DB().SetMaxOpenConns(100)
db.DB()

表定义

先来定义一个点赞表，这里面一条记录表示某个用户在某个时刻对某篇文章点了一个赞，用 ip + ua 来标识用户，title 标识文章标题数据库

type Like struct {
    ID        int    `gorm:"primary_key"`
    Ip        string `gorm:"type:varchar(20);not null;index:ip_idx"`
    Ua        string `gorm:"type:varchar(256);not null;"`
    Title     string `gorm:"type:varchar(128);not null;index:title_idx"`
    Hash      uint64 `gorm:"unique_index:hash_idx;"`
    CreatedAt time.Time
}

gorm 用 tag 的方式来标识 mysql 里面的约束后端

ip_idxtitle_idxunique_index
time.Time

建立表

if !db.HasTable(&Like{}) {
    if err := db.Set("gorm:table_options", "ENGINE=InnoDB DEFAULT CHARSET=utf8").CreateTable(&Like{}).Error; err != nil {
        panic(err)
    }
}
db.CreateTabledb.Set

插入

like := &Like{
    Ip:        ip,
    Ua:        ua,
    Title:     title,
    Hash:      murmur3.Sum64([]byte(strings.Join([]string{ip, ua, title}, "-"))) >> 1,
    CreatedAt: time.Now(),
}

if err := db.Create(like).Error; err != nil {
    return err
}
db.Create()

删除

if err := db.Where(&Like{Hash: hash}).Delete(Like{}).Error; err != nil {
    return err
}
db.Where()db.Delete()

查询

var count int
err := db.Model(&Like{}).Where(&Like{Ip: ip, Ua: ua, Title: title}).Count(&count).Error
if err != nil {
    return false, err
}
db.Model()db.Where()db.Count()db.Find(&Likes)db.First(&Like)

修改

db.Model(&user).Update("name", "hello")
db.Model(&user).Updates(User{Name: "hello", Age: 18})
db.Model(&user).Updates(User{Name: "", Age: 0, Actived: false}) // nothing update

我这个系统里面没有更新需求，这几个例子来自于官网，第一个是更新单条记录；第二个是更新整条记录，注意只有非空字段才会更新；第三个例子是不会更新的，在系统设计的时候要尽可能避免这些空值有特殊的含义，若是必定要更新，可使用第一种方式，设置单个值

错误处理

dbdb.Error

事务

func CreateAnimals(db *gorm.DB) err {
    tx := db.Begin()
    if err := tx.Create(&Animal{Name: "Giraffe"}).Error; err != nil {
        tx.Rollback()
        return err
    }
    if err := tx.Create(&Animal{Name: "Lion"}).Error; err != nil {
        tx.Rollback()
        return err
    }
    tx.Commit()
    return nil
}
db.Begin()tx.Commit()tx.Rollback()

其余

还可使用以下方式设置日志输出级别以及改变日志输出地方

db.LogMode(true)
db.SetLogger(gorm.Logger{revel.TRACE})
db.SetLogger(log.New(os.Stdout, "\r\n", 0))

也支持普通的 sql，可是建议尽可能不要使用

参考连接