hax/seek-tune
1
0
Fork 0
mirror of https://github.com/cgzirim/seek-tune.git synced 2025-12-17 08:54:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix race condition in by using atomic upsert operation.

Browse source
This commit is contained in:
Chigozirim Igweamaka 2024-06-18 01:01:57 +01:00
parent 3e3976992a
commit 2d13181fa8
1 changed files with 13 additions and 33 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
utils/dbClient.go
View file

@ -56,40 +56,20 @@ func (db *DbClient) StoreFingerprints(fingerprints map[uint32]models.Couple) err
collection := db.client.Database("song-recognition").Collection("fingerprints")
for address, couple := range fingerprints {
// Check if the address already exists in the database
var existingDoc bson.M
err := collection.FindOne(context.Background(), bson.M{"_id": address}).Decode(&existingDoc)
filter := bson.M{"_id": address}
update := bson.M{
"$push": bson.M{
"couples": bson.M{
"anchorTimeMs": couple.AnchorTimeMs,
"songID": couple.SongID,
},
},
}
opts := options.Update().SetUpsert(true)
_, err := collection.UpdateOne(context.Background(), filter, update, opts)
if err != nil {
if err == mongo.ErrNoDocuments {
// If address doesn't exist, insert a new document
doc := bson.M{
"_id": address,
"couples": []interface{}{
bson.M{
"anchorTimeMs": couple.AnchorTimeMs,
"songID": couple.SongID,
},
},
}
_, err := collection.InsertOne(context.Background(), doc)
if err != nil {
return fmt.Errorf("error inserting document: %s", err)
}
} else {
return fmt.Errorf("error checking if document exists: %s", err)
}
} else {
// If address exists, append the new couple to the existing couples list
_, err := collection.UpdateOne(
context.Background(),
bson.M{"_id": address},
bson.M{"$push": bson.M{"couples": bson.M{"anchorTimeMs": couple.AnchorTimeMs, "songID": couple.SongID}}},
)
if err != nil {
return fmt.Errorf("error updating document: %s", err)
}
return fmt.Errorf("error upserting document: %s", err)
}
}