mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-30 06:16:37 +08:00
ryan
aa4faddade
Result: {"status":"keep","total_issues":8,"golint_canonicalheader":0,"golint_errname":0,"golint_errorlint":1,"golint_exhaustive":0,"golint_forcetypeassert":0,"golint_gosec":0,"golint_intrange":0,"golint_modernize":3,"golint_nilnil":3,"golint_perfsprint":0,"golint_prealloc":0,"golint_recvcheck":1,"golint_usestdlibvars":0,"golint_wastedassign":0,"golint_total":8,"golint_test_testifylint":0,"golint_test_thelper":0,"golint_test_usetesting":0,"golint_test_total":0,"golint_vetx_total":0,"eslint_problems":0,"eslint_errors":0,"eslint_warnings":0,"tsc_errors":0,"vitest_failed":0,"vitest_total":116,"measure_s":81}
50 lines
1.4 KiB
Go
50 lines
1.4 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Package wsclient provides the agent-side WebSocket client for connecting to the OpenFlare server.
|
|
package wsclient
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/apps/agent/protocol"
|
|
edgews "github.com/Rain-kl/Wavelet/internal/apps/edge/wsclient"
|
|
)
|
|
|
|
// WSMessage is an alias for the WebSocket message type.
|
|
type WSMessage = edgews.WSMessage
|
|
|
|
// MessageHandler is an alias for the WebSocket message handler function type.
|
|
type MessageHandler = edgews.MessageHandler
|
|
|
|
// Connection is an alias for the AgentConnection interface.
|
|
type Connection = edgews.AgentConnection
|
|
|
|
// Client wraps the connection client for agent WebSockets.
|
|
type Client struct {
|
|
inner *edgews.Client
|
|
}
|
|
|
|
// New creates a new Client instance.
|
|
func New(baseURL, token string, timeout time.Duration) *Client {
|
|
return &Client{
|
|
inner: edgews.New(edgews.PresetAgent, baseURL, token, timeout),
|
|
}
|
|
}
|
|
|
|
// SetToken updates the client's token.
|
|
func (c *Client) SetToken(token string) {
|
|
c.inner.SetToken(token)
|
|
}
|
|
|
|
// URL returns the WebSocket client's target URL.
|
|
func (c *Client) URL() string {
|
|
return c.inner.URL()
|
|
}
|
|
|
|
// Connect establishes a WebSocket connection to the server and returns the connection handle.
|
|
func (c *Client) Connect(ctx context.Context) (protocol.WebSocketConnection, error) {
|
|
return c.inner.ConnectAgent(ctx)
|
|
}
|