149 lines
2.7 KiB
Go
149 lines
2.7 KiB
Go
package mongo
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go.mongodb.org/mongo-driver/v2/bson"
|
|
)
|
|
|
|
// Find is used to fetch the first page of data.
|
|
func (c *MongoClient) Find(ctx context.Context, database, name string, filter bson.M, limit int64) (bson.M, error) {
|
|
|
|
// 1. Prepare to query.
|
|
collection := c.GetCollection(database, name)
|
|
|
|
pageSize := max(limit, c.Limit)
|
|
|
|
sort := bson.M{"_id": 1}
|
|
|
|
if err := c.DiscriminatorCheckAndApplyToFilter(ctx, name, filter); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
pipeline := BuildPaginationPipeline(0, pageSize + 1, filter, sort)
|
|
|
|
// 2. Query
|
|
cursor, err := collection.Aggregate(ctx, pipeline)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer cursor.Close(ctx)
|
|
|
|
// 3. Build results
|
|
var facetResults []bson.M
|
|
if err = cursor.All(ctx, &facetResults); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
root := facetResults[0]
|
|
|
|
data := root["data"].(bson.A)
|
|
|
|
metadata := root["metadata"].(bson.A)
|
|
|
|
var totalValue any
|
|
if len(metadata) != 0 {
|
|
metadataRoot := metadata[0].(bson.M)
|
|
totalValue = metadataRoot["total"]
|
|
}
|
|
|
|
var total int64
|
|
switch v := totalValue.(type) {
|
|
case int32:
|
|
total = int64(v)
|
|
case int64:
|
|
total = v
|
|
default:
|
|
total = 0
|
|
}
|
|
|
|
hasMore := false
|
|
if int64(len(data)) > pageSize {
|
|
hasMore = true
|
|
data = data[:pageSize]
|
|
}
|
|
|
|
out := bson.M{
|
|
"data": data,
|
|
"has_more": hasMore,
|
|
"total": total,
|
|
}
|
|
|
|
if hasMore {
|
|
// next cursor
|
|
var last bson.M = data[len(data) - 1].(bson.M)
|
|
var nextCursor string
|
|
lastId := last["_id"]
|
|
|
|
nextCursor = EncodeCursor(lastId.(bson.ObjectID))
|
|
|
|
out["next_cursor"] = nextCursor
|
|
}
|
|
|
|
return out, nil
|
|
}
|
|
|
|
func (c *MongoClient) FindOffset(ctx context.Context, database, name string, filter bson.M, offset, limit int64) (bson.M, error) {
|
|
// 1. Prepare to query.
|
|
collection := c.GetCollection(database, name)
|
|
|
|
finalLimit := max(limit, c.Limit)
|
|
|
|
f := Mongofy(&Query{
|
|
Filter: filter,
|
|
})
|
|
|
|
pipeline := BuildPaginationPipeline(offset, finalLimit, f, nil)
|
|
|
|
// 2. Query
|
|
cursor, err := collection.Aggregate(ctx, pipeline)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer cursor.Close(ctx)
|
|
|
|
// 3. Build results
|
|
var facetResults []bson.M
|
|
if err = cursor.All(ctx, &facetResults); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
root := facetResults[0]
|
|
|
|
data := root["data"].(bson.A)
|
|
|
|
metadata := root["metadata"].(bson.A)
|
|
|
|
var totalValue any
|
|
if len(metadata) != 0 {
|
|
metadataRoot := metadata[0].(bson.M)
|
|
totalValue = metadataRoot["total"]
|
|
}
|
|
|
|
var total int64
|
|
switch v := totalValue.(type) {
|
|
case int32:
|
|
total = int64(v)
|
|
case int64:
|
|
total = v
|
|
default:
|
|
total = 0
|
|
}
|
|
|
|
hasMore := false
|
|
if int64(len(data)) > finalLimit {
|
|
hasMore = true
|
|
data = data[:finalLimit]
|
|
}
|
|
|
|
out := bson.M{
|
|
"data": data,
|
|
"offset": offset,
|
|
"limit": finalLimit,
|
|
"has_more": hasMore,
|
|
"total": total,
|
|
}
|
|
|
|
return out, nil
|
|
}
|