Andrew Beresford
ac99425002
Add configurable TCP keepalive (keepalive, keepalive.idle, keepalive.interval, keepalive.count) to all TCP-based transport types: tls, mtls, mtcp, ws/wss, mws/mwss, http2, ssh, sshd listeners and tls, mtls, mtcp, ws/wss, mws/mwss, ssh, sshd dialers. For types that already use "keepalive" for an app-level protocol (grpc listener/dialer, ws/mws dialers, ssh/sshd dialers), the TCP keepalive parameters are exposed under the tcp.keepalive.* prefix to avoid ambiguity. Also extract shared helpers into internal/net/keepalive.go: - WrapKeepaliveListener: applies KeepAliveConfig on every accepted TCPConn - ApplyKeepalive: applies KeepAliveConfig to a single TCPConn Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
44 lines
1.3 KiB
Go
44 lines
1.3 KiB
Go
package mtcp
|
|
|
|
import (
|
|
"time"
|
|
|
|
mdata "github.com/go-gost/core/metadata"
|
|
mdutil "github.com/go-gost/x/metadata/util"
|
|
"github.com/go-gost/x/internal/util/mux"
|
|
)
|
|
|
|
type metadata struct {
|
|
handshakeTimeout time.Duration
|
|
muxCfg *mux.Config
|
|
|
|
keepalive bool
|
|
keepaliveIdle time.Duration
|
|
keepaliveInterval time.Duration
|
|
keepaliveCount int
|
|
}
|
|
|
|
func (d *mtcpDialer) parseMetadata(md mdata.Metadata) (err error) {
|
|
d.md.handshakeTimeout = mdutil.GetDuration(md, "handshakeTimeout")
|
|
|
|
d.md.keepalive = mdutil.GetBool(md, "keepalive")
|
|
d.md.keepaliveIdle = mdutil.GetDuration(md, "keepalive.idle")
|
|
d.md.keepaliveInterval = mdutil.GetDuration(md, "keepalive.interval")
|
|
d.md.keepaliveCount = mdutil.GetInt(md, "keepalive.count")
|
|
|
|
d.md.muxCfg = &mux.Config{
|
|
Version: mdutil.GetInt(md, "mux.version"),
|
|
KeepAliveInterval: mdutil.GetDuration(md, "mux.keepaliveInterval"),
|
|
KeepAliveDisabled: mdutil.GetBool(md, "mux.keepaliveDisabled"),
|
|
KeepAliveTimeout: mdutil.GetDuration(md, "mux.keepaliveTimeout"),
|
|
MaxFrameSize: mdutil.GetInt(md, "mux.maxFrameSize"),
|
|
MaxReceiveBuffer: mdutil.GetInt(md, "mux.maxReceiveBuffer"),
|
|
MaxStreamBuffer: mdutil.GetInt(md, "mux.maxStreamBuffer"),
|
|
}
|
|
if d.md.muxCfg.Version == 0 {
|
|
d.md.muxCfg.Version = 2
|
|
}
|
|
|
|
return
|
|
}
|