io.javaoperatorsdk.operator.sample.MySQLSchemaReconciler Maven / Gradle / Ivy

Go to download
package io.javaoperatorsdk.operator.sample;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import io.fabric8.kubernetes.api.model.Secret;
import io.javaoperatorsdk.operator.api.reconciler.Context;
import io.javaoperatorsdk.operator.api.reconciler.ControllerConfiguration;
import io.javaoperatorsdk.operator.api.reconciler.ErrorStatusHandler;
import io.javaoperatorsdk.operator.api.reconciler.ErrorStatusUpdateControl;
import io.javaoperatorsdk.operator.api.reconciler.Reconciler;
import io.javaoperatorsdk.operator.api.reconciler.UpdateControl;
import io.javaoperatorsdk.operator.api.reconciler.dependent.Dependent;
import io.javaoperatorsdk.operator.sample.dependent.SchemaDependentResource;
import io.javaoperatorsdk.operator.sample.dependent.SecretDependentResource;
import io.javaoperatorsdk.operator.sample.schema.Schema;

import static io.javaoperatorsdk.operator.sample.dependent.SchemaDependentResource.decode;
import static io.javaoperatorsdk.operator.sample.dependent.SecretDependentResource.MYSQL_SECRET_USERNAME;
import static java.lang.String.format;

@ControllerConfiguration(
    dependents = {
        @Dependent(type = SecretDependentResource.class, name = SecretDependentResource.NAME),
        @Dependent(type = SchemaDependentResource.class, name = SchemaDependentResource.NAME,
            dependsOn = SecretDependentResource.NAME)
    })
public class MySQLSchemaReconciler
    implements Reconciler, ErrorStatusHandler {

  static final Logger log = LoggerFactory.getLogger(MySQLSchemaReconciler.class);


  @Override
  public UpdateControl reconcile(MySQLSchema schema, Context context) {
    // we only need to update the status if we just built the schema, i.e. when it's present in the
    // context
    Secret secret = context.getSecondaryResource(Secret.class).orElseThrow();

    return context.getSecondaryResource(Schema.class, SchemaDependentResource.NAME).map(s -> {
      updateStatusPojo(schema, s, secret.getMetadata().getName(),
          decode(secret.getData().get(MYSQL_SECRET_USERNAME)));
      log.info("Schema {} created - updating CR status", s.getName());
      return UpdateControl.patchStatus(schema);
    }).orElseGet(UpdateControl::noUpdate);
  }

  @Override
  public ErrorStatusUpdateControl updateErrorStatus(MySQLSchema schema,
      Context context,
      Exception e) {
    SchemaStatus status = new SchemaStatus();
    status.setUrl(null);
    status.setUserName(null);
    status.setSecretName(null);
    status.setStatus("ERROR: " + e.getMessage());
    schema.setStatus(status);
    return ErrorStatusUpdateControl.updateStatus(schema);
  }


  private void updateStatusPojo(MySQLSchema mySQLSchema, Schema schema, String secretName,
      String userName) {
    SchemaStatus status = new SchemaStatus();
    status.setUrl(
        format(
            "jdbc:mysql://%1$s/%2$s",
            System.getenv("MYSQL_HOST"), schema.getName()));
    status.setUserName(userName);
    status.setSecretName(secretName);
    status.setStatus("CREATED");
    mySQLSchema.setStatus(status);
  }
}