Go语言操作MongoDB文档

本文档详细介绍了如何使用Go语言对MongoDB进行基础操作。我们将学习如何使用Go连接MongoDB、插入文档、查询文档、更新文档和删除文档。以下是详细操作步骤：

1. 安装驱动

首先，使用以下命令安装官方的MongoDB Go驱动：

go get go.mongodb.org/mongo-driver/mongo

2. 导入依赖

导入必要的依赖包：

import (
	"context"
	"fmt"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
	"log"
	"time"
)

3. 建立连接

创建一个函数，用于连接并返回一个Mongo客户端：

func connect() (*mongo.Client, error) {
	client, err := mongo.NewClient(options.Client().ApplyURI("mongodb://localhost:27017"))

	if err != nil {
		return nil, err
	}

	ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
	err = client.Connect(ctx)

	if err != nil {
		return nil, err
	}

	return client, nil
}

4. 插入文档

创建一个函数，用于向集合中插入文档：

func insertDocument(client *mongo.Client, dbName, colName string, doc interface{}) error {
	collection := client.Database(dbName).Collection(colName)
	_, err := collection.InsertOne(context.Background(), doc)

	return err
}

5. 查询文档

创建一个函数，用于查询满足指定过滤器的文档：

func findDocuments(client *mongo.Client, dbName, colName string, filter bson.M) ([]bson.M, error) {
	collection := client.Database(dbName).Collection(colName)

	cur, err := collection.Find(context.Background(), filter)
	if err != nil {
		return nil, err
	}

	var results []bson.M
	err = cur.All(context.Background(), &results)

	return results, err
}

6. 更新文档

创建一个函数，用于更新满足指定过滤器的文档：

func updateDocument(client *mongo.Client, dbName, colName string, filter, update bson.M) error {
	collection := client.Database(dbName).Collection(colName)
	_, err := collection.UpdateMany(context.Background(), filter, bson.M{
		"$set": update,
	})

	return err
}

7. 删除文档

创建一个函数，用于删除满足指定过滤器的文档：

func deleteDocuments(client *mongo.Client, dbName, colName string, filter bson.M) error {
	collection := client.Database(dbName).Collection(colName)
	_, err := collection.DeleteMany(context.Background(), filter)

	return err
}

参考代码

在 main 函数中调用以上方法展示完整的CRUD操作：

package main

import (
	"context"
	"fmt"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
	"log"
	"time"
)

func main() {
	client, err := connect()
	if err != nil {
		log.Fatal(err)
	}
	defer client.Disconnect(context.Background())

	// 插入文档
	doc := bson.M{"username": "testuser", "email": "[email protected]", "age": 25}
	err = insertDocument(client, "mydb", "users", doc)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("插入文档成功")

	// 查询文档
	filter := bson.M{"username": "testuser"}
	docs, err := findDocuments(client, "mydb", "users", filter)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("查询结果：", docs)

	// 更新文档
	update := bson.M{"email": "[email protected]", "age": 26}
	err = updateDocument(client, "mydb", "users", filter, update)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("更新文档成功")

	// 查询更新后的文档
	docs, err = findDocuments(client, "mydb", "users", filter)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("更新后的查询结果：", docs)

	// 删除文档
	err = deleteDocuments(client, "mydb", "users", filter)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("删除文档成功")

	// 查询删除后的文档
	docs, err = findDocuments(client, "mydb", "users", filter)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("删除后的查询结果：", docs)
}

当你运行上述程序时，你将首先连接到MongoDB并插入一条文档，其中包括用户名，邮箱和年龄。接下来，程序查询用户名为testuser的文档，然后更新该文档的邮箱和年龄属性。再次查询文档以确认是否已更新，并最后删除该文档。完成这一系列操作后，对数据进行最终查询以确认文档是否已被删除。