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>
54 lines
1.9 KiB
Go
54 lines
1.9 KiB
Go
package grpc
|
|
|
|
import (
|
|
"time"
|
|
|
|
mdata "github.com/go-gost/core/metadata"
|
|
mdutil "github.com/go-gost/x/metadata/util"
|
|
)
|
|
|
|
type metadata struct {
|
|
insecure bool
|
|
host string
|
|
path string
|
|
keepalive bool
|
|
keepaliveTime time.Duration
|
|
keepaliveTimeout time.Duration
|
|
keepalivePermitWithoutStream bool
|
|
minConnectTimeout time.Duration
|
|
|
|
tcpKeepalive bool
|
|
tcpKeepaliveIdle time.Duration
|
|
tcpKeepaliveInterval time.Duration
|
|
tcpKeepaliveCount int
|
|
}
|
|
|
|
func (d *grpcDialer) parseMetadata(md mdata.Metadata) (err error) {
|
|
d.md.insecure = mdutil.GetBool(md, "grpc.insecure", "grpcInsecure", "insecure")
|
|
d.md.host = mdutil.GetString(md, "grpc.authority", "grpc.host", "host")
|
|
d.md.path = mdutil.GetString(md, "grpc.path", "path")
|
|
d.md.keepalive = mdutil.GetBool(md, "grpc.keepalive", "keepalive", "keepAlive")
|
|
if d.md.keepalive {
|
|
d.md.keepaliveTime = mdutil.GetDuration(md, "grpc.keepalive.time", "keepalive.time")
|
|
if d.md.keepaliveTime <= 0 {
|
|
d.md.keepaliveTime = 30 * time.Second
|
|
}
|
|
d.md.keepaliveTimeout = mdutil.GetDuration(md, "grpc.keepalive.timeout", "keepalive.timeout")
|
|
if d.md.keepaliveTimeout <= 0 {
|
|
d.md.keepaliveTimeout = 30 * time.Second
|
|
}
|
|
d.md.keepalivePermitWithoutStream = mdutil.GetBool(md, "grpc.keepalive.permitWithoutStream", "keepalive.permitWithoutStream")
|
|
}
|
|
d.md.minConnectTimeout = mdutil.GetDuration(md, "grpc.minConnectTimeout", "minConnectTimeout")
|
|
if d.md.minConnectTimeout <= 0 {
|
|
d.md.minConnectTimeout = 30 * time.Second
|
|
}
|
|
|
|
d.md.tcpKeepalive = mdutil.GetBool(md, "tcp.keepalive")
|
|
d.md.tcpKeepaliveIdle = mdutil.GetDuration(md, "tcp.keepalive.idle")
|
|
d.md.tcpKeepaliveInterval = mdutil.GetDuration(md, "tcp.keepalive.interval")
|
|
d.md.tcpKeepaliveCount = mdutil.GetInt(md, "tcp.keepalive.count")
|
|
|
|
return
|
|
}
|