public/grafana
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(unified-storage): check resource_api cfg for isHA (#102283)

Browse Source
This commit is contained in:
Jean-Philippe Quéméner
2025-03-18 12:33:27 +01:00
committed by GitHub
parent 321a886b8b
commit 08994304d8
2 changed files with 110 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/storage/unified/sql/server.go
+9 -2
View File
@@ -48,7 +48,8 @@ func NewResourceServer(db infraDB.DB, cfg *setting.Cfg,
return nil, err
}
isHA := isHighAvailabilityEnabled(cfg.SectionWithEnvOverrides("database"))
isHA := isHighAvailabilityEnabled(cfg.SectionWithEnvOverrides("database"),
cfg.SectionWithEnvOverrides("resource_api"))
withPruner := features.IsEnabledGlobally(featuremgmt.FlagUnifiedStorageHistoryPruner)
store, err := NewBackend(BackendOptions{
@@ -79,7 +80,13 @@ func NewResourceServer(db infraDB.DB, cfg *setting.Cfg,
// isHighAvailabilityEnabled determines if high availability mode should
// be enabled based on database configuration. High availability is enabled
// by default except for SQLite databases.
func isHighAvailabilityEnabled(dbCfg *setting.DynamicSection) bool {
func isHighAvailabilityEnabled(dbCfg, resourceAPICfg *setting.DynamicSection) bool {
// If the resource API is using a non-SQLite database, we assume it's in HA mode.
resourceDBType := resourceAPICfg.Key("db_type").String()
if resourceDBType != "" && resourceDBType != migrator.SQLite {
return true
}
// Check in the config if HA is enabled - by default we always assume a HA setup.
isHA := dbCfg.Key("high_availability").MustBool(true)
pkg/storage/unified/sql/server_test.go
+101 -59
View File
@@ -1,7 +1,6 @@
package sql
import (
"strconv"
"testing"
"github.com/grafana/grafana/pkg/services/sqlstore/migrator"
@@ -11,90 +10,133 @@ import (
func TestIsHighAvailabilityEnabled(t *testing.T) {
tests := []struct {
name string
dbType string
haConfigValue *bool
isHA bool
name string
cfg *setting.Cfg
isHA bool
}{
{
name: "SQLite should never have HA enabled",
dbType: migrator.SQLite,
haConfigValue: boolPtr(true),
isHA: false,
name: "SQLite should never have HA enabled",
cfg: func() *setting.Cfg {
cfg := setting.NewCfg()
dbSection := cfg.SectionWithEnvOverrides("database")
dbSection.Key("type").SetValue(migrator.SQLite)
dbSection.Key("high_availability").SetValue("true")
return cfg
}(),
isHA: false,
},
{
name: "MySQL with HA enabled in config should default to true",
dbType: migrator.MySQL,
haConfigValue: boolPtr(true),
isHA: true,
name: "MySQL with HA enabled in config should default to true",
cfg: func() *setting.Cfg {
cfg := setting.NewCfg()
dbSection := cfg.SectionWithEnvOverrides("database")
dbSection.Key("type").SetValue(migrator.MySQL)
dbSection.Key("high_availability").SetValue("true")
return cfg
}(),
isHA: true,
},
{
name: "MySQL with HA disabled in config should default to false",
dbType: migrator.MySQL,
haConfigValue: boolPtr(false),
isHA: false,
name: "MySQL with HA disabled in config should default to false",
cfg: func() *setting.Cfg {
cfg := setting.NewCfg()
dbSection := cfg.SectionWithEnvOverrides("database")
dbSection.Key("type").SetValue(migrator.MySQL)
dbSection.Key("high_availability").SetValue("false")
return cfg
}(),
isHA: false,
},
{
name: "MySQL with no HA config should default to true",
dbType: migrator.MySQL,
haConfigValue: nil,
isHA: true,
name: "MySQL with no HA config should default to true",
cfg: func() *setting.Cfg {
cfg := setting.NewCfg()
dbSection := cfg.SectionWithEnvOverrides("database")
dbSection.Key("type").SetValue(migrator.MySQL)
return cfg
}(),
isHA: true,
},
{
name: "Postgres with HA enabled in config should default to true",
dbType: migrator.Postgres,
haConfigValue: boolPtr(true),
isHA: true,
name: "Postgres with HA enabled in config should default to true",
cfg: func() *setting.Cfg {
cfg := setting.NewCfg()
dbSection := cfg.SectionWithEnvOverrides("database")
dbSection.Key("type").SetValue(migrator.Postgres)
dbSection.Key("high_availability").SetValue("true")
return cfg
}(),
isHA: true,
},
{
name: "Postgres with HA disabled in config should default to false",
dbType: migrator.Postgres,
haConfigValue: boolPtr(false),
isHA: false,
name: "Postgres with HA disabled in config should default to false",
cfg: func() *setting.Cfg {
cfg := setting.NewCfg()
dbSection := cfg.SectionWithEnvOverrides("database")
dbSection.Key("type").SetValue(migrator.Postgres)
dbSection.Key("high_availability").SetValue("false")
return cfg
}(),
isHA: false,
},
{
name: "Postgres with no HA config should default to true",
dbType: migrator.Postgres,
haConfigValue: nil,
isHA: true,
name: "Postgres with no HA config should default to true",
cfg: func() *setting.Cfg {
cfg := setting.NewCfg()
dbSection := cfg.SectionWithEnvOverrides("database")
dbSection.Key("type").SetValue(migrator.Postgres)
return cfg
}(),
isHA: true,
},
{
name: "No database type set should default to true",
dbType: "",
haConfigValue: nil,
isHA: true,
name: "No database type set should default to true",
cfg: func() *setting.Cfg {
cfg := setting.NewCfg()
_ = cfg.SectionWithEnvOverrides("database")
return cfg
}(),
isHA: true,
},
{
name: "No database type set with HA enabled in config should default to true",
dbType: "",
haConfigValue: boolPtr(true),
isHA: true,
name: "No database type set with HA enabled in config should default to true",
cfg: func() *setting.Cfg {
cfg := setting.NewCfg()
dbSection := cfg.SectionWithEnvOverrides("database")
dbSection.Key("high_availability").SetValue("true")
return cfg
}(),
isHA: true,
},
{
name: "No database type set with HA disabled in config should default to false",
dbType: "",
haConfigValue: boolPtr(false),
isHA: false,
name: "No database type set with HA disabled in config should default to false",
cfg: func() *setting.Cfg {
cfg := setting.NewCfg()
dbSection := cfg.SectionWithEnvOverrides("database")
dbSection.Key("high_availability").SetValue("false")
return cfg
}(),
isHA: false,
},
{
name: "Resource API with non-SQLite database type should default to true",
cfg: func() *setting.Cfg {
cfg := setting.NewCfg()
dbSection := cfg.SectionWithEnvOverrides("database")
dbSection.Key("type").SetValue(migrator.SQLite)
resourceAPISection := cfg.SectionWithEnvOverrides("resource_api")
resourceAPISection.Key("db_type").SetValue(migrator.Postgres)
return cfg
}(),
isHA: true,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
cfg := setting.NewCfg().SectionWithEnvOverrides("database")
if tt.dbType != "" {
cfg.Key("type").SetValue(tt.dbType)
}
if tt.haConfigValue != nil {
cfg.Key("high_availability").SetValue(strconv.FormatBool(*tt.haConfigValue))
}
result := isHighAvailabilityEnabled(cfg)
result := isHighAvailabilityEnabled(tt.cfg.SectionWithEnvOverrides("database"),
tt.cfg.SectionWithEnvOverrides("resource_api"))
require.Equal(t, tt.isHA, result)
})
}
}
func boolPtr(b bool) *bool {
return &b
}