Refactor character to types
This commit is contained in:
11
db/db.go
11
db/db.go
@@ -3,6 +3,7 @@ package db
|
||||
import (
|
||||
"go-eve-pi/esi"
|
||||
"go-eve-pi/options"
|
||||
"go-eve-pi/types"
|
||||
"path/filepath"
|
||||
|
||||
"gorm.io/driver/sqlite"
|
||||
@@ -14,8 +15,8 @@ import (
|
||||
type DB interface {
|
||||
DB() *gorm.DB
|
||||
Raw(sql string, args ...any) *gorm.DB
|
||||
GetCharacterByName(characterName string) (*esi.Character, error)
|
||||
SaveCharacter(character *esi.Character) error
|
||||
GetCharacterByName(characterName string) (*types.Character, error)
|
||||
SaveCharacter(character *types.Character) error
|
||||
AutoMigrate(dst ...interface{}) error
|
||||
GetCacheEntry(urlHash string) (*esi.CacheEntry, error)
|
||||
SaveCacheEntry(entry *esi.CacheEntry) error
|
||||
@@ -59,9 +60,9 @@ func (db *DBWrapper) DB() *gorm.DB {
|
||||
return db.db
|
||||
}
|
||||
|
||||
func (db *DBWrapper) GetCharacterByName(characterName string) (*esi.Character, error) {
|
||||
func (db *DBWrapper) GetCharacterByName(characterName string) (*types.Character, error) {
|
||||
logger.Debug("Fetching token for character %s from database", characterName)
|
||||
var char esi.Character
|
||||
var char types.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)
|
||||
@@ -71,7 +72,7 @@ func (db *DBWrapper) GetCharacterByName(characterName string) (*esi.Character, e
|
||||
return &char, nil
|
||||
}
|
||||
|
||||
func (db *DBWrapper) SaveCharacter(character *esi.Character) error {
|
||||
func (db *DBWrapper) SaveCharacter(character *types.Character) error {
|
||||
logger.Debug("Saving token for character %s to database", character.CharacterName)
|
||||
err := db.db.Save(character).Error
|
||||
if err != nil {
|
||||
|
||||
Reference in New Issue
Block a user