高级用法

错误处理 {#eh}

执行任何操作后,如果发生任何错误,GORM将其设置为*DBError字段

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

事物 {#t}

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

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

一个具体的例子

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

SQL构建 {#sb}

执行原生SQL

  1. db.Exec("DROP TABLE users;")
  2. db.Exec("UPDATE orders SET shipped_at=? WHERE id IN (?)", time.Now, []int64{11,22,33})
  3. // Scan
  4. type Result struct {
  5. Name string
  6. Age int
  7. }
  8. var result Result
  9. 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)
  3. rows, err := db.Model(&User{}).Where("name = ?", "jinzhu").Select("name, age, email").Rows() // (*sql.Rows, error)
  4. defer rows.Close()
  5. for rows.Next() {
  6. ...
  7. rows.Scan(&name, &age, &email)
  8. ...
  9. }
  10. // Raw SQL
  11. rows, err := db.Raw("select name, age, email from users where name = ?", "jinzhu").Rows() // (*sql.Rows, error)
  12. defer rows.Close()
  13. for rows.Next() {
  14. ...
  15. rows.Scan(&name, &age, &email)
  16. ...
  17. }

迭代中使用sql.Rows的Scan

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

通用数据库接口sql.DB {#g}

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

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

连接池

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

复合主键 {#cpk}

将多个字段设置为主键以启用复合主键

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

日志 {#l}

Gorm有内置的日志记录器支持,默认情况下,它会打印发生的错误

  1. // 启用Logger,显示详细日志
  2. db.LogMode(true)
  3. // 禁用日志记录器,不显示任何日志
  4. db.LogMode(false)
  5. // 调试单个操作,显示此操作的详细日志
  6. 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))