目录

go gorm 判断 db 对象是否存在 where 条件

目录

背景

最近有个极端的需求场景,在使用 gorm 操作数据库时,想着能不能判断一下 gorm db 对象是否已经存在 where 条件(ps:纯好奇,最后也没啥用,记录下)

在阅读源码后,发现可以判断几个对象从而确定是否存在 where 条件

  • len(db.Statement.Clauses)

  • len(db.Clauses)

测试代码如下:

package main
import (
  "gorm.io/driver/mysql"
  "gorm.io/gorm"
)
func main(){
  // 创建gorm句柄
  dsn := "user:pass@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
  db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
  // 没有任何where之前
  fmt.Println("start len:", len(db.Statement.Clauses))
  fmt.Println("start:", db.Statement.Clauses)
  // 添加where
  db = db.Where("1=?", 1)
  db = db.Where("id=?", 1)
  fmt.Println("where len:", len(db.Statement.Clauses))
  fmt.Println("where:", db.Statement.Clauses)
}

start len: 0 start: map[] where len: 1 where: map[WHERE:{WHERE {[{1=? [1] false} {id=? [1] false}]} }] — PASS: TestProcessorCheck/db_clone (480.75s)

从执行结果可以明显看出有一个where,因此可通过 len(db.Statement.Clauses) 来判断where条件是否存在