From 2ee9531096aeac72f06f629abe152ea75c6ef5b3 Mon Sep 17 00:00:00 2001 From: yzqzss Date: Sat, 20 Apr 2024 03:28:57 +0800 Subject: [PATCH] Update ClaimTask to use options.After for return document --- api_v1.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/api_v1.go b/api_v1.go index 9fd7ef6..b683725 100644 --- 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