From 9385a3389e5899057d43c9546555e7d8aac9fab3 Mon Sep 17 00:00:00 2001
From: TeoV <teofil.voivozeanu@itsyscom.com>
Date: Mon, 11 Nov 2019 16:50:14 +0200
Subject: [PATCH] Populate Timing when using AddBalance API

---
 apier/v1/accounts.go | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/apier/v1/accounts.go b/apier/v1/accounts.go
index 52c67586c..bea313852 100644
--- a/apier/v1/accounts.go
+++ b/apier/v1/accounts.go
@@ -490,6 +490,23 @@ func (self *ApierV1) modifyBalance(aType string, attr *AttrAddBalance, reply *st
 	if attr.TimingIds != nil {
 		a.Balance.TimingIDs = utils.StringMapPointer(utils.ParseStringMap(*attr.TimingIds))
 	}
+	if attr.TimingIds != nil && *attr.TimingIds != "" {
+		timingIds := strings.Split(*attr.TimingIds, utils.INFIELD_SEP)
+		for _, timingID := range timingIds {
+			timing, err := self.DataManager.GetTiming(timingID, false, utils.NonTransactional)
+			if err != nil {
+				return err
+			}
+			a.Balance.Timings = append(a.Balance.Timings, &engine.RITiming{
+				Years:     timing.Years,
+				Months:    timing.Months,
+				MonthDays: timing.MonthDays,
+				WeekDays:  timing.WeekDays,
+				StartTime: timing.StartTime,
+				EndTime:   timing.EndTime,
+			})
+		}
+	}
 	publishAction := &engine.Action{
 		ActionType: utils.MetaPublishBalance,
 	}