错误处理

执行任何操作后，如果发生任何错误，Gorm将其设置为*DB的Error字段

  1. if err := db.Where("name = ?", "jinzhu").First(&user).Error; err != nil {
  2.     // 错误处理...
  3. }
  5. // 如果有多个错误发生，用`GetErrors`获取所有的错误，它返回`[]error`
  6. db.First(&user).Limit(10).Find(&users).GetErrors()
  8. // 检查是否返回RecordNotFound错误
  9. db.Where("name = ?", "hello world").First(&user).RecordNotFound()
  11. if db.Model(&user).Related(&credit_card).RecordNotFound() {
  12.     // 没有信用卡被发现处理...
  13. }

事务

要在事务中执行一组操作，一般流程如下:

  1. // 开始事务
  2. tx := db.Begin()
  4. // 在事务中做一些数据库操作（从这一点使用'tx'，而不是'db'）
  5. tx.Create(...)
  7. // ...
  9. // 发生错误时回滚事务
  10. tx.Rollback()
  12. // 或提交事务
  13. tx.Commit()

举例

  1. func CreateAnimals(db *gorm.DB) err {
  2.   tx := db.Begin()
  3.   // 注意，一旦你在一个事务中，使用tx作为数据库句柄
  5.   if err := tx.Create(&Animal{Name: "Giraffe"}).Error; err != nil {
  6.      tx.Rollback()
  7.      return err
  8.   }
  10.   if err := tx.Create(&Animal{Name: "Lion"}).Error; err != nil {
  11.      tx.Rollback()
  12.      return err
  13.   }
  15.   tx.Commit()
  16.   return nil
  17. }

SQL构建

执行原生SQL

  1. db.Exec("DROP TABLE users;")
  2. db.Exec("UPDATE orders SET shipped_at=? WHERE id IN (?)", time.Now, []int64{11,22,33})
  4. // Scan
  5. type Result struct {
  6.     Name string
  7.     Age  int
  8. }
  10. var result Result
  11. db.Raw("SELECT name, age FROM users WHERE name = ?", 3).Scan(&result)

sql.Row & sql.Rows

获取查询结果为*sql.Row或*sql.Rows

  1. row := db.Table("users").Where("name = ?", "jinzhu").Select("name, age").Row() // (*sql.Row)
  2. row.Scan(&name, &age)
  4. rows, err := db.Model(&User{}).Where("name = ?", "jinzhu").Select("name, age, email").Rows() // (*sql.Rows, error)
  5. defer rows.Close()
  6. for rows.Next() {
  7.     ...
  8.     rows.Scan(&name, &age, &email)
  9.     ...
  10. }
  12. // Raw SQL
  13. rows, err := db.Raw("select name, age, email from users where name = ?", "jinzhu").Rows() // (*sql.Rows, error)
  14. defer rows.Close()
  15. for rows.Next() {
  16.     ...
  17.     rows.Scan(&name, &age, &email)
  18.     ...
  19. }

迭代中使用sql.Rows的Scan

  1. rows, err := db.Model(&User{}).Where("name = ?", "jinzhu").Select("name, age, email").Rows() // (*sql.Rows, error)
  2. defer rows.Close()
  4. for rows.Next() {
  5.   var user User
  6.   db.ScanRows(rows, &user)
  7.   // do something
  8. }

通用数据库接口DB

从*gorm.DB连接获取通用数据库接口*sql.DB

  1. // 获取通用数据库对象`*sql.DB`以使用其函数
  2. db.DB()
  4. // Ping
  5. db.DB().Ping()

连接池

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

复合主键

将多个字段设置为主键以启动符合主键

  1. type Product struct {
  2.     ID           string `gorm:"primary_key"`
  3.     LanguageCode string `gorm:"primary_key"`
  4. }

日志

  1. // 启用Logger，显示详细日志
  2. db.LogMode(true)
  4. // 禁用日志记录器，不显示任何日志
  5. db.LogMode(false)
  7. // 调试单个操作，显示此操作的详细日志
  8. db.Debug().Where("name = ?", "jinzhu").First(&User{})

自定义日志

参考GORM的默认记录器如何自定义它https://github.com/jinzhu/gorm/blob/master/logger.go

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