Go实现mongodb增删改查的工具类

以下是一个简化的Go语言中实现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"
)
 
var client *mongo.Client
var collection *mongo.Collection
 
func init() {
    var err error
    client, err = mongo.NewClient(options.Client().ApplyURI("mongodb://localhost:27017"))
    if err != nil {
        log.Fatal(err)
    }
 
    ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
    defer cancel()
    err = client.Connect(ctx)
    if err != nil {
        log.Fatal(err)
    }
 
    collection = client.Database("testdb").Collection("testcollection")
}
 
// 插入数据
func Insert(document interface{}) error {
    _, err := collection.InsertOne(context.Background(), document)
    return err
}
 
// 查询数据
func Find(filter interface{}) (*mongo.Cursor, error) {
    return collection.Find(context.Background(), filter)
}
 
// 更新数据
func Update(filter, update interface{}) (*mongo.UpdateResult, error) {
    return collection.UpdateOne(context.Background(), filter, update)
}
 
// 删除数据
func Delete(filter interface{}) (*mongo.DeleteResult, error) {
    return collection.DeleteOne(context.Background(), filter)
}
 
func main() {
    // 插入示例
    err := Insert(bson.M{"name": "Alice", "age": 30})
    if err != nil {
        log.Fatal(err)
    }
 
    // 查询示例
    cursor, err := Find(bson.M{"name": "Alice"})
    if err != nil {
        log.Fatal(err)
    }
    defer cursor.Close(context.Background())
    for cursor.Next(context.Background()) {
        var result bson.M
        err := cursor.Decode(&result)
        if err != nil {
            log.Fatal(err)
        }
        fmt.Println(result)
    }
 
    // 更新示例
    updateResult, err := Update(bson.M{"name": "Alice"}, bson.M{"$set": bson.M{"age": 31}})
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("Updated %v documents\n", updateResult.MatchedCount+updateResult.UpsertedCount)
 
    // 删除示例
    deleteResult, err := Delete(bson.M{"name": "Alice"})
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("Deleted %v documents\n", deleteResult.DeletedCount)
}

这段代码首先建立了与MongoDB的连接,然后定义了InsertFindUpdateDelete函数来进行数据的增删改查操作。在main函数中,我们演示了如何使用这些函数来插入、查询、更新和删除数据。这个例子提供了一个简洁的工具类,并且是一个可以直接运行的MongoDB操作示例。

最后修改于:2024年09月03日 17:35

评论已关闭

推荐阅读

DDPG 模型解析,附Pytorch完整代码
2024年11月24日
DQN 模型解析,附Pytorch完整代码
2024年11月24日
AIGC实战——Transformer模型
2024年12月01日
Socket TCP 和 UDP 编程基础(Python)
2024年11月30日
python , tcp , udp
如何使用 ChatGPT 进行学术润色?你需要这些指令
2024年12月01日
AI
最新 Python 调用 OpenAi 详细教程实现问答、图像合成、图像理解、语音合成、语音识别(详细教程)
2024年11月24日
ChatGPT 和 DALL·E 2 配合生成故事绘本
2024年12月01日
omegaconf,一个超强的 Python 库!
2024年11月24日
【视觉AIGC识别】误差特征、人脸伪造检测、其他类型假图检测
2024年12月01日
[超级详细]如何在深度学习训练模型过程中使用 GPU 加速
2024年11月29日
Python 物理引擎pymunk最完整教程
2024年11月27日
MediaPipe 人体姿态与手指关键点检测教程
2024年11月27日
深入了解 Taipy:Python 打造 Web 应用的全面教程
2024年11月26日
基于Transformer的时间序列预测模型
2024年11月25日
Python在金融大数据分析中的AI应用(股价分析、量化交易)实战
2024年11月25日
AIGC Gradio系列学习教程之Components
2024年12月01日
Python3 `asyncio` — 异步 I/O,事件循环和并发工具
2024年11月30日
llama-factory SFT系列教程:大模型在自定义数据集 LoRA 训练与部署
2024年12月01日
Python 多线程和多进程用法
2024年11月24日
Python socket详解,全网最全教程
2024年11月27日