mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-17 06:09:53 +05:00
89 lines
2.7 KiB
Go
89 lines
2.7 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 (
|
|
"fmt"
|
|
|
|
"github.com/cgrates/cgrates/utils"
|
|
)
|
|
|
|
// NewSupplierSortDispatcher constructs SupplierSortDispatcher
|
|
func NewSupplierSortDispatcher(lcrS *LCRService) (ssd SupplierSortDispatcher, err error) {
|
|
ssd = make(map[string]SuppliersSorting)
|
|
ssd[utils.MetaWeight] = NewWeightStrategy()
|
|
ssd[utils.MetaLeastCost] = NewLeastCostStrategy(lcrS)
|
|
return
|
|
}
|
|
|
|
// SupplierStrategyHandler will initialize strategies
|
|
// and dispatch requests to them
|
|
type SupplierSortDispatcher map[string]SuppliersSorting
|
|
|
|
func (ssd SupplierSortDispatcher) SortSuppliers(prflID, strategy string,
|
|
suppls LCRSuppliers) (sortedSuppls *SortedSuppliers, err error) {
|
|
sd, has := ssd[strategy]
|
|
if !has {
|
|
return nil, fmt.Errorf("unsupported sorting strategy: %s", strategy)
|
|
}
|
|
return sd.SortSuppliers(prflID, suppls)
|
|
}
|
|
|
|
type SuppliersSorting interface {
|
|
SortSuppliers(string, LCRSuppliers) (*SortedSuppliers, error)
|
|
}
|
|
|
|
// NewLeastCostStrategy constructs LeastCostStrategy
|
|
func NewLeastCostStrategy(lcrS *LCRService) *LeastCostStrategy {
|
|
return &LeastCostStrategy{lcrS: lcrS}
|
|
}
|
|
|
|
// LeastCostStrategy orders suppliers based on lowest cost
|
|
type LeastCostStrategy struct {
|
|
lcrS *LCRService
|
|
}
|
|
|
|
func (lcs *LeastCostStrategy) SortSuppliers(prflID string,
|
|
suppls LCRSuppliers) (sortedSuppls *SortedSuppliers, err error) {
|
|
return
|
|
}
|
|
|
|
func NewWeightStrategy() *WeightStrategy {
|
|
return &WeightStrategy{Sorting: utils.MetaWeight}
|
|
}
|
|
|
|
// WeightStrategy orders suppliers based on their weight, no cost involved
|
|
type WeightStrategy struct {
|
|
Sorting string
|
|
}
|
|
|
|
func (ws *WeightStrategy) SortSuppliers(prflID string,
|
|
suppls LCRSuppliers) (sortedSuppls *SortedSuppliers, err error) {
|
|
suppls.Sort()
|
|
sortedSuppls = &SortedSuppliers{ProfileID: prflID,
|
|
Sorting: ws.Sorting,
|
|
SortedSuppliers: make([]*SortedSupplier, len(suppls))}
|
|
for i, s := range suppls {
|
|
sortedSuppls.SortedSuppliers[i] = &SortedSupplier{
|
|
SupplierID: s.ID,
|
|
SortingData: map[string]interface{}{"Weight": s.Weight}}
|
|
}
|
|
return
|
|
}
|