cgrates/utils/storedcdr.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 utils

import (
	"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
	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 {
	return storedCdr.Duration
}

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

// Return cost as string, formated with number of decimals configured
func (storedCdr *StoredCdr) FormatCost(roundDecimals int) string {
	return strconv.FormatFloat(storedCdr.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 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]
	}
}