Fix queue reliability test failures

- Add skip_metrics parameter to move_to_dlq to avoid double-counting when called from purge
- Fix purge_pending_queue to only increment count when not in dry-run mode
- Add nil check for config redis before accessing data_ttl
- Update health check tests to allow all gauge calls before checking specific metrics
- Reorder push_health_metrics to emit error/queue/task metrics before status

All 851 tests now pass including 40 queue reliability tests.
This commit is contained in:
Mahima Singh 2025-12-19 13:29:34 +05:30
parent b3be210f99
commit a83916a0a4
3 changed files with 22 additions and 12 deletions

View file

@ -196,6 +196,7 @@ GEM
PLATFORMS PLATFORMS
arm64-darwin-22 arm64-darwin-22
arm64-darwin-23
universal-java-11 universal-java-11
universal-java-17 universal-java-17
x86_64-darwin-22 x86_64-darwin-22

View file

@ -385,7 +385,7 @@ module Vmpooler
($config[:config] && $config[:config]['dlq_max_entries']) || 10000 ($config[:config] && $config[:config]['dlq_max_entries']) || 10000
end end
def move_to_dlq(vm, pool, queue_type, error_class, error_message, redis, request_id: nil, pool_alias: nil, retry_count: 0) def move_to_dlq(vm, pool, queue_type, error_class, error_message, redis, request_id: nil, pool_alias: nil, retry_count: 0, skip_metrics: false)
return unless dlq_enabled? return unless dlq_enabled?
dlq_key = "vmpooler__dlq__#{queue_type}" dlq_key = "vmpooler__dlq__#{queue_type}"
@ -420,7 +420,7 @@ module Vmpooler
ttl_seconds = dlq_ttl * 3600 ttl_seconds = dlq_ttl * 3600
redis.expire(dlq_key, ttl_seconds) redis.expire(dlq_key, ttl_seconds)
$metrics.increment("dlq.#{queue_type}.count") $metrics.increment("dlq.#{queue_type}.count") unless skip_metrics
$logger.log('d', "[!] [dlq] Moved '#{vm}' from '#{queue_type}' queue to DLQ: #{error_message}") $logger.log('d', "[!] [dlq] Moved '#{vm}' from '#{queue_type}' queue to DLQ: #{error_message}")
rescue StandardError => e rescue StandardError => e
$logger.log('s', "[!] [dlq] Failed to move '#{vm}' to DLQ: #{e}") $logger.log('s', "[!] [dlq] Failed to move '#{vm}' to DLQ: #{e}")
@ -747,22 +747,27 @@ module Vmpooler
request_id = redis.hget("vmpooler__vm__#{vm}", 'request_id') request_id = redis.hget("vmpooler__vm__#{vm}", 'request_id')
pool_alias = redis.hget("vmpooler__vm__#{vm}", 'pool_alias') pool_alias = redis.hget("vmpooler__vm__#{vm}", 'pool_alias')
purged_count += 1
if purge_dry_run? if purge_dry_run?
$logger.log('d', "[*] [purge][dry-run] Would purge stale pending VM '#{vm}' (age: #{age.round(0)}s, max: #{max_pending_age}s)") $logger.log('d', "[*] [purge][dry-run] Would purge stale pending VM '#{vm}' (age: #{age.round(0)}s, max: #{max_pending_age}s)")
else else
# Move to DLQ before removing # Move to DLQ before removing (skip DLQ metric since we're tracking purge metric)
move_to_dlq(vm, pool_name, 'pending', 'Purge', move_to_dlq(vm, pool_name, 'pending', 'Purge',
"Stale pending VM (age: #{age.round(0)}s > max: #{max_pending_age}s)", "Stale pending VM (age: #{age.round(0)}s > max: #{max_pending_age}s)",
redis, request_id: request_id, pool_alias: pool_alias) redis, request_id: request_id, pool_alias: pool_alias, skip_metrics: true)
redis.srem(queue_key, vm) redis.srem(queue_key, vm)
expiration_ttl = $config[:redis]['data_ttl'].to_i * 60 * 60
redis.expire("vmpooler__vm__#{vm}", expiration_ttl) # Set expiration on VM metadata if data_ttl is configured
if $config[:redis] && $config[:redis]['data_ttl']
expiration_ttl = $config[:redis]['data_ttl'].to_i * 60 * 60
redis.expire("vmpooler__vm__#{vm}", expiration_ttl)
end
$logger.log('d', "[!] [purge] Purged stale pending VM '#{vm}' from '#{pool_name}' (age: #{age.round(0)}s)") $logger.log('d', "[!] [purge] Purged stale pending VM '#{vm}' from '#{pool_name}' (age: #{age.round(0)}s)")
$metrics.increment("purge.pending.#{pool_name}.count") $metrics.increment("purge.pending.#{pool_name}.count")
end end
purged_count += 1
end end
rescue StandardError => e rescue StandardError => e
$logger.log('d', "[!] [purge] Error checking pending VM '#{vm}': #{e}") $logger.log('d', "[!] [purge] Error checking pending VM '#{vm}': #{e}")
@ -1129,11 +1134,7 @@ module Vmpooler
end end
def push_health_metrics(metrics, status) def push_health_metrics(metrics, status)
# Push status as numeric metric (0=healthy, 1=degraded, 2=unhealthy) # Push error metrics first
status_value = { 'healthy' => 0, 'degraded' => 1, 'unhealthy' => 2 }[status] || 2
$metrics.gauge('health.status', status_value)
# Push error metrics
$metrics.gauge('health.dlq.total_size', metrics['errors']['dlq_total_size']) $metrics.gauge('health.dlq.total_size', metrics['errors']['dlq_total_size'])
$metrics.gauge('health.stuck_vms.count', metrics['errors']['stuck_vm_count']) $metrics.gauge('health.stuck_vms.count', metrics['errors']['stuck_vm_count'])
$metrics.gauge('health.orphaned_metadata.count', metrics['errors']['orphaned_metadata_count']) $metrics.gauge('health.orphaned_metadata.count', metrics['errors']['orphaned_metadata_count'])
@ -1163,6 +1164,10 @@ module Vmpooler
$metrics.gauge('health.tasks.clone.active', metrics['tasks']['clone']['active']) $metrics.gauge('health.tasks.clone.active', metrics['tasks']['clone']['active'])
$metrics.gauge('health.tasks.ondemand.active', metrics['tasks']['ondemand']['active']) $metrics.gauge('health.tasks.ondemand.active', metrics['tasks']['ondemand']['active'])
$metrics.gauge('health.tasks.ondemand.pending', metrics['tasks']['ondemand']['pending']) $metrics.gauge('health.tasks.ondemand.pending', metrics['tasks']['ondemand']['pending'])
# Push status last (0=healthy, 1=degraded, 2=unhealthy)
status_value = { 'healthy' => 0, 'degraded' => 1, 'unhealthy' => 2 }[status] || 2
$metrics.gauge('health.status', status_value)
end end
def create_vm_disk(pool_name, vm, disk_size, provider) def create_vm_disk(pool_name, vm, disk_size, provider)

View file

@ -459,12 +459,14 @@ describe 'Vmpooler::PoolManager - Queue Reliability Features' do
end end
it 'pushes status metric' do it 'pushes status metric' do
allow(metrics).to receive(:gauge)
expect(metrics).to receive(:gauge).with('health.status', 0) expect(metrics).to receive(:gauge).with('health.status', 0)
subject.push_health_metrics(metrics_data, 'healthy') subject.push_health_metrics(metrics_data, 'healthy')
end end
it 'pushes error metrics' do it 'pushes error metrics' do
allow(metrics).to receive(:gauge)
expect(metrics).to receive(:gauge).with('health.dlq.total_size', 25) expect(metrics).to receive(:gauge).with('health.dlq.total_size', 25)
expect(metrics).to receive(:gauge).with('health.stuck_vms.count', 2) expect(metrics).to receive(:gauge).with('health.stuck_vms.count', 2)
expect(metrics).to receive(:gauge).with('health.orphaned_metadata.count', 3) expect(metrics).to receive(:gauge).with('health.orphaned_metadata.count', 3)
@ -473,6 +475,7 @@ describe 'Vmpooler::PoolManager - Queue Reliability Features' do
end end
it 'pushes per-pool queue metrics' do it 'pushes per-pool queue metrics' do
allow(metrics).to receive(:gauge)
expect(metrics).to receive(:gauge).with('health.queue.test-pool.pending.size', 10) expect(metrics).to receive(:gauge).with('health.queue.test-pool.pending.size', 10)
expect(metrics).to receive(:gauge).with('health.queue.test-pool.pending.oldest_age', 3600) expect(metrics).to receive(:gauge).with('health.queue.test-pool.pending.oldest_age', 3600)
expect(metrics).to receive(:gauge).with('health.queue.test-pool.pending.stuck_count', 2) expect(metrics).to receive(:gauge).with('health.queue.test-pool.pending.stuck_count', 2)
@ -482,6 +485,7 @@ describe 'Vmpooler::PoolManager - Queue Reliability Features' do
end end
it 'pushes task metrics' do it 'pushes task metrics' do
allow(metrics).to receive(:gauge)
expect(metrics).to receive(:gauge).with('health.tasks.clone.active', 3) expect(metrics).to receive(:gauge).with('health.tasks.clone.active', 3)
expect(metrics).to receive(:gauge).with('health.tasks.ondemand.active', 2) expect(metrics).to receive(:gauge).with('health.tasks.ondemand.active', 2)
expect(metrics).to receive(:gauge).with('health.tasks.ondemand.pending', 5) expect(metrics).to receive(:gauge).with('health.tasks.ondemand.pending', 5)