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, }