usdtman

2026-02-03 00:01:41 +08:00
parent 3d3e814879
commit 79c1c710a2
4 changed files with 387 additions and 318 deletions
--- a/tron_usdt_monitor.go
+++ b/tron_usdt_monitor.go
@@ -11,95 +11,96 @@ import (
 	"time"
 )

-// TronUSDTMonitor 监听 USDT 收款
-type TronUSDTMonitor struct {
+// USDTMan USDT 监听管理器
+type USDTMan struct {
 	mu                sync.RWMutex
-	addresses         map[string]*AddressInfo // 地址映射
+	addresses         []string
 	apiKey            string
 	running           bool
 	ctx               context.Context
 	cancel            context.CancelFunc
-	onPaymentReceived func(*USDTPayment)
-}
-
-// AddressInfo 地址信息
-type AddressInfo struct {
-	Address       string
-	LastBlock     int64
-	TotalReceived float64
+	onPaymentComplete func(*USDTPayment)
+	minConfirmations  int64
+	processedTxns     map[string]bool // 已处理的交易
+	txnMutex          sync.RWMutex
 }

 // USDTPayment USDT 收款信息
 type USDTPayment struct {
-	Address     string
-	Amount      float64
-	TxID        string
-	BlockNumber int64
-	Timestamp   int64
-	From        string
+	Address       string
+	Amount        float64
+	TxID          string
+	BlockNumber   int64
+	Timestamp     int64
+	From          string
+	Confirmations int64
 }

-// TronGridEvent TronGrid 事件响应
-type TronGridEvent struct {
-	Success bool `json:"success"`
-	Data    []struct {
-		TransactionID   string                 `json:"transaction_id"`
-		BlockNumber     int64                  `json:"block_number"`
-		BlockTimestamp  int64                  `json:"block_timestamp"`
-		ContractAddress string                 `json:"contract_address"`
-		EventName       string                 `json:"event_name"`
-		Result          map[string]interface{} `json:"result"`
-	} `json:"data"`
-	Meta struct {
-		PageSize int `json:"page_size"`
+// TronGridTransaction 交易信息
+type TronGridTransaction struct {
+	Ret            []map[string]interface{} `json:"ret"`
+	TxID           string                   `json:"txID"`
+	BlockNumber    int64                    `json:"blockNumber"`
+	BlockTimeStamp int64                    `json:"block_timestamp"`
+	RawData        struct {
+		Contract []struct {
+			Parameter struct {
+				Value struct {
+					Amount int64  `json:"amount"`
+					To     string `json:"to_address"`
+					From   string `json:"owner_address"`
+				} `json:"value"`
+			} `json:"parameter"`
+		} `json:"contract"`
+	} `json:"raw_data"`
+}
+
+// TronGridAccountTransactions 账户交易列表
+type TronGridAccountTransactions struct {
+	Success bool                  `json:"success"`
+	Data    []TronGridTransaction `json:"data"`
+	Meta    struct {
+		PageSize    int    `json:"page_size"`
+		Fingerprint string `json:"fingerprint"`
 	} `json:"meta"`
 }

+// TronGridBlockInfo 区块信息
+type TronGridBlockInfo struct {
+	BlockHeader struct {
+		RawData struct {
+			Number int64 `json:"number"`
+		} `json:"raw_data"`
+	} `json:"block_header"`
+}
+
 const (
 	TronGridAPI         = "https://api.trongrid.io"
 	USDTContractAddress = "TR7NHqjeKQxGTCi8q8ZY4pL8otSzgjLj6t" // USDT TRC20
 )

-// NewTronUSDTMonitor 创建监听器
-func NewTronUSDTMonitor(apiKey string) *TronUSDTMonitor {
+// NewUSDTMan 创建监听管理器
+func NewUSDTMan(addresses []string, apiKey string) *USDTMan {
 	ctx, cancel := context.WithCancel(context.Background())
-	return &TronUSDTMonitor{
-		addresses: make(map[string]*AddressInfo),
-		apiKey:    apiKey,
-		ctx:       ctx,
-		cancel:    cancel,
+	return &USDTMan{
+		addresses:        addresses,
+		apiKey:           apiKey,
+		ctx:              ctx,
+		cancel:           cancel,
+		minConfirmations: 6,
+		processedTxns:    make(map[string]bool),
 	}
 }

-// AddAddress 添加监听地址
-func (m *TronUSDTMonitor) AddAddress(address string) {
+// OnPaymentComplete 设置收款回调
+func (m *USDTMan) OnPaymentComplete(callback func(*USDTPayment)) {
 	m.mu.Lock()
 	defer m.mu.Unlock()
-
-	if _, exists := m.addresses[address]; !exists {
-		m.addresses[address] = &AddressInfo{
-			Address:   address,
-			LastBlock: 0,
-		}
-	}
-}
-
-// RemoveAddress 移除监听地址
-func (m *TronUSDTMonitor) RemoveAddress(address string) {
-	m.mu.Lock()
-	defer m.mu.Unlock()
-	delete(m.addresses, address)
-}
-
-// SetPaymentCallback 设置收款回调
-func (m *TronUSDTMonitor) SetPaymentCallback(callback func(*USDTPayment)) {
-	m.mu.Lock()
-	defer m.mu.Unlock()
-	m.onPaymentReceived = callback
+	m.onPaymentComplete = callback
 }

 // Start 开始监听
-func (m *TronUSDTMonitor) Start() error {
+func (m *USDTMan) Start() error {
 	m.mu.Lock()
 	if m.running {
 		m.mu.Unlock()
@@ -113,7 +114,7 @@ func (m *TronUSDTMonitor) Start() error {
 }

 // Stop 停止监听
-func (m *TronUSDTMonitor) Stop() {
+func (m *USDTMan) Stop() {
 	m.mu.Lock()
 	if !m.running {
 		m.mu.Unlock()
@@ -126,8 +127,8 @@ func (m *TronUSDTMonitor) Stop() {
 }

 // monitorLoop 监听循环
-func (m *TronUSDTMonitor) monitorLoop() {
-	ticker := time.NewTicker(3 * time.Second)
+func (m *USDTMan) monitorLoop() {
+	ticker := time.NewTicker(5 * time.Second)
 	defer ticker.Stop()

 	for {
@@ -141,64 +142,61 @@ func (m *TronUSDTMonitor) monitorLoop() {
 }

 // checkAllAddresses 检查所有地址
-func (m *TronUSDTMonitor) checkAllAddresses() {
+func (m *USDTMan) checkAllAddresses() {
 	m.mu.RLock()
-	addresses := make([]*AddressInfo, 0, len(m.addresses))
-	for _, info := range m.addresses {
-		addresses = append(addresses, info)
-	}
+	addresses := make([]string, len(m.addresses))
+	copy(addresses, m.addresses)
 	m.mu.RUnlock()

-	for _, addrInfo := range addresses {
-		m.checkAddress(addrInfo)
+	for _, address := range addresses {
+		m.checkAddress(address)
 	}
 }

-// checkAddress 检查单个地址
-func (m *TronUSDTMonitor) checkAddress(addrInfo *AddressInfo) {
-	events, err := m.getUSDTEvents(addrInfo.Address, addrInfo.LastBlock)
+// checkAddress 检查单个地址的交易
+func (m *USDTMan) checkAddress(address string) {
+	transactions, err := m.getUSDTTransactions(address)
 	if err != nil {
-		fmt.Printf("Error checking address %s: %v\n", addrInfo.Address, err)
+		fmt.Printf("Error checking address %s: %v\n", address, err)
 		return
 	}

-	for _, event := range events {
-		if event.EventName != "Transfer" {
+	currentBlock, err := m.getCurrentBlock()
+	if err != nil {
+		fmt.Printf("Error getting current block: %v\n", err)
+		return
+	}
+
+	for _, txn := range transactions {
+		// 检查是否已处理
+		m.txnMutex.RLock()
+		processed := m.processedTxns[txn.TxID]
+		m.txnMutex.RUnlock()
+
+		if processed {
 			continue
 		}

-		to, ok := event.Result["to"].(string)
-		if !ok || !strings.EqualFold(to, addrInfo.Address) {
+		// 计算确认数
+		confirmations := currentBlock - txn.BlockNumber
+		if confirmations < m.minConfirmations {
 			continue
 		}

-		from, _ := event.Result["from"].(string)
-
-		valueStr, ok := event.Result["value"].(string)
-		if !ok {
+		// 解析交易数据
+		payment := m.parseTransaction(&txn, address, confirmations)
+		if payment == nil {
 			continue
 		}

-		amount := parseUSDTAmount(valueStr)
-		if amount <= 0 {
-			continue
-		}
-
-		payment := &USDTPayment{
-			Address:     addrInfo.Address,
-			Amount:      amount,
-			TxID:        event.TransactionID,
-			BlockNumber: event.BlockNumber,
-			Timestamp:   event.BlockTimestamp,
-			From:        from,
-		}
-
-		if event.BlockNumber > addrInfo.LastBlock {
-			addrInfo.LastBlock = event.BlockNumber
-		}
+		// 标记为已处理
+		m.txnMutex.Lock()
+		m.processedTxns[txn.TxID] = true
+		m.txnMutex.Unlock()

+		// 触发回调
 		m.mu.RLock()
-		callback := m.onPaymentReceived
+		callback := m.onPaymentComplete
 		m.mu.RUnlock()

 		if callback != nil {
@@ -207,16 +205,10 @@ func (m *TronUSDTMonitor) checkAddress(addrInfo *AddressInfo) {
 	}
 }

-// getUSDTEvents 获取 USDT 转账事件
-func (m *TronUSDTMonitor) getUSDTEvents(address string, sinceBlock int64) ([]struct {
-	TransactionID  string
-	BlockNumber    int64
-	BlockTimestamp int64
-	EventName      string
-	Result         map[string]interface{}
-}, error) {
-	url := fmt.Sprintf("%s/v1/contracts/%s/events?event_name=Transfer&only_to=true&min_block_timestamp=%d&limit=200",
-		TronGridAPI, USDTContractAddress, sinceBlock)
+// 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)

 	req, err := http.NewRequest("GET", url, nil)
 	if err != nil {
@@ -243,72 +235,121 @@ func (m *TronUSDTMonitor) getUSDTEvents(address string, sinceBlock int64) ([]str
 		return nil, fmt.Errorf("TronGrid API error: %d %s", resp.StatusCode, string(body))
 	}

-	var result TronGridEvent
+	var result TronGridAccountTransactions
 	if err := json.Unmarshal(body, &result); err != nil {
 		return nil, err
 	}

-	if !result.Success {
-		return nil, fmt.Errorf("TronGrid API returned success=false")
+	return result.Data, nil
+}
+
+// getCurrentBlock 获取当前区块高度
+func (m *USDTMan) getCurrentBlock() (int64, error) {
+	url := fmt.Sprintf("%s/wallet/getnowblock", TronGridAPI)
+
+	req, err := http.NewRequest("GET", url, nil)
+	if err != nil {
+		return 0, err
 	}

-	var filtered []struct {
-		TransactionID  string
-		BlockNumber    int64
-		BlockTimestamp int64
-		EventName      string
-		Result         map[string]interface{}
+	if m.apiKey != "" {
+		req.Header.Set("TRON-PRO-API-KEY", m.apiKey)
 	}

-	for _, event := range result.Data {
-		to, ok := event.Result["to"].(string)
-		if ok && strings.EqualFold(to, address) {
-			filtered = append(filtered, struct {
-				TransactionID  string
-				BlockNumber    int64
-				BlockTimestamp int64
-				EventName      string
-				Result         map[string]interface{}
-			}{
-				TransactionID:  event.TransactionID,
-				BlockNumber:    event.BlockNumber,
-				BlockTimestamp: event.BlockTimestamp,
-				EventName:      event.EventName,
-				Result:         event.Result,
-			})
+	client := &http.Client{Timeout: 10 * time.Second}
+	resp, err := client.Do(req)
+	if err != nil {
+		return 0, err
+	}
+	defer resp.Body.Close()
+
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return 0, err
+	}
+
+	if resp.StatusCode != 200 {
+		return 0, fmt.Errorf("TronGrid API error: %d", resp.StatusCode)
+	}
+
+	var blockInfo TronGridBlockInfo
+	if err := json.Unmarshal(body, &blockInfo); err != nil {
+		return 0, err
+	}
+
+	return blockInfo.BlockHeader.RawData.Number, nil
+}
+
+// parseTransaction 解析交易
+func (m *USDTMan) parseTransaction(txn *TronGridTransaction, targetAddress string, confirmations int64) *USDTPayment {
+	if len(txn.Ret) == 0 || txn.Ret[0]["contractRet"] != "SUCCESS" {
+		return nil
+	}
+
+	if len(txn.RawData.Contract) == 0 {
+		return nil
+	}
+
+	contract := txn.RawData.Contract[0]
+	value := contract.Parameter.Value
+
+	// 转换地址格式
+	to := value.To
+	from := value.From
+
+	// 检查是否是目标地址
+	if !strings.EqualFold(to, targetAddress) {
+		return nil
+	}
+
+	// USDT 是 6 位小数
+	amount := float64(value.Amount) / 1000000.0
+
+	return &USDTPayment{
+		Address:       targetAddress,
+		Amount:        amount,
+		TxID:          txn.TxID,
+		BlockNumber:   txn.BlockNumber,
+		Timestamp:     txn.BlockTimeStamp,
+		From:          from,
+		Confirmations: confirmations,
+	}
+}
+
+// GetAddresses 获取所有监听地址
+func (m *USDTMan) GetAddresses() []string {
+	m.mu.RLock()
+	defer m.mu.RUnlock()
+
+	addresses := make([]string, len(m.addresses))
+	copy(addresses, m.addresses)
+	return addresses
+}
+
+// AddAddress 添加监听地址
+func (m *USDTMan) AddAddress(address string) {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+
+	// 检查是否已存在
+	for _, addr := range m.addresses {
+		if addr == address {
+			return
 		}
 	}

-	return filtered, nil
+	m.addresses = append(m.addresses, address)
 }

-// parseUSDTAmount 解析 USDT 金额（6位小数）
-func parseUSDTAmount(valueStr string) float64 {
-	valueStr = strings.TrimPrefix(valueStr, "0x")
+// RemoveAddress 移除监听地址
+func (m *USDTMan) RemoveAddress(address string) {
+	m.mu.Lock()
+	defer m.mu.Unlock()

-	var value int64
-	fmt.Sscanf(valueStr, "%d", &value)
-
-	return float64(value) / 1000000.0
-}
-
-// GetAddressInfo 获取地址信息
-func (m *TronUSDTMonitor) GetAddressInfo(address string) (*AddressInfo, bool) {
-	m.mu.RLock()
-	defer m.mu.RUnlock()
-
-	info, exists := m.addresses[address]
-	return info, exists
-}
-
-// GetAllAddresses 获取所有监听地址
-func (m *TronUSDTMonitor) GetAllAddresses() []string {
-	m.mu.RLock()
-	defer m.mu.RUnlock()
-
-	addresses := make([]string, 0, len(m.addresses))
-	for addr := range m.addresses {
-		addresses = append(addresses, addr)
+	for i, addr := range m.addresses {
+		if addr == address {
+			m.addresses = append(m.addresses[:i], m.addresses[i+1:]...)
+			return
+		}
 	}
-	return addresses
 }