saveweb/solar_tracker
1
0
Fork 0
mirror of https://github.com/saveweb/solar_tracker.git synced 2024-09-19 11:15:29 -07:00
Code Issues Packages Projects Releases Wiki Activity

Update ClaimTask to use options.After for return document

Browse Source
This commit is contained in:
yzqzss 2024-04-20 03:28:57 +08:00
parent 48d32bacfb
commit 2ee9531096
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api_v1.go
View File

@ -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