Files
cgrates/engine/libattributes.go

105 lines
3.1 KiB
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 engine
import (
"sort"
"github.com/cgrates/cgrates/utils"
)
type Attribute struct {
FieldName string
Initial string
Substitute string
Append bool
}
type AttributeProfile struct {
Tenant string
ID string
Context string // bind this AttributeProfile to specific context
FilterIDs []string
ActivationInterval *utils.ActivationInterval // Activation interval
Attributes map[string]map[string]*Attribute // map[FieldName][InitialValue]*Attribute
Weight float64
}
func (als *AttributeProfile) TenantID() string {
return utils.ConcatenatedKey(als.Tenant, als.ID)
}
// AttributeProfiles is a sortable list of Attribute profiles
type AttributeProfiles []*AttributeProfile
// Sort is part of sort interface, sort based on Weight
func (aps AttributeProfiles) Sort() {
sort.Slice(aps, func(i, j int) bool { return aps[i].Weight > aps[j].Weight })
}
type ExternalAttributeProfile struct {
Tenant string
ID string
Context string // bind this AttributeProfile to specific context
FilterIDs []string
ActivationInterval *utils.ActivationInterval // Activation interval
Attributes []*Attribute
Weight float64
}
func (eap *ExternalAttributeProfile) AsAttributeProfile() *AttributeProfile {
alsPrf := &AttributeProfile{
Tenant: eap.Tenant,
ID: eap.ID,
Context: eap.Context,
FilterIDs: eap.FilterIDs,
ActivationInterval: eap.ActivationInterval,
Weight: eap.Weight,
}
alsMap := make(map[string]map[string]*Attribute)
for _, als := range eap.Attributes {
alsMap[als.FieldName] = make(map[string]*Attribute)
alsMap[als.FieldName][als.Initial] = als
}
alsPrf.Attributes = alsMap
return alsPrf
}
func NewExternalAttributeProfileFromAttributeProfile(alsPrf *AttributeProfile) *ExternalAttributeProfile {
extals := &ExternalAttributeProfile{
Tenant: alsPrf.Tenant,
ID: alsPrf.ID,
Context: alsPrf.Context,
ActivationInterval: alsPrf.ActivationInterval,
FilterIDs: alsPrf.FilterIDs,
Weight: alsPrf.Weight,
}
for key, val := range alsPrf.Attributes {
for key2, val2 := range val {
extals.Attributes = append(extals.Attributes, &Attribute{
FieldName: key,
Initial: key2,
Substitute: val2.Substitute,
Append: val2.Append,
})
}
}
return extals
}