摘要：最近在想給這個小站增加點贊和評論功能，第三方系統(tǒng)又有各種限制，就想自己弄個后端，實現(xiàn)類似的功能，對于個人來說，數(shù)據(jù)量不是很大，單機的足夠存下所有數(shù)據(jù)，作為底層存儲是個不錯的選擇之前在公司是直接用的訪問數(shù)據(jù)庫都是直接用寫，取出結(jié)果然后自己

最近在想給這個小站增加點贊和評論功能，第三方系統(tǒng)又有各種限制，就想自己弄個后端，實現(xiàn)類似的功能，對于個人來說，數(shù)據(jù)量不是很大，單機的 mysql 足夠存下所有數(shù)據(jù)，mysql 作為底層存儲是個不錯的選擇

之前在公司是直接用的 github.com/go-sql-driver/mysql 訪問數(shù)據(jù)庫都是直接用寫 sql，取出結(jié)果然后自己拼成對象，使用上面不是很方便，可讀性也不好。想起之前研究 php laravel 框架的時候，直接把數(shù)據(jù)庫層屏蔽了，用戶看到的只有對象，使用非常方便，java 里面這種操作方式基本上已經(jīng)成了標(biāo)準(zhǔn)做法，就去 github 上找了一下 golang 里面有沒有類似的東西，果然已經(jīng)有非常成熟的框架了，github.com/jinzhu/gorm 已經(jīng)有 7k+ 的 star 了

ORM（Object Relation Mapping），對象關(guān)系映射，實際上就是對數(shù)據(jù)庫的操作進(jìn)行封裝，對上層開發(fā)人員屏蔽數(shù)據(jù)操作的細(xì)節(jié)，開發(fā)人員看到的就是一個個對象，大大簡化了開發(fā)工作，提高了生產(chǎn)效率

好了，下面我以這個點贊評論系統(tǒng)為例，介紹一下 gorm 的簡單用法，以下使用的完整代碼：https://github.com/hatlonely/...

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", ":/?charset=utf8&parseTime=True&loc=Local")
    if err != nil {
        panic(err)
    }
}

連接比較簡單，直接調(diào)用 gorm.Open 傳入數(shù)據(jù)庫地址即可

github.com/jinzhu/gorm/dialects/mysql 是 golang 的 mysql 驅(qū)動，實際上就是 github.com/go-sql-driver/mysql 作者這里為了好記，重新弄了個名字

這里我用的 mysql，實際上支持基本上所有主流的關(guān)系數(shù)據(jù)庫，連接方式上略有不同

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

還可以使用 db.DB() 對象設(shè)置連接池信息

先來定義一個點贊表，這里面一條記錄表示某個用戶在某個時刻對某篇文章點了一個贊，用 ip + ua 來標(biāo)識用戶，title 標(biāo)識文章標(biāo)題

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 的方式來標(biāo)識 mysql 里面的約束

創(chuàng)建索引只需要直接指定列即可，這里創(chuàng)建了兩個索引，ip_idx 和 title_idx；如果需要多列組合索引，直接讓索引的名字相同即可；如果需要創(chuàng)建唯一索引，指定為 unique_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.CreateTable 就可以創(chuàng)建表了，非常方便，還可以通過 db.Set 設(shè)置一些額外的表屬性

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
}

先構(gòu)造已給對象，直接調(diào)用 db.Create() 就可以插入一條記錄了

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

先用 db.Where() 構(gòu)造查詢條件，再調(diào)用 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() 構(gòu)造查詢條件，后面可以使用 db.Count() 計算數(shù)量，如果要獲取對象，可以使用 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

我這個系統(tǒng)里面沒有更新需求，這幾個例子來自于官網(wǎng)，第一個是更新單條記錄；第二個是更新整條記錄，注意只有非空字段才會更新；第三個例子是不會更新的，在系統(tǒng)設(shè)計的時候要盡量避免這些空值有特殊的含義，如果一定要更新，可以使用第一種方式，設(shè)置單個值

其實你已經(jīng)看到了，這里基本上所有的函數(shù)都是鏈?zhǔn)降?，全部都返?db 對象，任何時候調(diào)用 db.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
}

事務(wù)的處理也很簡單，用 db.Begin() 聲明開啟事務(wù)，結(jié)束的時候調(diào)用 tx.Commit()，異常的時候調(diào)用 tx.Rollback()

還可以使用如下方式設(shè)置日志輸出級別以及改變?nèi)罩据敵龅胤?/p>

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

也支持普通的 sql，但是建議盡量不要使用

gorm 官方文檔: http://gorm.io/

gorm github: https://github.com/jinzhu/gorm

轉(zhuǎn)載請注明出處 
本文鏈接：http://hatlonely.github.io/20...