[PECOBLR-1147] Implement Client Manager for Per-Host Clients

telemetry/DESIGN.md

@@ -1744,7 +1744,8 @@ func BenchmarkInterceptor_Disabled(b *testing.B) {
 
 ### Phase 2: Per-Host Management
 - [x] Implement `featureflag.go` with caching and reference counting (PECOBLR-1146)
-- [ ] Implement `manager.go` for client management
+- [x] Implement `manager.go` for client management (PECOBLR-1147)
+- [x] Implement `client.go` with minimal telemetryClient stub (PECOBLR-1147)
 - [ ] Implement `circuitbreaker.go` with state machine
 - [ ] Add unit tests for all components
 

telemetry/client.go

@@ -0,0 +1,38 @@
+package telemetry
+
+import (
+	"net/http"
+)
+
+// telemetryClient represents a client for sending telemetry data to Databricks.
+// This is a minimal stub implementation that will be fully implemented in Phase 4.
+type telemetryClient struct {
+	host       string
+	httpClient *http.Client
+	cfg        *Config
+	started    bool
+	closed     bool
+}
+
+// newTelemetryClient creates a new telemetry client for the given host.
+func newTelemetryClient(host string, httpClient *http.Client, cfg *Config) *telemetryClient {
+	return &telemetryClient{
+		host:       host,
+		httpClient: httpClient,
+		cfg:        cfg,
+	}
+}
+
+// start starts the telemetry client's background operations.
+// This is a stub implementation that will be fully implemented in Phase 4.
+func (c *telemetryClient) start() error {
+	c.started = true
+	return nil
+}
+
+// close stops the telemetry client and flushes any pending data.
+// This is a stub implementation that will be fully implemented in Phase 4.
+func (c *telemetryClient) close() error {
+	c.closed = true
+	return nil
+}

telemetry/manager.go

@@ -0,0 +1,73 @@
+package telemetry
+
+import (
+	"net/http"
+	"sync"
+)
+
+// clientManager manages one telemetry client per host.
+// Prevents rate limiting by sharing clients across connections.
+type clientManager struct {
+	mu      sync.RWMutex
+	clients map[string]*clientHolder
+}
+
+// clientHolder holds a telemetry client and its reference count.
+type clientHolder struct {
+	client   *telemetryClient
+	refCount int
+}
+
+var (
+	managerOnce     sync.Once
+	managerInstance *clientManager
+)
+
+// getClientManager returns the singleton instance.
+func getClientManager() *clientManager {
+	managerOnce.Do(func() {
+		managerInstance = &clientManager{
+			clients: make(map[string]*clientHolder),
+		}
+	})
+	return managerInstance
+}
+
+// getOrCreateClient gets or creates a telemetry client for the host.
+// Increments reference count.
+func (m *clientManager) getOrCreateClient(host string, httpClient *http.Client, cfg *Config) *telemetryClient {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+
+	holder, exists := m.clients[host]
+	if !exists {
+		holder = &clientHolder{
+			client: newTelemetryClient(host, httpClient, cfg),
+		}
+		m.clients[host] = holder
+		_ = holder.client.start() // Start background flush goroutine
+	}
+	holder.refCount++
+	return holder.client
+}
+
+// releaseClient decrements reference count for the host.
+// Closes and removes client when ref count reaches zero.
+func (m *clientManager) releaseClient(host string) error {
+	m.mu.Lock()
+	holder, exists := m.clients[host]
+	if !exists {
+		m.mu.Unlock()
+		return nil
+	}
+
+	holder.refCount--
+	if holder.refCount <= 0 {
+		delete(m.clients, host)
+		m.mu.Unlock()
+		return holder.client.close() // Close and flush
+	}
+
+	m.mu.Unlock()
+	return nil
+}