mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-28 05:46:36 +08:00
ryan
0b34792709
- openflare/agent/relay/flared 协议类型改为 pkg/protocol 别名，删除重复 struct - 新增 edge/wsclient Preset 表，三组件 wsclient 改为薄包装 - 更新设计文档、计划与 changelog
30 lines
599 B
Go
30 lines
599 B
Go
package wsclient
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
edgews "github.com/Rain-kl/Wavelet/internal/apps/edge/wsclient"
|
|
)
|
|
|
|
type WSMessage = edgews.WSMessage
|
|
type MessageHandler = edgews.MessageHandler
|
|
type Connection = edgews.Connection
|
|
|
|
type Client struct {
|
|
inner *edgews.Client
|
|
}
|
|
|
|
func New(baseURL, token string, timeout time.Duration) *Client {
|
|
return &Client{
|
|
inner: edgews.New(edgews.PresetRelay, baseURL, token, timeout),
|
|
}
|
|
}
|
|
|
|
func (c *Client) SetToken(token string) {
|
|
c.inner.SetToken(token)
|
|
}
|
|
|
|
func (c *Client) Connect(ctx context.Context) (*Connection, error) {
|
|
return c.inner.Connect(ctx)
|
|
} |