cgrates/config/thresholdscfg.go

/*
Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
Copyright (C) ITsysCOM GmbH

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>
*/

package config

import (
	"slices"
	"time"

	"github.com/cgrates/cgrates/utils"
)

type ThresholdsOpts struct {
	ProfileIDs           []string
	ProfileIgnoreFilters bool
}

// ThresholdSCfg the threshold config section
type ThresholdSCfg struct {
	Enabled             bool
	IndexedSelects      bool
	StoreInterval       time.Duration // Dump regularly from cache into dataDB
	SessionSConns       []string
	ApierSConns         []string
	EEsExporterIDs      []string
	EEsConns            []string
	StringIndexedFields *[]string
	PrefixIndexedFields *[]string
	SuffixIndexedFields *[]string
	ExistsIndexedFields *[]string
	NestedFields        bool
	Opts                *ThresholdsOpts
}

func (thdOpts *ThresholdsOpts) loadFromJSONCfg(jsnCfg *ThresholdsOptsJson) {
	if jsnCfg == nil {
		return
	}
	if jsnCfg.ProfileIDs != nil {
		thdOpts.ProfileIDs = *jsnCfg.ProfileIDs
	}
	if jsnCfg.ProfileIgnoreFilters != nil {
		thdOpts.ProfileIgnoreFilters = *jsnCfg.ProfileIgnoreFilters
	}
}

func (t *ThresholdSCfg) loadFromJSONCfg(jsnCfg *ThresholdSJsonCfg) (err error) {
	if jsnCfg == nil {
		return nil
	}
	if jsnCfg.Enabled != nil {
		t.Enabled = *jsnCfg.Enabled
	}
	if jsnCfg.Indexed_selects != nil {
		t.IndexedSelects = *jsnCfg.Indexed_selects
	}
	if jsnCfg.Store_interval != nil {
		if t.StoreInterval, err = utils.ParseDurationWithNanosecs(*jsnCfg.Store_interval); err != nil {
			return err
		}
	}
	if len(jsnCfg.Sessions_conns) != 0 {
		t.SessionSConns = make([]string, len(jsnCfg.Sessions_conns))
		for idx, conn := range jsnCfg.Sessions_conns {
			// if we have the connection internal we change the name so we can have internal rpc for each subsystem
			t.SessionSConns[idx] = conn
			if conn == utils.MetaInternal {
				t.SessionSConns[idx] = utils.ConcatenatedKey(utils.MetaInternal, utils.MetaSessionS)
			}
		}
	}
	if len(jsnCfg.Apiers_conns) != 0 {
		t.ApierSConns = make([]string, len(jsnCfg.Apiers_conns))
		for idx, conn := range jsnCfg.Apiers_conns {
			// if we have the connection internal we change the name so we can have internal rpc for each subsystem
			t.ApierSConns[idx] = conn
			if conn == utils.MetaInternal {
				t.ApierSConns[idx] = utils.ConcatenatedKey(utils.MetaInternal, utils.MetaApier)
			}
		}
	}
	if jsnCfg.Ees_conns != nil {
		t.EEsConns = make([]string, len(*jsnCfg.Ees_conns))
		for idx, connID := range *jsnCfg.Ees_conns {
			// if we have the connection internal we change the name so we can have internal rpc for each subsystem
			t.EEsConns[idx] = connID
			if connID == utils.MetaInternal {
				t.EEsConns[idx] = utils.ConcatenatedKey(utils.MetaInternal, utils.MetaEEs)
			}
		}
	}
	if jsnCfg.Ees_exporter_ids != nil {
		t.EEsExporterIDs = append(t.EEsExporterIDs, *jsnCfg.Ees_exporter_ids...)
	}
	if jsnCfg.String_indexed_fields != nil {
		sif := make([]string, len(*jsnCfg.String_indexed_fields))
		copy(sif, *jsnCfg.String_indexed_fields)
		t.StringIndexedFields = &sif
	}
	if jsnCfg.Prefix_indexed_fields != nil {
		pif := make([]string, len(*jsnCfg.Prefix_indexed_fields))
		copy(pif, *jsnCfg.Prefix_indexed_fields)
		t.PrefixIndexedFields = &pif
	}
	if jsnCfg.Suffix_indexed_fields != nil {
		sif := make([]string, len(*jsnCfg.Suffix_indexed_fields))
		copy(sif, *jsnCfg.Suffix_indexed_fields)
		t.SuffixIndexedFields = &sif
	}
	if jsnCfg.ExistsIndexedFields != nil {
		eif := slices.Clone(*jsnCfg.ExistsIndexedFields)
		t.ExistsIndexedFields = &eif
	}
	if jsnCfg.Nested_fields != nil {
		t.NestedFields = *jsnCfg.Nested_fields
	}
	if jsnCfg.Opts != nil {
		t.Opts.loadFromJSONCfg(jsnCfg.Opts)
	}
	return nil
}

// AsMapInterface returns the config as a map[string]any
func (t *ThresholdSCfg) AsMapInterface() (initialMP map[string]any) {
	opts := map[string]any{
		utils.MetaProfileIDs:              t.Opts.ProfileIDs,
		utils.MetaProfileIgnoreFiltersCfg: t.Opts.ProfileIgnoreFilters,
	}
	initialMP = map[string]any{
		utils.EnabledCfg:        t.Enabled,
		utils.IndexedSelectsCfg: t.IndexedSelects,
		utils.NestedFieldsCfg:   t.NestedFields,
		utils.StoreIntervalCfg:  utils.EmptyString,
		utils.OptsCfg:           opts,
	}
	if t.StoreInterval != 0 {
		initialMP[utils.StoreIntervalCfg] = t.StoreInterval.String()
	}
	if t.SessionSConns != nil {
		sessionConns := make([]string, len(t.SessionSConns))
		for i, item := range t.SessionSConns {
			sessionConns[i] = item
			if item == utils.ConcatenatedKey(utils.MetaInternal, utils.MetaSessionS) {
				sessionConns[i] = utils.MetaInternal
			}
		}
		initialMP[utils.SessionSConnsCfg] = sessionConns
	}
	eesExporterIDs := make([]string, len(t.EEsExporterIDs))
	copy(eesExporterIDs, t.EEsExporterIDs)
	initialMP[utils.EEsExporterIDsCfg] = eesExporterIDs
	if t.ApierSConns != nil {
		apiersConns := make([]string, len(t.ApierSConns))
		for i, item := range t.ApierSConns {
			apiersConns[i] = item
			if item == utils.ConcatenatedKey(utils.MetaInternal, utils.MetaApier) {
				apiersConns[i] = utils.MetaInternal
			}
		}
		initialMP[utils.ApierSConnsCfg] = apiersConns
	}
	if t.EEsConns != nil {
		eesConns := make([]string, len(t.EEsConns))
		for i, item := range t.EEsConns {
			eesConns[i] = item
			if item == utils.ConcatenatedKey(utils.MetaInternal, utils.MetaEEs) {
				eesConns[i] = utils.MetaInternal
			}
		}
		initialMP[utils.EEsConnsCfg] = eesConns
	}
	if t.StringIndexedFields != nil {
		stringIndexedFields := make([]string, len(*t.StringIndexedFields))
		copy(stringIndexedFields, *t.StringIndexedFields)
		initialMP[utils.StringIndexedFieldsCfg] = stringIndexedFields
	}
	if t.PrefixIndexedFields != nil {
		prefixIndexedFields := make([]string, len(*t.PrefixIndexedFields))
		copy(prefixIndexedFields, *t.PrefixIndexedFields)
		initialMP[utils.PrefixIndexedFieldsCfg] = prefixIndexedFields
	}
	if t.SuffixIndexedFields != nil {
		suffixIndexedFields := make([]string, len(*t.SuffixIndexedFields))
		copy(suffixIndexedFields, *t.SuffixIndexedFields)
		initialMP[utils.SuffixIndexedFieldsCfg] = suffixIndexedFields
	}
	if t.ExistsIndexedFields != nil {
		eif := slices.Clone(*t.ExistsIndexedFields)
		initialMP[utils.ExistsIndexedFieldsCfg] = eif
	}
	return
}

func (thdOpts *ThresholdsOpts) Clone() *ThresholdsOpts {
	return &ThresholdsOpts{
		ProfileIDs:           slices.Clone(thdOpts.ProfileIDs),
		ProfileIgnoreFilters: thdOpts.ProfileIgnoreFilters,
	}
}

// Clone returns a deep copy of ThresholdSCfg
func (t ThresholdSCfg) Clone() (cln *ThresholdSCfg) {
	cln = &ThresholdSCfg{
		Enabled:        t.Enabled,
		IndexedSelects: t.IndexedSelects,
		StoreInterval:  t.StoreInterval,
		NestedFields:   t.NestedFields,
		Opts:           t.Opts.Clone(),
	}
	if t.SessionSConns != nil {
		cln.SessionSConns = make([]string, len(t.SessionSConns))
		copy(cln.SessionSConns, t.SessionSConns)
	}
	if t.ApierSConns != nil {
		cln.ApierSConns = make([]string, len(t.ApierSConns))
		copy(cln.ApierSConns, t.ApierSConns)
	}
	if t.EEsConns != nil {
		cln.EEsConns = make([]string, len(t.EEsConns))
		copy(cln.EEsConns, t.EEsConns)
	}
	if t.EEsExporterIDs != nil {
		cln.EEsExporterIDs = make([]string, len(t.EEsExporterIDs))
		copy(cln.EEsExporterIDs, t.EEsExporterIDs)
	}
	if t.StringIndexedFields != nil {
		idx := make([]string, len(*t.StringIndexedFields))
		copy(idx, *t.StringIndexedFields)
		cln.StringIndexedFields = &idx
	}
	if t.PrefixIndexedFields != nil {
		idx := make([]string, len(*t.PrefixIndexedFields))
		copy(idx, *t.PrefixIndexedFields)
		cln.PrefixIndexedFields = &idx
	}
	if t.SuffixIndexedFields != nil {
		idx := make([]string, len(*t.SuffixIndexedFields))
		copy(idx, *t.SuffixIndexedFields)
		cln.SuffixIndexedFields = &idx
	}
	if t.ExistsIndexedFields != nil {
		idx := slices.Clone(*t.ExistsIndexedFields)
		cln.ExistsIndexedFields = &idx
	}
	return
}