Hallucinate hella logs
This commit is contained in:
27
db.go
27
db.go
@@ -5,6 +5,8 @@ import (
|
||||
|
||||
"gorm.io/driver/sqlite"
|
||||
"gorm.io/gorm"
|
||||
|
||||
logger "git.site.quack-lab.dev/dave/cylogger"
|
||||
)
|
||||
|
||||
type DB interface {
|
||||
@@ -25,9 +27,11 @@ func GetDB() (DB, error) {
|
||||
var err error
|
||||
|
||||
if db != nil {
|
||||
logger.Debug("Returning existing database connection")
|
||||
return db, nil
|
||||
}
|
||||
|
||||
logger.Debug("Creating new database connection to %s", options.DBPath)
|
||||
dbFile := filepath.Join(options.DBPath)
|
||||
db, err := gorm.Open(sqlite.Open(dbFile), &gorm.Config{
|
||||
// SkipDefaultTransaction: true,
|
||||
@@ -35,9 +39,11 @@ func GetDB() (DB, error) {
|
||||
// Logger: gormlogger.Default.LogMode(gormlogger.Silent),
|
||||
})
|
||||
if err != nil {
|
||||
logger.Error("Failed to open database: %v", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
logger.Info("Database connection established successfully")
|
||||
return &DBWrapper{db: db}, nil
|
||||
}
|
||||
|
||||
@@ -51,18 +57,35 @@ func (db *DBWrapper) DB() *gorm.DB {
|
||||
}
|
||||
|
||||
func (db *DBWrapper) GetTokenForCharacter(characterName string) (*Token, error) {
|
||||
logger.Debug("Fetching token for character %s from database", characterName)
|
||||
var token Token
|
||||
err := db.db.Where("character_name = ?", characterName).First(&token).Error
|
||||
if err != nil {
|
||||
logger.Debug("No token found for character %s: %v", characterName, err)
|
||||
return nil, err
|
||||
}
|
||||
logger.Debug("Token found for character %s, expires at %v", characterName, token.ExpiresAt)
|
||||
return &token, nil
|
||||
}
|
||||
|
||||
func (db *DBWrapper) SaveTokenForCharacter(token *Token) error {
|
||||
return db.db.Save(token).Error
|
||||
logger.Debug("Saving token for character %s to database", token.CharacterName)
|
||||
err := db.db.Save(token).Error
|
||||
if err != nil {
|
||||
logger.Error("Failed to save token for character %s: %v", token.CharacterName, err)
|
||||
return err
|
||||
}
|
||||
logger.Debug("Token saved successfully for character %s", token.CharacterName)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (db *DBWrapper) AutoMigrate(dst ...interface{}) error {
|
||||
return db.db.AutoMigrate(dst...)
|
||||
logger.Debug("Running database auto-migration")
|
||||
err := db.db.AutoMigrate(dst...)
|
||||
if err != nil {
|
||||
logger.Error("Database auto-migration failed: %v", err)
|
||||
return err
|
||||
}
|
||||
logger.Debug("Database auto-migration completed successfully")
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user