Class: Nonnative::FaultInjectionProxy

Inherits:
Proxy
  • Object
show all
Defined in:
lib/nonnative/fault_injection_proxy.rb

Overview

Fault-injection proxy for TCP services.

This proxy accepts incoming TCP connections and forwards traffic to the configured upstream (service.proxy.host / service.proxy.port) via a socket-pair implementation. It can also inject failures to help validate client resilience.

This class exposes a small public control surface for tests:

  • #close_all: close connections immediately on accept
  • #reset_peer: reset connections immediately on accept (TCP RST rather than a graceful close)
  • #delay: delay reads by a configured duration (default: 2 seconds)
  • #timeout: accept connections and keep them silent until clients time out
  • #invalid_data: forward requests unchanged and mutate upstream responses before they reach clients
  • #bandwidth: throttle forwarded throughput to a configured rate (KB/s)
  • #reset: return to healthy pass-through behavior

State changes terminate any active connections so new connections observe the new behavior.

Wiring

When enabled, your test/client should connect to the service host and port (the proxy endpoint), and the proxy will forward traffic to the upstream target exposed by #host:#port.

Configuration

The proxy is configured via the service's proxy hash:

  • kind: "fault_injection"
  • host / port: upstream target behind the proxy (exposed via #host/#port)
  • log: file path used by this proxy’s internal logger
  • wait: sleep interval (seconds) applied after state changes
  • options:
    • delay: delay duration in seconds used by #delay
    • jitter: optional random offset (seconds) added in -jitter..jitter to each delay (a negative value uses its magnitude), so clients see variable latency instead of a flat value

See Also:

Defined Under Namespace

Classes: Connection

Instance Method Summary collapse

Constructor Details

#initialize(service) ⇒ FaultInjectionProxy

Returns a new instance of FaultInjectionProxy.

Parameters:



59
60
61
62
63
64
65
66
 
# File 'lib/nonnative/fault_injection_proxy.rb', line 59

def initialize(service)
  @connections = Concurrent::Hash.new
  @logger = Logger.new(service.proxy.log)
  @mutex = Mutex.new
  @state = :none

  super
end

Instance Method Details

#bandwidth ⇒ void

This method returns an undefined value.

Throttles forwarded throughput to a configured rate.

The rate is controlled by service.proxy.options[:rate] (kilobytes per second); when it is absent or not positive the connection forwards at full speed.



149
150
151
 
# File 'lib/nonnative/fault_injection_proxy.rb', line 149

def bandwidth
  apply_state :bandwidth
end

#close_all ⇒ void

This method returns an undefined value.

Forces new connections to be closed immediately.



102
103
104
 
# File 'lib/nonnative/fault_injection_proxy.rb', line 102

def close_all
  apply_state :close_all
end

#delay ⇒ void

This method returns an undefined value.

Delays reads before forwarding.

The delay duration is controlled by service.proxy.options[:delay] and defaults to 2 seconds.



121
122
123
 
# File 'lib/nonnative/fault_injection_proxy.rb', line 121

def delay
  apply_state :delay
end

#host ⇒ String

Returns the upstream host behind this proxy.

Returns:

  • (String) 


163
164
165
 
# File 'lib/nonnative/fault_injection_proxy.rb', line 163

def host
  service.proxy.host
end

#invalid_data ⇒ void

This method returns an undefined value.

Mutates upstream responses while forwarding client requests unchanged.



139
140
141
 
# File 'lib/nonnative/fault_injection_proxy.rb', line 139

def invalid_data
  apply_state :invalid_data
end

#port ⇒ Integer

Returns the upstream port behind this proxy.

Returns:

  • (Integer) 


170
171
172
 
# File 'lib/nonnative/fault_injection_proxy.rb', line 170

def port
  service.proxy.port
end

#reset ⇒ void

This method returns an undefined value.

Resets the proxy back to healthy pass-through behavior.



156
157
158
 
# File 'lib/nonnative/fault_injection_proxy.rb', line 156

def reset
  apply_state :none
end

#reset_peer ⇒ void

This method returns an undefined value.

Forces new connections to be reset immediately.

Unlike #close_all, which closes the socket gracefully (FIN), this closes the accepted socket with a zero linger timeout so clients observe a TCP reset (Errno::ECONNRESET).



112
113
114
 
# File 'lib/nonnative/fault_injection_proxy.rb', line 112

def reset_peer
  apply_state :reset_peer
end

#start ⇒ void

This method returns an undefined value.

Starts the proxy accept loop in a background thread.

This binds a TCP server on the service host and port. Clients connect to that service endpoint, while upstream traffic is forwarded to #host:#port.



74
75
76
77
78
79
 
# File 'lib/nonnative/fault_injection_proxy.rb', line 74

def start
  @tcp_server = ::TCPServer.new(service.host, service.port)
  @thread = Thread.new { perform_start }

  Nonnative.logger.info "started with host '#{service.host}' and port '#{service.port}' for proxy 'fault_injection'"
end

#stop ⇒ void

This method returns an undefined value.

Stops the proxy, closes active connections, and closes its listening socket.



84
85
86
87
88
89
90
91
92
93
94
95
96
97
 
# File 'lib/nonnative/fault_injection_proxy.rb', line 84

def stop
  server = tcp_server
  server&.close

  listener_thread = thread
  listener_thread&.join

  @tcp_server = nil
  @thread = nil

  close_connections

  Nonnative.logger.info "stopped with host '#{service.host}' and port '#{service.port}' for proxy 'fault_injection'"
end

#timeout ⇒ void

This method returns an undefined value.

Accepts connections and stalls without forwarding bytes.

This simulates a dependency that accepts a TCP connection but leaves clients waiting until their own read timeout fires. The proxy keeps the connection silent until reset or stop closes active connections.



132
133
134
 
# File 'lib/nonnative/fault_injection_proxy.rb', line 132

def timeout
  apply_state :timeout
end