fix update

tron_usdt_monitor.go

@@ -6,23 +6,38 @@ import (
 	"fmt"
 	"io"
 	"net/http"
+	"net/url"
 	"strings"
 	"sync"
 	"time"
 )
 
+// Config USDTMan 配置
+type Config struct {
+	Addresses        []string          // 监听地址列表
+	APIKey           string            // TronGrid API Key
+	QueryInterval    time.Duration     // 查询间隔（秒），默认 5 秒
+	MinConfirmations int64             // 最小确认数，默认 6
+	MaxHistoryTxns   int               // 每次查询的最大历史交易数，默认 20
+	ProxyURL         string            // HTTP/SOCKS5 代理地址，例如 "http://127.0.0.1:7890" 或 "socks5://127.0.0.1:1080"
+	Transport        http.RoundTripper // 自定义 Transport（优先级高于 ProxyURL）
+}
+
 // USDTMan USDT 监听管理器
 type USDTMan struct {
 	mu                sync.RWMutex
 	addresses         []string
 	apiKey            string
+	queryInterval     time.Duration
 	running           bool
 	ctx               context.Context
 	cancel            context.CancelFunc
 	onPaymentComplete func(*USDTPayment)
 	minConfirmations  int64
+	maxHistoryTxns    int
 	processedTxns     map[string]bool // 已处理的交易
 	txnMutex          sync.RWMutex
+	httpClient        *http.Client // HTTP 客户端
 }
 
 // USDTPayment USDT 收款信息
@@ -80,14 +95,50 @@ const (
 )
 
 // NewUSDTMan 创建监听管理器
-func NewUSDTMan(addresses []string, apiKey string) *USDTMan {
+func NewUSDTMan(config Config) *USDTMan {
 	ctx, cancel := context.WithCancel(context.Background())
+
+	// 设置默认值
+	if config.QueryInterval == 0 {
+		config.QueryInterval = 5 * time.Second
+	}
+	if config.MinConfirmations == 0 {
+		config.MinConfirmations = 6
+	}
+	if config.MaxHistoryTxns == 0 {
+		config.MaxHistoryTxns = 20
+	}
+
+	// 创建 HTTP 客户端
+	httpClient := &http.Client{
+		Timeout: 10 * time.Second,
+	}
+
+	// 配置代理
+	if config.Transport != nil {
+		// 使用自定义 Transport
+		httpClient.Transport = config.Transport
+	} else if config.ProxyURL != "" {
+		// 使用代理 URL
+		proxyURL, err := url.Parse(config.ProxyURL)
+		if err == nil {
+			httpClient.Transport = &http.Transport{
+				Proxy: http.ProxyURL(proxyURL),
+			}
+		} else {
+			fmt.Printf("⚠️  代理 URL 解析失败: %v，使用直连\n", err)
+		}
+	}
+
 	return &USDTMan{
-		addresses:        addresses,
-		apiKey:           apiKey,
+		addresses:        config.Addresses,
+		apiKey:           config.APIKey,
+		queryInterval:    config.QueryInterval,
+		minConfirmations: config.MinConfirmations,
+		maxHistoryTxns:   config.MaxHistoryTxns,
+		httpClient:       httpClient,
 		ctx:              ctx,
 		cancel:           cancel,
-		minConfirmations: 6,
 		processedTxns:    make(map[string]bool),
 	}
 }
@@ -128,7 +179,7 @@ func (m *USDTMan) Stop() {
 
 // monitorLoop 监听循环
 func (m *USDTMan) monitorLoop() {
-	ticker := time.NewTicker(5 * time.Second)
+	ticker := time.NewTicker(m.queryInterval)
 	defer ticker.Stop()
 
 	for {
@@ -207,8 +258,8 @@ func (m *USDTMan) checkAddress(address string) {
 
 // getUSDTTransactions 获取地址的 USDT 交易
 func (m *USDTMan) getUSDTTransactions(address string) ([]TronGridTransaction, error) {
-	url := fmt.Sprintf("%s/v1/accounts/%s/transactions/trc20?limit=20&contract_address=%s&only_to=true",
-		TronGridAPI, address, USDTContractAddress)
+	url := fmt.Sprintf("%s/v1/accounts/%s/transactions/trc20?limit=%d&contract_address=%s&only_to=true",
+		TronGridAPI, address, m.maxHistoryTxns, USDTContractAddress)
 
 	req, err := http.NewRequest("GET", url, nil)
 	if err != nil {
@@ -219,8 +270,7 @@ func (m *USDTMan) getUSDTTransactions(address string) ([]TronGridTransaction, er
 		req.Header.Set("TRON-PRO-API-KEY", m.apiKey)
 	}
 
-	client := &http.Client{Timeout: 10 * time.Second}
-	resp, err := client.Do(req)
+	resp, err := m.httpClient.Do(req)
 	if err != nil {
 		return nil, err
 	}
@@ -256,8 +306,7 @@ func (m *USDTMan) getCurrentBlock() (int64, error) {
 		req.Header.Set("TRON-PRO-API-KEY", m.apiKey)
 	}
 
-	client := &http.Client{Timeout: 10 * time.Second}
-	resp, err := client.Do(req)
+	resp, err := m.httpClient.Do(req)
 	if err != nil {
 		return 0, err
 	}