Go语言ORM框架：快捷查询函数的使用

站长

2022年09月16日 10:54 · 阅读数 31

快捷查询函数

大多数情况下您都是在使用worm支提供的Model方式（持结构体字段映射）来访问数据库，但有些时候使用Model方式显然有些笨重。例如您可能只需要查询一个记录的一个字段的值，这种情况下若使用Model方式则比较繁琐。worm提供了一些快捷查询函数来应对这种需求，例如DbTable的GetString()函数，该函数克用于查询一个记录的一个字符串类型的字段。这些快捷查询函数可以用于原生SQL以及SQLBuilder模式，这些快捷函数的用法以上两种模式下的用法基本上是一致的，本文中使用SQLBuilder模式来说明这些快捷函数的使用方法。

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.ShowSqlLogworm.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`)
);

查询某个记录是否存在

有些情况下您不需要获取一个记录的数据，而是只需要查询数据库中是否存在某个记录，这种情况下可以使用Exist()函数：

func SQLBuilderExist() {
    has, err := worm.Table("user").ID(1).Exist()
    if err != nil {
        log.Error(err)
        return
    }
    //select * from user where id=? limit 1
}

查询一个记录的指定字段

有些情况下您可能只需要获取一条记录的某几个字段，这种情况在没必要定义一个struct，然后通过Model方式来获取字段的数据。您可以使用DbTable的Get函数：

func SQLBuilderGetValue() {
    name := ""; age := 0
    _, err := worm.Table("user").Select("name,age").ID(1).Get(&name, &age)
    if err != nil {
        log.Error(err)
        return
    }
    //select name,age from user where id=? limit 1
}

说明：

Get函数的参数是可变长参数，参数的数量与类型必须与Select()中的字段一一对应。
Get函数的第一个返回值是一个bool类型，用于表示是否存在该记录。

查询一个记录的一个字符串字段

GetString()函数用于查询一个记录的一个字符串字段

func SQLBuilderGetValue() {
    val, err := worm.Table("user").Select("name").ID(1).GetString()
    if err != nil {
        log.Error(err)
        return
    }
    //select name from user where id=? limit 1
}

说明：

Select()中必须只有一个字段，并且字段的类型为字符串。
GetString()的第一个返回值为sql.NullString类型，其中的Valid变量用于说明是否是一个有效的值。

查询一个记录的一个整形字段

GetInt()函数用于查询一个记录的一个整形字段

func SQLBuilderGetInt() {
    val, err := worm.Table("user").Select("age").ID(1).GetInt()
    if err != nil {
        log.Error(err)
        return
    }
    //select age from user where id=? limit 1
}

说明：

Select()中必须只有一个字段，并且字段的类型为整形。
GetInt()的第一个返回值为sql.NullInt64类型，其中的Valid变量用于说明是否是一个有效的值。

查询一个记录的一个浮点型字段

GetFloat()函数用于查询一个记录的一个浮点型字段

func SQLBuilderGetFloat() {
    val, err := worm.Table("user").Select("weight").ID(1).GetFloat()
    if err != nil {
        log.Error(err)
        return
    }
    //select weight from user where id=? limit 1
}

说明：

Select()中必须只有一个字段，并且字段的类型为浮点型。
GetFloat()的第一个返回值为sql.NullFloat64类型，其中的Valid变量用于说明是否是一个有效的值。

查询一个记录

GetFloat()函数用于查询一个记录的一个浮点型字段

func SQLBuilderGetFloat() {
    val, err := worm.Table("user").Select("weight").ID(1).GetFloat()
    if err != nil {
        log.Error(err)
        return
    }
    //select weight from user where id=? limit 1
}

说明：

Select()中必须只有一个字段，并且字段的类型为浮点型。
GetFloat()的第一个返回值为sql.NullFloat64类型，其中的Valid变量用于说明是否是一个有效的值。

查询满足条件的记录数

Count()函数用于查询查询满足条件的记录数

func SQLBuilderCount() {
    val, err := worm.Table("user").Where("id>?", 0).Count()
    if err != nil {
        log.Error(err)
        return
    }
    //select count(1) from user where id>?
}

说明：

Count()若没有参数，则使用count(1)的计数，若需要您也可以指定字段名称来计数：

func SQLBuilderCount() {
  val, err := worm.Table("user").Where("id>?", 0).Count("name")
  if err != nil {
      log.Error(err)
      return
  }
  //select count(name) from user where id>?
}

count_distinct查询

DistinctCount()函数用于统计满足条件的字段的排重后的数量：

func SQLBuilderDistinctCount() {
    val, err := worm.Table("user").Where("id>?", 0).DistinctCount("name")
    if err != nil {
        log.Error(err)
        return
    }
    //select count(distinct name) from user where id>?
}

查询字符串字段到数组

FindString()函数用于查询满足条件的记录，并将字符串字段的值添加到字符串数组：

func SQLBuilderFindString() {
    val, err := worm.Table("user").Select("name").Where("id>?", 0).FindString()
    if err != nil {
        log.Error(err)
        return
    }
    // select name from user where id>?
}

说明：

Select()中必须只有一个字段，并且字段的类型为字符串。

查询整形字段到数组

FindInt()函数用于查询满足条件的记录，并将整形字段的值添加到整形数组：

func SQLBuilderFindInt() {
    val, err := worm.Table("user").Select("age").Where("id>?", 0).FindInt()
    if err != nil {
        log.Error(err)
        return
    }
    log.Debug(val)
    // select age from user where id>?
}

说明：

Select()中必须只有一个字段，并且字段的类型为整形。

查询浮点型字段到数组

FindInt()函数用于查询满足条件的记录，并将浮点型字段的值添加到浮点型数组：

func SQLBuilderFindFloat() {
    val, err := worm.Table("user").Select("weight").Where("id>?", 0).FindFloat()
    if err != nil {
        log.Error(err)
        return
    }
    // select weight from user where id>?
}

说明：

Select()中必须只有一个字段，并且字段的类型为浮点型。