Add redis driver

Signed-off-by: Asai Neko <sugar@sne.moe>
This commit is contained in:
2025-12-25 14:51:45 +08:00
parent 9e51414a13
commit 32a27d974a
9 changed files with 90 additions and 4 deletions

View File

@@ -3,11 +3,13 @@ package data
import (
"nixcn-cms/data/drivers"
"github.com/redis/go-redis/v9"
log "github.com/sirupsen/logrus"
"github.com/spf13/viper"
)
var Database *drivers.DBClient
var Redis *redis.UniversalClient
func Init() {
// Init database
@@ -34,6 +36,20 @@ func Init() {
if err != nil {
log.Error("[Database] Error migrating database: ", err)
}
Database = db
// Init redis conection
rdbAddress := viper.GetStringSlice("cache.hosts")
dsn := drivers.RedisDSN{
Hosts: rdbAddress,
Master: viper.GetString("cache.master"),
Username: viper.GetString("cache.username"),
Password: viper.GetString("cache.password"),
DB: viper.GetInt("cache.db"),
}
rdb, err := drivers.Redis(dsn)
if err != nil {
log.Fatal("[Redis] Error connecting to Redis: ", err)
}
Redis = rdb
}