add a go-routime

Author: rushmash91

pkg/runtime/reconciler.go

@@ -57,6 +57,9 @@ const (
 	// resource if the CARM cache is not synced yet, or if the roleARN is not
 	// available.
 	roleARNNotAvailableRequeueDelay = 15 * time.Second
+	// gracefulShutdownTimeout is the default duration to allow reconciliation
+	// to complete during controller shutdown
+	gracefulShutdownTimeout = 10 * time.Second
 	// adoptOrCreate is an annotation field that decides whether to create the
 	// resource if it doesn't exist, or adopt the resource if it exists.
 	// value comes from getAdoptionPolicy
@@ -207,7 +210,7 @@ func (r *reconciler) WriteToSecret(
 }
 
 // Reconcile implements `controller-runtime.Reconciler` and handles reconciling
-// a CR CRUD request
+// a CR CRUD request with graceful shutdown support.
 func (r *resourceReconciler) Reconcile(ctx context.Context, req ctrlrt.Request) (ctrlrt.Result, error) {
 	desired, err := r.getAWSResource(ctx, req)
 	if err != nil {
@@ -285,8 +288,26 @@ func (r *resourceReconciler) Reconcile(ctx context.Context, req ctrlrt.Request)
 	if err != nil {
 		return ctrlrt.Result{}, err
 	}
+
+	// Perform normal reconciliation with original context
 	latest, err := r.reconcile(ctx, rm, desired)
-	return r.HandleReconcileError(ctx, desired, latest, err)
+
+	// Use graceful context for final status patch only if original context is cancelled
+	finalCtx := ctx
+	if ctx.Err() != nil {
+		// Original context cancelled - create graceful context for final status patch
+		gracefulCtx, cancel := context.WithTimeout(context.Background(), gracefulShutdownTimeout)
+		defer cancel()
+
+		// Transfer important values from the original context to the graceful context
+		gracefulCtx = context.WithValue(gracefulCtx, ackrtlog.ContextKey, rlog)
+		gracefulCtx = context.WithValue(gracefulCtx, "resourceNamespace", req.Namespace)
+
+		finalCtx = gracefulCtx
+		rlog.Info("using graceful context for final status patch due to controller shutdown")
+	}
+
+	return r.HandleReconcileError(finalCtx, desired, latest, err)
 }
 
 func (r *resourceReconciler) handleCacheError(