bbgo_origin/pkg/service/account.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}
}

// 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
}