Q58Bot/service/binance/symbols.go

package binance

import (
	"context"
	"fmt"
	"log"
	"strings"
	"sync"
	"time"

	"github.com/adshao/go-binance/v2"
	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
)

var (
	symbolsMu sync.RWMutex
)

// LoadSymbols 初始化并缓存所有交易对
func LoadSymbols() error {
	client := binance.NewClient("", "")
	exchangeInfo, err := client.NewExchangeInfoService().Do(context.Background())
	if err != nil {
		return err
	}

	symbolsMu.Lock()
	defer symbolsMu.Unlock()

	symbols = nil // 清空旧的符号列表
	for _, symbol := range exchangeInfo.Symbols {
		if symbol.Status == "TRADING" && symbol.QuoteAsset == "USDT" {
			symbols = append(symbols, symbol.Symbol) // 使用完整的交易对名称
		}
	}

	log.Printf("Loaded %d valid USDT trading pairs", len(symbols))
	return nil
}

// GetAllSymbols 获取缓存的交易对列表
func GetAllSymbols() []string {
	symbolsMu.RLock()
	defer symbolsMu.RUnlock()
	return symbols
}

// StartSymbolRefresh 每小时刷新一次交易对缓存
func StartSymbolRefresh(interval time.Duration) {
	ticker := time.NewTicker(interval)
	go func() {
		for range ticker.C {
			log.Println("Refreshing trading pairs...")
			if err := LoadSymbols(); err != nil {
				log.Printf("Failed to refresh symbols: %v", err)
			}
		}
	}()
}

// HandleSymbolQuery 处理虚拟币名查询
func HandleSymbolQuery(bot *tgbotapi.BotAPI, message *tgbotapi.Message) {
	symbols := GetAllSymbols()
	msg := strings.TrimSpace(message.Text)

	for _, symbol := range symbols {
		// 移除USDT后缀
		coinName := strings.TrimSuffix(symbol, "USDT")

		// 不区分大小写的完全匹配检查
		if strings.EqualFold(msg, coinName) {
			info, err := getTickerInfo(symbol)
			if err != nil {
				log.Printf("Error getting ticker info for %s: %v", symbol, err)
				return
			}
			replyMessage := fmt.Sprintf("*%s*\n价格: $%.7f\n24h 涨跌: %s\n",
				info.symbol,
				info.last,
				formatChange(info.changePercent))
			replyMsg := tgbotapi.NewMessage(message.Chat.ID, replyMessage)
			replyMsg.ParseMode = "Markdown"
			bot.Send(replyMsg)
			return
		}
	}
}