cgrates/console/balance.go

/* Implementing balance related console commands.
 */
package console

import (
	"fmt"
)

// Data being sent to the rpc responder
type ArgsGetBalance struct {
	Tenant    string
	User      string
	Direction string
	BalanceId string
}

// Received back from query
type ReplyGetBalance struct {
	Tenant    string
	User      string
	Direction string
	BalanceId string
	Balance   float64
}

// Commander implementation
type CmdGetBalance struct {
	rpcMethod        string
	rpcParams        *ArgsGetBalance
	rpcResult        *ReplyGetBalance
	idxArgsToRpcPrms map[int]string
}

// name should be exec's name
func (self *CmdGetBalance) Usage(name string) string {
	return fmt.Sprintf("\n\tUsage: %s [cfg_opts...{-h}] get_balance <tenant> <user> [<balanceid> [<direction>]]")
}

// set param defaults
func (self *CmdGetBalance) defaults() error {
	self.idxArgsToRpcPrms = map[int]string{2: "Tenant", 3: "User", 4: "BalanceId", 5: "Direction"}
	self.rpcMethod = "Responder.GetBalance"
	self.rpcParams = &ArgsGetBalance{BalanceId: "MONETARY", Direction: "OUT"}
	return nil
}

// Parses command line args and builds CmdBalance value
func (self *CmdGetBalance) FromArgs(args []string) error {
	if len(args) < 4 {
		return fmt.Errorf(self.Usage(""))
	}
	// Args look OK, set defaults before going further	
	self.defaults()
	self.rpcParams = &ArgsGetBalance{
		Tenant:    args[2],
		User:      args[3],
		BalanceId: args[4],
		Direction: args[5],
	}
	return nil
}

func (self *CmdGetBalance) RpcMethod() string {
	return self.rpcMethod
}

func (self *CmdGetBalance) RpcParams() interface{} {
	return self.rpcParams
}

func (self *CmdGetBalance) RpcResult() interface{} {
	self.rpcResult = &ReplyGetBalance{}
	return self.rpcResult
}