golang简单封装使用mongodb

shoulai_

已于 2022-07-23 13:08:24 修改

阅读量922

点赞数

分类专栏： golang mongodb 文章标签： mongodb golang 数据库

于 2022-07-22 11:31:36 首次发布

本文链接：https://blog.csdn.net/xslde_com/article/details/125928803

版权

这篇博客展示了如何在Golang中封装和使用MongoDB，包括连接数据库、插入单条和多条记录、查询及删除操作。测试结果显示所有操作成功。

摘要由CSDN通过智能技术生成

golang简单封装使用mongodb

一、封装代码
二、测试代码

一、封装代码

var client *mongo.Client

func NewMongoDB(ctx context.Context, appUri string) error {
   
	//连接到mongodb
	c, err := mongo.Connect(ctx, options.Client().ApplyURI(appUri))
	if err != nil {
   
		return err
	}
	//检查连接
	err = c.Ping(ctx, nil)
	if err != nil {
   
		return err
	}
	client = c
	log.Println("mongodb连接成功")
	return nil
}

type MongoDB[T any] struct {
   
	database   string
	collection string
}

func NewMGDB[T any](database string, collection string) *MongoDB[T] {
   
	if client == nil {
   
		log.Fatalln("mongo.Client Not initialized !")
	}
	return &MongoDB[T]{
   
		database,
		collection,
	}
}

//新增一条记录
func (mg *MongoDB[T]) InsertOne(ctx context.Context, value T) *mongo.InsertOneResult {
   
	result, err := mg.getCollection().InsertOne(ctx, value)
	if err != nil {
   
		panic(err)
	}
	return result
}

//新增多条记录
func (mg *MongoDB[T]) InsertMultiple(ctx context.Context, data []T) *mongo.InsertManyResult {
   
	var array []interface{
   }
	for i := 0; i < len(data); i++ {
   
		array = append(array, data[i])
	}
	result, err := mg.getCollection().InsertMany(ctx, array)
	if err != nil {
   
		panic(err)
	}
	return result
}

//根据字段名和值查询一条记录
func (mg *MongoDB[T]) FindOne(ctx context.Context, filter filter) (T, error) {
   
	var t T
	err := mg.getCollection().FindOne(ctx, filter).Decode(&t)
	if err != nil {
   
		return t, err
	}
	return t, nil
}

//根据条件查询多条记录
func (mg *MongoDB[T]) Find(ctx context.Context, filter filter, limit int64) ([]T, error) {
   
	findOpts := options.Find()
	findOpts.SetLimit(limit)
	cursor, err := mg.getCollection().Find(ctx, filter, findOpts)
	var ts []T
	if err != nil {
   
		return ts, err
	}
	for cursor.Next(ctx)