sendnrw/netbird
2
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2026-04-20 09:16:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

use gorm cache

Browse Source
This commit is contained in:
Pascal Fischer
2025-10-13 16:29:04 +02:00
parent 0d2e67983a
commit f0e8cd578d
15 changed files with 498 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
management/server/store/sql_store.go
View File

@@ -15,6 +15,8 @@ import (
"sync"
"time"
"github.com/go-gorm/caches/v4"
"github.com/redis/go-redis/v9"
log "github.com/sirupsen/logrus"
"gorm.io/driver/mysql"
"gorm.io/driver/postgres"
@@ -30,6 +32,7 @@ import (
networkTypes "github.com/netbirdio/netbird/management/server/networks/types"
nbpeer "github.com/netbirdio/netbird/management/server/peer"
"github.com/netbirdio/netbird/management/server/posture"
"github.com/netbirdio/netbird/management/server/store/cache"
"github.com/netbirdio/netbird/management/server/telemetry"
"github.com/netbirdio/netbird/management/server/types"
"github.com/netbirdio/netbird/management/server/util"
@@ -46,11 +49,15 @@ const (
accountAndIDsQueryCondition = "account_id = ? AND id IN ?"
accountIDCondition = "account_id = ?"
peerNotFoundFMT = "peer %s not found"
storeCacheEnabledEnv = "NB_STORE_CACHE_ENABLED"
storeCacheRedisAddrEnv = "NB_STORE_CACHE_REDIS_ADDR"
)
// SqlStore represents an account storage backed by a Sql DB persisted to disk
type SqlStore struct {
db *gorm.DB
uncachedDB *gorm.DB
globalAccountLock sync.Mutex
metrics telemetry.AppMetrics
installationPK int
@@ -66,6 +73,13 @@ type migrationFunc func(*gorm.DB) error
// NewSqlStore creates a new SqlStore instance.
func NewSqlStore(ctx context.Context, db *gorm.DB, storeEngine types.Engine, metrics telemetry.AppMetrics, skipMigration bool) (*SqlStore, error) {
if os.Getenv(storeCacheEnabledEnv) == "true" {
err := configureStoreCache(ctx, db)
if err != nil {
return nil, fmt.Errorf("failed to configure store cache: %w", err)
}
}
sql, err := db.DB()
if err != nil {
return nil, err
@@ -116,6 +130,26 @@ func NewSqlStore(ctx context.Context, db *gorm.DB, storeEngine types.Engine, met
return &SqlStore{db: db, storeEngine: storeEngine, metrics: metrics, installationPK: 1}, nil
}
func configureStoreCache(ctx context.Context, db *gorm.DB) error {
var cacher caches.Cacher = &cache.MemoryCacher{}
if addr := os.Getenv(storeCacheRedisAddrEnv); addr != "" {
opt, err := redis.ParseURL(addr)
if err != nil {
return fmt.Errorf("failed to parse redis url from %s: %w", addr, err)
}
cacher = cache.NewRedisCacher(redis.NewClient(opt))
log.WithContext(ctx).Infof("using redis store cache at %s", addr)
} else {
log.WithContext(ctx).Infof("using in-memory store cache")
}
cachesPlugin := &caches.Caches{Conf: &caches.Config{
Cacher: cacher,
}}
return db.Use(cachesPlugin)
}
func GetKeyQueryCondition(s *SqlStore) string {
if s.storeEngine == types.MysqlStoreEngine {
return mysqlKeyQueryCondition