gatewayd-io / gatewayd / 12449272098

60%

Build Type

push

github

Committed by web-flow

Commit Message

Add Raft State Management for Load Balancers (#641)

* feat(raft): add round-robin state management

- Implemented round-robin state to enhance leader election and task distribution.
- Updated Raft state machine to incorporate round-robin logic.
- Added tests to ensure correct round-robin behavior in various scenarios.

This change improves the efficiency and fairness of task handling within the Raft cluster.

* Fix race condition in server test by adding mutex for proxy state checks

- Introduced a mutex (`proxyStateMutex`) to synchronize access to proxy state checks in `testProxy` function.
- Modified the `testProxy` function to lock the mutex before checking the state of `AvailableConnections` and `busyConnections`.
- Updated the test logic to ensure that one of the proxies is in the expected state, preventing race conditions where the second goroutine could access the connection state prematurely.
- Removed the `proxy` parameter from `testProxy` function calls as it is no longer needed.

This change addresses a race condition that could cause the second goroutine to access connection states before they are properly synchronized, ensuring reliable test results.

* feat: Add weighted round-robin state to Raft

Add support for storing weighted round-robin load balancer state in Raft FSM
to ensure consistency across cluster nodes. Changes include:

- Add WeightedProxy and WeightedRRPayload structs for state management
- Store proxy weights in Raft FSM using weightedRRStates map
- Update WeightedRoundRobin to use Raft for weight tracking
- Add new CommandUpdateWeightedRR command type
- Remove local weight tracking in favor of distributed state

This change ensures that proxy weights remain consistent across cluster nodes
during failover and leader changes.

* perf(network): optimize weighted round-robin with batch updates

Improve performance of WeightedRoundRobin.NextProxy by reducing Raft operations:
- Replace multiple individual Raft updates with a ... (continued)

Run Details

176 of 212 new or added lines in 4 files covered. (83.02%)

2 existing lines in 1 file now uncovered.

5044 of 8077 relevant lines covered (62.45%)

17.44 hits per line