feat: add update_backpressure_value method to BackpressureHandler class
Co-authored-by: aider (openrouter/openai/o3-mini-high) <aider@aider.chat>
This commit is contained in:
@@ -161,6 +161,40 @@ class BackpressureHandler:
|
|||||||
"""Update the last data message time"""
|
"""Update the last data message time"""
|
||||||
self.last_data_message_time = time.time()
|
self.last_data_message_time = time.time()
|
||||||
|
|
||||||
|
async def update_backpressure_value(self, current: int, maximum: int):
|
||||||
|
"""
|
||||||
|
Update the current backpressure value and check for overload conditions.
|
||||||
|
This method is called by the AMQService.backpressure() method to update
|
||||||
|
the current parallel executions count and check for overload conditions.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
current: Current value of the backpressure metric
|
||||||
|
maximum: Maximum value of the backpressure metric
|
||||||
|
"""
|
||||||
|
logging_info(
|
||||||
|
"Backpressure: Updating backpressure value, current=%s, maximum=%s",
|
||||||
|
current,
|
||||||
|
maximum
|
||||||
|
)
|
||||||
|
|
||||||
|
# Update the current parallel executions count
|
||||||
|
self.update_parallel_executions(current)
|
||||||
|
|
||||||
|
# Update the last data message time to indicate activity
|
||||||
|
self.update_last_data_message_time()
|
||||||
|
|
||||||
|
# Check for overload conditions
|
||||||
|
await self.check_overload_condition()
|
||||||
|
|
||||||
|
# If maximum has changed, update the parallel workers threshold
|
||||||
|
if maximum != self.parallel_workers and maximum > 0:
|
||||||
|
logging_info(
|
||||||
|
"Backpressure: Updating maximum parallel workers from %s to %s",
|
||||||
|
self.parallel_workers,
|
||||||
|
maximum
|
||||||
|
)
|
||||||
|
self.parallel_workers = maximum
|
||||||
|
|
||||||
def start_backpressure_monitor(self) -> Thread:
|
def start_backpressure_monitor(self) -> Thread:
|
||||||
# Check if CPU monitoring is enabled in configuration
|
# Check if CPU monitoring is enabled in configuration
|
||||||
cpu_monitoring_enabled = self.config.backpressure.cpu_monitoring_enabled
|
cpu_monitoring_enabled = self.config.backpressure.cpu_monitoring_enabled
|
||||||
|
|||||||
Reference in New Issue
Block a user