mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-14 02:53:50 +00:00
38 lines
911 B
Go
38 lines
911 B
Go
package service
|
|
|
|
import (
|
|
"github.com/c9s/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}
|
|
}
|
|
|
|
func (s *AccountService) InsertAsset(time time.Time, name types.ExchangeName, account 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 ( exchange, subaccount, time, currency, balance_in_usd, balance_in_btc,
|
|
balance,available,locked)
|
|
values (?,?,?,?,?,?,?,?,?);
|
|
`, name, account, time, v.Currency, v.InUSD, v.InBTC, v.Total, v.Available, v.Locked)
|
|
|
|
err = multierr.Append(err, _err) // successful request
|
|
|
|
}
|
|
return err
|
|
}
|