mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-20 22:58:44 +05:00
added sql storage
This commit is contained in:
482
rater/storage_sql.go
Normal file
482
rater/storage_sql.go
Normal file
@@ -0,0 +1,482 @@
|
||||
/*
|
||||
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) GetTPDestination(tpid, destTag string) (*Destination, error) {
|
||||
rows, err := sql.Db.Query(fmt.Sprintf("SELECT prefix FROM tp_destinatins WHERE id='%s' AND tag='%s'", 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) 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 tp_destinations WHERE tpid=%s", 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 tp_rates WHERE tpid=%s", 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 string
|
||||
var connect_fee, rate, priced_units, rate_increments float64
|
||||
if err := rows.Scan(&id, &tpid, &tag, &destinations_tag, &connect_fee, &rate, &priced_units, &rate_increments); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
r := &Rate{
|
||||
DestinationsTag: destinations_tag,
|
||||
ConnectFee: connect_fee,
|
||||
Price: rate,
|
||||
PricedUnits: priced_units,
|
||||
RateIncrements: rate_increments,
|
||||
}
|
||||
|
||||
rts[tag] = append(rts[tag], r)
|
||||
}
|
||||
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 tp_timings WHERE tpid=%s", 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) GetTpRateTimings(tpid, tag string) ([]*RateTiming, error) {
|
||||
var rts []*RateTiming
|
||||
q := fmt.Sprintf("SELECT * FROM tp_rate_timings WHERE tpid=%s", 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, rates_tag, timings_tag string
|
||||
if err := rows.Scan(&id, &tpid, &tag, &rates_tag, &timings_tag, &weight); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
rt := &RateTiming{
|
||||
Tag: tag,
|
||||
RatesTag: 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 tp_rate_profiles WHERE tpid=%s", 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 tp_actions WHERE tpid=%s", 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 tp_action_timings WHERE tpid=%s", 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 tp_action_triggers WHERE tpid=%s", 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 tp_account_actions WHERE tpid=%s", 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()
|
||||
}
|
||||
Reference in New Issue
Block a user