initial commit

dialer/mws/metadata.go

@@ -0,0 +1,87 @@
+package mws
+
+import (
+	"net/http"
+	"time"
+
+	mdata "github.com/go-gost/gost/v3/pkg/metadata"
+)
+
+const (
+	defaultPath = "/ws"
+)
+
+type metadata struct {
+	host string
+	path string
+
+	handshakeTimeout  time.Duration
+	readHeaderTimeout time.Duration
+	readBufferSize    int
+	writeBufferSize   int
+	enableCompression bool
+
+	muxKeepAliveDisabled bool
+	muxKeepAliveInterval time.Duration
+	muxKeepAliveTimeout  time.Duration
+	muxMaxFrameSize      int
+	muxMaxReceiveBuffer  int
+	muxMaxStreamBuffer   int
+
+	header    http.Header
+	keepAlive time.Duration
+}
+
+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"
+		keepAlive = "keepAlive"
+
+		muxKeepAliveDisabled = "muxKeepAliveDisabled"
+		muxKeepAliveInterval = "muxKeepAliveInterval"
+		muxKeepAliveTimeout  = "muxKeepAliveTimeout"
+		muxMaxFrameSize      = "muxMaxFrameSize"
+		muxMaxReceiveBuffer  = "muxMaxReceiveBuffer"
+		muxMaxStreamBuffer   = "muxMaxStreamBuffer"
+	)
+
+	d.md.host = mdata.GetString(md, host)
+
+	d.md.path = mdata.GetString(md, path)
+	if d.md.path == "" {
+		d.md.path = defaultPath
+	}
+
+	d.md.muxKeepAliveDisabled = mdata.GetBool(md, muxKeepAliveDisabled)
+	d.md.muxKeepAliveInterval = mdata.GetDuration(md, muxKeepAliveInterval)
+	d.md.muxKeepAliveTimeout = mdata.GetDuration(md, muxKeepAliveTimeout)
+	d.md.muxMaxFrameSize = mdata.GetInt(md, muxMaxFrameSize)
+	d.md.muxMaxReceiveBuffer = mdata.GetInt(md, muxMaxReceiveBuffer)
+	d.md.muxMaxStreamBuffer = mdata.GetInt(md, muxMaxStreamBuffer)
+
+	d.md.handshakeTimeout = mdata.GetDuration(md, handshakeTimeout)
+	d.md.readHeaderTimeout = mdata.GetDuration(md, readHeaderTimeout)
+	d.md.readBufferSize = mdata.GetInt(md, readBufferSize)
+	d.md.writeBufferSize = mdata.GetInt(md, writeBufferSize)
+	d.md.enableCompression = mdata.GetBool(md, enableCompression)
+
+	if m := mdata.GetStringMapString(md, header); len(m) > 0 {
+		h := http.Header{}
+		for k, v := range m {
+			h.Add(k, v)
+		}
+		d.md.header = h
+	}
+	d.md.keepAlive = mdata.GetDuration(md, keepAlive)
+
+	return
+}