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