目錄
1. 高級用法
1.1. 錯誤處理
執行任何操作后，如果發生任何錯誤，GORM將其設置為*DB的Error字段
if err := db.Where("name = ?", "jinzhu").First(&user).Error; err != nil {
// 錯誤處理...
}
// 如果有多個錯誤發生，用`GetErrors`獲取所有的錯誤，它返回`[]error`
db.First(&user).Limit(10).Find(&users).GetErrors()
// 檢查是否返回RecordNotFound錯誤
db.Where("name = ?", "hello world").First(&user).RecordNotFound()
if db.Model(&user).Related(&credit_card).RecordNotFound() {
// 沒有信用卡被發現處理...
}
1.2. 事物
要在事務中執行一組操作，一般流程如下。
// 開始事務
tx := db.Begin()
// 在事務中做一些數據庫操作（從這一點使用'tx'，而不是'db'）
tx.Create(...)
// ...
// 發生錯誤時回滾事務
tx.Rollback()
// 或提交事務
tx.Commit()
1.2.1. 一個具體的例子
func CreateAnimals(db *gorm.DB) err {
tx := db.Begin()
// 注意，一旦你在一個事務中，使用tx作為數據庫句柄
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
}
1.3. SQL構建
1.3.1. 執行原生SQL
db.Exec("DROP TABLE users;")
db.Exec("UPDATE orders SET shipped_at=? WHERE id IN (?)", time.Now, []int64{11,22,33})
// Scan
type Result struct {
Name string
Age int
}
var result Result
db.Raw("SELECT name, age FROM users WHERE name = ?", 3).Scan(&result)
1.3.2. sql.Row & sql.Rows
獲取查詢結果為*sql.Row或*sql.Rows
row := db.Table("users").Where("name = ?", "jinzhu").Select("name, age").Row() // (*sql.Row)
row.Scan(&name, &age)
rows, err := db.Model(&User{}).Where("name = ?", "jinzhu").Select("name, age, email").Rows() // (*sql.Rows, error)
defer rows.Close()
for rows.Next() {
...
rows.Scan(&name, &age, &email)
...
}
// Raw SQL
rows, err := db.Raw("select name, age, email from users where name = ?", "jinzhu").Rows() // (*sql.Rows, error)
defer rows.Close()
for rows.Next() {
...
rows.Scan(&name, &age, &email)
...
}
1.3.3. 迭代中使用sql.Rows的Scan
rows, err := db.Model(&User{}).Where("name = ?", "jinzhu").Select("name, age, email").Rows() // (*sql.Rows, error)
defer rows.Close()
for rows.Next() {
var user User
db.ScanRows(rows, &user)
// do something
}
1.4. 通用數據庫接口sql.DB
從*gorm.DB連接獲取通用數據庫接口*sql.DB
// 獲取通用數據庫對象`*sql.DB`以使用其函數
db.DB()
// Ping
db.DB().Ping()
1.4.1. 連接池
db.DB().SetMaxIdleConns(10)
db.DB().SetMaxOpenConns(100)
1.5. 復合主鍵
將多個字段設置為主鍵以啟用復合主鍵
type Product struct {
ID string `gorm:"primary_key"`
LanguageCode string `gorm:"primary_key"`
}
1.6. 日志
Gorm有內置的日志記錄器支持，默認情況下，它會打印發生的錯誤
// 啟用Logger，顯示詳細日志
db.LogMode(true)
// 禁用日志記錄器，不顯示任何日志
db.LogMode(false)
// 調試單個操作，顯示此操作的詳細日志
db.Debug().Where("name = ?", "jinzhu").First(&User{})
1.6.1. 自定義日志
參考GORM的默認記錄器如何自定義它https://github.com/jinzhu/gorm/blob/master/logger.go
db.SetLogger(gorm.Logger{revel.TRACE})
db.SetLogger(log.New(os.Stdout, "\r\n", 0))