Class: ToggleCraft::ConnectionPool

Inherits:
Object
  • Object
show all
Defined in:
lib/togglecraft/connection_pool.rb

Overview

Connection pool manager for SSE connections Ensures that multiple SDK instances can share the same SSE connection

Class Method Summary collapse

Class Method Details

.clear(force: false) ⇒ void

This method returns an undefined value.

Clear all connections

Parameters:

  • force (Boolean) (defaults to: false) —

    Force disconnect all connections



82
83
84
85
 
# File 'lib/togglecraft/connection_pool.rb', line 82

def clear(force: false)
  connections.each_value(&:force_disconnect) if force
  connections.clear
end

.connection_count ⇒ Integer

Get the number of active connections

Returns:

  • (Integer) 


98
99
100
 
# File 'lib/togglecraft/connection_pool.rb', line 98

def connection_count
  connections.size
end

.connections ⇒ Concurrent::Map

Get the connections map

Returns:

  • (Concurrent::Map) 


10
11
12
 
# File 'lib/togglecraft/connection_pool.rb', line 10

def connections
  @connections ||= Concurrent::Map.new
end

.generate_pool_key(config) ⇒ String

Generate a pool key from configuration

Parameters:

  • config (Hash) —

    Client configuration

Returns:

  • (String) —

    Pool key for the connection



51
52
53
54
55
56
57
58
59
 
# File 'lib/togglecraft/connection_pool.rb', line 51

def generate_pool_key(config)
  # Allow custom pool key for advanced users
  return config[:connection_pool_key] if config[:connection_pool_key]

  # Generate key from connection parameters
  url = config[:url] || 'https://sse.togglecraft.io'
  sdk_key = config[:sdk_key]
  "#{url}:#{sdk_key}"
end

.get_connection(config) ⇒ SharedSSEConnection?

Get or create a shared connection

Parameters:

  • config (Hash) —

    Client configuration

Returns:



17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
 
# File 'lib/togglecraft/connection_pool.rb', line 17

def get_connection(config)
  # If connection sharing is disabled, return nil to create a dedicated connection
  unless config[:share_connection]
    log 'Connection sharing disabled for this client'
    return nil
  end

  pool_key = generate_pool_key(config)
  log "Getting connection for pool key: #{pool_key}"

  # Check if connection already exists
  connection = connections[pool_key]

  if connection
    log "Reusing existing connection for #{pool_key} (#{connection.client_count} clients)"
  else
    # Create new shared SSE connection
    log "Creating new shared SSE connection for #{pool_key}"
    connection = SharedSSEConnection.new(config, pool_key)
    connections[pool_key] = connection

    # Set up cleanup when connection has no more clients
    connection.on_empty do
      log "Removing empty connection for #{pool_key}"
      connections.delete(pool_key)
    end
  end

  connection
end

.get_connection_by_key(pool_key) ⇒ SharedSSEConnection?

Get connection by pool key (for testing/debugging)

Parameters:

  • pool_key (String)

Returns:



118
119
120
 
# File 'lib/togglecraft/connection_pool.rb', line 118

def get_connection_by_key(pool_key)
  connections[pool_key]
end

.has_connection?(pool_key) ⇒ Boolean

Check if a specific pool key has an active connection

Parameters:

  • pool_key (String)

Returns:

  • (Boolean) 


111
112
113
 
# File 'lib/togglecraft/connection_pool.rb', line 111

def has_connection?(pool_key)
  connections.key?(pool_key)
end

.set_debug(enabled) ⇒ void

This method returns an undefined value.

Enable or disable debug logging

Parameters:

  • enabled (Boolean) 


90
91
92
93
94
 
# File 'lib/togglecraft/connection_pool.rb', line 90

def set_debug(enabled)
  @debug = enabled
  # Also enable debug on existing connections
  connections.each_value { |conn| conn.set_debug(enabled) }
end

.stats ⇒ Hash

Get statistics about the connection pool

Returns:

  • (Hash) —

    Pool statistics



63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
 
# File 'lib/togglecraft/connection_pool.rb', line 63

def stats
  conn_stats = []
  connections.each do |key, conn|
    conn_stats << {
      key: key,
      client_count: conn.client_count,
      connected: conn.connected?
    }
  end

  {
    total_connections: connections.size,
    connections: conn_stats
  }
end

.total_client_count ⇒ Integer

Get the total number of clients across all connections

Returns:

  • (Integer) 


104
105
106
 
# File 'lib/togglecraft/connection_pool.rb', line 104

def total_client_count
  connections.values.sum(&:client_count)
end