refactor(edge): Server 协议统一与 wsclient 收敛(Phase 3 Batch 3)

- openflare/agent/relay/flared 协议类型改为 pkg/protocol 别名,删除重复 struct
- 新增 edge/wsclient Preset 表,三组件 wsclient 改为薄包装
- 更新设计文档、计划与 changelog
This commit is contained in:
ryan
2026-06-19 15:00:46 +08:00
parent db9a9f98fd
commit 0b34792709
16 changed files with 328 additions and 484 deletions
+11 -58
View File
@@ -5,78 +5,31 @@ import (
"time"
"github.com/Rain-kl/Wavelet/internal/apps/agent/protocol"
shared "github.com/Rain-kl/Wavelet/pkg/wsclient"
edgews "github.com/Rain-kl/Wavelet/internal/apps/edge/wsclient"
)
type WSMessage = shared.WSMessage
type MessageHandler = shared.MessageHandler
type WSMessage = edgews.WSMessage
type MessageHandler = edgews.MessageHandler
type Connection = edgews.AgentConnection
type Client struct {
sharedClient *shared.Client
inner *edgews.Client
}
type Connection struct {
sharedConn *shared.Connection
}
func New(baseURL string, token string, timeout time.Duration) *Client {
func New(baseURL, token string, timeout time.Duration) *Client {
return &Client{
sharedClient: shared.New(shared.Config{
BaseURL: baseURL,
Token: token,
Timeout: timeout,
HeaderKey: "X-Agent-Token",
WSPath: "/api/v1/agent/ws",
}),
inner: edgews.New(edgews.PresetAgent, baseURL, token, timeout),
}
}
func (c *Client) SetToken(token string) {
c.sharedClient.SetToken(token)
c.inner.SetToken(token)
}
func (c *Client) URL() string {
return c.sharedClient.URL()
return c.inner.URL()
}
func (c *Client) Connect(ctx context.Context) (protocol.WebSocketConnection, error) {
conn, err := c.sharedClient.Connect(ctx)
if err != nil {
return nil, err
}
return &Connection{sharedConn: conn}, nil
}
func (conn *Connection) URL() string {
if conn == nil || conn.sharedConn == nil {
return ""
}
return conn.sharedConn.URL
}
func (conn *Connection) SendStatus(payload protocol.NodePayload) error {
return conn.sharedConn.SendMessage(protocol.WSMessageTypeStatus, payload)
}
func (conn *Connection) SendPong() error {
return conn.sharedConn.SendMessage(protocol.WSMessageTypePong, nil)
}
func (conn *Connection) Receive() (protocol.WSMessage, error) {
var message protocol.WSMessage
if err := conn.sharedConn.Receive(&message); err != nil {
return message, err
}
return message, nil
}
func (conn *Connection) RunReceiveLoop(ctx context.Context, handler shared.MessageHandler) error {
return conn.sharedConn.RunReceiveLoop(ctx, handler)
}
func (conn *Connection) Close() error {
if conn == nil || conn.sharedConn == nil {
return nil
}
return conn.sharedConn.Close()
}
return c.inner.ConnectAgent(ctx)
}