66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
|
package service
|
||
|
|
|
||
|
|
import (
|
||
|
|
"git.qtrade.icu/lychiyu/bbgo/pkg/types"
|
||
|
|
"github.com/jmoiron/sqlx"
|
||
|
|
"go.uber.org/multierr"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type AccountService struct {
|
||
|
|
DB *sqlx.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewAccountService(db *sqlx.DB) *AccountService {
|
||
|
|
return &AccountService{DB: db}
|
||
|
|
}
|
||
|
|
|
||
|
|
// TODO: should pass bbgo.ExchangeSession to this function, but that might cause cyclic import
|
||
|
|
func (s *AccountService) InsertAsset(time time.Time, session string, name types.ExchangeName, account string, isMargin bool, isIsolatedMargin bool, isolatedMarginSymbol string, assets types.AssetMap) error {
|
||
|
|
if s.DB == nil {
|
||
|
|
// skip db insert when no db connection setting.
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
var err error
|
||
|
|
for _, v := range assets {
|
||
|
|
_, _err := s.DB.Exec(`
|
||
|
|
INSERT INTO nav_history_details (
|
||
|
|
session,
|
||
|
|
exchange,
|
||
|
|
subaccount,
|
||
|
|
time,
|
||
|
|
currency,
|
||
|
|
net_asset_in_usd,
|
||
|
|
net_asset_in_btc,
|
||
|
|
balance,
|
||
|
|
available,
|
||
|
|
locked,
|
||
|
|
borrowed,
|
||
|
|
net_asset,
|
||
|
|
price_in_usd,
|
||
|
|
is_margin, is_isolated, isolated_symbol)
|
||
|
|
values (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?);`,
|
||
|
|
session,
|
||
|
|
name,
|
||
|
|
account,
|
||
|
|
time,
|
||
|
|
v.Currency,
|
||
|
|
v.InUSD,
|
||
|
|
v.InBTC,
|
||
|
|
v.Total,
|
||
|
|
v.Available,
|
||
|
|
v.Locked,
|
||
|
|
v.Borrowed,
|
||
|
|
v.NetAsset,
|
||
|
|
v.PriceInUSD,
|
||
|
|
isMargin,
|
||
|
|
isIsolatedMargin,
|
||
|
|
isolatedMarginSymbol)
|
||
|
|
|
||
|
|
err = multierr.Append(err, _err) // successful request
|
||
|
|
|
||
|
|
}
|
||
|
|
return err
|
||
|
|
}
|