Go实现mongodb增删改查的工具类
文章目录
-
- 1、驱动下载
- 2、实现代码
-
- 2.1 Mongodb工具类代码
- 2.2 使用例子
- 2.3 运行效果
1、驱动下载
mongodb官方go介绍
使用例子https://www.mongodb.com/docs/drivers/go/current/fundamentals/connection/#connection-example
快速入门https://www.mongodb.com/docs/drivers/go/current/quick-start/
要对mongodb进行增删改查，需要添加对应的go驱动依赖，安装命令
go get go.mongodb.org/mongo-driver/mongo
2、实现代码
2.1 Mongodb工具类代码
封装的工具类mongodao.go
package dao
import (
"context"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
"log"
"time"
)
type MondoDao struct {
client *mongo.Client
db *mongo.Database
collection *mongo.Collection
}
func (dao *MondoDao) Connect(url string, db string, collection string) {
var err error
dao.client, err = mongo.Connect(context.TODO(), options.Client().SetConnectTimeout(time.Second*10).ApplyURI(url))
if err != nil {
log.Fatal(err)
}
dao.db = dao.client.Database(db)
dao.collection = dao.db.Collection(collection)
}
// 插入一行
func (dao *MondoDao) InsertOne(obj any) (interface{}, error) {
result, err := dao.collection.InsertOne(context.TODO(), obj)
if err == nil {
return result.InsertedID, err
}
return nil, err
}
// 插入多行
func (dao *MondoDao) InsertMany(objs []any) ([]interface{}, error) {
result, err := dao.collection.InsertMany(context.TODO(), objs)
if err != nil {
return nil, err
}
return result.InsertedIDs, nil
}
// 查询一行
func (dao *MondoDao) FindOne(filter bson.M, obj any) error {
result := dao.collection.FindOne(context.TODO(), filter)
if result.Err() == nil {
err := result.Decode(obj)
return err
}
return result.Err()
}
// 查询多行
func (dao *MondoDao) FindMany(filter bson.M, obj any) error {
cursor, err := dao.collection.Find(context.TODO(), filter)
defer cursor.Close(context.TODO())
if err == nil {
err = cursor.All(context.TODO(), obj)
}
return err
}
// 分布查询多行
func (dao *MondoDao) FindManyByPage(filter bson.M, obj any, page int64, pagesize int64, sort bson.M) error {
options := options.Find().SetSkip((page - 1) * pagesize).SetLimit(pagesize).SetSort(sort)
cursor, err := dao.collection.Find(context.TODO(), filter, options)
defer cursor.Close(context.TODO())
if err == nil {
err = cursor.All(context.TODO(), obj)
}
return err
}
// 删除一行
func (dao *MondoDao) DeleteOne(filter bson.M) (int64, error) {
result, err := dao.collection.DeleteOne(context.TODO(), filter)
if err == nil {
return result.DeletedCount, err
}
return 0, err
}
// 删除多行
func (dao *MondoDao) DeleteMany(filter bson.M) (int64, error) {
result, err := dao.collection.DeleteMany(context.TODO(), filter)
if err == nil {
return result.DeletedCount, err
}
return 0, err
}
// 删除集合
func (dao *MondoDao) DropCollection() error {
err := dao.collection.Drop(context.TODO())
//if err = nil {
// log.Fatal("drop collection %v", err)
//}
return err
}
// 更新一行
func (dao *MondoDao) UpdateOne(filter bson.M, update bson.M) (int64, error) {
result, err := dao.collection.UpdateOne(context.TODO(), filter, update)
if err == nil {
return result.ModifiedCount, err
}
return 0, err
}
// 更新多行
func (dao *MondoDao) UpdateMany(filter bson.M, update bson.M) (int64, error) {
result, err := dao.collection.UpdateMany(context.TODO(), filter, update)
if err == nil {
return result.ModifiedCount, err
}
return 0, err
}
/**
// 条件查询
$lt 小于 bson.M{"age": bson.M{"$lt": 20}}
$gt 大于 bson.M{"age": bson.M{"$gt": 20}}
$lte 小于等于 bson.M{"age": bson.M{"$lte": 20}} bson.D{{"age", bson.D{{"$lte", 20}}}}
$gte 大于等于 bson.M{"age": bson.M{"$gte": 20}}
$ne 不等于 bson.M{"age": bson.M{"$ne": 20}}
$eq 等于，可以省略这个符号 bson.M{"age": bson.M{"$eq": 20}}，bson.M{"age": 20}
$in 在范围内 bson.M{"age": bson.M{"$in": []int{16, 33}}}
$nin 不在范围内 bson.M{"age": bson.M{"$nin": []int{16, 33}}}
//
$inc 对给定数值字段数加减 bson.M{"$inc": bson.M{"age": -5}}
$set 设置字段值，如果字段不存在则创建 bson.M{"$set": bson.M{"age": 100}}
$unset 移除字段 {'$unset':{'Name':""}}
*/
2.2 使用例子
使用例子
package main
import (
"fmt"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"log"
"mymongodb/dao"
)
type User struct {
ID primitive.ObjectID `bson:"_id,omitempty"`
Name string
Age int
}
func (s User) String() string {
return fmt.Sprintf("%v %v %v", s.ID.Hex(), s.Name, s.Age)
}
func main() {
dao := dao.MondoDao{}
url := "mongodb://127.0.0.1:27017"
dao.Connect(url, "gotest", "user")
user1 := User{Name: "user1", Age: 66}
id, _ := dao.InsertOne(user1)
fmt.Printf("InsertOne id:%v\n", (id.(primitive.ObjectID)).Hex())
user2 := []any{
User{Name: "user2", Age: 66},
User{Name: "user3_3", Age: 76},
User{Name: "user3_4", Age: 88},
}
ids, _ := dao.InsertMany(user2)
for index, id := range ids {
fmt.Printf("InsertMany id:%v=%v\n", index, (id.(primitive.ObjectID)).Hex())
}
filter3 := bson.M{"name": "user2", "age": 66}
var user3 User
err := dao.FindOne(filter3, &user3)
fmt.Printf("Findone user3:%v, %v\n", user3, err)
var user4 []User
filter4 := bson.M{"age": bson.M{"$gt": 60}}
err4 := dao.FindMany(filter4, &user4)
if err4 == nil {
for index, v := range user4 {
fmt.Printf("FindMany user4:%v, %v, \n", index, v)
}
}
var users5 []User
filter5 := bson.M{"age": bson.M{"$gt": 65}}
sort := bson.M{"age": 1}
err5 := dao.FindManyByPage(filter5, &users5, 1, 10, sort)
if err5 == nil {
for index, v := range users5 {
fmt.Printf("FindManyByPage user5:%v, %v\n", index, v.String())
}
}
filter8 := bson.M{"name": "user2"}
update8 := bson.M{"$inc": bson.M{"age": +6}}
count8, err8 := dao.UpdateOne(filter8, update8)
fmt.Printf("UpdateOne count:%v, %v\n", count8, err8)
filter9 := bson.M{"$or": []bson.M{bson.M{"name": "user1"}, bson.M{"name": "user2"}}}
update9 := bson.M{"$set": bson.M{"age": 90}}
count9, err9 := dao.UpdateMany(filter9, update9)
fmt.Printf("UpdateMany count:%v, %v\n", count9, err9)
filter6 := bson.M{"name": "user1", "age": 90}
deletecount6, err6 := dao.DeleteOne(filter6)
fmt.Printf("DeleteOne count:%v, %v\n", deletecount6, err6)
filter7 := bson.M{"age": bson.M{"$gte": 66}}
deletecount7, err7 := dao.DeleteMany(filter7)
fmt.Printf("DeleteMany count:%v, %v\n", deletecount7, err7)
err10 := dao.DropCollection()
if err10 == nil {
fmt.Println("drop Collection success!")
} else {
log.Fatal("drop collection error,%v", err)
}
}
2.3 运行效果
D:\project\go\gotest\sql\mongodb>go run main.go
InsertOne id:64d32053a61a1841b70e129a
InsertMany id:0=64d32054a61a1841b70e129b
InsertMany id:1=64d32054a61a1841b70e129c
InsertMany id:2=64d32054a61a1841b70e129d
Findone user3:64d32054a61a1841b70e129b user2 66,
FindMany user4:0, 64d32053a61a1841b70e129a user1 66,
FindMany user4:1, 64d32054a61a1841b70e129b user2 66,
FindMany user4:2, 64d32054a61a1841b70e129c user3_3 76,
FindMany user4:3, 64d32054a61a1841b70e129d user3_4 88,
FindManyByPage user5:0, 64d32053a61a1841b70e129a user1 66
FindManyByPage user5:1, 64d32054a61a1841b70e129b user2 66
FindManyByPage user5:2, 64d32054a61a1841b70e129c user3_3 76
FindManyByPage user5:3, 64d32054a61a1841b70e129d user3_4 88
UpdateOne count:1,
UpdateMany count:2,
DeleteOne count:1,
DeleteMany count:3,
drop Collection success!