x/dialer/kcp/metadata.go
2022-03-14 20:27:14 +08:00

40 lines
773 B
Go

package kcp
import (
"encoding/json"
"time"
mdata "github.com/go-gost/gost/v3/pkg/metadata"
kcp_util "github.com/go-gost/x/internal/util/kcp"
)
type metadata struct {
handshakeTimeout time.Duration
config *kcp_util.Config
}
func (d *kcpDialer) parseMetadata(md mdata.Metadata) (err error) {
const (
config = "config"
handshakeTimeout = "handshakeTimeout"
)
if m := mdata.GetStringMap(md, config); len(m) > 0 {
b, err := json.Marshal(m)
if err != nil {
return err
}
cfg := &kcp_util.Config{}
if err := json.Unmarshal(b, cfg); err != nil {
return err
}
d.md.config = cfg
}
if d.md.config == nil {
d.md.config = kcp_util.DefaultConfig
}
d.md.handshakeTimeout = mdata.GetDuration(md, handshakeTimeout)
return
}