true_casey/usdtman
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix update

Browse Source
This commit is contained in:
TQCasey
2026-02-03 00:10:43 +08:00
parent 79c1c710a2
commit 8965b96f0e
3 changed files with 102 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
README.md
View File

@@ -12,8 +12,25 @@ TRON USDT TRC20 收款监听服务 - 基于交易记录扫描 + 区块确认数
## API 使用方式
```go
// 创建监听器
uman := usdtman.NewUSDTMan([]string{"地址1", "地址2"}, "API_KEY")
// 创建监听器（配置对象方式）
uman := usdtman.NewUSDTMan(usdtman.Config{
Addresses: []string{"地址1", "地址2"},
APIKey: "YOUR_API_KEY",
QueryInterval: 5 * time.Second, // 查询间隔（可选，默认 5 秒）
MinConfirmations: 6, // 最小确认数（可选，默认 6）
MaxHistoryTxns: 20, // 查询历史交易数（可选，默认 20）
ProxyURL: "http://127.0.0.1:7890", // HTTP/SOCKS5 代理（可选）
})
// 或者使用自定义 Transport
uman := usdtman.NewUSDTMan(usdtman.Config{
Addresses: []string{"地址1"},
APIKey: "YOUR_API_KEY",
Transport: &http.Transport{
Proxy: http.ProxyURL(proxyURL),
// 其他自定义配置...
},
})
// 设置收款回调
uman.OnPaymentComplete(func(payment *usdtman.USDTPayment) {
@@ -25,6 +42,10 @@ uman.Start()
// 停止监听
uman.Stop()
// 动态添加/移除地址
uman.AddAddress("新地址")
uman.RemoveAddress("旧地址")
```
## 运行

21
cmd/server/main.go
View File

@@ -7,6 +7,7 @@ import (
"net/http"
"os"
"sync"
"time"
"usdtman"
"github.com/gorilla/websocket"
@@ -34,8 +35,24 @@ func main() {
apiKey = "da1e77dc-b35b-4458-846a-5a551b9df4b2"
}
// 初始化 USDTMan (无初始地址)
uman = usdtman.NewUSDTMan([]string{}, apiKey)
// 初始化 USDTMan
proxyURL := os.Getenv("PROXY_URL") // 例如: http://127.0.0.1:7890
config := usdtman.Config{
Addresses: []string{}, // 初始无地址，可通过 API 添加
APIKey: apiKey,
QueryInterval: 5 * time.Second, // 每 5 秒查询一次
MinConfirmations: 6, // 6 个确认
MaxHistoryTxns: 20, // 每次查询最多 20 条历史交易
}
// 如果设置了代理
if proxyURL != "" {
config.ProxyURL = proxyURL
log.Printf("✅ 使用代理: %s", proxyURL)
}
uman = usdtman.NewUSDTMan(config)
// 设置收款回调
uman.OnPaymentComplete(func(payment *usdtman.USDTPayment) {

71
tron_usdt_monitor.go
View File

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