mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-12-04 03:12:19 +08:00
82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
package mgotool
|
|
|
|
import (
|
|
"context"
|
|
"github.com/OpenIMSDK/tools/errs"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
)
|
|
|
|
type Pagination interface {
|
|
GetPageNumber() int32
|
|
GetShowNumber() int32
|
|
}
|
|
|
|
func Anys[T any](ts []T) []any {
|
|
val := make([]any, len(ts))
|
|
for i := range ts {
|
|
val[i] = ts[i]
|
|
}
|
|
return val
|
|
}
|
|
|
|
func InsertMany[T any](ctx context.Context, coll *mongo.Collection, val []T, opts ...*options.InsertManyOptions) error {
|
|
_, err := coll.InsertMany(ctx, Anys(val), opts...)
|
|
if err != nil {
|
|
return errs.Wrap(err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func UpdateOne(ctx context.Context, coll *mongo.Collection, filter any, update any, notMatchedErr bool, opts ...*options.UpdateOptions) error {
|
|
res, err := coll.UpdateOne(ctx, filter, update, opts...)
|
|
if err != nil {
|
|
return errs.Wrap(err)
|
|
}
|
|
if notMatchedErr && res.MatchedCount == 0 {
|
|
return errs.Wrap(mongo.ErrNoDocuments)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func Find[T any](ctx context.Context, coll *mongo.Collection, filter any, opts ...*options.FindOptions) ([]T, error) {
|
|
cur, err := coll.Find(ctx, filter, opts...)
|
|
if err != nil {
|
|
return nil, errs.Wrap(err)
|
|
}
|
|
defer cur.Close(ctx)
|
|
var res []T
|
|
if err := cur.All(ctx, &res); err != nil {
|
|
return nil, errs.Wrap(err)
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func FindOne[T any](ctx context.Context, coll *mongo.Collection, filter any, opts ...*options.FindOneOptions) (res T, err error) {
|
|
cur := coll.FindOne(ctx, filter, opts...)
|
|
if err := cur.Err(); err != nil {
|
|
return res, errs.Wrap(err)
|
|
}
|
|
if err := cur.Decode(&res); err != nil {
|
|
return res, errs.Wrap(err)
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
func FindPage[T any](ctx context.Context, coll *mongo.Collection, filter any, pagination Pagination, opts ...*options.FindOptions) (int64, []T, error) {
|
|
count, err := Count[T](ctx, coll, filter)
|
|
if err != nil {
|
|
return 0, nil, err
|
|
}
|
|
opt := options.Find().SetSkip(int64(pagination.GetPageNumber() * pagination.GetShowNumber())).SetLimit(int64(pagination.GetShowNumber()))
|
|
res, err := Find[T](ctx, coll, filter, append(opts, opt)...)
|
|
if err != nil {
|
|
return 0, nil, err
|
|
}
|
|
return count, res, nil
|
|
}
|
|
|
|
func Count(ctx context.Context, coll *mongo.Collection, filter any, opts ...*options.CountOptions) (int64, error) {
|
|
return coll.CountDocuments(ctx, filter, opts...)
|
|
}
|