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

Save 'tables' as 'couples'

Browse source
This commit is contained in:
Chigozirim Igweamaka 2024-05-24 10:45:00 +01:00
parent caa628c33e
commit a6a86eb95c
1 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
utils/dbClient.go
View file

@ -80,7 +80,7 @@ func (db *DbClient) StoreFingerprints(fingerprints map[uint32]models.Couple) err
return fmt.Errorf("error checking if document exists: %s", err) return fmt.Errorf("error checking if document exists: %s", err)
} }
} else { } else {
// If address exists, append the new couple to the existing tables list // If address exists, append the new couple to the existing couples list
_, err := collection.UpdateOne( _, err := collection.UpdateOne(
context.Background(), context.Background(),
@ -99,7 +99,7 @@ func (db *DbClient) StoreFingerprints(fingerprints map[uint32]models.Couple) err
func (db *DbClient) GetCouples(addresses []uint32) (map[uint32][]models.Couple, error) { func (db *DbClient) GetCouples(addresses []uint32) (map[uint32][]models.Couple, error) {
collection := db.client.Database("song-recognition").Collection("fingerprints") collection := db.client.Database("song-recognition").Collection("fingerprints")
tables := make(map[uint32][]models.Couple) couples := make(map[uint32][]models.Couple)
for _, address := range addresses { for _, address := range addresses {
// Find the document corresponding to the address // Find the document corresponding to the address
@ -112,14 +112,14 @@ func (db *DbClient) GetCouples(addresses []uint32) (map[uint32][]models.Couple,
return nil, fmt.Errorf("error retrieving document for address %d: %s", address, err) return nil, fmt.Errorf("error retrieving document for address %d: %s", address, err)
} }
// Extract tables from the document and append them to the tables map // Extract couples from the document and append them to the couples map
var docCouples []models.Couple var docCouples []models.Couple
tableArray, ok := result["tables"].(primitive.A) couplesList, ok := result["couples"].(primitive.A)
if !ok { if !ok {
return nil, fmt.Errorf("tables field in document for address %d is not valid", address) return nil, fmt.Errorf("couples field in document for address %d is not valid", address)
} }
for _, item := range tableArray { for _, item := range couplesList {
itemMap, ok := item.(primitive.M) itemMap, ok := item.(primitive.M)
if !ok { if !ok {
return nil, fmt.Errorf("invalid couple format in document for address %d", address) return nil, fmt.Errorf("invalid couple format in document for address %d", address)
@ -131,10 +131,10 @@ func (db *DbClient) GetCouples(addresses []uint32) (map[uint32][]models.Couple,
} }
docCouples = append(docCouples, couple) docCouples = append(docCouples, couple)
} }
tables[address] = docCouples couples[address] = docCouples
} }
return tables, nil return couples, nil
} }
func (db *DbClient) TotalSongs() (int, error) { func (db *DbClient) TotalSongs() (int, error) {