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:
@@ -0,0 +1,89 @@
|
||||
// spmux 包示例:单端口多路复用——同一端口上按报文特征把流量分发给 HTTP / HTTPS / TCP 三个 Listener
|
||||
package main
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"net"
|
||||
"net/http"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"git.zeroonesoft.cn/golib/zogo/spmux"
|
||||
)
|
||||
|
||||
// stdoutLogger 适配 spmux.Logger 接口(*logrus.Logger 天然满足,无需适配)
|
||||
type stdoutLogger struct{}
|
||||
|
||||
func (stdoutLogger) Debugf(format string, a ...any) { log.Printf("DEBUG "+format, a...) }
|
||||
func (stdoutLogger) Errorf(format string, a ...any) { log.Printf("ERROR "+format, a...) }
|
||||
|
||||
func main() {
|
||||
// 找一个空闲端口
|
||||
l, err := net.Listen("tcp", "127.0.0.1:0")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
port := l.Addr().(*net.TCPAddr).Port
|
||||
l.Close()
|
||||
|
||||
// 创建端口复用器(logger 实现 spmux.Logger 即可,也可传 nil 静默)
|
||||
pm := spmux.NewPortMux(port, stdoutLogger{})
|
||||
if err := pm.Start(); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
// HTTP 流量交给 HttpListener
|
||||
go http.Serve(pm.HttpListener(), http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
fmt.Fprintln(w, "hello from http over spmux")
|
||||
}))
|
||||
|
||||
// 其他 TCP 流量交给 TcpListener(echo 服务,回显内容并补换行)
|
||||
tcpL := pm.TcpListener()
|
||||
go func() {
|
||||
for {
|
||||
c, err := tcpL.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)
|
||||
}
|
||||
}()
|
||||
|
||||
time.Sleep(100 * time.Millisecond)
|
||||
|
||||
// HTTP 请求会被识别并路由到 HttpListener
|
||||
resp, err := http.Get(fmt.Sprintf("http://127.0.0.1:%d/", port))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
body, _ := io.ReadAll(resp.Body)
|
||||
resp.Body.Close()
|
||||
fmt.Print("HTTP 响应:", string(body))
|
||||
|
||||
// 普通数据会被路由到 TcpListener
|
||||
raw, err := net.Dial("tcp", fmt.Sprintf("127.0.0.1:%d", port))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
fmt.Fprint(raw, "ping")
|
||||
line, _ := bufio.NewReader(raw).ReadString('\n')
|
||||
fmt.Print("TCP 响应:", line)
|
||||
raw.Close()
|
||||
|
||||
_ = os.Stdout
|
||||
}
|
||||
Reference in New Issue
Block a user