112 lines
2.9 KiB
Go
112 lines
2.9 KiB
Go
package transport
|
|
|
|
import (
|
|
"context"
|
|
"sync/atomic"
|
|
"time"
|
|
)
|
|
|
|
// Metrics struct extensions for server metrics
|
|
type ServerMetrics struct {
|
|
Metrics
|
|
ServerStarted uint64
|
|
ServerErrored uint64
|
|
ServerStopped uint64
|
|
}
|
|
|
|
// Connection represents a connection to a remote endpoint
|
|
type Connection interface {
|
|
// Execute executes a function with the underlying connection
|
|
Execute(func(interface{}) error) error
|
|
|
|
// Close closes the connection
|
|
Close() error
|
|
|
|
// Address returns the remote endpoint address
|
|
Address() string
|
|
|
|
// Status returns the connection status
|
|
Status() ConnectionStatus
|
|
}
|
|
|
|
// ConnectionStatus represents the status of a connection
|
|
type ConnectionStatus struct {
|
|
Connected bool
|
|
LastActivity time.Time
|
|
ErrorCount int
|
|
RequestCount int
|
|
LatencyAvg time.Duration
|
|
}
|
|
|
|
// TransportManager is an interface for managing transport layer operations
|
|
type TransportManager interface {
|
|
// Start starts the transport manager
|
|
Start() error
|
|
|
|
// Stop stops the transport manager
|
|
Stop(ctx context.Context) error
|
|
|
|
// Connect connects to a remote endpoint
|
|
Connect(ctx context.Context, address string) (Connection, error)
|
|
}
|
|
|
|
// ExtendedMetricsCollector extends the basic metrics collector with server metrics
|
|
type ExtendedMetricsCollector struct {
|
|
BasicMetricsCollector
|
|
serverStarted uint64
|
|
serverErrored uint64
|
|
serverStopped uint64
|
|
}
|
|
|
|
// NewMetrics creates a new extended metrics collector with a given transport name
|
|
func NewMetrics(transport string) *ExtendedMetricsCollector {
|
|
return &ExtendedMetricsCollector{
|
|
BasicMetricsCollector: BasicMetricsCollector{
|
|
avgLatencyByType: make(map[string]time.Duration),
|
|
requestCountByType: make(map[string]uint64),
|
|
},
|
|
}
|
|
}
|
|
|
|
// ServerStarted increments the server started counter
|
|
func (c *ExtendedMetricsCollector) ServerStarted() {
|
|
atomic.AddUint64(&c.serverStarted, 1)
|
|
}
|
|
|
|
// ServerErrored increments the server errored counter
|
|
func (c *ExtendedMetricsCollector) ServerErrored() {
|
|
atomic.AddUint64(&c.serverErrored, 1)
|
|
}
|
|
|
|
// ServerStopped increments the server stopped counter
|
|
func (c *ExtendedMetricsCollector) ServerStopped() {
|
|
atomic.AddUint64(&c.serverStopped, 1)
|
|
}
|
|
|
|
// ConnectionOpened records a connection opened event
|
|
func (c *ExtendedMetricsCollector) ConnectionOpened() {
|
|
atomic.AddUint64(&c.connections, 1)
|
|
}
|
|
|
|
// ConnectionFailed records a connection failed event
|
|
func (c *ExtendedMetricsCollector) ConnectionFailed() {
|
|
atomic.AddUint64(&c.connectionFailures, 1)
|
|
}
|
|
|
|
// ConnectionClosed records a connection closed event
|
|
func (c *ExtendedMetricsCollector) ConnectionClosed() {
|
|
// No specific counter for closed connections yet
|
|
}
|
|
|
|
// GetExtendedMetrics returns the current extended metrics
|
|
func (c *ExtendedMetricsCollector) GetExtendedMetrics() ServerMetrics {
|
|
baseMetrics := c.GetMetrics()
|
|
|
|
return ServerMetrics{
|
|
Metrics: baseMetrics,
|
|
ServerStarted: atomic.LoadUint64(&c.serverStarted),
|
|
ServerErrored: atomic.LoadUint64(&c.serverErrored),
|
|
ServerStopped: atomic.LoadUint64(&c.serverStopped),
|
|
}
|
|
}
|