zogo/conn/README.md

conn

net.Conn 装饰器：在读写路径上透明叠加带宽限速（rate）与流量统计（flow）， 可选关联 sync.WaitGroup 管理连接生命周期。

迁移自 go-hua/conn，内部依赖指向本仓库 flow/rate。

用法

import (
    "net"
    "sync"

    "git.zeroonesoft.cn/golib/zogo/conn"
    "git.zeroonesoft.cn/golib/zogo/flow"
    "git.zeroonesoft.cn/golib/zogo/rate"
)

var wg sync.WaitGroup
wg.Add(1)
f := &flow.Flow{}
r := rate.NewRate(1024 * 512) // 512 KB/s，0 表示不限速
r.Start()

c := conn.NewConn(rawConn, r, f, &wg)
defer c.Close()

// 之后像普通 net.Conn 一样使用：Read/Write 已自动限速并累计流量
c.Write(buf)
n, _ := c.Read(buf)

send, recv := f.Get() // 读写流量统计

完整可运行例程：examples/conn/main.go

注意

• rate 传 nil 表示不限速；flow 传 nil 表示不统计。
• Close 幂等，并会 wg.Done()（前提是创建时传入了 wg）。