80 lines
2.0 KiB
Go
80 lines
2.0 KiB
Go
|
package logic
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"strconv"
|
|||
|
|
|
|||
|
|
"git.apinb.com/quant/coin/internal/impl"
|
|||
|
|
"git.apinb.com/quant/coin/internal/models"
|
|||
|
|
"github.com/adshao/go-binance/v2"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// refreshStepSizes 从 exchangeInfo 拉取各 symbol 的 LOT_SIZE.stepSize，供卖出数量格式化。
|
|||
|
|
func refreshStepSizes(ctx context.Context, client *binance.Client) error {
|
|||
|
|
watch := spotWatchesFromConfig()
|
|||
|
|
syms := make([]string, 0, len(watch))
|
|||
|
|
for _, w := range watch {
|
|||
|
|
syms = append(syms, w.Symbol)
|
|||
|
|
}
|
|||
|
|
info, err := client.NewExchangeInfoService().Symbols(syms...).Do(ctx)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
for _, s := range info.Symbols {
|
|||
|
|
lot := s.LotSizeFilter()
|
|||
|
|
if lot == nil || lot.StepSize == "" {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
stepSizes[s.Symbol] = lot.StepSize
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// loadPortfolio 从 GORM 读入全表 spot_positions，填充内存 map（键为 BaseAsset）。
|
|||
|
|
func loadPortfolio() error {
|
|||
|
|
portfolioMu.Lock()
|
|||
|
|
defer portfolioMu.Unlock()
|
|||
|
|
if impl.DBService == nil {
|
|||
|
|
portfolio = models.NewSpotPortfolioSnapshot()
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
var rows []models.SpotPosition
|
|||
|
|
if err := impl.DBService.Find(&rows).Error; err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
portfolio = models.NewSpotPortfolioSnapshot()
|
|||
|
|
for i := range rows {
|
|||
|
|
p := new(models.SpotPosition)
|
|||
|
|
*p = rows[i]
|
|||
|
|
portfolio.Positions[p.BaseAsset] = p
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// savePortfolioLocked 将内存中各 SpotPosition 以 Save 写回数据库（有主键则更新，无则插入）。
|
|||
|
|
// 调用方必须已持有 portfolioMu。
|
|||
|
|
func savePortfolioLocked() error {
|
|||
|
|
if impl.DBService == nil {
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
for _, st := range portfolio.Positions {
|
|||
|
|
if st == nil {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
if err := impl.DBService.Save(st).Error; err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// balanceFree 从账户余额列表里解析某资产的可用数量（Free 字段为字符串）。
|
|||
|
|
func balanceFree(balances []binance.Balance, asset string) (float64, error) {
|
|||
|
|
for _, b := range balances {
|
|||
|
|
if b.Asset == asset {
|
|||
|
|
return strconv.ParseFloat(b.Free, 64)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return 0, nil
|
|||
|
|
}
|