feat: make EventSourceRegistry available from Context

Author: metacosm

operator-framework-core/src/main/java/io/javaoperatorsdk/operator/api/reconciler/Context.java

@@ -1,9 +1,13 @@
 package io.javaoperatorsdk.operator.api.reconciler;
 
+import io.fabric8.kubernetes.api.model.HasMetadata;
+import io.javaoperatorsdk.operator.processing.event.source.EventSourceRegistry;
 import java.util.Optional;
 
-public interface Context {
+public interface Context<P extends HasMetadata> {
 
   Optional<RetryInfo> getRetryInfo();
 
+  EventSourceRegistry<P> getEventSourceRegistry();
+
 }

operator-framework-core/src/main/java/io/javaoperatorsdk/operator/api/reconciler/DefaultContext.java

@@ -1,17 +1,27 @@
 package io.javaoperatorsdk.operator.api.reconciler;
 
+import io.fabric8.kubernetes.api.model.HasMetadata;
+import io.javaoperatorsdk.operator.processing.event.source.EventSourceRegistry;
 import java.util.Optional;
 
-public class DefaultContext implements Context {
+public class DefaultContext<P extends HasMetadata> implements Context<P> {
 
   private final RetryInfo retryInfo;
+  private final EventSourceRegistry<P> registry;
 
-  public DefaultContext(RetryInfo retryInfo) {
+  public DefaultContext(RetryInfo retryInfo,
+      EventSourceRegistry<P> registry) {
     this.retryInfo = retryInfo;
+    this.registry = registry;
   }
 
   @Override
   public Optional<RetryInfo> getRetryInfo() {
     return Optional.ofNullable(retryInfo);
   }
+
+  @Override
+  public EventSourceRegistry<P> getEventSourceRegistry() {
+    return registry;
+  }
 }

operator-framework-core/src/main/java/io/javaoperatorsdk/operator/api/reconciler/Reconciler.java

@@ -28,7 +28,7 @@ public interface Reconciler<R extends HasMetadata> {
    *         finalizer to indicate that the resource should not be deleted after all, in which case
    *         the controller should restore the resource's state appropriately.
    */
-  default DeleteControl cleanup(R resource, Context context) {
+  default DeleteControl cleanup(R resource, Context<R> context) {
     return DeleteControl.defaultDelete();
   }
 
@@ -46,5 +46,5 @@ default DeleteControl cleanup(R resource, Context context) {
    *         be skipped. <b>However we will always call an update if there is no finalizer on object
    *         and it's not marked for deletion.</b>
    */
-  UpdateControl<R> reconcile(R resource, Context context);
+  UpdateControl<R> reconcile(R resource, Context<R> context);
 }

operator-framework-core/src/main/java/io/javaoperatorsdk/operator/processing/Controller.java

@@ -285,4 +285,8 @@ public void stop() {
       eventSourceManager.stop();
     }
   }
+
+  public EventSourceRegistry<R> getEventSourceRegistry() {
+    return eventSourceManager;
+  }
 }

operator-framework-core/src/main/java/io/javaoperatorsdk/operator/processing/event/ReconciliationDispatcher.java

@@ -67,8 +67,8 @@ private PostExecutionControl<R> handleDispatch(ExecutionScope<R> executionScope)
       return PostExecutionControl.defaultDispatch();
     }
 
-    Context context =
-        new DefaultContext(executionScope.getRetryInfo());
+    Context<R> context =
+        new DefaultContext<>(executionScope.getRetryInfo(), controller.getEventSourceRegistry());
     if (markedForDeletion) {
       return handleCleanup(resource, context);
     } else {
@@ -91,7 +91,7 @@ private boolean shouldNotDispatchToDelete(R resource) {
   }
 
   private PostExecutionControl<R> handleReconcile(
-      ExecutionScope<R> executionScope, R originalResource, Context context) {
+      ExecutionScope<R> executionScope, R originalResource, Context<R> context) {
     if (configuration().useFinalizer()
         && !originalResource.hasFinalizer(configuration().getFinalizer())) {
       /*
@@ -121,7 +121,7 @@ private PostExecutionControl<R> handleReconcile(
    * resource is changed during an execution, and it's much cleaner to have to original resource in
    * place for status update.
    */
-  private R cloneResourceForErrorStatusHandlerIfNeeded(R resource, Context context) {
+  private R cloneResourceForErrorStatusHandlerIfNeeded(R resource, Context<R> context) {
     if (isErrorStatusHandlerPresent() ||
         shouldUpdateObservedGenerationAutomatically(resource)) {
       return configuration().getConfigurationService().getResourceCloner().clone(resource);
@@ -131,7 +131,7 @@ private R cloneResourceForErrorStatusHandlerIfNeeded(R resource, Context context
   }
 
   private PostExecutionControl<R> reconcileExecution(ExecutionScope<R> executionScope,
-      R resourceForExecution, R originalResource, Context context) {
+      R resourceForExecution, R originalResource, Context<R> context) {
     log.debug(
         "Executing createOrUpdate for resource {} with version: {} with execution scope: {}",
         getName(resourceForExecution),
@@ -161,7 +161,7 @@ && shouldUpdateObservedGenerationAutomatically(resourceForExecution)) {
     return createPostExecutionControl(updatedCustomResource, updateControl);
   }
 
-  private void handleErrorStatusHandler(R resource, Context context,
+  private void handleErrorStatusHandler(R resource, Context<R> context,
       RuntimeException e) {
     if (isErrorStatusHandlerPresent()) {
       try {
@@ -238,7 +238,7 @@ private void updatePostExecutionControlWithReschedule(
     baseControl.getScheduleDelay().ifPresent(postExecutionControl::withReSchedule);
   }
 
-  private PostExecutionControl<R> handleCleanup(R resource, Context context) {
+  private PostExecutionControl<R> handleCleanup(R resource, Context<R> context) {
     log.debug(
         "Executing delete for resource: {} with version: {}",
         getName(resource),