cgrates/utils/storedcdr.go

/*
Real-time Charging System for Telecom & ISP environments
Copyright (C) 2012-2014 ITsysCOM GmbH

This program is free software: you can Storagetribute 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 WITH*out 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 utils

import (
	"math"
	"net/url"
	"strconv"
	"time"
)

func NewStoredCdrFromRawCDR(rawcdr RawCDR) (*StoredCdr, error) {
	var err error
	rtCdr := new(StoredCdr)
	rtCdr.CgrId = rawcdr.GetCgrId()
	rtCdr.AccId = rawcdr.GetAccId()
	rtCdr.CdrHost = rawcdr.GetCdrHost()
	rtCdr.CdrSource = rawcdr.GetCdrSource()
	rtCdr.ReqType = rawcdr.GetReqType()
	rtCdr.Direction = rawcdr.GetDirection()
	rtCdr.Tenant = rawcdr.GetTenant()
	rtCdr.TOR = rawcdr.GetTOR()
	rtCdr.Account = rawcdr.GetAccount()
	rtCdr.Subject = rawcdr.GetSubject()
	rtCdr.Destination = rawcdr.GetDestination()
	if rtCdr.SetupTime, err = rawcdr.GetSetupTime(); err != nil {
		return nil, err
	}
	if rtCdr.AnswerTime, err = rawcdr.GetAnswerTime(); err != nil {
		return nil, err
	}
	rtCdr.Duration, _ = rawcdr.GetDuration()
	rtCdr.ExtraFields = rawcdr.GetExtraFields()
	rtCdr.MediationRunId = DEFAULT_RUNID
	rtCdr.Cost = -1
	return rtCdr, nil
}

// Rated CDR as extracted from StorDb. Kinda standard of internal CDR, complies to CDR interface also
type StoredCdr struct {
	CgrId          string
	OrderId        int64 // Stor order id used as export order id
	AccId          string
	CdrHost        string
	CdrSource      string
	ReqType        string
	Direction      string
	Tenant         string
	TOR            string
	Account        string
	Subject        string
	Destination    string
	SetupTime      time.Time
	AnswerTime     time.Time
	Duration       time.Duration
	ExtraFields    map[string]string
	MediationRunId string
	Cost           float64
}

// Methods maintaining RawCDR interface

func (storedCdr *StoredCdr) GetCgrId() string {
	return storedCdr.CgrId
}

func (storedCdr *StoredCdr) GetAccId() string {
	return storedCdr.AccId
}

func (storedCdr *StoredCdr) GetCdrHost() string {
	return storedCdr.CdrHost
}

func (storedCdr *StoredCdr) GetCdrSource() string {
	return storedCdr.CdrSource
}

func (storedCdr *StoredCdr) GetDirection() string {
	return storedCdr.Direction
}

func (storedCdr *StoredCdr) GetSubject() string {
	return storedCdr.Subject
}

func (storedCdr *StoredCdr) GetAccount() string {
	return storedCdr.Account
}

func (storedCdr *StoredCdr) GetDestination() string {
	return storedCdr.Destination
}

func (storedCdr *StoredCdr) GetTOR() string {
	return storedCdr.TOR
}

func (storedCdr *StoredCdr) GetTenant() string {
	return storedCdr.Tenant
}

func (storedCdr *StoredCdr) GetReqType() string {
	return storedCdr.ReqType
}

func (storedCdr *StoredCdr) GetSetupTime() (time.Time, error) {
	return storedCdr.SetupTime, nil
}

func (storedCdr *StoredCdr) GetAnswerTime() (time.Time, error) {
	return storedCdr.AnswerTime, nil
}

func (storedCdr *StoredCdr) GetDuration() (time.Duration, error) {
	return storedCdr.Duration, nil
}

func (storedCdr *StoredCdr) GetExtraFields() map[string]string {
	return storedCdr.ExtraFields
}

// Return cost as string, formated with number of decimals configured
func (storedCdr *StoredCdr) FormatCost(shiftDecimals, roundDecimals int) string {
	cost := storedCdr.Cost
	if shiftDecimals != 0 {
		cost = cost * math.Pow10(shiftDecimals)
	}
	return strconv.FormatFloat(cost, 'f', roundDecimals, 64)
}

func (storedCdr *StoredCdr) AsStoredCdr(runId, reqTypeFld, directionFld, tenantFld, torFld, accountFld, subjectFld, destFld, setupTimeFld, answerTimeFld, durationFld string, extraFlds []string, fieldsMandatory bool) (*StoredCdr, error) {
	return storedCdr, nil
}

// Converts part of the rated Cdr as httpForm used to post remotely to CDRS
func (storedCdr *StoredCdr) AsRawCdrHttpForm() url.Values {
	v := url.Values{}
	v.Set(ACCID, storedCdr.AccId)
	v.Set(CDRHOST, storedCdr.CdrHost)
	v.Set(CDRSOURCE, storedCdr.CdrSource)
	v.Set(REQTYPE, storedCdr.ReqType)
	v.Set(DIRECTION, storedCdr.Direction)
	v.Set(TENANT, storedCdr.Tenant)
	v.Set(TOR, storedCdr.TOR)
	v.Set(ACCOUNT, storedCdr.Account)
	v.Set(SUBJECT, storedCdr.Subject)
	v.Set(DESTINATION, storedCdr.Destination)
	v.Set(SETUP_TIME, storedCdr.SetupTime.String())
	v.Set(ANSWER_TIME, storedCdr.AnswerTime.String())
	v.Set(DURATION, strconv.FormatFloat(storedCdr.Duration.Seconds(), 'f', -1, 64))
	for fld, val := range storedCdr.ExtraFields {
		v.Set(fld, val)
	}
	return v
}

// Used to export fields as string, primary fields are const labeled
func (storedCdr *StoredCdr) ExportFieldValue(fldName string) string {
	switch fldName {
	case CGRID:
		return storedCdr.CgrId
	case ORDERID:
		return strconv.FormatInt(storedCdr.OrderId, 10)
	case ACCID:
		return storedCdr.AccId
	case CDRHOST:
		return storedCdr.CdrHost
	case CDRSOURCE:
		return storedCdr.CdrSource
	case REQTYPE:
		return storedCdr.ReqType
	case DIRECTION:
		return storedCdr.Direction
	case TENANT:
		return storedCdr.Tenant
	case TOR:
		return storedCdr.TOR
	case ACCOUNT:
		return storedCdr.Account
	case SUBJECT:
		return storedCdr.Subject
	case DESTINATION:
		return storedCdr.Destination
	case SETUP_TIME:
		return storedCdr.SetupTime.String()
	case ANSWER_TIME:
		return storedCdr.AnswerTime.String()
	case DURATION:
		return strconv.FormatFloat(storedCdr.Duration.Seconds(), 'f', -1, 64)
	case MEDI_RUNID:
		return storedCdr.MediationRunId
	case COST:
		return strconv.FormatFloat(storedCdr.Cost, 'f', -1, 64) // Recommended to use FormatCost
	default:
		return storedCdr.ExtraFields[fldName]
	}
}