ZBLOG

golang 操作 mongodb 分批删除数据

在Golang中,你可以使用官方提供的MongoDB驱动程序(如”go.mongodb.org/mongo-driver”)来操作MongoDB数据库。以下是一个示例代码片段,演示如何使用分批方式删除数据:

package main

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

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

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

	db := client.Database("mydatabase")
	coll := db.Collection("mycollection")

	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()

	filter := bson.M{} // 设置筛选条件,这里留空表示删除所有文档

	opts := options.Delete()
	opts.SetCollation(&options.Collation{Locale: "en", Strength: 1}) // 设置排序规则和大小写敏感度等选项

	batchSize := 1000   // 每次删除的文档数量
	maxDeletes := 5000 // 最多删除的总文档数量

	totalDeletes := 0 // 已经删除的文档数量
	for totalDeletes < maxDeletes {
		result, err := coll.DeleteMany(ctx, filter, opts.SetLimit(int64(batchSize)))
		if err != nil {
			log.Fatal(err)
		}

		fmt.Printf("Deleted %v documents\n", result.DeletedCount)

		totalDeletes += int(result.DeletedCount)
		
        if result.DeletedCount == 0 || result.DeletedCount < int64(batchSize) {
			// 如果没有更多文档可删除或者已经达到最大删除数量,结束循环
			break
		}
	}

	fmt.Printf("Total deleted: %v\n", totalDeletes)
}

请根据实际情况修改MongoDB连接字符串、数据库名称、集合名称以及筛选条件等参数。注意,在实际使用时,请谨慎操作删除数据,确保备份重要数据以避免意外损失。


内容由零声教学AI助手提供,问题来源于学员提问

本站部分文章来源于网络,版权归原作者所有,如有侵权请联系站长删除。
转载请注明出处:https://golang.0voice.com/?id=14992

分享:
扫描分享到社交APP
上一篇
下一篇
发表列表
游客 游客
此处应有掌声~
评论列表

还没有评论,快来说点什么吧~

联系我们

在线咨询: 点击这里给我发消息

微信号:3007537140

上班时间: 10:30-22:30

关注我们
x

注册

已经有帐号?