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

Rename 'table' to 'couple'

Browse source
This commit is contained in:
Chigozirim Igweamaka 2024-05-15 05:05:34 +01:00
parent ffe26e5c87
commit e4d6a099eb
1 changed files with 29 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

44
utils/dbClient.go
View file

@ -13,7 +13,17 @@ import (
"go.mongodb.org/mongo-driver/mongo/options" "go.mongodb.org/mongo-driver/mongo/options"
) )
const dbUri string = "mongodb://localhost:27017" // godotenv.Load(".env")
var (
dbUsername = GetEnv("DB_USER")
dbPassword = GetEnv("DB_PASS")
dbName = GetEnv("DB_NAME")
dbHost = GetEnv("DB_HOST")
dbPort = GetEnv("DB_PORT")
dbUri = "mongodb://" + dbUsername + ":" + dbPassword + "@" + dbHost + ":" + dbPort + "/" + dbName
)
// DbClient represents a MongoDB client // DbClient represents a MongoDB client
type DbClient struct { type DbClient struct {
@ -22,6 +32,10 @@ type DbClient struct {
// NewDbClient creates a new instance of DbClient // NewDbClient creates a new instance of DbClient
func NewDbClient() (*DbClient, error) { func NewDbClient() (*DbClient, error) {
if dbUsername == "" || dbPassword == "" {
dbUri = "mongodb://localhost:27017"
}
clientOptions := options.Client().ApplyURI(dbUri) clientOptions := options.Client().ApplyURI(dbUri)
client, err := mongo.Connect(context.Background(), clientOptions) client, err := mongo.Connect(context.Background(), clientOptions)
if err != nil { if err != nil {
@ -38,10 +52,10 @@ func (db *DbClient) Close() error {
return nil return nil
} }
func (db *DbClient) StoreFingerprints(fingerprints map[uint32]models.Table) error { func (db *DbClient) StoreFingerprints(fingerprints map[uint32]models.Couple) error {
collection := db.client.Database("song-recognition").Collection("fingerprints") collection := db.client.Database("song-recognition").Collection("fingerprints")
for address, table := range fingerprints { for address, couple := range fingerprints {
// Check if the address already exists in the database // Check if the address already exists in the database
var existingDoc bson.M var existingDoc bson.M
err := collection.FindOne(context.Background(), bson.M{"_id": address}).Decode(&existingDoc) err := collection.FindOne(context.Background(), bson.M{"_id": address}).Decode(&existingDoc)
@ -50,10 +64,10 @@ func (db *DbClient) StoreFingerprints(fingerprints map[uint32]models.Table) erro
// If address doesn't exist, insert a new document // If address doesn't exist, insert a new document
doc := bson.M{ doc := bson.M{
"_id": address, "_id": address,
"tables": []interface{}{ "couples": []interface{}{
bson.M{ bson.M{
"anchorTimeMs": table.AnchorTimeMs, "anchorTimeMs": couple.AnchorTimeMs,
"songID": table.SongID, "songID": couple.SongID,
}, },
}, },
} }
@ -66,12 +80,12 @@ func (db *DbClient) StoreFingerprints(fingerprints map[uint32]models.Table) erro
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 table to the existing tables list // If address exists, append the new couple to the existing tables list
_, err := collection.UpdateOne( _, err := collection.UpdateOne(
context.Background(), context.Background(),
bson.M{"_id": address}, bson.M{"_id": address},
bson.M{"$push": bson.M{"tables": bson.M{"anchorTimeMs": table.AnchorTimeMs, "songID": table.SongID}}}, bson.M{"$push": bson.M{"couples": bson.M{"anchorTimeMs": couple.AnchorTimeMs, "songID": couple.SongID}}},
) )
if err != nil { if err != nil {
return fmt.Errorf("error updating document: %s", err) return fmt.Errorf("error updating document: %s", err)
@ -82,10 +96,10 @@ func (db *DbClient) StoreFingerprints(fingerprints map[uint32]models.Table) erro
return nil return nil
} }
func (db *DbClient) GetTables(addresses []uint32) (map[uint32][]models.Table, 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.Table) tables := 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
@ -99,7 +113,7 @@ func (db *DbClient) GetTables(addresses []uint32) (map[uint32][]models.Table, er
} }
// Extract tables from the document and append them to the tables map // Extract tables from the document and append them to the tables map
var docTables []models.Table var docCouples []models.Couple
tableArray, ok := result["tables"].(primitive.A) tableArray, ok := result["tables"].(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("tables field in document for address %d is not valid", address)
@ -108,16 +122,16 @@ func (db *DbClient) GetTables(addresses []uint32) (map[uint32][]models.Table, er
for _, item := range tableArray { for _, item := range tableArray {
itemMap, ok := item.(primitive.M) itemMap, ok := item.(primitive.M)
if !ok { if !ok {
return nil, fmt.Errorf("invalid table format in document for address %d", address) return nil, fmt.Errorf("invalid couple format in document for address %d", address)
} }
table := models.Table{ couple := models.Couple{
AnchorTimeMs: uint32(itemMap["anchorTimeMs"].(int64)), AnchorTimeMs: uint32(itemMap["anchorTimeMs"].(int64)),
SongID: uint32(itemMap["songID"].(int64)), SongID: uint32(itemMap["songID"].(int64)),
} }
docTables = append(docTables, table) docCouples = append(docCouples, couple)
} }
tables[address] = docTables tables[address] = docCouples
} }
return tables, nil return tables, nil