cgrates/utils/ratedcdr.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 NewRatedCDRFromRawCDR(rawcdr RawCDR) (*RatedCDR, error) {
	var err error
	rtCdr := new(RatedCDR)
	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.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 RatedCDR struct {
	CgrId          string
	AccId          string
	CdrHost        string
	CdrSource      string
	ReqType        string
	Direction      string
	Tenant         string
	TOR            string
	Account        string
	Subject        string
	Destination    string
	AnswerTime     time.Time
	Duration       time.Duration
	ExtraFields    map[string]string
	MediationRunId string
	Cost           float64
}

// Methods maintaining RawCDR interface

func (ratedCdr *RatedCDR) GetCgrId() string {
	return ratedCdr.CgrId
}

func (ratedCdr *RatedCDR) GetAccId() string {
	return ratedCdr.AccId
}

func (ratedCdr *RatedCDR) GetCdrHost() string {
	return ratedCdr.CdrHost
}

func (ratedCdr *RatedCDR) GetCdrSource() string {
	return ratedCdr.CdrSource
}

func (ratedCdr *RatedCDR) GetDirection() string {
	return ratedCdr.Direction
}

func (ratedCdr *RatedCDR) GetSubject() string {
	return ratedCdr.Subject
}

func (ratedCdr *RatedCDR) GetAccount() string {
	return ratedCdr.Account
}

func (ratedCdr *RatedCDR) GetDestination() string {
	return ratedCdr.Destination
}

func (ratedCdr *RatedCDR) GetTOR() string {
	return ratedCdr.TOR
}

func (ratedCdr *RatedCDR) GetTenant() string {
	return ratedCdr.Tenant
}

func (ratedCdr *RatedCDR) GetReqType() string {
	return ratedCdr.ReqType
}

func (ratedCdr *RatedCDR) GetAnswerTime() (time.Time, error) {
	return ratedCdr.AnswerTime, nil
}

func (ratedCdr *RatedCDR) GetDuration() time.Duration {
	return ratedCdr.Duration
}

func (ratedCdr *RatedCDR) GetExtraFields() map[string]string {
	return ratedCdr.ExtraFields
}

func (ratedCdr *RatedCDR) AsRatedCdr(runId, reqTypeFld, directionFld, tenantFld, torFld, accountFld, subjectFld, destFld, answerTimeFld, durationFld string, extraFlds []string, fieldsMandatory bool) (*RatedCDR, error) {
	return ratedCdr, nil
}

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