修复 准点播报功能

This commit is contained in:
wood chen 2024-09-20 16:18:15 +08:00
parent 2af6b41866
commit ff668a0ef7
2 changed files with 14 additions and 19 deletions

View File

@ -76,7 +76,7 @@ func sendPriceUpdate() {
message := fmt.Sprintf("市场更新 - %s (SGT)\n\n", now.Format("2006-01-02 15:04:05")) message := fmt.Sprintf("市场更新 - %s (SGT)\n\n", now.Format("2006-01-02 15:04:05"))
for _, symbol := range symbols { for _, symbol := range symbols {
info, err := getTickerInfo(symbol) // 直接使用完整的交易对名称 info, err := getTickerInfo(symbol)
if err != nil { if err != nil {
log.Printf("Error getting ticker info for %s: %v", symbol, err) log.Printf("Error getting ticker info for %s: %v", symbol, err)
continue continue
@ -118,9 +118,9 @@ func RunBinance() {
symbols = core.Symbols symbols = core.Symbols
singaporeTZ = core.SingaporeTZ singaporeTZ = core.SingaporeTZ
// 初始化并加载交易对 // 初始化并加载所有交易对
if err := LoadSymbols(); err != nil { if err := LoadAllSymbols(); err != nil {
log.Fatalf("Failed to load trading pairs: %v", err) log.Fatalf("Failed to load all trading pairs: %v", err)
} }
// 启动每小时刷新交易对缓存 // 启动每小时刷新交易对缓存

View File

@ -14,10 +14,11 @@ import (
var ( var (
symbolsMu sync.RWMutex symbolsMu sync.RWMutex
allSymbols []string
) )
// LoadSymbols 初始化并缓存所有交易对 // LoadSymbols 初始化并缓存所有交易对
func LoadSymbols() error { func LoadAllSymbols() error {
client := binance.NewClient("", "") client := binance.NewClient("", "")
exchangeInfo, err := client.NewExchangeInfoService().Do(context.Background()) exchangeInfo, err := client.NewExchangeInfoService().Do(context.Background())
if err != nil { if err != nil {
@ -27,47 +28,41 @@ func LoadSymbols() error {
symbolsMu.Lock() symbolsMu.Lock()
defer symbolsMu.Unlock() defer symbolsMu.Unlock()
symbols = nil // 清空旧的符号列表 allSymbols = nil // 清空旧的符号列表
for _, symbol := range exchangeInfo.Symbols { for _, symbol := range exchangeInfo.Symbols {
if symbol.Status == "TRADING" && symbol.QuoteAsset == "USDT" { if symbol.Status == "TRADING" && symbol.QuoteAsset == "USDT" {
symbols = append(symbols, symbol.Symbol) // 使用完整的交易对名称 allSymbols = append(allSymbols, symbol.Symbol)
} }
} }
log.Printf("Loaded %d valid USDT trading pairs", len(symbols)) log.Printf("Loaded %d valid USDT trading pairs", len(allSymbols))
return nil return nil
} }
// GetAllSymbols 获取缓存的交易对列表
func GetAllSymbols() []string { func GetAllSymbols() []string {
symbolsMu.RLock() symbolsMu.RLock()
defer symbolsMu.RUnlock() defer symbolsMu.RUnlock()
return symbols return allSymbols
} }
// StartSymbolRefresh 每小时刷新一次交易对缓存
func StartSymbolRefresh(interval time.Duration) { func StartSymbolRefresh(interval time.Duration) {
ticker := time.NewTicker(interval) ticker := time.NewTicker(interval)
go func() { go func() {
for range ticker.C { for range ticker.C {
log.Println("Refreshing trading pairs...") log.Println("Refreshing trading pairs...")
if err := LoadSymbols(); err != nil { if err := LoadAllSymbols(); err != nil {
log.Printf("Failed to refresh symbols: %v", err) log.Printf("Failed to refresh symbols: %v", err)
} }
} }
}() }()
} }
// HandleSymbolQuery 处理虚拟币名查询
func HandleSymbolQuery(bot *tgbotapi.BotAPI, message *tgbotapi.Message) { func HandleSymbolQuery(bot *tgbotapi.BotAPI, message *tgbotapi.Message) {
symbols := GetAllSymbols() allSymbols := GetAllSymbols()
msg := strings.TrimSpace(message.Text) msg := strings.TrimSpace(message.Text)
for _, symbol := range symbols { for _, symbol := range allSymbols {
// 移除USDT后缀
coinName := strings.TrimSuffix(symbol, "USDT") coinName := strings.TrimSuffix(symbol, "USDT")
// 不区分大小写的完全匹配检查
if strings.EqualFold(msg, coinName) { if strings.EqualFold(msg, coinName) {
info, err := getTickerInfo(symbol) info, err := getTickerInfo(symbol)
if err != nil { if err != nil {