From 50301e1771d8d100db0164f437fab72812fe0789 Mon Sep 17 00:00:00 2001 From: lzx <393768033@qq.com> Date: Wed, 6 Aug 2025 17:16:17 +0800 Subject: [PATCH] update change pool address send authorize info --- README.md | 63 ++++++++++++++++++++++++++++++++++++++++- go.mod | 5 +--- go.sum | 2 -- internal/miner/miner.go | 14 ++++----- internal/msg/msg.go | 1 + internal/proxy.go | 14 +++++---- internal/zmq/zmq.go | 6 ++-- 7 files changed, 83 insertions(+), 22 deletions(-) diff --git a/README.md b/README.md index 74eadfc..d341c33 100644 --- a/README.md +++ b/README.md @@ -1 +1,62 @@ -# 代理 \ No newline at end of file +# 代理项目 + +Go语言编写的矿工代理服务，支持矿工连接代理并动态切换矿池地址。 + +## 功能 + +- 矿工连接代理服务 +- 默认代理到指定矿池地址 +- 通过RabbitMQ消息动态切换矿池 +- 支持多种加密货币 + +## 快速开始 + +### 环境要求 +- Go 1.23.1+ + +### 编译运行 + +```bash +# 编译代理程序 +go build -o proxy.exe cmd/proxy.go + +# 运行代理程序 +./proxy.exe +``` + +### 配置文件 + +创建 `config.json`： + +```json +{ + "coin": "your_coin", + "zmqAddr": "amqp://localhost:5672", + "tcpAddr": ":3333", + "proxyAddr": "pool.example.com:3333", + "rabbitTopic": "proxy_queue" +} +``` + +## 项目结构 + +``` +├── cmd/proxy.go # 代理程序入口 +├── internal/ +│ ├── miner/miner.go # 矿工连接管理 +│ ├── msg/msg.go # 消息结构 +│ ├── proxy.go # 代理服务核心 +│ └── zmq/zmq.go # RabbitMQ处理 +└── go.mod # 依赖管理 +``` + +## 使用说明 + +- **proxy.exe**: 矿工代理服务 (监听TCP连接) + +## 工作流程 + +1. 矿工连接代理服务 +2. 代理使用默认矿池地址 +3. 收到RabbitMQ消息可切换矿池 +4. 持续转发矿工和矿池消息 \ No newline at end of file diff --git a/go.mod b/go.mod index ee127d8..f3dfbf3 100644 --- a/go.mod +++ b/go.mod @@ -2,7 +2,4 @@ module proxy go 1.23.1 -require ( - github.com/rabbitmq/amqp091-go v1.10.0 - github.com/zeromq/goczmq v4.1.0+incompatible -) +require github.com/rabbitmq/amqp091-go v1.10.0 diff --git a/go.sum b/go.sum index b3fc2a5..024eebe 100644 --- a/go.sum +++ b/go.sum @@ -1,6 +1,4 @@ github.com/rabbitmq/amqp091-go v1.10.0 h1:STpn5XsHlHGcecLmMFCtg7mqq0RnD+zFr4uzukfVhBw= github.com/rabbitmq/amqp091-go v1.10.0/go.mod h1:Hy4jKW5kQART1u+JkDTF9YYOQUHXqMuhrgxOEeS7G4o= -github.com/zeromq/goczmq v4.1.0+incompatible h1:cGVQaU6kIwwrGso0Pgbl84tzAz/h7FJ3wYQjSonjFFc= -github.com/zeromq/goczmq v4.1.0+incompatible/go.mod h1:1uZybAJoSRCvZMH2rZxEwWBSmC4T7CB/xQOfChwPEzg= go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto= go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE= diff --git a/internal/miner/miner.go b/internal/miner/miner.go index fb00c82..1b698ca 100644 --- a/internal/miner/miner.go +++ b/internal/miner/miner.go @@ -35,8 +35,8 @@ func NewMiner(coin string, poolAddress string, minerConn net.Conn) (*Miner, erro }, nil } -// 动态切换矿池地址并建立新连接 -func (m *Miner) ChangePoolAddress(newAddress string) { +// 动态切换矿池地址并建立新连接,切换时由proxy构造，由于每个池子的authorize认证不同，需单独处理 +func (m *Miner) ChangePoolAddress(newAddress string, authorizeMsg string) { m.Lock() defer m.Unlock() @@ -54,7 +54,7 @@ func (m *Miner) ChangePoolAddress(newAddress string) { m.PoolConn = newConn m.PoolAddress = newAddress - + m.PoolConn.Write([]byte(authorizeMsg + "\n")) // 发送认证信息 fmt.Println("成功切换矿池地址为:", newAddress) } @@ -78,10 +78,10 @@ func (m *Miner) HandleMinerMsg(ch chan string) { case "nexa": // default: - var msg msg.Authorize_msg - if err := json.Unmarshal([]byte(msgStr), &msg); err == nil && - msg.Method == "mining.authorize" && len(msg.Params) >= 1 && !sent { - parts := strings.Split(msg.Params[0], ".") + var _msg msg.Authorize_msg + if err := json.Unmarshal([]byte(msgStr), &_msg); err == nil && + _msg.Method == "mining.authorize" && len(_msg.Params) >= 1 && !sent { + parts := strings.Split(_msg.Params[0], ".") if len(parts) >= 2 { userSign := parts[0] + "-" + parts[1] select { diff --git a/internal/msg/msg.go b/internal/msg/msg.go index 204f3a6..dce3d9e 100644 --- a/internal/msg/msg.go +++ b/internal/msg/msg.go @@ -4,6 +4,7 @@ type ZmqMsg struct { MethodID int `json:"methodId"` //0(add), 1(delete) ID string `json:"id"` // user-miner Address string `json:"address"` // 转发目标地址 + Msg string `json:"msg"` // 转发矿池的mining.authorize方法配置参数 } type Authorize_msg struct { diff --git a/internal/proxy.go b/internal/proxy.go index ad3a5a1..89f0712 100644 --- a/internal/proxy.go +++ b/internal/proxy.go @@ -6,6 +6,7 @@ import ( "net" "os" "proxy/internal/miner" + "proxy/internal/msg" "proxy/internal/zmq" "sync" "time" @@ -18,6 +19,7 @@ type Config struct { ZmqAddr string `json:"zmqAddr"` TcpAddr string `json:"tcpAddr"` DefaultAddr string `json:"proxyAddr"` + RabbitTopic string `json:"rabbitTopic"` } type ProxyCtx struct { @@ -26,7 +28,7 @@ type ProxyCtx struct { Cfg Config Listener net.Listener MinerConn []*miner.Miner - ProxyList map[string]string + ProxyList map[string]msg.ZmqMsg } func initConfig() Config { @@ -52,7 +54,7 @@ func newProxy() *ProxyCtx { Cfg: cfg, Listener: listener, MinerConn: []*miner.Miner{}, - ProxyList: make(map[string]string), + ProxyList: make(map[string]msg.ZmqMsg), } } @@ -79,10 +81,12 @@ func (p *ProxyCtx) handleMinerConnect() { select { case userSign := <-ch: p.mu.RLock() - addr, ok := p.ProxyList[userSign] + zmqMsg, ok := p.ProxyList[userSign] p.mu.RUnlock() if ok { - minerObj.ChangePoolAddress(addr) + addr := zmqMsg.Address + authorizeMsg := zmqMsg.Msg + minerObj.ChangePoolAddress(addr, authorizeMsg) } p.mu.Lock() @@ -101,7 +105,7 @@ func (p *ProxyCtx) handleMinerConnect() { } func (p *ProxyCtx) startZMQ() { - zmq.StartZMQ(p.Cfg.ZmqAddr, "testQueue", p.ProxyList, &p.mu) + zmq.StartZMQ(p.Cfg.ZmqAddr, p.Cfg.RabbitTopic, p.ProxyList, &p.mu) } func StartProxy() { diff --git a/internal/zmq/zmq.go b/internal/zmq/zmq.go index abbd8a0..93c8d40 100644 --- a/internal/zmq/zmq.go +++ b/internal/zmq/zmq.go @@ -101,7 +101,7 @@ func initRabbitConsumer(rabbitURL, queueName string) (*amqp.Connection, *amqp.Ch return conn, ch, msgs, nil } -func handleRabbitMsg(body []byte, proxyList map[string]string, proxyListLock *sync.RWMutex) { +func handleRabbitMsg(body []byte, proxyList map[string]msg.ZmqMsg, proxyListLock *sync.RWMutex) { var data msg.ZmqMsg if err := json.Unmarshal(body, &data); err != nil { fmt.Println("[rabbitmq]: json unmarshal error:", err) @@ -110,13 +110,13 @@ func handleRabbitMsg(body []byte, proxyList map[string]string, proxyListLock *sy proxyListLock.Lock() defer proxyListLock.Unlock() if data.MethodID == 0 { - proxyList[data.ID] = data.Address + proxyList[data.ID] = data } else { delete(proxyList, data.ID) } } -func StartZMQ(rabbitURL, queueName string, proxyList map[string]string, proxyListLock *sync.RWMutex) { +func StartZMQ(rabbitURL, queueName string, proxyList map[string]msg.ZmqMsg, proxyListLock *sync.RWMutex) { conn, ch, msgs, err := initRabbitConsumer(rabbitURL, queueName) if err != nil { fmt.Println("[rabbitmq]:", err)