feat(spmux): 自 go-hua 迁入单端口多路复用并解耦日志依赖

- go-hua/log 已弃用: *log.Logger 参数改为两方法接口 spmux.Logger
  (*logrus.Logger 天然满足), 传 nil 静默(logErrorf/logDebugf 判空)
- 协议识别改用 zogo/httpprobe; 测试与例程同步适配
- 附测试、examples/spmux 例程(含 Logger 适配演示)与包 README
This commit is contained in:
2026-09-20 12:37:10 +08:00
parent f120ab24ed
commit 96c63fb153
6 changed files with 557 additions and 0 deletions
+128
View File
@@ -0,0 +1,128 @@
package spmux
import (
"bufio"
"fmt"
"io"
"net"
"net/http"
"testing"
"time"
)
// freePort 找一个空闲端口(有轻微竞态,测试环境可接受)
func freePort(t *testing.T) int {
l, err := net.Listen("tcp", "127.0.0.1:0")
if err != nil {
t.Fatal(err)
}
port := l.Addr().(*net.TCPAddr).Port
l.Close()
return port
}
func TestPortMuxHttpRouting(t *testing.T) {
port := freePort(t)
pm := NewPortMux(port, nil)
if err := pm.Start(); err != nil {
t.Fatalf("Start: %v", err)
}
httpDone := make(chan struct{})
go func() {
defer close(httpDone)
http.Serve(pm.HttpListener(), http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
fmt.Fprint(w, "mux-http-ok")
}))
}()
// 先 Close 解除 Serve 阻塞,再等待协程退出(顺序不能反,否则死锁)
defer func() {
pm.Close()
<-httpDone
}()
time.Sleep(100 * time.Millisecond)
resp, err := http.Get(fmt.Sprintf("http://127.0.0.1:%d/", port))
if err != nil {
t.Fatalf("HTTP 请求失败: %v", err)
}
body, _ := io.ReadAll(resp.Body)
resp.Body.Close()
if string(body) != "mux-http-ok" {
t.Errorf("HTTP 响应 = %q", string(body))
}
}
func TestPortMuxTcpRouting(t *testing.T) {
port := freePort(t)
pm := NewPortMux(port, nil)
if err := pm.Start(); err != nil {
t.Fatalf("Start: %v", err)
}
tcpDone := make(chan struct{})
go func() {
defer close(tcpDone)
for {
c, err := pm.TcpListener().Accept()
if err != nil {
return
}
go func(cc net.Conn) {
defer cc.Close()
buf := make([]byte, 1024)
for {
n, err := cc.Read(buf)
if n > 0 {
cc.Write(append(buf[:n:n], '\n'))
}
if err != nil {
return
}
}
}(c)
}
}()
// 先 Close 解除 Accept 阻塞,再等待协程退出(顺序不能反,否则死锁)
defer func() {
pm.Close()
<-tcpDone
}()
time.Sleep(100 * time.Millisecond)
// 非 HTTP 数据(超过 8 字节)应路由到 TcpListener
raw, err := net.Dial("tcp", fmt.Sprintf("127.0.0.1:%d", port))
if err != nil {
t.Fatalf("TCP 连接失败: %v", err)
}
fmt.Fprint(raw, "ping-ping")
raw.SetReadDeadline(time.Now().Add(3 * time.Second))
line, err := bufio.NewReader(raw).ReadString('\n')
if err != nil {
t.Fatalf("读取 echo: %v", err)
}
if line != "ping-ping\n" {
t.Errorf("echo = %q", line)
}
raw.Close()
}
func TestPortListenerAddr(t *testing.T) {
port := freePort(t)
pm := NewPortMux(port, nil)
if err := pm.Start(); err != nil {
t.Fatalf("Start: %v", err)
}
defer pm.Close()
if pm.HttpListener().Addr().String() == "" {
t.Error("HttpListener Addr 不应为空")
}
if pm.TcpListener().Addr().String() == "" {
t.Error("TcpListener Addr 不应为空")
}
}