uppal/cgrates
1
0
Fork 1
mirror of https://github.com/cgrates/cgrates.git synced 2026-02-11 18:16:24 +05:00
Code Issues Packages Projects Releases Wiki Activity

Add context.Context in SetCDR func signature

Browse Source
This commit is contained in:
ionutboangiu
2023-12-08 08:43:28 -05:00
committed by Dan Christian Bogos
parent ab26ba2d02
commit 2ccd6ca74d
4 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
engine/cdrs.go
View File

@@ -324,13 +324,13 @@ func (cdrS *CDRServer) processEvents(ctx *context.Context, evs []*utils.CGREvent
if err != nil {
return nil, fmt.Errorf("retrieving %s option failed: %w", utils.MetaRerate, err)
}
if err := cdrS.db.SetCDR(cgrEv, false); err != nil {
if err := cdrS.db.SetCDR(ctx, cgrEv, false); err != nil {
if err != utils.ErrExists || !rerate {
// ToDo: add refund logic
return nil, fmt.Errorf("storing CDR %s failed: %w", utils.ToJSON(cgrEv), err)
}
if err = cdrS.db.SetCDR(cgrEv, true); err != nil {
if err = cdrS.db.SetCDR(ctx, cgrEv, true); err != nil {
utils.Logger.Warning(
fmt.Sprintf("<%s> error: <%s> updating CDR %+v",
utils.CDRs, err.Error(), utils.ToJSON(cgrEv)))

2
engine/storage_interface.go
View File

@@ -108,7 +108,7 @@ type DataDBDriver interface {
type StorDB interface {
Storage
SetCDR(*utils.CGREvent, bool) error
SetCDR(*context.Context, *utils.CGREvent, bool) error
GetCDRs(*context.Context, []*Filter, map[string]interface{}) ([]*CDR, error)
RemoveCDRs(*context.Context, []*Filter) error
}

2
engine/storage_internal_stordb.go
View File

@@ -28,7 +28,7 @@ import (
)
// SetCDR for ManagerDB interface. SetCDR will set a single CDR in internal based on the CGREvent
func (iDB *InternalDB) SetCDR(cgrEv *utils.CGREvent, allowUpdate bool) error {
func (iDB *InternalDB) SetCDR(_ *context.Context, cgrEv *utils.CGREvent, allowUpdate bool) error {
uniqueID := utils.IfaceAsString(cgrEv.APIOpts[utils.MetaCDRID])
if !allowUpdate {
if _, has := iDB.db.Get(utils.MetaCDRs, uniqueID); has {

2
engine/storage_sql.go
View File

@@ -141,7 +141,7 @@ func (sqls *SQLStorage) RemoveVersions(vrs Versions) (err error) {
return
}
func (sqls *SQLStorage) SetCDR(cdr *utils.CGREvent, allowUpdate bool) error {
func (sqls *SQLStorage) SetCDR(_ *context.Context, cdr *utils.CGREvent, allowUpdate bool) error {
tx := sqls.db.Begin()
if tx.Error != nil {
return tx.Error