mirror of
https://github.com/aclindsa/moneygo.git
synced 2024-10-30 07:40:05 -04:00
Add Trading account type and method to automatically create them
This commit is contained in:
parent
2088065603
commit
df1912f69b
68
accounts.go
68
accounts.go
@ -18,6 +18,7 @@ const (
|
||||
Investment = 5
|
||||
Income = 6
|
||||
Expense = 7
|
||||
Trading = 8
|
||||
)
|
||||
|
||||
type Account struct {
|
||||
@ -93,6 +94,73 @@ func GetAccounts(userid int64) (*[]Account, error) {
|
||||
return &accounts, nil
|
||||
}
|
||||
|
||||
// Get (and attempt to create if it doesn't exist) the security/currency
|
||||
// trading account for the supplied security/currency
|
||||
func GetTradingAccount(userid int64, securityid int64) (*Account, error) {
|
||||
var tradingAccounts []Account //top-level 'Trading' account(s)
|
||||
var tradingAccount Account
|
||||
var accounts []Account //second-level security-specific trading account(s)
|
||||
var account Account
|
||||
|
||||
transaction, err := DB.Begin()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// Try to find the top-level trading account
|
||||
_, err = transaction.Select(&tradingAccounts, "SELECT * from accounts where UserId=? AND Name='Trading' AND ParentAccountId=-1 AND Type=? ORDER BY AccountId ASC LIMIT 1", userid, Trading)
|
||||
if err != nil {
|
||||
transaction.Rollback()
|
||||
return nil, err
|
||||
}
|
||||
if len(tradingAccounts) == 1 {
|
||||
tradingAccount = tradingAccounts[0]
|
||||
} else {
|
||||
tradingAccount.UserId = userid
|
||||
tradingAccount.Name = "Trading"
|
||||
tradingAccount.ParentAccountId = -1
|
||||
tradingAccount.SecurityId = 840 /*USD*/ //FIXME SecurityId shouldn't matter for top-level trading account, but maybe we should grab the user's default
|
||||
tradingAccount.Type = Trading
|
||||
|
||||
err = transaction.Insert(&tradingAccount)
|
||||
if err != nil {
|
||||
transaction.Rollback()
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
// Now, try to find the security-specific trading account
|
||||
_, err = transaction.Select(&accounts, "SELECT * from accounts where UserId=? AND SecurityId=? AND ParentAccountId=? ORDER BY AccountId ASC LIMIT 1", userid, securityid, tradingAccount.AccountId)
|
||||
if err != nil {
|
||||
transaction.Rollback()
|
||||
return nil, err
|
||||
}
|
||||
if len(accounts) == 1 {
|
||||
account = accounts[0]
|
||||
} else {
|
||||
security := GetSecurity(securityid)
|
||||
account.UserId = userid
|
||||
account.Name = security.Name
|
||||
account.ParentAccountId = tradingAccount.AccountId
|
||||
account.SecurityId = securityid
|
||||
account.Type = Trading
|
||||
|
||||
err = transaction.Insert(&account)
|
||||
if err != nil {
|
||||
transaction.Rollback()
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
err = transaction.Commit()
|
||||
if err != nil {
|
||||
transaction.Rollback()
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &account, nil
|
||||
}
|
||||
|
||||
type ParentAccountMissingError struct{}
|
||||
|
||||
func (pame ParentAccountMissingError) Error() string {
|
||||
|
@ -133,7 +133,8 @@ const AccountType = {
|
||||
Liability: 4,
|
||||
Investment: 5,
|
||||
Income: 6,
|
||||
Expense: 7
|
||||
Expense: 7,
|
||||
Trading: 8
|
||||
}
|
||||
var AccountTypeList = [];
|
||||
for (var type in AccountType) {
|
||||
|
Loading…
Reference in New Issue
Block a user