60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
|
package helpers
|
||
|
|
|
||
|
|
import (
|
||
|
|
"code.lowsec.club/okawari/go-discord-bot/lib/models"
|
||
|
|
"context"
|
||
|
|
"github.com/rs/zerolog/log"
|
||
|
|
"gorm.io/driver/postgres"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
"os"
|
||
|
|
)
|
||
|
|
|
||
|
|
func CreateConnection(config *gorm.Config) (*gorm.DB, error) {
|
||
|
|
log.Debug().Msg("Connecting to DB")
|
||
|
|
return gorm.Open(postgres.Open(os.Getenv("DATABASE_URL")), config)
|
||
|
|
}
|
||
|
|
|
||
|
|
func AddDb(ctx context.Context, db *gorm.DB) context.Context {
|
||
|
|
return context.WithValue(ctx, "db", db)
|
||
|
|
}
|
||
|
|
|
||
|
|
func DB(ctx context.Context) *gorm.DB {
|
||
|
|
if val, ok := ctx.Value("db").(*gorm.DB); ok {
|
||
|
|
return val
|
||
|
|
}
|
||
|
|
log.Fatal().Msg("Could not find DB object")
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func AutoMigrate(db *gorm.DB) {
|
||
|
|
hasError := false
|
||
|
|
|
||
|
|
logger := log.With().Str("module", "sql-migrator").Logger()
|
||
|
|
|
||
|
|
for _, model := range []any{
|
||
|
|
models.Commodity{},
|
||
|
|
} {
|
||
|
|
err := db.AutoMigrate(model)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
logger.Error().Err(err).Interface("model", model).Msg("Failed to auto migrate")
|
||
|
|
hasError = true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if hasError {
|
||
|
|
log.Panic().Msg("Exiting")
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func FindCommodity(db *gorm.DB, term string) (*models.Commodity, bool, error) {
|
||
|
|
result := models.Commodity{}
|
||
|
|
tx := db.Model(models.Commodity{}).Where("ticker = ?", term).First(&result)
|
||
|
|
|
||
|
|
if tx.Error != nil {
|
||
|
|
return nil, false, tx.Error
|
||
|
|
}
|
||
|
|
return &result, tx.RowsAffected > 0, nil
|
||
|
|
}
|