v0.0.131

mongoext/aggregation.go

@@ -0,0 +1,57 @@
+package mongoext
+
+import (
+	"context"
+	"go.mongodb.org/mongo-driver/mongo"
+	"gogs.mikescher.com/BlackForestBytes/goext/rfctime"
+	"inopart/backend/common/cursortoken"
+	"inopart/backend/models"
+)
+
+func HandleMongoCursorWithToken[T models.Entity](ctx context.Context, cursor *mongo.Cursor, pageSize *int) ([]T, string, error) {
+	var tokenStr string
+	var docs []T
+	numDocs := cursor.RemainingBatchLength()
+
+	if pageSize == nil || (numDocs <= *pageSize) && numDocs > 0 {
+		docs = make([]T, numDocs)
+
+		err := cursor.All(ctx, &docs)
+		if err != nil {
+			return []T{}, "", err
+		}
+
+	} else if numDocs == *pageSize+1 {
+
+		docs = make([]T, numDocs-1)
+
+		for i := 0; i < numDocs-1; i++ {
+			cursor.Next(ctx)
+			err := cursor.Decode(&docs[i])
+			if err != nil {
+				return []T{}, "", err
+			}
+		}
+
+		var nextPageDoc T
+		cursor.Next(ctx)
+		err := cursor.Decode(&nextPageDoc)
+		if err != nil {
+			return []T{}, "", err
+		}
+
+		nextPageToken := cursortoken.CursorToken{
+			Timestamp: rfctime.RFC3339NanoTime{},
+			Id:        nextPageDoc.GetId(),
+			Offset:    0,
+			PageSize:  *pageSize,
+		}
+
+		tokenStr, err = nextPageToken.Token()
+		if err != nil {
+			return nil, "", err
+		}
+	}
+
+	return docs, tokenStr, nil
+}

rfctime/rfc3339.go

@@ -81,7 +81,7 @@ func (t *RFC3339Time) UnmarshalBSONValue(bt bsontype.Type, data []byte) error {
 		return errors.New(fmt.Sprintf("cannot unmarshal %v into RFC3339Time", bt))
 	}
 	var tt time.Time
-	err := bson.RawValue{Type: bt, Value: data}.Unmarshal(&tt)
+	err := bson.Unmarshal(data, &tt)
 	if err != nil {
 		return err
 	}
@@ -116,12 +116,6 @@ func (t RFC3339Time) DecodeValue(dc bsoncodec.DecodeContext, vr bsonrw.ValueRead
 		return err
 	}
 
-	if val.Kind() == reflect.Ptr {
-		val.Set(reflect.ValueOf(&t))
-	} else {
-		val.Set(reflect.ValueOf(t))
-	}
-
 	return nil
 }