mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
553 lines
15 KiB
Go
553 lines
15 KiB
Go
/*
|
|
Rating system designed to be used in VoIP Carriers 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 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 rater
|
|
|
|
import (
|
|
"database/sql"
|
|
"encoding/json"
|
|
"fmt"
|
|
"github.com/cgrates/cgrates/utils"
|
|
)
|
|
|
|
type SQLStorage struct {
|
|
Db *sql.DB
|
|
}
|
|
|
|
func (sql *SQLStorage) Close() {}
|
|
|
|
func (sql *SQLStorage) Flush() (err error) {
|
|
return
|
|
}
|
|
|
|
func (sql *SQLStorage) GetRatingProfile(string) (rp *RatingProfile, err error) {
|
|
/*row := sql.Db.QueryRow(fmt.Sprintf("SELECT * FROM ratingprofiles WHERE id='%s'", id))
|
|
err = row.Scan(&rp, &cc.Direction, &cc.Tenant, &cc.TOR, &cc.Subject, &cc.Destination, &cc.Cost, &cc.ConnectFee, ×pansJson)
|
|
err = json.Unmarshal([]byte(timespansJson), cc.Timespans)*/
|
|
return
|
|
}
|
|
|
|
func (sql *SQLStorage) SetRatingProfile(rp *RatingProfile) (err error) {
|
|
return
|
|
}
|
|
|
|
func (sql *SQLStorage) GetDestination(string) (d *Destination, err error) {
|
|
return
|
|
}
|
|
|
|
func (sql *SQLStorage) SetDestination(d *Destination) (err error) {
|
|
return
|
|
}
|
|
|
|
// Extracts destinations from StorDB on specific tariffplan id
|
|
func (sql *SQLStorage) GetTPDestinationIds(tpid string) ([]string, error) {
|
|
rows, err := sql.Db.Query(fmt.Sprintf("SELECT DISTINCT tag FROM %s where tpid='%s'", utils.TBL_TP_DESTINATIONS, tpid))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
ids := []string{}
|
|
i := 0
|
|
for rows.Next() {
|
|
i++ //Keep here a reference so we know we got at least one prefix
|
|
var id string
|
|
err = rows.Scan(&id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ids = append(ids, id)
|
|
}
|
|
if i == 0 {
|
|
return nil, nil
|
|
}
|
|
return ids, nil
|
|
}
|
|
|
|
func (sql *SQLStorage) ExistsTPDestination(tpid, destTag string) (bool, error) {
|
|
var exists bool
|
|
err := sql.Db.QueryRow(fmt.Sprintf("SELECT EXISTS (SELECT 1 FROM %s WHERE tpid='%s' AND tag='%s')", utils.TBL_TP_DESTINATIONS, tpid, destTag)).Scan(&exists)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return exists, nil
|
|
}
|
|
|
|
// Extracts destinations from StorDB on specific tariffplan id
|
|
func (sql *SQLStorage) GetTPDestination(tpid, destTag string) (*Destination, error) {
|
|
rows, err := sql.Db.Query(fmt.Sprintf("SELECT prefix FROM %s WHERE tpid='%s' AND tag='%s'", utils.TBL_TP_DESTINATIONS, tpid, destTag))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
d := &Destination{Id: destTag}
|
|
i := 0
|
|
for rows.Next() {
|
|
i++ //Keep here a reference so we know we got at least one prefix
|
|
var pref string
|
|
err = rows.Scan(&pref)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
d.Prefixes = append(d.Prefixes, pref)
|
|
}
|
|
if i == 0 {
|
|
return nil, nil
|
|
}
|
|
return d, nil
|
|
}
|
|
|
|
func (sql *SQLStorage) SetTPDestination(tpid string, dest *Destination) error {
|
|
for _, prefix := range dest.Prefixes {
|
|
if _, err := sql.Db.Exec(fmt.Sprintf("INSERT INTO %s (tpid, tag, prefix) VALUES( '%s','%s','%s')", utils.TBL_TP_DESTINATIONS, tpid, dest.Id, prefix)); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (sql *SQLStorage) GetActions(string) (as Actions, err error) {
|
|
return
|
|
}
|
|
|
|
func (sql *SQLStorage) SetActions(key string, as Actions) (err error) { return }
|
|
|
|
func (sql *SQLStorage) GetUserBalance(string) (ub *UserBalance, err error) { return }
|
|
|
|
func (sql *SQLStorage) SetUserBalance(ub *UserBalance) (err error) { return }
|
|
|
|
func (sql *SQLStorage) GetActionTimings(key string) (ats ActionTimings, err error) { return }
|
|
|
|
func (sql *SQLStorage) SetActionTimings(key string, ats ActionTimings) (err error) { return }
|
|
|
|
func (sql *SQLStorage) GetAllActionTimings() (ats map[string]ActionTimings, err error) {
|
|
return
|
|
}
|
|
|
|
func (sql *SQLStorage) LogCallCost(uuid, source string, cc *CallCost) (err error) {
|
|
if sql.Db == nil {
|
|
//timespans.Logger.Warning("Cannot write log to database.")
|
|
return
|
|
}
|
|
tss, err := json.Marshal(cc.Timespans)
|
|
if err != nil {
|
|
Logger.Err(fmt.Sprintf("Error marshalling timespans to json: %v", err))
|
|
}
|
|
_, err = sql.Db.Exec(fmt.Sprintf("INSERT INTO callcosts VALUES ('NULL','%s', '%s','%s', '%s', '%s', '%s', '%s', '%s', %v, %v, '%s')",
|
|
uuid,
|
|
source,
|
|
cc.Direction,
|
|
cc.Tenant,
|
|
cc.TOR,
|
|
cc.Subject,
|
|
cc.Account,
|
|
cc.Destination,
|
|
cc.Cost,
|
|
cc.ConnectFee,
|
|
tss))
|
|
if err != nil {
|
|
Logger.Err(fmt.Sprintf("failed to execute insert statement: %v", err))
|
|
}
|
|
return
|
|
}
|
|
|
|
func (sql *SQLStorage) GetCallCostLog(uuid, source string) (cc *CallCost, err error) {
|
|
row := sql.Db.QueryRow(fmt.Sprintf("SELECT * FROM callcosts WHERE uuid='%s' AND source='%s'", uuid, source))
|
|
var uuid_found string
|
|
var timespansJson string
|
|
err = row.Scan(&uuid_found, &cc.Direction, &cc.Tenant, &cc.TOR, &cc.Subject, &cc.Destination, &cc.Cost, &cc.ConnectFee, ×pansJson)
|
|
err = json.Unmarshal([]byte(timespansJson), cc.Timespans)
|
|
return
|
|
}
|
|
|
|
func (sql *SQLStorage) LogActionTrigger(ubId, source string, at *ActionTrigger, as Actions) (err error) {
|
|
return
|
|
}
|
|
func (sql *SQLStorage) LogActionTiming(source string, at *ActionTiming, as Actions) (err error) {
|
|
return
|
|
}
|
|
func (sql *SQLStorage) LogError(uuid, source, errstr string) (err error) { return }
|
|
|
|
func (sql *SQLStorage) SetCdr(cdr utils.CDR) (err error) {
|
|
startTime, err := cdr.GetAnswerTime()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = sql.Db.Exec(fmt.Sprintf("INSERT INTO cdrs_primary VALUES (NULL, '%s', '%s','%s', '%s', '%s', '%s', '%s', '%s', '%s', '%s', '%s', %d)",
|
|
cdr.GetCgrId(),
|
|
cdr.GetAccId(),
|
|
cdr.GetCdrHost(),
|
|
cdr.GetReqType(),
|
|
cdr.GetDirection(),
|
|
cdr.GetTenant(),
|
|
cdr.GetTOR(),
|
|
cdr.GetAccount(),
|
|
cdr.GetSubject(),
|
|
cdr.GetDestination(),
|
|
startTime,
|
|
cdr.GetDuration(),
|
|
))
|
|
if err != nil {
|
|
Logger.Err(fmt.Sprintf("failed to execute cdr insert statement: %v", err))
|
|
}
|
|
extraFields, err := json.Marshal(cdr.GetExtraFields())
|
|
if err != nil {
|
|
Logger.Err(fmt.Sprintf("Error marshalling cdr extra fields to json: %v", err))
|
|
}
|
|
_, err = sql.Db.Exec(fmt.Sprintf("INSERT INTO cdrs_extra VALUES ('NULL','%s', '%s')",
|
|
cdr.GetCgrId(),
|
|
extraFields,
|
|
))
|
|
if err != nil {
|
|
Logger.Err(fmt.Sprintf("failed to execute cdr insert statement: %v", err))
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (sql *SQLStorage) SetRatedCdr(cdr utils.CDR, cc *CallCost) (err error) {
|
|
_, err = sql.Db.Exec(fmt.Sprintf("INSERT INTO rated_cdrs VALUES ('%s', '%s', '%s', '%s')",
|
|
cdr.GetCgrId(),
|
|
cc.Cost,
|
|
"cgrcostid",
|
|
"cdrsrc",
|
|
))
|
|
if err != nil {
|
|
Logger.Err(fmt.Sprintf("failed to execute cdr insert statement: %v", err))
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (sql *SQLStorage) GetAllRatedCdr() ([]utils.CDR, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (sql *SQLStorage) GetTpDestinations(tpid, tag string) ([]*Destination, error) {
|
|
var dests []*Destination
|
|
q := fmt.Sprintf("SELECT * FROM %s WHERE tpid=%s", utils.TBL_TP_DESTINATIONS, tpid)
|
|
if tag != "" {
|
|
q += "AND tag=" + tag
|
|
}
|
|
rows, err := sql.Db.Query(q, tpid)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for rows.Next() {
|
|
var id int
|
|
var tpid, tag, prefix string
|
|
if err := rows.Scan(id, tpid, &tag, &prefix); err != nil {
|
|
return nil, err
|
|
}
|
|
var dest *Destination
|
|
for _, d := range dests {
|
|
if d.Id == tag {
|
|
dest = d
|
|
break
|
|
}
|
|
}
|
|
if dest == nil {
|
|
dest = &Destination{Id: tag}
|
|
dests = append(dests, dest)
|
|
}
|
|
dest.Prefixes = append(dest.Prefixes, prefix)
|
|
}
|
|
return dests, rows.Err()
|
|
}
|
|
|
|
func (sql *SQLStorage) GetTpRates(tpid, tag string) (map[string]*Rate, error) {
|
|
rts := make(map[string]*Rate)
|
|
q := fmt.Sprintf("SELECT * FROM %s WHERE tpid=%s", utils.TBL_TP_RATES, tpid)
|
|
if tag != "" {
|
|
q += "AND tag=" + tag
|
|
}
|
|
rows, err := sql.Db.Query(q, tpid)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for rows.Next() {
|
|
var id int
|
|
var tpid, tag string
|
|
var connect_fee, rate, priced_units, rate_increments float64
|
|
if err := rows.Scan(&id, &tpid, &tag, &connect_fee, &rate, &priced_units, &rate_increments); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
r := &Rate{
|
|
Tag: tag,
|
|
ConnectFee: connect_fee,
|
|
Price: rate,
|
|
PricedUnits: priced_units,
|
|
RateIncrements: rate_increments,
|
|
}
|
|
|
|
rts[tag] = r
|
|
}
|
|
return rts, rows.Err()
|
|
}
|
|
|
|
func (sql *SQLStorage) GetTpDestinationRates(tpid, tag string) (map[string][]*DestinationRate, error) {
|
|
rts := make(map[string][]*DestinationRate)
|
|
q := fmt.Sprintf("SELECT * FROM %s WHERE tpid=%s", utils.TBL_TP_DESTINATION_RATES, tpid)
|
|
if tag != "" {
|
|
q += "AND tag=" + tag
|
|
}
|
|
rows, err := sql.Db.Query(q, tpid)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for rows.Next() {
|
|
var id int
|
|
var tpid, tag, destinations_tag, rate_tag string
|
|
if err := rows.Scan(&id, &tpid, &tag, destinations_tag, rate_tag); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
dr := &DestinationRate{
|
|
Tag: tag,
|
|
DestinationsTag: destinations_tag,
|
|
RateTag: rate_tag,
|
|
}
|
|
|
|
rts[tag] = append(rts[tag], dr)
|
|
}
|
|
return rts, rows.Err()
|
|
}
|
|
|
|
func (sql *SQLStorage) GetTpTimings(tpid, tag string) (map[string]*Timing, error) {
|
|
tms := make(map[string]*Timing)
|
|
q := fmt.Sprintf("SELECT * FROM %s WHERE tpid=%s", utils.TBL_TP_TIMINGS, tpid)
|
|
if tag != "" {
|
|
q += "AND tag=" + tag
|
|
}
|
|
rows, err := sql.Db.Query(q, tpid)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for rows.Next() {
|
|
var id int
|
|
var tpid, tag, years, months, month_days, week_days, start_time string
|
|
if err := rows.Scan(&id, &tpid, &tag, &years, &months, &month_days, &week_days, &start_time); err != nil {
|
|
return nil, err
|
|
}
|
|
tms[tag] = NewTiming(years, months, month_days, week_days, start_time)
|
|
}
|
|
return tms, rows.Err()
|
|
}
|
|
|
|
func (sql *SQLStorage) GetTpDestinationRateTimings(tpid, tag string) ([]*DestinationRateTiming, error) {
|
|
var rts []*DestinationRateTiming
|
|
q := fmt.Sprintf("SELECT * FROM %s WHERE tpid=%s", utils.TBL_TP_DESTINATION_RATE_TIMINGS, tpid)
|
|
if tag != "" {
|
|
q += "AND tag=" + tag
|
|
}
|
|
rows, err := sql.Db.Query(q, tpid)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for rows.Next() {
|
|
var id int
|
|
var weight float64
|
|
var tpid, tag, destination_rates_tag, timings_tag string
|
|
if err := rows.Scan(&id, &tpid, &tag, &destination_rates_tag, &timings_tag, &weight); err != nil {
|
|
return nil, err
|
|
}
|
|
rt := &DestinationRateTiming{
|
|
Tag: tag,
|
|
DestinationRatesTag: destination_rates_tag,
|
|
Weight: weight,
|
|
TimingsTag: timings_tag,
|
|
}
|
|
rts = append(rts, rt)
|
|
}
|
|
return rts, rows.Err()
|
|
}
|
|
|
|
func (sql *SQLStorage) GetTpRatingProfiles(tpid, tag string) (map[string]*RatingProfile, error) {
|
|
rpfs := make(map[string]*RatingProfile)
|
|
q := fmt.Sprintf("SELECT * FROM %s WHERE tpid=%s", utils.TBL_TP_RATE_PROFILES, tpid)
|
|
if tag != "" {
|
|
q += "AND tag=" + tag
|
|
}
|
|
rows, err := sql.Db.Query(q, tpid)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for rows.Next() {
|
|
var id int
|
|
var tpid, tenant, tor, direction, subject, fallback_subject, rates_timing_tag, activation_time string
|
|
|
|
if err := rows.Scan(&id, &tpid, &tenant, &tor, &direction, &subject, &fallback_subject, &rates_timing_tag, &activation_time); err != nil {
|
|
return nil, err
|
|
}
|
|
key := fmt.Sprintf("%s:%s:%s:%s", direction, tenant, tor, subject)
|
|
rp, ok := rpfs[key]
|
|
if !ok {
|
|
rp = &RatingProfile{Id: key}
|
|
rpfs[key] = rp
|
|
}
|
|
rp.ratesTimingTag = rates_timing_tag
|
|
rp.activationTime = activation_time
|
|
if fallback_subject != "" {
|
|
rp.FallbackKey = fmt.Sprintf("%s:%s:%s:%s", direction, tenant, tor, fallback_subject)
|
|
}
|
|
}
|
|
return rpfs, rows.Err()
|
|
}
|
|
func (sql *SQLStorage) GetTpActions(tpid, tag string) (map[string][]*Action, error) {
|
|
as := make(map[string][]*Action)
|
|
q := fmt.Sprintf("SELECT * FROM %s WHERE tpid=%s", utils.TBL_TP_ACTIONS, tpid)
|
|
if tag != "" {
|
|
q += "AND tag=" + tag
|
|
}
|
|
rows, err := sql.Db.Query(q, tpid)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for rows.Next() {
|
|
var id int
|
|
var units, rate, minutes_weight, weight float64
|
|
var tpid, tag, action, balance_tag, direction, destinations_tag, rate_type string
|
|
if err := rows.Scan(&id, &tpid, &tag, &action, &balance_tag, &direction, &units, &destinations_tag, &rate_type, &rate, &minutes_weight, &weight); err != nil {
|
|
return nil, err
|
|
}
|
|
var a *Action
|
|
if balance_tag != MINUTES {
|
|
a = &Action{
|
|
ActionType: action,
|
|
BalanceId: balance_tag,
|
|
Direction: direction,
|
|
Units: units,
|
|
}
|
|
} else {
|
|
var percent, price float64
|
|
if rate_type == PERCENT {
|
|
percent = rate
|
|
}
|
|
if rate_type == ABSOLUTE {
|
|
price = rate
|
|
}
|
|
a = &Action{
|
|
Id: utils.GenUUID(),
|
|
ActionType: action,
|
|
BalanceId: balance_tag,
|
|
Direction: direction,
|
|
Weight: weight,
|
|
MinuteBucket: &MinuteBucket{
|
|
Seconds: units,
|
|
Weight: minutes_weight,
|
|
Price: price,
|
|
Percent: percent,
|
|
DestinationId: destinations_tag,
|
|
},
|
|
}
|
|
}
|
|
as[tag] = append(as[tag], a)
|
|
}
|
|
return as, rows.Err()
|
|
}
|
|
|
|
func (sql *SQLStorage) GetTpActionTimings(tpid, tag string) (ats map[string][]*ActionTiming, err error) {
|
|
ats = make(map[string][]*ActionTiming)
|
|
q := fmt.Sprintf("SELECT * FROM %s WHERE tpid=%s", utils.TBL_TP_ACTION_TIMINGS, tpid)
|
|
if tag != "" {
|
|
q += "AND tag=" + tag
|
|
}
|
|
rows, err := sql.Db.Query(q, tpid)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for rows.Next() {
|
|
var id int
|
|
var weight float64
|
|
var tpid, tag, actions_tag, timings_tag string
|
|
if err := rows.Scan(&id, &tpid, &tag, &actions_tag, &timings_tag, &weight); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
at := &ActionTiming{
|
|
Id: utils.GenUUID(),
|
|
Tag: timings_tag,
|
|
Weight: weight,
|
|
ActionsId: actions_tag,
|
|
}
|
|
ats[tag] = append(ats[tag], at)
|
|
}
|
|
return ats, rows.Err()
|
|
}
|
|
|
|
func (sql *SQLStorage) GetTpActionTriggers(tpid, tag string) (map[string][]*ActionTrigger, error) {
|
|
ats := make(map[string][]*ActionTrigger)
|
|
q := fmt.Sprintf("SELECT * FROM %s WHERE tpid=%s", utils.TBL_TP_ACTION_TRIGGERS, tpid)
|
|
if tag != "" {
|
|
q += "AND tag=" + tag
|
|
}
|
|
rows, err := sql.Db.Query(q, tpid)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for rows.Next() {
|
|
var id int
|
|
var threshold, weight float64
|
|
var tpid, tag, balances_tag, direction, destinations_tag, actions_tag string
|
|
if err := rows.Scan(&id, &tpid, &tag, &balances_tag, &direction, &threshold, &destinations_tag, &actions_tag, &weight); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
at := &ActionTrigger{
|
|
Id: utils.GenUUID(),
|
|
BalanceId: balances_tag,
|
|
Direction: direction,
|
|
ThresholdValue: threshold,
|
|
DestinationId: destinations_tag,
|
|
ActionsId: actions_tag,
|
|
Weight: weight,
|
|
}
|
|
ats[tag] = append(ats[tag], at)
|
|
}
|
|
return ats, rows.Err()
|
|
}
|
|
|
|
func (sql *SQLStorage) GetTpAccountActions(tpid, tag string) ([]*AccountAction, error) {
|
|
var acs []*AccountAction
|
|
q := fmt.Sprintf("SELECT * FROM %s WHERE tpid=%s", utils.TBL_TP_ACCOUNT_ACTIONS, tpid)
|
|
if tag != "" {
|
|
q += "AND tag=" + tag
|
|
}
|
|
rows, err := sql.Db.Query(q, tpid)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for rows.Next() {
|
|
var id int
|
|
var tpid, tenant, account, direction, action_timings_tag, action_triggers_tag string
|
|
if err := rows.Scan(&id, &tpid, &tenant, &account, &direction, &action_timings_tag, &action_triggers_tag); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
aa := &AccountAction{
|
|
Tenant: tenant,
|
|
Account: account,
|
|
Direction: direction,
|
|
ActionTimingsTag: action_timings_tag,
|
|
ActionTriggersTag: action_triggers_tag,
|
|
}
|
|
acs = append(acs, aa)
|
|
}
|
|
return acs, rows.Err()
|
|
}
|