Add caching_delay flag to cgr-loader

This commit is contained in:
arberkatellari
2023-12-22 10:20:31 -05:00
committed by Dan Christian Bogos
parent a5361485d9
commit d5d873b851
8 changed files with 31 additions and 4 deletions

View File

@@ -25,6 +25,7 @@ import (
"log"
"os"
"strings"
"time"
"github.com/cgrates/cgrates/config"
"github.com/cgrates/cgrates/engine"
@@ -96,7 +97,8 @@ var (
cachingArg = cgrLoaderFlags.String(utils.CachingArgCgr, utils.EmptyString,
"Caching strategy used when loading TP")
tpid = cgrLoaderFlags.String(utils.TpIDCfg, dfltCfg.LoaderCgrCfg().TpID,
cachingDlay = cgrLoaderFlags.Duration(utils.CachingDlayCfg, 0, "Adds delay before cache reload")
tpid = cgrLoaderFlags.String(utils.TpIDCfg, dfltCfg.LoaderCgrCfg().TpID,
"The tariff plan ID from the database")
dataPath = cgrLoaderFlags.String(utils.PathCfg, dfltCfg.LoaderCgrCfg().DataPath,
"The path to folder containing the data files")
@@ -294,6 +296,9 @@ func loadConfig() (ldrCfg *config.CGRConfig) {
if *cachingArg != utils.EmptyString {
ldrCfg.GeneralCfg().DefaultCaching = *cachingArg
}
if *cachingDlay != 0 {
ldrCfg.GeneralCfg().CachingDelay = *cachingDlay
}
return
}
@@ -409,6 +414,14 @@ func main() {
}
}
// delay if needed before cache reload
if ldrCfg.GeneralCfg().CachingDelay != 0 {
if *verbose {
log.Printf("Delaying %v", ldrCfg.GeneralCfg().CachingDelay)
}
time.Sleep(ldrCfg.GeneralCfg().CachingDelay)
}
// reload cache
if err = tpReader.ReloadCache(ldrCfg.GeneralCfg().DefaultCaching, *verbose, map[string]any{
utils.OptsAPIKey: *apiKey,

View File

@@ -42,6 +42,7 @@ const CGRATES_CFG_JSON = `
"default_tenant": "cgrates.org", // default tenant to consider when missing from requests
"default_timezone": "Local", // default timezone for timestamps where not specified <""|UTC|Local|$IANA_TZ_DB>
"default_caching": "*reload", // default actions to do when caching items
"caching_delay": "0s", // add delay before cache reload
"min_call_duration": "0s", // only authorize calls with allowed duration higher than this
"max_call_duration": "3h", // maximum call duration a prepaid call can last
"connect_attempts": 5, // initial server connect attempts

View File

@@ -42,6 +42,7 @@ func TestDfGeneralJsonCfg(t *testing.T) {
Default_category: utils.StringPointer("call"),
Default_tenant: utils.StringPointer("cgrates.org"),
Default_caching: utils.StringPointer(utils.MetaReload),
Caching_delay: utils.StringPointer("0s"),
Default_timezone: utils.StringPointer("Local"),
Connect_attempts: utils.IntPointer(5),
Reconnects: utils.IntPointer(-1),

File diff suppressed because one or more lines are too long

View File

@@ -41,6 +41,7 @@ type GeneralCfg struct {
DefaultTenant string // set default tenant
DefaultTimezone string // default timezone for timestamps where not specified <""|UTC|Local|$IANA_TZ_DB>
DefaultCaching string
CachingDelay time.Duration // use to add delay before cache reload
ConnectAttempts int // number of initial connection attempts before giving up
Reconnects int // number of reconnect attempts in case of connection lost <-1 for infinite | nb>
MaxReconnectInterval time.Duration // time to wait in between reconnect attempts
@@ -124,6 +125,11 @@ func (gencfg *GeneralCfg) loadFromJSONCfg(jsnGeneralCfg *GeneralJsonCfg) (err er
if jsnGeneralCfg.Default_caching != nil {
gencfg.DefaultCaching = *jsnGeneralCfg.Default_caching
}
if jsnGeneralCfg.Caching_delay != nil {
if gencfg.CachingDelay, err = utils.ParseDurationWithNanosecs(*jsnGeneralCfg.Caching_delay); err != nil {
return err
}
}
if jsnGeneralCfg.Locking_timeout != nil {
if gencfg.LockingTimeout, err = utils.ParseDurationWithNanosecs(*jsnGeneralCfg.Locking_timeout); err != nil {
return err
@@ -161,6 +167,7 @@ func (gencfg *GeneralCfg) AsMapInterface() (initialMP map[string]any) {
utils.DefaultTenantCfg: gencfg.DefaultTenant,
utils.DefaultTimezoneCfg: gencfg.DefaultTimezone,
utils.DefaultCachingCfg: gencfg.DefaultCaching,
utils.CachingDlayCfg: gencfg.CachingDelay,
utils.ConnectAttemptsCfg: gencfg.ConnectAttempts,
utils.ReconnectsCfg: gencfg.Reconnects,
utils.MaxReconnectIntervalCfg: "0",

View File

@@ -165,6 +165,7 @@ func TestGeneralCfgAsMapInterface(t *testing.T) {
utils.DefaultTenantCfg: "cgrates.org",
utils.DefaultTimezoneCfg: "Local",
utils.DefaultCachingCfg: "*reload",
utils.CachingDlayCfg: 0 * time.Second,
utils.ConnectAttemptsCfg: 5,
utils.ReconnectsCfg: -1,
utils.MaxReconnectIntervalCfg: "1s",
@@ -209,6 +210,7 @@ func TestGeneralCfgAsMapInterface1(t *testing.T) {
utils.DefaultTenantCfg: "cgrates.org",
utils.DefaultTimezoneCfg: "Local",
utils.DefaultCachingCfg: "*reload",
utils.CachingDlayCfg: 0 * time.Second,
utils.ConnectAttemptsCfg: 5,
utils.ReconnectsCfg: -1,
utils.MaxReconnectIntervalCfg: "0",

View File

@@ -38,6 +38,7 @@ type GeneralJsonCfg struct {
Default_tenant *string
Default_timezone *string
Default_caching *string
Caching_delay *string
Connect_attempts *int
Reconnects *int
Max_reconnect_interval *string

View File

@@ -1978,6 +1978,7 @@ const (
DefaultTenantCfg = "default_tenant"
DefaultTimezoneCfg = "default_timezone"
DefaultCachingCfg = "default_caching"
CachingDlayCfg = "caching_delay"
ConnectAttemptsCfg = "connect_attempts"
ReconnectsCfg = "reconnects"
MaxReconnectIntervalCfg = "max_reconnect_interval"