Go GORM是時候升級新版本了 2.0新特性介紹（2）

版本升級遇到的問題

• 老版本 Count() 回傳int，新版本 Count() 回傳int64
• 老版本 Update() 的用法相當于新版本 Save() 的用法

Update

• 老版本傳入結構體指標

func LogIp(admin *service.Admin) {
	DB.Table("admin").Where("id = ?", admin.ID).Update(admin)
}

• 新版本傳入key value 更新

db.Model(&User{}).Where("active = ?", true).Update("name", "hello")

鉤子函式

• 鉤子函式（Hook）是在創建、查詢、更新、洗掉等操作之前、之后呼叫的函式，
• 常用的鉤子函式有：

創建

// 開始事務
BeforeSave
BeforeCreate
// 關聯前的 save
// 插入記錄至 db
// 關聯后的 save
AfterCreate
AfterSave
// 提交或回滾事務

更新

// 開始事務
BeforeSave
BeforeUpdate
// 關聯前的 save
// 更新 db
// 關聯后的 save
AfterUpdate
AfterSave
// 提交或回滾事務

洗掉

// 開始事務
BeforeDelete
// 洗掉 db 中的資料
AfterDelete
// 提交或回滾事務

查詢

// 從 db 中加載資料
// Preloading (eager loading)
AfterFind

查詢

檢查 ErrRecordNotFound 錯誤

// 檢查 ErrRecordNotFound 錯誤
errors.Is(result.Error, gorm.ErrRecordNotFound)

檢索單個物件

var user User

// 可以
db.First(&user)
// SELECT * FROM `users` ORDER BY `users`.`id` LIMIT 1

// 可以
result := map[string]interface{}{}
db.Model(&User{}).First(&result)
// SELECT * FROM `users` ORDER BY `users`.`id` LIMIT 1

// 不行
result := map[string]interface{}{}
db.Table("users").First(&result)

// 但可以配合 Take 使用
result := map[string]interface{}{}
db.Table("users").Take(&result)

// 根據第一個欄位排序
type Language struct {
  Code string
  Name string
}
db.First(&Language{})
// SELECT * FROM `languages` ORDER BY `languages`.`code` LIMIT 1

查詢注意問題

• 結構體查詢會忽略零值條件，如果需要把零值作為篩選條件需要用map

db.Where(&User{Name: "xiaoming", Age: 0}).Find(&users)
// SELECT * FROM users WHERE name = "jinzhu";

db.Where(map[string]interface{}{"Name": "xiaoming", "Age": 0}).Find(&users)
// SELECT * FROM users WHERE name = "jinzhu" AND age = 0;

執行原生SQL

// 原生 SQL
db.Raw("SELECT name, age FROM users WHERE name = ?", "xxx").Scan(&result)

子查詢

db.Where("amount > (?)", db.Table("orders").Select("AVG(amount)")).Find(&orders)
// SELECT * FROM "orders" WHERE amount > (SELECT AVG(amount) FROM "orders");

subQuery := db.Select("AVG(age)").Where("name LIKE ?", "name%").Table("users")
db.Select("AVG(age) as avgage").Group("name").Having("AVG(age) > (?)", subQuery).Find(&results)
// SELECT AVG(age) as avgage FROM `users` GROUP BY `name` HAVING AVG(age) > (SELECT AVG(age) FROM `users` WHERE name LIKE "name%")

高級查詢

1. FirstOrInit ,獲取第一條匹配的記錄，或者根據給定的條件初始化一個實體（僅支持 sturct 和 map 條件,不會保存再資料庫

// 未找到 user，根據條件和 Assign 屬性創建記錄
db.Where(User{Name: "non_existing"}).Attrs(User{Age: 20}).FirstOrCreate(&user)
// SELECT * FROM users WHERE name = 'non_existing' ORDER BY id LIMIT 1;
// INSERT INTO "users" (name, age) VALUES ("non_existing", 20);
// user -> User{ID: 112, Name: "non_existing", Age: 20}

// 找到了 `name` = `jinzhu` 的 user，則忽略 Attrs
db.Where(User{Name: "jinzhu"}).Attrs(User{Age: 20}).FirstOrCreate(&user)
// SELECT * FROM users WHERE name = 'jinzhu' ORDER BY id LIMIT 1;
// user -> User{ID: 111, Name: "jinzhu", Age: 18}

2. FirstOrCreate ,獲取第一條匹配的記錄，或者根據給定的條件創建一條新紀錄（僅支持 sturct 和 map 條件）,不管是否找到記錄，Assign 都會將屬性賦值給 struct，并將結果寫回資料庫

// 未找到 user，根據條件和 Assign 屬性創建記錄
db.Where(User{Name: "non_existing"}).Assign(User{Age: 20}).FirstOrCreate(&user)
// SELECT * FROM users WHERE name = 'non_existing' ORDER BY id LIMIT 1;
// INSERT INTO "users" (name, age) VALUES ("non_existing", 20);
// user -> User{ID: 112, Name: "non_existing", Age: 20}

// 找到了 `name` = `jinzhu` 的 user，依然會根據 Assign 更新記錄
db.Where(User{Name: "jinzhu"}).Assign(User{Age: 20}).FirstOrCreate(&user)
// SELECT * FROM users WHERE name = 'jinzhu' ORDER BY id LIMIT 1;
// UPDATE users SET age=20 WHERE id = 111;
// user -> User{ID: 111, Name: "jinzhu", Age: 20}

3. 迭代：支持通過行進行迭代

rows, err := db.Model(&User{}).Where("name = ?", "xxx").Rows()
defer rows.Close()

for rows.Next() {
  var user User
  // ScanRows 方法用于將一行記錄掃描至結構體
  db.ScanRows(rows, &user)

  // 業務邏輯...
}

4. 查詢鉤子:在查詢完成之后呼叫

func (u *User) AfterFind(tx *gorm.DB) (err error) {
  if u.Role == "" {
    u.Role = "user"
  }
  return
}

更新

使用sql運算式更新

• gorm.Expr()

// product 的 ID 是 `3`
db.Model(&product).Update("price", gorm.Expr("price * ? + ?", 2, 100))
// UPDATE "products" SET "price" = price * 2 + 100, "updated_at" = '2013-11-17 21:34:10' WHERE "id" = 3;

子查詢更新

• Update(“欄位”,子查詢陳述句)

db.Model(&user).Update("company_name", db.Model(&Company{}).Select("name").Where("companies.id = users.company_id"))
// UPDATE "users" SET "company_name" = (SELECT name FROM companies WHERE companies.id = users.company_id);

db.Table("users as u").Where("name = ?", "jinzhu").Update("company_name", db.Table("companies as c").Select("name").Where("c.id = u.company_id"))

db.Table("users as u").Where("name = ?", "jinzhu").Updates(map[string]interface{}{}{"company_name": db.Table("companies as c").Select("name").Where("c.id = u.company_id")})

通用的回傳結果

• result.RowsAffected // 更新的記錄數
• result.Error // 更新的錯誤

// 通過 `RowsAffected` 得到更新的記錄數
result := db.Model(User{}).Where("role = ?", "admin").Updates(User{Name: "hello", Age: 18})
// UPDATE users SET name='hello', age=18 WHERE role = 'admin;

result.RowsAffected // 更新的記錄數
result.Error        // 更新的錯誤

上一篇

Go gorm是時候升級新版本了 2.0新特性介紹（1） ｜Go主題月

下一篇

添加和洗掉的知識點下一篇整理