mirror of
https://github.com/cgzirim/seek-tune.git
synced 2025-12-17 17:04:22 +00:00
54 lines
1.4 KiB
Go
54 lines
1.4 KiB
Go
package db
|
|
|
|
import (
|
|
"fmt"
|
|
"song-recognition/models"
|
|
"song-recognition/utils"
|
|
)
|
|
|
|
type DBClient interface {
|
|
Close() error
|
|
StoreFingerprints(fingerprints map[uint32]models.Couple) error
|
|
GetCouples(addresses []uint32) (map[uint32][]models.Couple, error)
|
|
TotalSongs() (int, error)
|
|
RegisterSong(songTitle, songArtist, ytID string) (uint32, error)
|
|
GetSong(filterKey string, value interface{}) (Song, bool, error)
|
|
GetSongByID(songID uint32) (Song, bool, error)
|
|
GetSongByYTID(ytID string) (Song, bool, error)
|
|
GetSongByKey(key string) (Song, bool, error)
|
|
DeleteSongByID(songID uint32) error
|
|
DeleteCollection(collectionName string) error
|
|
}
|
|
|
|
type Song struct {
|
|
Title string
|
|
Artist string
|
|
YouTubeID string
|
|
}
|
|
|
|
var DBtype = utils.GetEnv("DB_TYPE", "sqlite") // Can be "sqlite" or "mongo"
|
|
|
|
func NewDBClient() (DBClient, error) {
|
|
switch DBtype {
|
|
case "mongo":
|
|
var (
|
|
dbUsername = utils.GetEnv("DB_USER")
|
|
dbPassword = utils.GetEnv("DB_PASS")
|
|
dbName = utils.GetEnv("DB_NAME")
|
|
dbHost = utils.GetEnv("DB_HOST")
|
|
dbPort = utils.GetEnv("DB_PORT")
|
|
|
|
dbUri = "mongodb://" + dbUsername + ":" + dbPassword + "@" + dbHost + ":" + dbPort + "/" + dbName
|
|
)
|
|
if dbUsername == "" || dbPassword == "" {
|
|
dbUri = "mongodb://localhost:27017"
|
|
}
|
|
return NewMongoClient(dbUri)
|
|
|
|
case "sqlite":
|
|
return NewSQLiteClient("db/db.sqlite3")
|
|
|
|
default:
|
|
return nil, fmt.Errorf("unsupported database type: %s", DBtype)
|
|
}
|
|
}
|