原生 SQL

原生 SQL 查询

  1. type Result struct {
  2. ID int
  3. Name string
  4. Age int
  5. }
  6. var result Result
  7. db.Raw("SELECT id, name, age FROM users WHERE name = ?", 3).Scan(&result)
  8. db.Raw("SELECT id, name, age FROM users WHERE name = ?", 3).Scan(&result)
  9. var age int
  10. DB.Raw("select sum(age) from users where role = ?", "admin").Scan(&age)

执行原生 SQL

  1. db.Exec("DROP TABLE users")
  2. db.Exec("UPDATE orders SET shipped_at=? WHERE id IN ?", time.Now(), []int64{1,2,3})
  3. // SQL 表达式
  4. DB.Exec("update users set money=? where name = ?", gorm.Expr("money * ? + ?", 10000, 1), "jinzhu")

注意 GORM 允许缓存准备好的语句来提高性能,详情请参考 性能

Row & Rows

获取 *sql.Row 结果

  1. // 使用 GORM API 构建 SQL
  2. row := db.Table("users").Where("name = ?", "jinzhu").Select("name", "age").Row()
  3. row.Scan(&name, &age)
  4. // 使用原生 SQL
  5. row := db.Raw("select name, age, email from users where name = ?", "jinzhu").Row()
  6. row.Scan(&name, &age, &email)

获取 *sql.Rows 结果

  1. // 使用 GORM API 构建 SQL
  2. rows, err := db.Model(&User{}).Where("name = ?", "jinzhu").Select("name, age, email").Rows()
  3. defer rows.Close()
  4. for rows.Next() {
  5. rows.Scan(&name, &age, &email)
  6. // 业务逻辑...
  7. }
  8. // 原生 SQL
  9. rows, err := db.Raw("select name, age, email from users where name = ?", "jinzhu").Rows()
  10. defer rows.Close()
  11. for rows.Next() {
  12. rows.Scan(&name, &age, &email)
  13. // 业务逻辑...
  14. }

转到 FindInBatches 获取如何在批量中查询和处理记录的信息, 转到 Group 条件 获取如何构建复杂 SQL 查询的信息

命名参数

GORM 支持 sql.NamedArgmap[string]interface{}{} 命名参数,例如:

  1. DB.Where("name1 = @name OR name2 = @name", sql.Named("name", "jinzhu")).Find(&user)
  2. // SELECT * FROM `users` WHERE name1 = "jinzhu" OR name2 = "jinzhu"
  3. DB.Where("name1 = @name OR name2 = @name", map[string]interface{}{"name": "jinzhu2"}).First(&result3)
  4. // SELECT * FROM `users` WHERE name1 = "jinzhu2" OR name2 = "jinzhu2" ORDER BY `users`.`id` LIMIT 1
  5. DB.Raw("SELECT * FROM users WHERE name1 = @name OR name2 = @name2 OR name3 = @name", sql.Named("name", "jinzhu1"), sql.Named("name2", "jinzhu2")).Find(&user)
  6. // SELECT * FROM users WHERE name1 = "jinzhu1" OR name2 = "jinzhu2" OR name3 = "jinzhu1"
  7. DB.Exec("UPDATE users SET name1 = @name, name2 = @name2, name3 = @name", sql.Named("name", "jinzhunew"), sql.Named("name2", "jinzhunew2"))
  8. // UPDATE users SET name1 = "jinzhunew", name2 = "jinzhunew2", name3 = "jinzhunew"
  9. DB.Raw("SELECT * FROM users WHERE (name1 = @name AND name3 = @name) AND name2 = @name2", map[string]interface{}{"name": "jinzhu", "name2": "jinzhu2"}).Find(&user)
  10. // SELECT * FROM users WHERE (name1 = "jinzhu" AND name3 = "jinzhu") AND name2 = "jinzhu2"

sql.Rows 扫描至 struct

  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. // ScanRows 将一行扫描至 user
  6. db.ScanRows(rows, &user)
  7. // 业务逻辑...
  8. }

DryRun 模式

在不执行的情况下生成 SQL ,可以用于准备或测试生成的 SQL,详情请参考 Session

  1. stmt := DB.Session(&Session{DryRun: true}).First(&user, 1).Statement
  2. stmt.SQL.String() //=> SELECT * FROM `users` WHERE `id` = $1 ORDER BY `id`
  3. stmt.Vars //=> []interface{}{1}

高级

Clauses

GORM 内部使用 SQL builder 生成 SQL。对于每个操作,GORM 都会创建一个 *gorm.Statement 对象,所有的 GORM API 都是在为 statement 添加/修改 Clause,最后,GORM 会根据这些 Clause 生成 SQL

例如,当通过 First 进行查询时,它会在 Statement 中添加以下 Clause

  1. clause.Select{Columns: "*"}
  2. clause.From{Tables: clause.CurrentTable}
  3. clause.Limit{Limit: 1}
  4. clause.OrderByColumn{
  5. Column: clause.Column{Table: clause.CurrentTable, Name: clause.PrimaryKey},
  6. }

然后 GORM 在回调中构建最终的查询 SQL,像这样:

  1. Statement.Build("SELECT", "FROM", "WHERE", "GROUP BY", "ORDER BY", "LIMIT", "FOR")

生成 SQL:

  1. SELECT * FROM `users` ORDER BY `users`.`id` LIMIT 1

您可以自定义 Clause 并与 GORM 一起使用,这需要实现 Interface 接口

可以参考 示例

Clause 构建器

不同的数据库, Clause 可能会生成不同的 SQL,例如:

  1. db.Offset(10).Limit(5).Find(&users)
  2. // SQL Server 会生成
  3. // SELECT * FROM "users" OFFSET 10 ROW FETCH NEXT 5 ROWS ONLY
  4. // MySQL 会生成
  5. // SELECT * FROM `users` LIMIT 5 OFFSET 10

之所以支持 Clause,是因为 GORM 允许数据库驱动程序通过注册 Clause Builder 来取代默认值,这儿有一个 Limit 的示例

Clause 选项

GORM 定义了很多 Clause,其中一些 Clause 提供了你可能会用到的选项

尽管很少会用到它们,但如果你发现 GORM API 与你的预期不符合。这可能可以很好地检查它们,例如:

  1. DB.Clauses(clause.Insert{Modifier: "IGNORE"}).Create(&user)
  2. // INSERT IGNORE INTO users (name,age...) VALUES ("jinzhu",18...);

StatementModifier

GORM 提供了 StatementModifier 接口,允许您修改语句,使其符合您的要求,这儿有一个 Hint 示例

  1. import "gorm.io/hints"
  2. DB.Clauses(hints.New("hint")).Find(&User{})
  3. // SELECT * /*+ hint */ FROM `users`