Update ClaimTask to use options.After for return document

saveweb · Apr 19, 2024 · 2ee9531 · 2ee9531
1 parent 48d32ba
commit 2ee9531
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/api_v1.go b/api_v1.go
@@ -10,6 +10,7 @@ import (
 	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/bson/primitive"
 	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
 )
 
 const DEFAULT_DOC_ID_NAME = "id"
@@ -58,7 +59,8 @@ func ClaimTask(queue *mongo.Collection, from_status string, archivist string) *p
 		}}
 
 	var task primitive.M
-	err := queue.FindOneAndUpdate(context.TODO(), filter, update).Decode(&task)
+	opts := options.FindOneAndUpdate().SetReturnDocument(options.After)
+	err := queue.FindOneAndUpdate(context.TODO(), filter, update, opts).Decode(&task)
 	if err != nil {
 		if err == mongo.ErrNoDocuments {
 			return nil