Refactor "Token" to "Character" which is what it actually is
This commit is contained in:
22
db.go
22
db.go
@@ -12,8 +12,8 @@ import (
|
||||
type DB interface {
|
||||
DB() *gorm.DB
|
||||
Raw(sql string, args ...any) *gorm.DB
|
||||
GetTokenForCharacter(characterName string) (*Token, error)
|
||||
SaveTokenForCharacter(token *Token) error
|
||||
GetCharacterByName(characterName string) (*Character, error)
|
||||
SaveCharacter(character *Character) error
|
||||
AutoMigrate(dst ...interface{}) error
|
||||
}
|
||||
|
||||
@@ -55,25 +55,25 @@ func (db *DBWrapper) DB() *gorm.DB {
|
||||
return db.db
|
||||
}
|
||||
|
||||
func (db *DBWrapper) GetTokenForCharacter(characterName string) (*Token, error) {
|
||||
func (db *DBWrapper) GetCharacterByName(characterName string) (*Character, error) {
|
||||
logger.Debug("Fetching token for character %s from database", characterName)
|
||||
var token Token
|
||||
err := db.db.Where("character_name = ?", characterName).First(&token).Error
|
||||
var char Character
|
||||
err := db.db.Where("character_name = ?", characterName).First(&char).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
|
||||
logger.Debug("Token found for character %s, expires at %v", characterName, char.ExpiresAt)
|
||||
return &char, nil
|
||||
}
|
||||
|
||||
func (db *DBWrapper) SaveTokenForCharacter(token *Token) error {
|
||||
logger.Debug("Saving token for character %s to database", token.CharacterName)
|
||||
err := db.db.Save(token).Error
|
||||
func (db *DBWrapper) SaveCharacter(character *Character) error {
|
||||
logger.Debug("Saving token for character %s to database", character.CharacterName)
|
||||
err := db.db.Save(character).Error
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
logger.Debug("Token saved successfully for character %s", token.CharacterName)
|
||||
logger.Debug("Token saved successfully for character %s", character.CharacterName)
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user