使用 mongo-go-pagination 实现 MongoDB 分页查询与聚合管道

mongo-go-pagination 是一个专门为 Go 语言 MongoDB 驱动设计的轻量级分页库，它支持基本查询和聚合管道的分页功能。下面我将详细介绍如何使用这个库。

安装

首先安装 mongo-go-pagination 库：

go get github.com/gobeam/mongo-go-pagination

基本查询分页

package main

import (
	"context"
	"fmt"
	"log"
	"time"

	"github.com/gobeam/mongo-go-pagination"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

type Product struct {
	Name      string    `bson:"name"`
	Price     float64   `bson:"price"`
	CreatedAt time.Time `bson:"created_at"`
}

func main() {
	// 连接 MongoDB
	client, err := mongo.Connect(context.Background(), options.Client().ApplyURI("mongodb://localhost:27017"))
	if err != nil {
		log.Fatal(err)
	}
	defer client.Disconnect(context.Background())

	collection := client.Database("testdb").Collection("products")

	// 创建一些测试数据
	createTestData(collection)

	// 基本查询分页
	basicPagination(collection)
}

func createTestData(collection *mongo.Collection) {
	// 这里可以插入一些测试数据
	// ...
}

func basicPagination(collection *mongo.Collection) {
	filter := bson.M{"price": bson.M{"$gt": 10}} // 查询条件

	// 第1页，每页5条记录
	paginatedData, err := pagination.New(collection).Limit(5).Page(1).Filter(filter).Find()
	if err != nil {
		log.Fatal(err)
	}

	// 解码结果
	var products []Product
	for _, raw := range paginatedData.Data {
		var product Product
		if err := bson.Unmarshal(raw, &product); err != nil {
			log.Fatal(err)
		}
		products = append(products, product)
	}

	fmt.Printf("Total Records: %d\n", paginatedData.Pagination.Total)
	fmt.Printf("Page: %d\n", paginatedData.Pagination.Page)
	fmt.Printf("Per Page: %d\n", paginatedData.Pagination.PerPage)
	fmt.Printf("Total Pages: %d\n", paginatedData.Pagination.TotalPage)
	fmt.Println("Products:")
	for _, p := range products {
		fmt.Printf("- %s: $%.2f\n", p.Name, p.Price)
	}
}

聚合管道分页

func aggregatePagination(collection *mongo.Collection) {
	// 定义聚合管道
	pipeline := []bson.M{
		{
			"$match": bson.M{
				"price": bson.M{"$gt": 10},
			},
		},
		{
			"$group": bson.M{
				"_id":   "$category",
				"count": bson.M{"$sum": 1},
				"total": bson.M{"$sum": "$price"},
			},
		},
		{
			"$sort": bson.M{"total": -1},
		},
	}

	// 第1页，每页3条记录
	paginatedData, err := pagination.New(collection).Limit(3).Page(1).Aggregate(pipeline)
	if err != nil {
		log.Fatal(err)
	}

	fmt.Printf("\nAggregation Results (Page %d):\n", paginatedData.Pagination.Page)
	for _, raw := range paginatedData.Data {
		var result struct {
			ID    string  `bson:"_id"`
			Count int     `bson:"count"`
			Total float64 `bson:"total"`
		}
		if err := bson.Unmarshal(raw, &result); err != nil {
			log.Fatal(err)
		}
		fmt.Printf("- Category: %s, Count: %d, Total: $%.2f\n", result.ID, result.Count, result.Total)
	}
}

高级功能

1. 排序

// 添加排序条件
paginatedData, err := pagination.New(collection).
    Limit(5).
    Page(1).
    Sort("price", -1). // -1 表示降序
    Filter(filter).
    Find()

2. 选择字段

// 只选择特定字段
paginatedData, err := pagination.New(collection).
    Limit(5).
    Page(1).
    Select(bson.M{"name": 1, "price": 1}).
    Filter(filter).
    Find()

3. 自定义解码

// 自定义解码方式
var products []Product
paginatedData, err := pagination.New(collection).
    Limit(5).
    Page(1).
    Filter(filter).
    Decode(&products)

性能考虑

1. 对于大型集合，确保查询字段有适当的索引
2. 避免在分页查询中使用 $where 或 JavaScript 表达式
3. 考虑使用 hint() 强制使用特定索引

总结

mongo-go-pagination 提供了简单易用的 API 来实现 MongoDB 分页功能，主要特点包括：

• 支持基本查询和聚合管道的分页
• 提供丰富的分页元数据（总记录数、总页数等）
• 灵活的排序和字段选择
• 与官方 MongoDB Go 驱动无缝集成

对于大多数分页需求，这个库都能提供简洁高效的解决方案。