From 351d7d9f553f82c684f0b43d5766db5a909750a4 Mon Sep 17 00:00:00 2001 From: Andrew Delph Date: Fri, 23 Feb 2024 14:47:03 -0500 Subject: [PATCH] kpa: simplify computeActiveCondition logic Making changes to the pa active status has becomes difficult. This change breaks down some of the logic so that it is more readable. No changes to test cases were made as it doesn't actually change the logical cases. --- pkg/reconciler/autoscaling/kpa/kpa.go | 63 ++++++++++++++-------- pkg/reconciler/autoscaling/kpa/kpa_test.go | 25 ++++++++- 2 files changed, 64 insertions(+), 24 deletions(-) diff --git a/pkg/reconciler/autoscaling/kpa/kpa.go b/pkg/reconciler/autoscaling/kpa/kpa.go index 6c185949d534..b06d95cebe93 100644 --- a/pkg/reconciler/autoscaling/kpa/kpa.go +++ b/pkg/reconciler/autoscaling/kpa/kpa.go @@ -259,40 +259,57 @@ func reportMetrics(pa *autoscalingv1alpha1.PodAutoscaler, pc podCounts) { // | -1 | >= min | >0 | activating | active | // | -1 | >= min | >0 | active | active | func computeActiveCondition(ctx context.Context, pa *autoscalingv1alpha1.PodAutoscaler, pc podCounts) { + minReady := activeThreshold(ctx, pa) - if pc.ready >= minReady && pa.Status.ServiceName != "" { + isServiceCreated := pa.Status.ServiceName != "" + + // Mark the pa as Initialized if the service is created and reached the initial scale. + if pc.ready >= minReady && isServiceCreated { pa.Status.MarkScaleTargetInitialized() } - switch { - // Need to check for minReady = 0 because in the initialScale 0 case, pc.want will be -1. - case pc.want == 0 || minReady == 0: - if pa.Status.IsActivating() && minReady > 0 { - // We only ever scale to zero while activating if we fail to activate within the progress deadline. - pa.Status.MarkInactive("TimedOut", "The target could not be activated.") - } else { + // If the service is not created yet, it is either queued or inctive if initial scale zero. + if !isServiceCreated { + if minReady == 0 { pa.Status.MarkInactive(noTrafficReason, "The target is not receiving traffic.") - } - - case pc.ready < minReady: - if pc.want > 0 || !pa.Status.IsInactive() { + } else { pa.Status.MarkActivating( "Queued", "Requests to the target are being buffered as resources are provisioned.") + } + return + } + + // If the target is not initialized, it will be queued. + if !pa.Status.IsScaleTargetInitialized() { + pa.Status.MarkActivating( + "Queued", "Requests to the target are being buffered as resources are provisioned.") + return + } + + switch { + case pc.want == 0: + // The kpa is trying to scale to 0. + if pa.Status.IsActivating() { + // The pa is stuck in activating and the progressDeadline is forcing a scale to zero. + pa.Status.MarkInactive("TimedOut", "The target could not be activated.") } else { - // This is for the initialScale 0 case. In the first iteration, minReady is 0, - // but for the following iterations, minReady is 1. pc.want will continue being - // -1 until we start receiving metrics, so we will end up here. - // Even though PA is already been marked as inactive in the first iteration, we - // still need to set it again. Otherwise reconciliation will fail with NewObservedGenFailure - // because we cannot go through one iteration of reconciliation without setting - // some status. + // The pa is no longer receiving traffic. pa.Status.MarkInactive(noTrafficReason, "The target is not receiving traffic.") } - case pc.ready >= minReady: - if pc.want > 0 || !pa.Status.IsInactive() { - pa.Status.MarkActive() - } + case pc.ready >= minReady && pc.ready > 0: + // To be active there must be atleast 1 active pod. + // This takes into consideration when InitialScale is zero. + pa.Status.MarkActive() + + case pc.want < 0: + // The decider is not receiving metrics because the pa scaled to zero. + pa.Status.MarkInactive(noTrafficReason, "The target is not receiving traffic.") + + default: + // The desired scale has not been reached yet. + pa.Status.MarkActivating( + "Queued", "Requests to the target are being buffered as resources are provisioned.") } } diff --git a/pkg/reconciler/autoscaling/kpa/kpa_test.go b/pkg/reconciler/autoscaling/kpa/kpa_test.go index f16450cc44f2..4946b79e657c 100644 --- a/pkg/reconciler/autoscaling/kpa/kpa_test.go +++ b/pkg/reconciler/autoscaling/kpa/kpa_test.go @@ -1171,7 +1171,30 @@ func TestReconcile(t *testing.T) { WantUpdates: []clientgotesting.UpdateActionImpl{{ Object: defaultProxySKS, }}, - }} + }, { + Name: "image pull error - stay queued", // The service is created and not initialized + Key: key, + Ctx: context.WithValue(context.Background(), deciderKey{}, + decider(testNamespace, testRevision, -1, /* desiredScale */ + 0 /* ebc */)), + Objects: []runtime.Object{ + kpa(testNamespace, testRevision, WithPASKSNotReady(""), withScales(0, -1), WithReachabilityReachable, + WithPAMetricsService(privateSvc)), + // SKS won't be ready bc no ready endpoints, but private service name will be populated. + sks(testNamespace, testRevision, WithProxyMode, WithDeployRef(deployName), WithPrivateService, WithPubService), + metric(testNamespace, testRevision), + deploy(testNamespace, testRevision, func(d *appsv1.Deployment) { + d.Spec.Replicas = ptr.Int32(0) + }), + }, + WantStatusUpdates: []clientgotesting.UpdateActionImpl{{ + Object: kpa(testNamespace, testRevision, WithPASKSNotReady(""), + WithBufferedTraffic, + withScales(0, -1), WithReachabilityReachable, + WithPAMetricsService(privateSvc), WithObservedGeneration(1), + WithPAStatusService(testRevision), + ), + }}}} table.Test(t, MakeFactory(func(ctx context.Context, listers *Listers, cmw configmap.Watcher) controller.Reconciler { retryAttempted = false