feat: status updates

Signed-off-by: Attila Mészáros <[email protected]>

Author: csviri

src/main/java/io/javaoperatorsdk/operator/glue/customresource/glue/RelatedResourceSpec.java

@@ -3,6 +3,7 @@
 import java.util.List;
 import java.util.Objects;
 
+import io.fabric8.crd.generator.annotation.PreserveUnknownFields;
 import io.fabric8.generator.annotation.Required;
 
 public class RelatedResourceSpec {
@@ -19,6 +20,9 @@ public class RelatedResourceSpec {
   private boolean clusterScoped = Boolean.FALSE;
   private List<String> resourceNames;
 
+  @PreserveUnknownFields
+  private Object statusPatch;
+  private String statusPatchTemplate;
 
   public String getApiVersion() {
     return apiVersion;
@@ -73,14 +77,16 @@ public boolean equals(Object o) {
       return false;
     RelatedResourceSpec that = (RelatedResourceSpec) o;
     return clusterScoped == that.clusterScoped && Objects.equals(name, that.name)
-        && Objects.equals(apiVersion, that.apiVersion) && Objects.equals(kind, that.kind)
-        && Objects.equals(namespace, that.namespace)
-        && Objects.equals(resourceNames, that.resourceNames);
+        && Objects.equals(namespace, that.namespace) && Objects.equals(apiVersion, that.apiVersion)
+        && Objects.equals(kind, that.kind) && Objects.equals(resourceNames, that.resourceNames)
+        && Objects.equals(statusPatch, that.statusPatch)
+        && Objects.equals(statusPatchTemplate, that.statusPatchTemplate);
   }
 
   @Override
   public int hashCode() {
-    return Objects.hash(name, apiVersion, kind, clusterScoped, namespace, resourceNames);
+    return Objects.hash(name, namespace, apiVersion, kind, clusterScoped, resourceNames,
+        statusPatch, statusPatchTemplate);
   }
 
   public boolean isClusterScoped() {
@@ -90,4 +96,20 @@ public boolean isClusterScoped() {
   public void setClusterScoped(boolean clusterScoped) {
     this.clusterScoped = clusterScoped;
   }
+
+  public Object getStatusPatch() {
+    return statusPatch;
+  }
+
+  public void setStatusPatch(Object statusPatch) {
+    this.statusPatch = statusPatch;
+  }
+
+  public String getStatusPatchTemplate() {
+    return statusPatchTemplate;
+  }
+
+  public void setStatusPatchTemplate(String statusPatchTemplate) {
+    this.statusPatchTemplate = statusPatchTemplate;
+  }
 }

src/main/java/io/javaoperatorsdk/operator/glue/reconciler/glue/GlueReconciler.java

@@ -10,6 +10,7 @@
 import io.fabric8.kubernetes.client.KubernetesClientException;
 import io.fabric8.kubernetes.client.dsl.base.PatchContext;
 import io.fabric8.kubernetes.client.dsl.base.PatchType;
+import io.fabric8.kubernetes.client.utils.Serialization;
 import io.javaoperatorsdk.operator.api.reconciler.*;
 import io.javaoperatorsdk.operator.glue.Utils;
 import io.javaoperatorsdk.operator.glue.conditions.JavaScripCondition;
@@ -47,17 +48,19 @@ public class GlueReconciler implements Reconciler<Glue>, Cleaner<Glue>, ErrorSta
 
   private final ValidationAndErrorHandler validationAndErrorHandler;
   private final InformerRegister informerRegister;
+  private final GenericTemplateHandler templateHandler;
 
   private final KubernetesResourceDeletedCondition deletePostCondition =
       new KubernetesResourceDeletedCondition();
 
   private final GenericTemplateHandler genericTemplateHandler;
 
   public GlueReconciler(ValidationAndErrorHandler validationAndErrorHandler,
-      InformerRegister informerRegister,
+      InformerRegister informerRegister, GenericTemplateHandler templateHandler,
       GenericTemplateHandler genericTemplateHandler) {
     this.validationAndErrorHandler = validationAndErrorHandler;
     this.informerRegister = informerRegister;
+    this.templateHandler = templateHandler;
     this.genericTemplateHandler = genericTemplateHandler;
   }
 
@@ -89,6 +92,7 @@ public UpdateControl<Glue> reconcile(Glue primary,
     cleanupRemovedResourcesFromWorkflow(context, primary);
     informerRegister.deRegisterInformerOnResourceFlowChange(context, primary);
     result.throwAggregateExceptionIfErrorsPresent();
+    patchRelatedResourcesStatus(context, primary);
     return UpdateControl.noUpdate();
   }
 
@@ -222,6 +226,35 @@ private GenericDependentResource createDependentResource(DependentResourceSpec s
     }
   }
 
+  // todo add workflow result?
+  private void patchRelatedResourcesStatus(Context<Glue> context,
+      Glue primary) {
+
+    var targetRelatedResources = primary.getSpec().getRelatedResources().stream()
+        .filter(r -> r.getStatusPatch() != null || r.getStatusPatchTemplate() != null)
+        .toList();
+
+    if (!targetRelatedResources.isEmpty()) {
+      return;
+    }
+    var actualData = genericTemplateHandler.createDataWithResources(primary, context);
+
+    targetRelatedResources.forEach(r -> {
+      var relatedResources = Utils.getRelatedResources(primary, r, context);
+
+      var template = r.getStatusPatchTemplate() != null ? r.getStatusPatchTemplate()
+          : Serialization.asYaml(r.getStatusPatch());
+      var resultTemplate =
+          genericTemplateHandler.processTemplate(actualData, template);
+      var statusObjectMap = GenericTemplateHandler.parseTemplateToMapObject(resultTemplate);
+      relatedResources.forEach((n, kr) -> {
+        kr.setAdditionalProperty("status", statusObjectMap);
+        context.getClient().resource(kr).patchStatus();
+      });
+    });
+
+  }
+
   @SuppressWarnings({"rawtypes"})
   private Condition toCondition(ConditionSpec condition) {
     if (condition instanceof ReadyConditionSpec readyConditionSpec) {

src/main/java/io/javaoperatorsdk/operator/glue/templating/GenericTemplateHandler.java

@@ -29,20 +29,26 @@ public String processTemplate(Map<String, Map<?, ?>> data, String template) {
 
   public String processInputAndTemplate(Map<String, GenericKubernetesResource> data,
       String template) {
-    Map<String, Map<?, ?>> res = new HashMap<>();
-    data.forEach((key, value) -> res.put(key,
-        value == null ? null : objectMapper.convertValue(value, Map.class)));
+    Map<String, Map<?, ?>> res =
+        genericKubernetesResourceDataToGenericData(data);
     return processTemplate(res, template);
   }
 
   public String processTemplate(String template, Glue primary,
       Context<Glue> context) {
-
     var data = createDataWithResources(primary, context);
     return processTemplate(data, template);
   }
 
-  private static Map<String, Map<?, ?>> createDataWithResources(Glue primary,
+  private static Map<String, Map<?, ?>> genericKubernetesResourceDataToGenericData(
+      Map<String, GenericKubernetesResource> data) {
+    Map<String, Map<?, ?>> res = new HashMap<>();
+    data.forEach((key, value) -> res.put(key,
+        value == null ? null : objectMapper.convertValue(value, Map.class)));
+    return res;
+  }
+
+  public static Map<String, Map<?, ?>> createDataWithResources(Glue primary,
       Context<Glue> context) {
     Map<String, Map<?, ?>> res = new HashMap<>();
     var actualResourcesByName = Utils.getActualResourcesByNameInWorkflow(context, primary);
@@ -55,4 +61,11 @@ public String processTemplate(String template, Glue primary,
 
     return res;
   }
+
+  // todo unit test
+  @SuppressWarnings("unchecked")
+  public static Map<String, ?> parseTemplateToMapObject(String template) {
+    return objectMapper.convertValue(template, Map.class);
+  }
+
 }