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的连接,然后定义了Insert
、Find
、Update
和Delete
函数来进行数据的增删改查操作。在main
函数中,我们演示了如何使用这些函数来插入、查询、更新和删除数据。这个例子提供了一个简洁的工具类,并且是一个可以直接运行的MongoDB操作示例。
评论已关闭