aclindsa/moneygo
1
0
Fork 0
mirror of https://github.com/aclindsa/moneygo.git synced 2025-07-01 12:08:37 -04:00
Code Issues Releases Wiki Activity

lua: Query account balances at dates

Browse Source
This commit is contained in:
Aaron Lindsay
2017-01-30 21:04:18 -05:00
parent b7d2273b1c
commit cec769b6b8
4 changed files with 112 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
accounts_lua.go
View File

@ -4,6 +4,7 @@ import (
"context"
"errors"
"github.com/yuin/gopher-lua"
"math/big"
)
const luaAccountTypeName = "account"
@ -61,9 +62,9 @@ func luaRegisterAccounts(L *lua.LState) {
L.SetField(mt, "__tostring", L.NewFunction(luaAccount__tostring))
L.SetField(mt, "__eq", L.NewFunction(luaAccount__eq))
L.SetField(mt, "__metatable", lua.LString("protected"))
getAccountsFn := L.NewFunction(luaGetAccounts)
L.SetField(mt, "get_all", getAccountsFn)
// also register the get_accounts function as a global in its own right
L.SetGlobal("get_accounts", getAccountsFn)
}
@ -121,27 +122,7 @@ func luaAccount__index(L *lua.LState) int {
case "Type", "type":
L.Push(lua.LNumber(float64(a.Type)))
case "Balance", "balance":
ctx := L.Context()
user, ok := ctx.Value(userContextKey).(*User)
if !ok {
panic("Couldn't find User in lua's Context")
}
security_map, err := luaContextGetSecurities(L)
if err != nil {
panic("account.security couldn't fetch securities")
}
security, ok := security_map[a.SecurityId]
if !ok {
panic("SecurityId not in lua security_map")
}
rat, err := GetAccountBalance(user, a.AccountId)
if err != nil {
panic("Failed to GetAccountBalance:" + err.Error())
}
var b Balance
b.Security = security
b.Amount = rat
L.Push(BalanceToLua(L, &b))
L.Push(L.NewFunction(luaAccountBalance))
default:
L.ArgError(2, "unexpected account attribute: "+field)
}
@ -149,6 +130,45 @@ func luaAccount__index(L *lua.LState) int {
return 1
}
func luaAccountBalance(L *lua.LState) int {
a := luaCheckAccount(L, 1)
ctx := L.Context()
user, ok := ctx.Value(userContextKey).(*User)
if !ok {
panic("Couldn't find User in lua's Context")
}
security_map, err := luaContextGetSecurities(L)
if err != nil {
panic("account.security couldn't fetch securities")
}
security, ok := security_map[a.SecurityId]
if !ok {
panic("SecurityId not in lua security_map")
}
date := luaWeakCheckTime(L, 2)
var b Balance
var rat *big.Rat
if date != nil {
end := luaWeakCheckTime(L, 3)
if end != nil {
rat, err = GetAccountBalanceDateRange(user, a.AccountId, date, end)
} else {
rat, err = GetAccountBalanceDate(user, a.AccountId, date)
}
} else {
rat, err = GetAccountBalance(user, a.AccountId)
}
if err != nil {
panic("Failed to GetAccountBalance:" + err.Error())
}
b.Amount = rat
b.Security = security
L.Push(BalanceToLua(L, &b))
return 1
}
func luaAccount__tostring(L *lua.LState) int {
a := luaCheckAccount(L, 1)