使用sqlc实现高效的行计数查询-优快云博客

本文链接：https://blog.youkuaiyun.com/gitblog_00393/article/details/148377574

使用sqlc实现高效的行计数查询

sqlc 项目地址: https://gitcode.com/gh_mirrors/sql/sqlc

前言

在数据库应用中，行计数是最基础但非常重要的操作之一。sqlc作为一个强大的SQL转Go代码工具，能够帮助我们高效地实现这类功能。本文将详细介绍如何使用sqlc来实现行计数查询，包括简单的总数统计和分组统计。

基础表结构

首先我们需要定义一个基础表结构，这里以作者表为例：

CREATE TABLE authors (
  id       SERIAL PRIMARY KEY,
  hometown text   NOT NULL
);

这个表包含两个字段：

id：自增主键
hometown：作者的家乡，非空字段

简单行计数

SQL查询

最基本的行计数查询是统计表中的总记录数：

-- name: CountAuthors :one
SELECT count(*) FROM authors;

这里使用了sqlc的特殊注释语法：

-- name: CountAuthors 定义了查询名称
:one 表示这个查询返回单行结果

生成的Go代码

sqlc会自动生成对应的Go代码：

const countAuthors = `-- name: CountAuthors :one
SELECT count(*) FROM authors
`

func (q *Queries) CountAuthors(ctx context.Context) (int, error) {
	row := q.db.QueryRowContext(ctx, countAuthors)
	var i int
	err := row.Scan(&i)
	return i, err
}

这段代码实现了：

定义SQL查询字符串
创建CountAuthors方法
执行查询并返回结果

使用场景

这种简单计数适用于：

显示总记录数
分页时计算总页数
数据统计概览

分组行计数

SQL查询

更复杂的分组计数可以按特定字段分组统计：

-- name: CountAuthorsByTown :many
SELECT hometown, count(*) FROM authors
GROUP BY 1
ORDER BY 1;

关键点：

:many 表示返回多行结果
GROUP BY 1 表示按第一个字段(hometown)分组
ORDER BY 1 表示按第一个字段排序

生成的Go代码

const countAuthorsByTown = `-- name: CountAuthorsByTown :many
SELECT hometown, count(*) FROM authors
GROUP BY 1
ORDER BY 1
`

type CountAuthorsByTownRow struct {
	Hometown string
	Count    int
}

func (q *Queries) CountAuthorsByTown(ctx context.Context) ([]CountAuthorsByTownRow, error) {
	rows, err := q.db.QueryContext(ctx, countAuthorsByTown)
	if err != nil {
		return nil, err
	}
	defer rows.Close()
	items := []CountAuthorsByTownRow{}
	for rows.Next() {
		var i CountAuthorsByTownRow
		if err := rows.Scan(&i.Hometown, &i.Count); err != nil {
			return nil, err
		}
		items = append(items, i)
	}
	// 错误处理...
	return items, nil
}

这段代码实现了：