mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
305 lines
8.2 KiB
Go
305 lines
8.2 KiB
Go
/*
|
|
Rating system designed to be used in VoIP Carriems World
|
|
Copyright (C) 2013 ITsysCOM
|
|
|
|
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 vemsion 3 of the License, or
|
|
(at your option) any later vemsion.
|
|
|
|
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 (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"strings"
|
|
"time"
|
|
"github.com/cgrates/cgrates/cache2go"
|
|
"github.com/cgrates/cgrates/history"
|
|
"github.com/cgrates/cgrates/utils"
|
|
)
|
|
|
|
type MapStorage struct {
|
|
dict map[string][]byte
|
|
ms Marshaler
|
|
}
|
|
|
|
func NewMapStorage() (DataStorage, error) {
|
|
return &MapStorage{dict: make(map[string][]byte), ms: NewCodecMsgpackMarshaler()}, nil
|
|
}
|
|
|
|
func (ms *MapStorage) Close() {}
|
|
|
|
func (ms *MapStorage) Flush() error {
|
|
ms.dict = make(map[string][]byte)
|
|
return nil
|
|
}
|
|
|
|
func (ms *MapStorage) PreCache(dKeys, rpKeys, rpfKeys, actKeys []string) error {
|
|
if dKeys == nil && rpKeys == nil && rpfKeys == nil && actKeys == nil {
|
|
cache2go.Flush()
|
|
}
|
|
cache2go.RemPrefixKey(DESTINATION_PREFIX)
|
|
for k, _ := range ms.dict {
|
|
if strings.HasPrefix(k, DESTINATION_PREFIX) {
|
|
if _, err := ms.GetDestination(k[len(DESTINATION_PREFIX):]); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if strings.HasPrefix(k, RATING_PLAN_PREFIX) {
|
|
cache2go.RemKey(k)
|
|
if _, err := ms.GetRatingPlan(k[len(RATING_PLAN_PREFIX):], true); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if strings.HasPrefix(k, RATING_PROFILE_PREFIX) {
|
|
cache2go.RemKey(k)
|
|
if _, err := ms.GetRatingProfile(k[len(RATING_PROFILE_PREFIX):], true); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if strings.HasPrefix(k, ACTION_PREFIX) {
|
|
cache2go.RemKey(k)
|
|
if _, err := ms.GetActions(k[len(ACTION_PREFIX):], true); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Used to check if specific subject is stored using prefix key attached to entity
|
|
func (ms *MapStorage) ExistsData(categ, subject string) (bool, error) {
|
|
switch categ {
|
|
case DESTINATION_PREFIX:
|
|
_, exists := ms.dict[DESTINATION_PREFIX+subject]
|
|
return exists, nil
|
|
case RATING_PLAN_PREFIX:
|
|
_, exists := ms.dict[RATING_PLAN_PREFIX+subject]
|
|
return exists, nil
|
|
}
|
|
return false, errors.New("Unsupported category")
|
|
}
|
|
|
|
func (ms *MapStorage) GetRatingPlan(key string, checkDb bool) (rp *RatingPlan, err error) {
|
|
key = RATING_PLAN_PREFIX + key
|
|
if x, err := cache2go.GetCached(key); err == nil {
|
|
return x.(*RatingPlan), nil
|
|
}
|
|
if !checkDb {
|
|
return nil, errors.New(utils.ERR_NOT_FOUND)
|
|
}
|
|
if values, ok := ms.dict[key]; ok {
|
|
rp = new(RatingPlan)
|
|
|
|
err = ms.ms.Unmarshal(values, rp)
|
|
cache2go.Cache(key, rp)
|
|
} else {
|
|
return nil, errors.New("not found")
|
|
}
|
|
return
|
|
}
|
|
|
|
func (ms *MapStorage) SetRatingPlan(rp *RatingPlan) (err error) {
|
|
result, err := ms.ms.Marshal(rp)
|
|
ms.dict[RATING_PLAN_PREFIX+rp.Id] = result
|
|
response := 0
|
|
go historyScribe.Record(&history.Record{RATING_PLAN_PREFIX + rp.Id, rp}, &response)
|
|
cache2go.Cache(RATING_PLAN_PREFIX+rp.Id, rp)
|
|
return
|
|
}
|
|
|
|
func (ms *MapStorage) GetRatingProfile(key string, checkDb bool) (rpf *RatingProfile, err error) {
|
|
key = RATING_PROFILE_PREFIX + key
|
|
if x, err := cache2go.GetCached(key); err == nil {
|
|
return x.(*RatingProfile), nil
|
|
}
|
|
if !checkDb {
|
|
return nil, errors.New(utils.ERR_NOT_FOUND)
|
|
}
|
|
if values, ok := ms.dict[key]; ok {
|
|
rpf = new(RatingProfile)
|
|
|
|
err = ms.ms.Unmarshal(values, rpf)
|
|
cache2go.Cache(key, rpf)
|
|
} else {
|
|
return nil, errors.New("not found")
|
|
}
|
|
return
|
|
}
|
|
|
|
func (ms *MapStorage) SetRatingProfile(rpf *RatingProfile) (err error) {
|
|
result, err := ms.ms.Marshal(rpf)
|
|
ms.dict[RATING_PROFILE_PREFIX+rpf.Id] = result
|
|
response := 0
|
|
go historyScribe.Record(&history.Record{RATING_PROFILE_PREFIX + rpf.Id, rpf}, &response)
|
|
cache2go.Cache(RATING_PROFILE_PREFIX+rpf.Id, rpf)
|
|
return
|
|
}
|
|
|
|
func (ms *MapStorage) GetDestination(key string) (dest *Destination, err error) {
|
|
key = DESTINATION_PREFIX + key
|
|
if values, ok := ms.dict[key]; ok {
|
|
dest = &Destination{Id: key}
|
|
err = ms.ms.Unmarshal(values, dest)
|
|
// create optimized structure
|
|
for _, p := range dest.Prefixes {
|
|
var ids []string
|
|
if x, err := cache2go.GetCached(DESTINATION_PREFIX + p); err == nil {
|
|
ids = x.([]string)
|
|
}
|
|
ids = append(ids, dest.Id)
|
|
cache2go.Cache(DESTINATION_PREFIX+p, ids)
|
|
}
|
|
} else {
|
|
return nil, errors.New("not found")
|
|
}
|
|
return
|
|
}
|
|
|
|
func (ms *MapStorage) SetDestination(dest *Destination) (err error) {
|
|
result, err := ms.ms.Marshal(dest)
|
|
ms.dict[DESTINATION_PREFIX+dest.Id] = result
|
|
response := 0
|
|
go historyScribe.Record(&history.Record{DESTINATION_PREFIX + dest.Id, dest}, &response)
|
|
cache2go.Cache(DESTINATION_PREFIX+dest.Id, dest)
|
|
return
|
|
}
|
|
|
|
func (ms *MapStorage) GetActions(key string, checkDb bool) (as Actions, err error) {
|
|
if values, ok := ms.dict[ACTION_PREFIX+key]; ok {
|
|
err = ms.ms.Unmarshal(values, &as)
|
|
} else {
|
|
return nil, errors.New("not found")
|
|
}
|
|
return
|
|
|
|
key = ACTION_PREFIX + key
|
|
if x, err := cache2go.GetCached(key); err == nil {
|
|
return x.(Actions), nil
|
|
}
|
|
if !checkDb {
|
|
return nil, errors.New(utils.ERR_NOT_FOUND)
|
|
}
|
|
if values, ok := ms.dict[key]; ok {
|
|
err = ms.ms.Unmarshal(values, &as)
|
|
cache2go.Cache(key, as)
|
|
} else {
|
|
return nil, errors.New("not found")
|
|
}
|
|
return
|
|
}
|
|
|
|
func (ms *MapStorage) SetActions(key string, as Actions) (err error) {
|
|
result, err := ms.ms.Marshal(&as)
|
|
ms.dict[ACTION_PREFIX+key] = result
|
|
cache2go.Cache(ACTION_PREFIX+key, as)
|
|
return
|
|
}
|
|
|
|
func (ms *MapStorage) GetUserBalance(key string) (ub *UserBalance, err error) {
|
|
if values, ok := ms.dict[USER_BALANCE_PREFIX+key]; ok {
|
|
ub = &UserBalance{Id: key}
|
|
err = ms.ms.Unmarshal(values, ub)
|
|
} else {
|
|
return nil, errors.New("not found")
|
|
}
|
|
return
|
|
}
|
|
|
|
func (ms *MapStorage) SetUserBalance(ub *UserBalance) (err error) {
|
|
result, err := ms.ms.Marshal(ub)
|
|
ms.dict[USER_BALANCE_PREFIX+ub.Id] = result
|
|
return
|
|
}
|
|
|
|
func (ms *MapStorage) GetActionTimings(key string) (ats ActionTimings, err error) {
|
|
if values, ok := ms.dict[ACTION_TIMING_PREFIX+key]; ok {
|
|
err = ms.ms.Unmarshal(values, &ats)
|
|
} else {
|
|
return nil, errors.New("not found")
|
|
}
|
|
return
|
|
}
|
|
|
|
func (ms *MapStorage) SetActionTimings(key string, ats ActionTimings) (err error) {
|
|
if len(ats) == 0 {
|
|
// delete the key
|
|
delete(ms.dict, ACTION_TIMING_PREFIX+key)
|
|
return
|
|
}
|
|
result, err := ms.ms.Marshal(&ats)
|
|
ms.dict[ACTION_TIMING_PREFIX+key] = result
|
|
return
|
|
}
|
|
|
|
func (ms *MapStorage) GetAllActionTimings() (ats map[string]ActionTimings, err error) {
|
|
ats = make(map[string]ActionTimings)
|
|
for key, value := range ms.dict {
|
|
if !strings.Contains(key, ACTION_TIMING_PREFIX) {
|
|
continue
|
|
}
|
|
var tempAts ActionTimings
|
|
err = ms.ms.Unmarshal(value, &tempAts)
|
|
ats[key[len(ACTION_TIMING_PREFIX):]] = tempAts
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (ms *MapStorage) LogCallCost(uuid, source string, cc *CallCost) error {
|
|
result, err := ms.ms.Marshal(cc)
|
|
ms.dict[LOG_CALL_COST_PREFIX+source+"_"+uuid] = result
|
|
return err
|
|
}
|
|
|
|
func (ms *MapStorage) GetCallCostLog(uuid, source string) (cc *CallCost, err error) {
|
|
if values, ok := ms.dict[LOG_CALL_COST_PREFIX+source+"_"+uuid]; ok {
|
|
err = ms.ms.Unmarshal(values, &cc)
|
|
} else {
|
|
return nil, errors.New("not found")
|
|
}
|
|
return
|
|
}
|
|
|
|
func (ms *MapStorage) LogActionTrigger(ubId, source string, at *ActionTrigger, as Actions) (err error) {
|
|
mat, err := ms.ms.Marshal(at)
|
|
if err != nil {
|
|
return
|
|
}
|
|
mas, err := ms.ms.Marshal(&as)
|
|
if err != nil {
|
|
return
|
|
}
|
|
ms.dict[LOG_ACTION_TRIGGER_PREFIX+source+"_"+time.Now().Format(time.RFC3339Nano)] = []byte(fmt.Sprintf("%s*%s*%s", ubId, string(mat), string(mas)))
|
|
return
|
|
}
|
|
|
|
func (ms *MapStorage) LogActionTiming(source string, at *ActionTiming, as Actions) (err error) {
|
|
mat, err := ms.ms.Marshal(at)
|
|
if err != nil {
|
|
return
|
|
}
|
|
mas, err := ms.ms.Marshal(&as)
|
|
if err != nil {
|
|
return
|
|
}
|
|
ms.dict[LOG_ACTION_TIMMING_PREFIX+source+"_"+time.Now().Format(time.RFC3339Nano)] = []byte(fmt.Sprintf("%s*%s", string(mat), string(mas)))
|
|
return
|
|
}
|
|
|
|
func (ms *MapStorage) LogError(uuid, source, errstr string) (err error) {
|
|
ms.dict[LOG_ERR+source+"_"+uuid] = []byte(errstr)
|
|
return nil
|
|
}
|