86 lines
2.1 KiB
Go
86 lines
2.1 KiB
Go
package mws
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
|
|
mdata "github.com/go-gost/core/metadata"
|
|
mdutil "github.com/go-gost/core/metadata/util"
|
|
"github.com/go-gost/x/internal/util/mux"
|
|
)
|
|
|
|
const (
|
|
defaultPath = "/ws"
|
|
defaultKeepalivePeriod = 15 * time.Second
|
|
)
|
|
|
|
type metadata struct {
|
|
host string
|
|
path string
|
|
|
|
handshakeTimeout time.Duration
|
|
readHeaderTimeout time.Duration
|
|
readBufferSize int
|
|
writeBufferSize int
|
|
enableCompression bool
|
|
|
|
header http.Header
|
|
keepaliveInterval time.Duration
|
|
muxCfg *mux.Config
|
|
}
|
|
|
|
func (d *mwsDialer) parseMetadata(md mdata.Metadata) (err error) {
|
|
const (
|
|
host = "host"
|
|
path = "path"
|
|
|
|
handshakeTimeout = "handshakeTimeout"
|
|
readHeaderTimeout = "readHeaderTimeout"
|
|
readBufferSize = "readBufferSize"
|
|
writeBufferSize = "writeBufferSize"
|
|
enableCompression = "enableCompression"
|
|
|
|
header = "header"
|
|
)
|
|
|
|
d.md.host = mdutil.GetString(md, host)
|
|
|
|
d.md.path = mdutil.GetString(md, path)
|
|
if d.md.path == "" {
|
|
d.md.path = defaultPath
|
|
}
|
|
|
|
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"),
|
|
}
|
|
|
|
d.md.handshakeTimeout = mdutil.GetDuration(md, handshakeTimeout)
|
|
d.md.readHeaderTimeout = mdutil.GetDuration(md, readHeaderTimeout)
|
|
d.md.readBufferSize = mdutil.GetInt(md, readBufferSize)
|
|
d.md.writeBufferSize = mdutil.GetInt(md, writeBufferSize)
|
|
d.md.enableCompression = mdutil.GetBool(md, enableCompression)
|
|
|
|
if m := mdutil.GetStringMapString(md, header); len(m) > 0 {
|
|
h := http.Header{}
|
|
for k, v := range m {
|
|
h.Add(k, v)
|
|
}
|
|
d.md.header = h
|
|
}
|
|
|
|
if mdutil.GetBool(md, "keepalive") {
|
|
d.md.keepaliveInterval = mdutil.GetDuration(md, "ttl", "keepalive.interval")
|
|
if d.md.keepaliveInterval <= 0 {
|
|
d.md.keepaliveInterval = defaultKeepalivePeriod
|
|
}
|
|
}
|
|
|
|
return
|
|
}
|