apigatewaymanagementapi

type Connection struct {
	ConnectedAt    time.Time `json:"connectedAt"`
	LastActiveAt   time.Time `json:"lastActiveAt"`
	ConnectionID   string    `json:"connectionId"`
	SourceIP       string    `json:"sourceIp"`
	UserAgent      string    `json:"userAgent"`
	PostedMessages int       `json:"postedMessages"`
	BytesSent      int64     `json:"bytesSent"`
}

type EventType string

type Handler struct {
	Backend StorageBackend
}

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

type LifecycleEvent struct {
	At     time.Time `json:"at"`
	Type   EventType `json:"type"`
	Detail string    `json:"detail,omitempty"`
	Bytes  int       `json:"bytes,omitempty"`
}

type PostedMessage struct {
	ReceivedAt   time.Time `json:"receivedAt"`
	ConnectionID string    `json:"connectionId"`
	Data         []byte    `json:"data"`
}

type Provider struct{}

type Stats struct {
	ActiveConnections   int   `json:"activeConnections"`
	BufferedMessages    int   `json:"bufferedMessages"`
	TotalConnections    int64 `json:"totalConnections"`
	TotalDisconnections int64 `json:"totalDisconnections"`
	TotalMessages       int64 `json:"totalMessages"`
	TotalBroadcasts     int64 `json:"totalBroadcasts"`
	TotalBytesSent      int64 `json:"totalBytesSent"`
	TotalRejected       int64 `json:"totalRejected"`
}

type StorageBackend interface {
	// PostToConnection sends data to the specified WebSocket connection.
	PostToConnection(connectionID string, data []byte) error
	// GetConnection retrieves metadata for the specified WebSocket connection.
	GetConnection(connectionID string) (*Connection, error)
	// DeleteConnection deletes the specified WebSocket connection.
	DeleteConnection(connectionID string) error
	// CreateConnection creates a new simulated WebSocket connection.
	CreateConnection(connectionID, sourceIP, userAgent string, downstream chan []byte) (*Connection, error)
	// ListConnections returns all active WebSocket connections.
	ListConnections() []Connection
	// FilterConnections returns connections whose ID, IP, or user-agent contain query.
	FilterConnections(query string) []Connection
	// GetMessages returns all messages posted to the given connection.
	GetMessages(connectionID string) []PostedMessage
	// ClearMessages drops all stored messages for the connection.
	ClearMessages(connectionID string) error
	// GetTimeline returns lifecycle events for the given connection.
	GetTimeline(connectionID string) []LifecycleEvent
	// PingConnection refreshes LastActiveAt without storing payload data.
	PingConnection(connectionID string) error
	// Broadcast posts data to every active connection and returns the count delivered.
	Broadcast(data []byte) (int, error)
	// PruneIdle removes connections idle longer than threshold.
	PruneIdle(threshold time.Duration) []string
	// Stats returns cumulative counters and active state.
	Stats() Stats
	// Snapshot serialises backend state to JSON for persistence.
	Snapshot(ctx context.Context) []byte
	// Restore loads backend state from a JSON snapshot.
	Restore(ctx context.Context, data []byte) error
	// Reset clears all in-memory state for test isolation.
	Reset()
}