axon

type Client[T any] struct {
	// contains filtered or unexported fields
}

type ClientOptions struct {
	// DialOptions for the underlying connection
	DialOptions

	// Reconnection configuration
	Reconnect *ReconnectConfig

	// QueueSize is the maximum number of messages to queue during disconnection
	// 0 disables queuing
	QueueSize int

	// QueueTimeout is how long queued messages are valid
	// Default: 30 seconds
	QueueTimeout time.Duration

	// OnError is called when an error occurs
	OnError func(error)

	// OnStateChange is called when the connection state changes
	OnStateChange StateHandler
}

type CloseCode int

type CloseError struct {
	Code   CloseCode
	Reason string
}

type CompressionManager struct {
	// contains filtered or unexported fields
}

type Conn[T any] struct {
	// contains filtered or unexported fields
}

type ConnectionState int32

type DialOptions struct {
	// HandshakeTimeout is the maximum time to wait for the handshake to complete.
	// Default is 30 seconds.
	HandshakeTimeout time.Duration

	// ReadBufferSize sets the size of the read buffer in bytes.
	// Default is 4096 bytes.
	ReadBufferSize int

	// WriteBufferSize sets the size of the write buffer in bytes.
	// Default is 4096 bytes.
	WriteBufferSize int

	// MaxFrameSize sets the maximum frame size in bytes.
	// Default is 4096 bytes.
	MaxFrameSize int

	// MaxMessageSize sets the maximum message size in bytes.
	// Default is 1048576 bytes (1MB).
	MaxMessageSize int

	// ReadDeadline sets the read deadline for connections.
	// Default is no deadline.
	ReadDeadline time.Duration

	// WriteDeadline sets the write deadline for connections.
	// Default is no deadline.
	WriteDeadline time.Duration

	// PingInterval sets the interval for sending ping frames.
	// If zero, pings are disabled.
	PingInterval time.Duration

	// PongTimeout sets the timeout for waiting for a pong response.
	// If zero, pong timeout is disabled.
	PongTimeout time.Duration

	// Subprotocols sets the list of supported subprotocols.
	// Default is nil (no subprotocols).
	Subprotocols []string

	// Compression enables per-message compression (RFC 7692).
	// Default is false (disabled).
	Compression bool

	// CompressionThreshold sets the minimum message size to compress.
	// Messages smaller than this will not be compressed.
	// Default is 256 bytes.
	CompressionThreshold int

	// Headers sets additional HTTP headers for the handshake request.
	Headers http.Header

	// TLSConfig specifies the TLS configuration to use for wss:// connections.
	// If nil, the default configuration is used.
	TLSConfig *tls.Config

	// NetDialer specifies the dialer to use for creating the network connection.
	// If nil, a default dialer is used.
	NetDialer *net.Dialer
}

type Dialer struct {
	// contains filtered or unexported fields
}

type Frame struct {
	Fin     bool
	Rsv1    bool
	Rsv2    bool
	Rsv3    bool
	Opcode  byte
	Masked  bool
	MaskKey []byte
	Payload []byte
}

type MessageQueue[T any] struct {
	// contains filtered or unexported fields
}

type MessageQueueStats struct {
	CurrentSize int
	MaxSize     int
	Dropped     int64
	Enqueued    int64
	Sent        int64
}

type Metrics struct {
	// Connection metrics
	ActiveConnections atomic.Int64
	TotalConnections  atomic.Int64
	ClosedConnections atomic.Int64

	// Message metrics
	MessagesRead    atomic.Int64
	MessagesWritten atomic.Int64
	BytesRead       atomic.Int64
	BytesWritten    atomic.Int64

	// Error metrics
	ReadErrors      atomic.Int64
	WriteErrors     atomic.Int64
	FrameErrors     atomic.Int64
	HandshakeErrors atomic.Int64

	// Performance metrics
	ReadLatency  atomic.Int64 // nanoseconds
	WriteLatency atomic.Int64 // nanoseconds

	// Reconnection metrics
	ReconnectAttempts  atomic.Int64
	ReconnectSuccesses atomic.Int64
	ReconnectFailures  atomic.Int64

	// Queue metrics
	QueueEnqueued atomic.Int64
	QueueSent     atomic.Int64
	QueueDropped  atomic.Int64

	// Compression metrics
	CompressedMessages   atomic.Int64
	DecompressedMessages atomic.Int64
	CompressionSaved     atomic.Int64 // bytes saved by compression
}

type MetricsSnapshot struct {
	ActiveConnections int64
	TotalConnections  int64
	ClosedConnections int64
	MessagesRead      int64
	MessagesWritten   int64
	BytesRead         int64
	BytesWritten      int64
	ReadErrors        int64
	WriteErrors       int64
	FrameErrors       int64
	HandshakeErrors   int64
	AvgReadLatency    time.Duration
	AvgWriteLatency   time.Duration

	// Reconnection metrics
	ReconnectAttempts  int64
	ReconnectSuccesses int64
	ReconnectFailures  int64

	// Queue metrics
	QueueEnqueued int64
	QueueSent     int64
	QueueDropped  int64

	// Compression metrics
	CompressedMessages   int64
	DecompressedMessages int64
	CompressionSaved     int64
}

type ReconnectConfig struct {
	// Enabled determines if automatic reconnection is active
	Enabled bool

	// MaxAttempts is the maximum number of reconnection attempts (0 = unlimited)
	MaxAttempts int

	// InitialDelay is the initial delay before the first reconnection attempt
	// Default: 1 second
	InitialDelay time.Duration

	// MaxDelay is the maximum delay between reconnection attempts
	// Default: 30 seconds
	MaxDelay time.Duration

	// BackoffMultiplier is the multiplier for exponential backoff
	// Default: 2.0
	BackoffMultiplier float64

	// Jitter adds randomness to delay to prevent thundering herd
	// Default: true
	Jitter bool

	// ResetAfter resets the attempt counter after being connected for this duration
	// Default: 60 seconds
	ResetAfter time.Duration

	// OnReconnecting is called when a reconnection attempt starts
	OnReconnecting func(attempt int, delay time.Duration)

	// OnReconnected is called when reconnection succeeds
	OnReconnected func(attempt int)

	// OnReconnectFailed is called when reconnection fails
	OnReconnectFailed func(attempt int, err error)

	// ShouldReconnect is called to determine if reconnection should be attempted
	// If nil, always attempts to reconnect (within MaxAttempts)
	ShouldReconnect func(err error, attempt int) bool
}

type StateChange struct {
	From      ConnectionState
	To        ConnectionState
	Time      time.Time
	Err       error  // Error that caused the transition (if any)
	Attempt   int    // Reconnection attempt number (if applicable)
	SessionID string // Current session identifier
}

type StateHandler func(StateChange)

type UpgradeOptions struct {
	// ReadBufferSize sets the size of the read buffer in bytes.
	// Default is 4096 bytes.
	ReadBufferSize int

	// WriteBufferSize sets the size of the write buffer in bytes.
	// Default is 4096 bytes.
	WriteBufferSize int

	// MaxFrameSize sets the maximum frame size in bytes.
	// Frames exceeding this size will result in ErrFrameTooLarge.
	// Default is 4096 bytes.
	MaxFrameSize int

	// MaxMessageSize sets the maximum message size in bytes.
	// Messages exceeding this size will result in ErrMessageTooLarge.
	// Default is 1048576 bytes (1MB).
	MaxMessageSize int

	// ReadDeadline sets the read deadline for connections.
	// Default is no deadline.
	ReadDeadline time.Duration

	// WriteDeadline sets the write deadline for connections.
	// Default is no deadline.
	WriteDeadline time.Duration

	// PingInterval sets the interval for sending ping frames.
	// If zero, pings are disabled.
	// Default is 0 (disabled).
	PingInterval time.Duration

	// PongTimeout sets the timeout for waiting for a pong response.
	// If zero, pong timeout is disabled.
	// Default is 0 (disabled).
	PongTimeout time.Duration

	// CheckOrigin sets a function to validate the origin header.
	// If nil, all origins are allowed.
	// Default is nil (all origins allowed).
	CheckOrigin func(r *http.Request) bool

	// Subprotocols sets the list of supported subprotocols.
	// The client's requested subprotocol must match one of these.
	// Default is nil (no subprotocols).
	Subprotocols []string

	// Compression enables per-message compression (RFC 7692).
	// Default is false (disabled).
	Compression bool
}

type Upgrader struct {
	// contains filtered or unexported fields
}