Go语言ORM包中使用worm构造查询条件的实例详解

构造查询条件

worm是一款方便易用的Go语言ORM库。worm支Model方式（持结构体字段映射）、原生SQL以及SQLBuilder三种模式来操作数据库，并且Model方式、原生SQL以及SQLBuilder可混合使用。

Model方式、SQL builder支持链式API，可使用Where, And, Or, ID, In, Limit, GroupBy, OrderBy, Having等函数构造查询条件。也可以可通过Join、LeftJoin、RightJoin来进行数据库表之间的关联查询。

本文通过一些例子来说明如何使用worm来构造查询条件。

main函数

package main
import (
   "database/sql"
   _ "github.com/go-sql-driver/mysql"
   log "github.com/haming123/wego/dlog"
   "github.com/haming123/wego/worm"
)
func mysql_open(cnnstr string) (*sql.DB, error) {
   db, err := sql.Open("mysql", cnnstr)
   if err != nil {
       return nil, err
   }
   err = db.Ping()
   if err != nil {
       return nil, err
   }
   return db, nil
}
func main() {
   //创建数据连接池
   cnnstr := "user:passwd@tcp(127.0.0.1:3306)/dbname?charset=utf8&parseTime=True"
   db_cnn, err := mysql_open(cnnstr)
   if err != nil {
       log.Error(err)
       return
   }
   //初始化ORM
   worm.InitMysql(db_cnn)
   //显示SQL语句log
   worm.ShowSqlLog(true)
}

说明：

• worm代码的下载
  go get github.com/haming123/wego

• worm.ShowSqlLog
  worm.ShowSqlLog用于控制sql日志的显示，建议测试环境下打开sql日志的显示的开关，这样可以看到每个数据库操作的sql语句以及执行时间，方便快速定位问题。

• 数据库的支持
  目前worm支持的数据库有：mysql、postgres、sqlite、sqlserver， 本文的例子中采用了mysql数据库。

数据库表与数据模型

//建表语句
CREATE TABLE `user` (
 `id` bigint(20) NOT NULL AUTO_INCREMENT,
 `name` varchar(30) DEFAULT NULL,
 `age` int(11) DEFAULT NULL,
 `passwd` varchar(32) DEFAULT NULL,
 `created` datetime DEFAULT NULL,
 PRIMARY KEY (`id`)
);

数据库表user对应的实体类的定义如下：

type User struct {
   Id          int64       `db:"id;autoincr"`
   Name        string      `db:"name"`
   Age         int64       `db:"age"`
   Passwd      string      `db:"passwd"`
   Created     time.Time    `db:"created;n_update"`
}
func (ent *User) TableName() string {
   return "user"
}

说明：

• worm使用名称为"db"的Tag映射数据库字段，"db"后面是字段的名称，autoincr用于说明该字段是自增ID，n_update用于说明该字段不可用于update语句中。

通过ID来查询数据

若数据库表存在 id 字段，则可以通过ID函数来查询一条数据据记录：

func DemoGetById() {
   var user model.User
   _, err := worm.Model(&user).ID(1).Get()
   if err != nil {
       log.Error(err)
       return
   }
   log.Debug(user)
}
//select id,name,age,passwd,created from user where id=? limit 1

执行该函数后的sql日志为：

[S] select id,name,age,passwd,created from user where id=1 limit 1
[S] DB: time=18.816ms

通过Where函数来查询数据

Where函数的使用类似Sprintf函数，函数的第一个参数是sql语句(where语句)模板，后面的参数是模板变量的值。

func DemoWhere() {
   var users []model.User
   err := worm.Model(&model.User{}).Where("id>? and age>?", 1, 10).Find(&users)
   if err != nil {
       log.Error(err)
       return
   }
}
//对应的sql语句为：
//select id,name,age,passwd,created from user where id>? and age>?

说明：

• worm占位符统一使用?，worm会根据数据库类型,自动替换占位符，例如postgresql数据库把?替换成$1,$2...

• 可以在Where函数使用多个变量进行查询，这种方式比较直观，与数据库查询中的sql语句的写法比较类似。但是当查询条件比较多时，建议使用And、OR函数进行适当的分割，防止将查询变量与变量的值对应错误。例如：

func DemoWhere2() {
   var users []model.User
   err := worm.Model(&model.User{}).Where("id>?", 1).And("age>?", 10).Find(&users)
   if err != nil {
       log.Error(err)
       return
   }
}
//对应的sql语句为：
//select id,name,age,passwd,created from user where id>? and age>?

• like查询的写法

例如查询用户的姓名中包含： demo 的数据库记录：

func DemoWhereLike() {
 var users []model.User
 err := worm.Model(&model.User{}).Where("name like ?", "％demo％").Find(&users)
 if err != nil {
     log.Error(err)
     return
 }
}
//对应的sql语句为：
//select id,name,age,passwd,created from user where name like '％demo％'

XXXIf查询

有些情况加我们会根据变量的值来判断使用使用一个变量来作为查询条件来查询书库，例如，若用户的姓名不为空时通过用户姓名来查询数据库。常规的写法如下：

func DemoWhereIf(name string) {
   var users []model.User
   var err error
   if name == "" {
       err = worm.Model(&model.User{}).Find(&users)
   } else {
       err = worm.Model(&model.User{}).Where("name=?", name).Find(&users)
   }
   if err != nil {
       log.Error(err)
       return
   }
}

worm提供了更为简单的方法（提供了WhereIf、AndIf、OrIf函数）来支持这种查询需求：

func DemoWhereIf(name string) {
   var users []model.User
   err := worm.Model(&model.User{}).WhereIf(name != "", "name=?", name).Find(&users)
   if err != nil {
       log.Error(err)
       return
   }
}

说明：

• WhereIf函数的第一个参数时一个bool变量，若该变量为true，则会添加查询条件，否则忽略该查询条件。

in、not in查询

worm提供了AndIn、AndNotIn、OrIn、OrNotIn函数来支持sql语句中的in、not in查询。例如：

func DemoWhereIn() {
   var users []model.User
   err := worm.Model(&model.User{}).Where("").AndIn("id", 11, 12, 13, 14).Find(&users)
   if err != nil {
       log.Error(err)
       return
   }
}
//对应的sql语句为：
select id,name,age,passwd,created from user where id in (?,?,?,?)

XXXIn、XXXNotIn的第二个参数时一个变长参数，您可以将需要查询的值作为变长参数传入，也可以将查询的值放到一个数组中进行查询：

func DemoWhereIn() {
   var users []model.User
   arr_id := []int64{11, 12, 13, 14}
   err := worm.Model(&model.User{}).Where("").AndIn("id", arr_id).Find(&users)
   if err != nil {
       log.Error(err)
       return
   }
}

说明：

• 若使用数组方式，则可变长参数智能时一个参数，并且该参数为数组类型。

嵌套查询语句

worm支持嵌套查询语句，例如查询为： age>10 and (name='demo1' or name='demo2') ， 则使用worm的方式如下：

func DemoWhereExp() {
   var users []model.User
   sqlw := worm.SQLW("name=?", "demo1").Or("name=?", "demo2")
   err := worm.Model(&model.User{}).Where("age>?", 10).AndExp(sqlw).Find(&users)
   if err != nil {
       log.Error(err)
       return
   }
}
//对应的sql语句为：
//select id,name,age,passwd,created from user where age>? and (name=? or name=?)

Limit与Offset

在MySQL语句中可以使用Limit与Offset来查询数据库，这种查询通常用于WEB的分页查询中。worm也支持mysql的Limit与Offset语句：

func DemoQueryPage(plen int64, pcur int64) {
   var users []model.User
   err := worm.Model(&model.User{}).Where("age>?", 10).Limit(plen).Offset(plen * pcur).Find(&users)
   if err != nil {
       log.Error(err)
       return
   }
}
//对应的sql语句为：
//select id,name,age,passwd,created from user where age>? limit ?, ?

orderby查询

OrderBy函数对应sql语句中的order by语句：

func DemoQueryOrderBy(orderby string) {
   var users []model.User
   err := worm.Model(&model.User{}).Where("age>?", 10).OrderBy(orderby).Find(&users)
   if err != nil {
       log.Error(err)
       return
   }
}
//对应的sql语句为：
//select id,name,age,passwd,created from user where age>? order by created desc

来源：https://segmentfault.com/a/1190000042183799