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
}