Files
cgrates/engine/libattributes.go
2019-04-04 10:52:41 +02:00

115 lines
3.4 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/config"
"github.com/cgrates/cgrates/utils"
)
type Attribute struct {
FilterIDs []string
FieldName string
Type string
Substitute config.RSRParsers
}
type AttributeProfile struct {
Tenant string
ID string
Contexts []string // bind this AttributeProfile to multiple contexts
FilterIDs []string
ActivationInterval *utils.ActivationInterval // Activation interval
Attributes []*Attribute
Blocker bool // blocker flag to stop processing on multiple runs
Weight float64
}
func (ap *AttributeProfile) compileSubstitutes() (err error) {
for _, attr := range ap.Attributes {
if err = attr.Substitute.Compile(); err != nil {
return
}
}
return
}
// Compile is a wrapper for convenience setting up the AttributeProfile
func (ap *AttributeProfile) Compile() error {
return ap.compileSubstitutes()
}
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 ExternalAttribute struct {
FilterIDs []string
FieldName string
Substitute string
}
type ExternalAttributeProfile struct {
Tenant string
ID string
Contexts []string // bind this AttributeProfile to multiple contexts
FilterIDs []string
ActivationInterval *utils.ActivationInterval // Activation interval
Attributes []*ExternalAttribute
Blocker bool // blocker flag to stop processing on multiple runs
Weight float64
}
func (ext *ExternalAttributeProfile) AsAttributeProfile() (attr *AttributeProfile, err error) {
attr = new(AttributeProfile)
if len(ext.Attributes) == 0 {
return nil, utils.NewErrMandatoryIeMissing("Attributes")
}
attr.Attributes = make([]*Attribute, len(ext.Attributes))
for i, extAttr := range ext.Attributes {
if len(extAttr.Substitute) == 0 {
return nil, utils.NewErrMandatoryIeMissing("Substitute")
}
attr.Attributes[i] = new(Attribute)
if attr.Attributes[i].Substitute, err = config.NewRSRParsers(extAttr.Substitute, true, utils.INFIELD_SEP); err != nil {
return nil, err
}
attr.Attributes[i].FilterIDs = extAttr.FilterIDs
attr.Attributes[i].FieldName = extAttr.FieldName
}
attr.Tenant = ext.Tenant
attr.ID = ext.ID
attr.Contexts = ext.Contexts
attr.FilterIDs = ext.FilterIDs
attr.ActivationInterval = ext.ActivationInterval
attr.Blocker = ext.Blocker
attr.Weight = ext.Weight
return
}