当前位置：首页 > news >正文

Go基础：用Go语言操作MongoDB详解

news 2025/10/6 6:49:15

文章目录

- 一、安装Go MongoDB驱动
- 二、连接MongoDB
- - 2.1 基本连接
  - 2.2 连接池模式
  - 2.3 断开与MongoDB的连接
- 三、CRUD操作
- - 3.1 插入文档
  - 3.2 查询文档
  - 3.3 更新文档
  - 3.4 删除文档
  - 3.5 完整案例
- 四、BSON解析

一、安装Go MongoDB驱动

Go语言操作MongoDB是一种常见的开发需求，特别是在构建高性能、分布式系统时。如需更详细的信息，可以参考以下来源：

Go MongoDB驱动官方文档

在Go语言中操作MongoDB，首先需要安装官方的MongoDB驱动。可以使用以下命令进行安装：

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

该驱动提供了丰富的API，支持连接池、CRUD操作、事务处理等功能。

二、连接MongoDB

2.1 基本连接

以下是连接MongoDB的示例代码：

package main
import ("context""fmt""log""time""go.mongodb.org/mongo-driver/mongo""go.mongodb.org/mongo-driver/mongo/options"
)
func main() {// 设置客户端连接配置clientOptions := options.Client().ApplyURI("mongodb://localhost:27017")// 连接到MongoDBclient, err := mongo.Connect(context.TODO(), clientOptions)if err != nil {log.Fatal(err)}// 检查连接err = client.Ping(context.TODO(), nil)if err != nil {log.Fatal(err)}fmt.Println("Connected to MongoDB!")
}

2.2 连接池模式

MongoDB驱动支持连接池，可以通过设置MaxPoolSize和MinPoolSize来优化连接管理：

clientOptions := options.Client().ApplyURI("mongodb://localhost:27017").SetMaxPoolSize(100). // 最大连接数SetMinPoolSize(10)   // 最小连接数

连接池模式能有效提高数据库操作的性能，特别是在高并发场景下。

2.3 断开与MongoDB的连接

完成操作后，应断开与MongoDB的连接以释放资源：

err = client.Disconnect(context.TODO())
if err != nil {log.Fatal(err)
}
fmt.Println("Connection to MongoDB closed.")

三、CRUD操作

3.1 插入文档

以下代码演示如何向MongoDB集合中插入文档：

collection := client.Database("testdb").Collection("users")
// 定义要插入的文档
document := bson.D{{"name", "Alice"},{"age", 30},{"city", "New York"},
}
// 插入文档
insertResult, err := collection.InsertOne(context.TODO(), document)
if err != nil {log.Fatal(err)
}
fmt.Println("Inserted document ID:", insertResult.InsertedID)

3.2 查询文档

查询文档可以使用FindOne或Find方法：

var result bson.M
err = collection.FindOne(context.TODO(), bson.D{{"name", "Alice"}}).Decode(&result)
if err != nil {log.Fatal(err)
}
fmt.Println("Found document:", result)

3.3 更新文档

更新文档可以使用UpdateOne或UpdateMany方法：

filter := bson.D{{"name", "Alice"}}
update := bson.D{{"$set", bson.D{{"age", 31},}},
}
updateResult, err := collection.UpdateOne(context.TODO(), filter, update)
if err != nil {log.Fatal(err)
}
fmt.Println("Updated document count:", updateResult.ModifiedCount)

3.4 删除文档

删除文档可以使用DeleteOne或DeleteMany方法：

deleteResult, err := collection.DeleteOne(context.TODO(), bson.D{{"name", "Alice"}})
if err != nil {log.Fatal(err)
}
fmt.Println("Deleted document count:", deleteResult.DeletedCount)

3.5 完整案例

package mainimport ("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{"name": "Alice", "age": 28, "email": "alice@example.com"}err = insertDocument(client, "testdb", "users", doc)if err != nil {log.Fatal(err)}fmt.Println("插入文档成功")// 查询文档filter := bson.M{"name": "Alice"}docs, err := findDocuments(client, "testdb", "users", filter)if err != nil {log.Fatal(err)}fmt.Println("查询结果:", docs)// 更新文档update := bson.M{"age": 30}err = updateDocument(client, "testdb", "users", bson.M{"name": "Alice"}, update)if err != nil {log.Fatal(err)}fmt.Println("更新文档成功")// 查询更新后的文档docs, err = findDocuments(client, "testdb", "users", filter)if err != nil {log.Fatal(err)}fmt.Println("更新后的查询结果:", docs)// 删除文档err = deleteDocument(client, "testdb", "users", bson.M{"name": "Alice"})if err != nil {log.Fatal(err)}fmt.Println("删除文档成功")// 查询删除后的文档docs, err = findDocuments(client, "testdb", "users", filter)if err != nil {log.Fatal(err)}fmt.Println("删除后的查询结果:", docs)
}

四、BSON解析

MongoDB使用BSON（二进制JSON）格式存储数据。Go语言中的bson包提供了将Go数据结构转换为BSON格式的功能。例如：

type User struct {Name string `bson:"name"`Age  int    `bson:"age"`City string `bson:"city"`
}
user := User{Name: "Bob", Age: 25, City: "Los Angeles"}
// 将结构体转换为BSON
bsonData, err := bson.Marshal(user)
if err != nil {log.Fatal(err)
}
fmt.Println("BSON data:", bsonData)