wings/internal/database/database.go

62 lines
1.5 KiB
Go
Raw Normal View History

2022-07-24 15:43:48 +00:00
package database
import (
2022-10-05 23:24:11 +00:00
"path/filepath"
"time"
2022-07-24 15:43:48 +00:00
"emperror.dev/errors"
"github.com/glebarez/sqlite"
2022-10-06 15:58:42 +00:00
"gorm.io/gorm"
"gorm.io/gorm/logger"
2022-07-24 15:43:48 +00:00
"github.com/pterodactyl/wings/config"
"github.com/pterodactyl/wings/internal/models"
"github.com/pterodactyl/wings/system"
)
2022-10-05 23:24:11 +00:00
var (
o system.AtomicBool
db *gorm.DB
)
2022-07-24 15:43:48 +00:00
// Initialize configures the local SQLite database for Wings and ensures that the models have
// been fully migrated.
func Initialize() error {
if !o.SwapIf(true) {
panic("database: attempt to initialize more than once during application lifecycle")
}
p := filepath.Join(config.Get().System.RootDirectory, "wings.db")
instance, err := gorm.Open(sqlite.Open(p), &gorm.Config{
Logger: logger.Default.LogMode(logger.Silent),
})
2022-07-24 15:43:48 +00:00
if err != nil {
return errors.Wrap(err, "database: could not open database file")
}
db = instance
if sql, err := db.DB(); err != nil {
return errors.WithStack(err)
} else {
sql.SetMaxOpenConns(1)
sql.SetConnMaxLifetime(time.Hour)
}
if tx := db.Exec("PRAGMA synchronous = OFF"); tx.Error != nil {
return errors.WithStack(tx.Error)
}
if tx := db.Exec("PRAGMA journal_mode = MEMORY"); tx.Error != nil {
return errors.WithStack(tx.Error)
}
2022-07-24 15:43:48 +00:00
if err := db.AutoMigrate(&models.Activity{}); err != nil {
return errors.WithStack(err)
}
return nil
}
// Instance returns the gorm database instance that was configured when the application was
// booted.
func Instance() *gorm.DB {
if db == nil {
panic("database: attempt to access instance before initialized")
}
return db
}