conn

package

v0.5.6 Latest Latest Go to latest Published: Oct 23, 2026 License: MIT Imports: 7 Imported by: 0

Details

Package conn 提供连接管理相关的工具。

This section is empty.

View Source

var ErrConnectionUnavailable = errors.New("connection unavailable")

ErrConnectionUnavailable 在连接不可用时由 Write 方法返回。

func Exponential(d time.Duration) time.Duration

Exponential 计算指数退避时间（带 ±50% 抖动），上限 1 分钟。用于避免重连风暴（thundering herd）。

type AfterFunc func(time.Duration) <-chan time.Time

AfterFunc 模仿 time.After，用于控制重连退避时间。

type Dialer func(network, address string) (net.Conn, error)

Dialer 模仿 net.Dial，创建网络连接。返回的连接必须是并发安全的。

type Manager struct {
	// contains filtered or unexported fields
}

Manager 管理一个 net.Conn 的生命周期。

通过 Take 获取连接，通过 Put 报告错误。当 Put 收到非 nil error 时，当前连接失效，Manager 自动重连（指数退避）。

func NewDefaultManager(network, address string) *Manager

NewDefaultManager 创建一个使用默认配置的连接管理器。使用 net.Dial 和 time.After。

func NewManager(d Dialer, network, address string, after AfterFunc) *Manager

NewManager 创建一个连接管理器。

参数：

func (m *Manager) Put(err error)

Put 报告连接使用中产生的错误。非 nil error 会导致当前连接被关闭并触发重连。 Stop 后调用是空操作。

func (m *Manager) Stop()

Stop 停止管理器，关闭当前连接，解除所有阻塞的 Take/Put 调用。

func (m *Manager) Take() net.Conn

Take 返回当前连接，可能为 nil（连接断开或尚未建立）。 Stop 后调用返回 nil。

func (m *Manager) Write(b []byte) (int, error)

Write 是便捷方法：自动 Take → Write → Put。当连接不可用时返回 ErrConnectionUnavailable。

注意：Write 不是并发安全的。Take/Put 设计为单 goroutine 使用，多 goroutine 调用 Write 会导致连接竞争。并发场景应由调用方加锁。