130 lines
2.6 KiB
Go
130 lines
2.6 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/redis/go-redis/v9"
|
|
|
|
"go.uber.org/zap"
|
|
|
|
"git.lumeweb.com/LumeWeb/portal/config"
|
|
|
|
"git.lumeweb.com/LumeWeb/portal/db/models"
|
|
"github.com/go-gorm/caches/v4"
|
|
"go.uber.org/fx"
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type DatabaseParams struct {
|
|
fx.In
|
|
Config *config.Manager
|
|
Logger *zap.Logger
|
|
LoggerLevel *zap.AtomicLevel
|
|
}
|
|
|
|
var Module = fx.Module("db",
|
|
fx.Options(
|
|
fx.Provide(NewDatabase),
|
|
),
|
|
)
|
|
|
|
func NewDatabase(lc fx.Lifecycle, params DatabaseParams) *gorm.DB {
|
|
username := params.Config.Config().Core.DB.Username
|
|
password := params.Config.Config().Core.DB.Password
|
|
host := params.Config.Config().Core.DB.Host
|
|
port := params.Config.Config().Core.DB.Port
|
|
dbname := params.Config.Config().Core.DB.Name
|
|
charset := params.Config.Config().Core.DB.Charset
|
|
|
|
dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=%s&parseTime=True&loc=Local", username, password, host, port, dbname, charset)
|
|
|
|
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
|
|
Logger: newLogger(params.Logger, params.LoggerLevel),
|
|
})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
cacher := getCacher(params.Config, params.Logger)
|
|
if cacher != nil {
|
|
cache := &caches.Caches{Conf: &caches.Config{
|
|
Cacher: cacher,
|
|
}}
|
|
err := db.Use(cache)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
}
|
|
|
|
lc.Append(fx.Hook{
|
|
OnStart: func(ctx context.Context) error {
|
|
return db.AutoMigrate(
|
|
&models.APIKey{},
|
|
&models.Blocklist{},
|
|
&models.DNSLink{},
|
|
&models.Download{},
|
|
&models.EmailVerification{},
|
|
&models.PasswordReset{},
|
|
&models.Pin{},
|
|
&models.PublicKey{},
|
|
&models.Upload{},
|
|
&models.User{},
|
|
&models.S5Challenge{},
|
|
&models.TusLock{},
|
|
&models.TusUpload{},
|
|
)
|
|
},
|
|
})
|
|
|
|
return db
|
|
}
|
|
|
|
func getCacheMode(cm *config.Manager, logger *zap.Logger) string {
|
|
|
|
if cm.Config().Core.DB.Cache == nil {
|
|
return "none"
|
|
}
|
|
|
|
switch cm.Config().Core.DB.Cache.Mode {
|
|
case "", "none":
|
|
return "none"
|
|
case "memory":
|
|
return "memory"
|
|
case "redis":
|
|
return "redis"
|
|
default:
|
|
logger.Fatal("invalid cache mode", zap.String("mode", cm.Config().Core.DB.Cache.Mode))
|
|
}
|
|
|
|
return "none"
|
|
}
|
|
|
|
func getCacher(cm *config.Manager, logger *zap.Logger) caches.Cacher {
|
|
mode := getCacheMode(cm, logger)
|
|
|
|
switch mode {
|
|
case "none":
|
|
return nil
|
|
|
|
case "memory":
|
|
return &memoryCacher{}
|
|
case "redis":
|
|
rcfg, ok := cm.Config().Core.DB.Cache.Options.(config.RedisConfig)
|
|
if !ok {
|
|
logger.Fatal("invalid redis config")
|
|
return nil
|
|
}
|
|
return &redisCacher{
|
|
redis.NewClient(&redis.Options{
|
|
Addr: rcfg.Address,
|
|
Password: rcfg.Password,
|
|
DB: rcfg.DB,
|
|
}),
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|