Gorm 高级查询

原创

王小明_HIT

发布于 2022-02-13 14:43:06

2.3K0

发布于 2022-02-13 14:43:06

文章被收录于专栏：程序员奇点程序员奇点

前言

越高级，越复杂的查询，也同时意味着高耗，但是平时有一些数据少，但是业务复杂的场景，可以使用下。这里主要说明的是 go 中使用 gorm 进务查询。

gorm import 依赖

 "database/sql"
 "fmt"
 "github.com/jinzhu/gorm"
 _ "github.com/jinzhu/gorm/dialects/mysql"

将查询结果放到一个 struct 中

// 根据主键查询第一条记录
db.First(&user)
//// SELECT * FROM users ORDER BY id LIMIT 1;

// 随机获取一条记录
db.Take(&user)
//// SELECT * FROM users LIMIT 1;

// 根据主键查询最后一条记录
db.Last(&user)
//// SELECT * FROM users ORDER BY id DESC LIMIT 1;

// 查询所有的记录
db.Find(&users)
//// SELECT * FROM users;

// 查询指定的某条记录(仅当主键为整型时可用)
db.First(&user, 10)
//// SELECT * FROM users WHERE id = 10;

查询条件是map 或者 struct 查询

有时候代码是可需要映射到一个 map 结构，不需要映射到一个结构体中，可以写成如下：

for update

在涉及并发的场景，往往需要加锁互斥，和 Java 类似， Go 中也有加行锁的方式，加 for update 即可。

一般写法如下：

// 为查询 SQL 添加额外的 SQL 操作
db.Set("gorm:query_option", "FOR UPDATE").First(&user, 10)
//// SELECT * FROM users WHERE id = 10 FOR UPDATE;

示例代码：

err := db.Model(&XXX{}).Set("gorm:query_option", "FOR UPDATE").Where("XXX=?", XXX).First(&XXX).Error
 if err != nil {
  if err == gorm.ErrRecordNotFound {
   logs.Warn("xxx")
   return nil, nil
  }
  logs.Error("XXX")
 }

Count 查询

有时候，我们需要进行简单的数据统计，比如查询到结果有多少行，

var count int64
db.Model(&User{}).Where("name = ?","jinzhu").Or("name = ?","jinzhu 2").Count(&count)
// SELECT count(*) FROM users WHERE name = 'jinzhu' OR name = 'jinzhu 2'
db.Model(&User{}).Where("name = ?","jinzhu").Count(&count)
// SELECT count(*) FROM users WHERE name = 'jinzhu'; (count)

分组计数

有时候也需要分组统计行数

// 分组计数
users :=[]User{
{Name:"name1"},
{Name:"name2"},
{Name:"name3"},
{Name:"name3"},
}
DB.Model(&User{}).Group("name").Count(&count)
count // => 3

去重统计

// 去重计数
DB.Model(&User{}).Distinct("name").Count(&count)
// SELECT COUNT(DISTINCT(`name`)) FROM `users`

Group & Having

有时候我们会使用到数据统计的功能，比如根据数据库字段 batch_no 进行分组，然后统计总金额，总笔数。

分组查询统计一般的写法如下：

db.Table("orders").Select("date(created_at) as date, sum(amount) as total").Group("date(created_at)").Having("sum(amount) > ?", 100).Scan(&results)

代码示例：

type Sum struct {
 BatchNo      string `gorm:"column:batch_no" json:"batch_no"`
 TotalCounts  int64  `gorm:"column:total_counts" json:"total_counts"`
 TotalAmounts int64  `gorm:"column:total_amounts" json:"total_amounts"`
}


var result []*Sum
 db := db.Model(&Voucher{}).Select("batch_no, count(1) as totalCounts, sum(amount) as totalAmounts")
 status := []string{10,20,40}

 db = db.Where("no >= ?", startVoucherNo)
 db = db.Where("no <= ?", endVoucherNo)
 db = db.Where("batch_no IN ?", batchNos)
 db = db.Where("status IN ?", status)

 if shardingKey >= 0 {
  db = db.Where("sharding_key = ", shardingKey)
 }

 db = db.Group("batch_no")
 err := db.Scan(&result).Error
 if err != nil {
  if err == gorm.ErrRecordNotFound {
   logs.CtxWarn("xxx")
   return nil, nil
  }
  logs.CtxError(ctx, "xxx", err)
 }

Join 查询

一般来说，很少使用关联查询，但是如果要使用关联查询，可以如下：

db.Table("users").Select("users.name, emails.email").Joins("left join emails on emails.user_id = users.id").Scan(&results)

// 多连接及参数
db.Joins("JOIN emails ON emails.user_id = users.id AND emails.email = ?", "jinzhu@example.org").Joins("JOIN credit_cards ON credit_cards.user_id = users.id").Where("credit_cards.number = ?", "411111111111").Find(&user)

查询指定函数

Scopes允许你指定常用的查询，可以在调用方法时引用这些查询，也就是说，可以在查询中使用函数。

举个例子：

func AmountGreaterThan1000(db *gorm.DB)*gorm.DB {
return db.Where("amount > ?",1000)
}

func PaidWithCreditCard(db *gorm.DB)*gorm.DB {
return db.Where("pay_mode_sign = ?","C")
}
func PaidWithCod(db *gorm.DB)*gorm.DB {
return db.Where("pay_mode_sign = ?","C")
}

db.Scopes(AmountGreaterThan1000,PaidWithCreditCard).Find(&orders)
// 查找所有金额大于 1000 的信用卡订单


db.Scopes(AmountGreaterThan1000,PaidWithCod).Find(&orders)
// 查找所有金额大于 1000 的货到付款订单