From ddf57df2ebc31c6565005311f768ab6d29a26f09 Mon Sep 17 00:00:00 2001 From: Radu Ioan Fericean Date: Thu, 28 Jan 2016 15:34:40 +0200 Subject: [PATCH] added DebitBalance api --- apier/v1/accounts.go | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/apier/v1/accounts.go b/apier/v1/accounts.go index 2290a13db..fa2991869 100644 --- a/apier/v1/accounts.go +++ b/apier/v1/accounts.go @@ -394,6 +394,13 @@ type AttrAddBalance struct { } func (self *ApierV1) AddBalance(attr *AttrAddBalance, reply *string) error { + return self.modifyBalance(engine.TOPUP, attr, reply) +} +func (self *ApierV1) DebitBalance(attr *AttrAddBalance, reply *string) error { + return self.modifyBalance(engine.DEBIT, attr, reply) +} + +func (self *ApierV1) modifyBalance(aType string, attr *AttrAddBalance, reply *string) error { if missing := utils.MissingStructFields(attr, []string{"Tenant", "Account", "BalanceType"}); len(missing) != 0 { return utils.NewErrMandatoryIeMissing(missing...) } @@ -416,9 +423,8 @@ func (self *ApierV1) AddBalance(attr *AttrAddBalance, reply *string) error { at := &engine.ActionTiming{} at.SetAccountIDs(utils.StringMap{accID: true}) - aType := engine.TOPUP if attr.Overwrite { - aType = engine.TOPUP_RESET + aType += "_reset" // => *topup_reset/*debit_reset } at.SetActions(engine.Actions{ &engine.Action{