48 lines
782 B
Go
48 lines
782 B
Go
package trade
|
||
|
||
import (
|
||
"sync"
|
||
)
|
||
|
||
var (
|
||
AccountsAssets *Assets
|
||
)
|
||
|
||
type AccountsBalance struct {
|
||
AccountEquity float64 // 账户权益(保证金币种)，包含未实现盈亏（根据mark price计算）
|
||
Available float64 // 账户可用数量
|
||
}
|
||
|
||
// lock
|
||
type Assets struct {
|
||
sync.RWMutex
|
||
Data map[string]*AccountsBalance
|
||
}
|
||
|
||
func NewAccounts() {
|
||
AccountsAssets = &Assets{
|
||
Data: make(map[string]*AccountsBalance),
|
||
}
|
||
}
|
||
|
||
func (ac *Assets) Set(assets string, balance *AccountsBalance) {
|
||
ac.Lock()
|
||
defer ac.Unlock()
|
||
|
||
ac.Data[assets] = balance
|
||
}
|
||
|
||
func (ac *Assets) SetData(data map[string]*AccountsBalance) {
|
||
ac.Lock()
|
||
defer ac.Unlock()
|
||
|
||
ac.Data = data
|
||
}
|
||
|
||
func (ac *Assets) Get(assets string) *AccountsBalance {
|
||
ac.Lock()
|
||
defer ac.Unlock()
|
||
|
||
return ac.Data[assets]
|
||
}
|