49 lines
1.5 KiB
Go
49 lines
1.5 KiB
Go
package gormstore
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"git.itzana.me/strafesnet/data-service/internal/datastore"
|
|
"git.itzana.me/strafesnet/data-service/internal/model"
|
|
"git.itzana.me/strafesnet/utils/logger"
|
|
"github.com/eko/gocache/lib/v4/cache"
|
|
gocache_store "github.com/eko/gocache/store/go_cache/v4"
|
|
gocache "github.com/patrickmn/go-cache"
|
|
log "github.com/sirupsen/logrus"
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func New(migrate bool) (datastore.Datastore, error) {
|
|
db, err := gorm.Open(postgres.Open(fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s", os.Getenv("PG_HOST"), os.Getenv("PG_USER"), os.Getenv("PG_PASS"), os.Getenv("PG_DB"), os.Getenv("PG_PORT"))), &gorm.Config{
|
|
Logger: logger.New()})
|
|
if err != nil {
|
|
log.WithFields(log.Fields{
|
|
"PG_USER": os.Getenv("PG_USER"),
|
|
"PG_HOST": os.Getenv("PG_HOST"),
|
|
"PG_PORT": os.Getenv("PG_PORT"),
|
|
"PG_DB": os.Getenv("PG_DB"),
|
|
"error": err,
|
|
}).Errorln("failed to connect to database")
|
|
return nil, err
|
|
}
|
|
|
|
sqlDB, err := db.DB()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
sqlDB.SetMaxIdleConns(10)
|
|
sqlDB.SetMaxOpenConns(25)
|
|
|
|
if migrate {
|
|
if err := db.AutoMigrate(&model.Time{}, &model.User{}, &model.Bot{}, &model.Map{}, &model.Event{}, &model.Server{}, &model.Transaction{}, &model.Rank{}, &model.RankCalc{}); err != nil {
|
|
log.WithField("error", err).Errorln("database migration failed")
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return &Gormstore{db, cache.New[[]byte](gocache_store.NewGoCache(gocache.New(5*time.Minute, 10*time.Minute)))}, nil
|
|
}
|