io.github.microcks.operator.base.resources.MongoDBDeploymentDependentResource Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of microcks-kube-operator Show documentation
The newest version!
/*
 * Copyright The Microcks Authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.github.microcks.operator.base.resources;

import io.github.microcks.operator.MicrocksOperatorConfig;
import io.github.microcks.operator.api.base.v1alpha1.Microcks;
import io.github.microcks.operator.api.base.v1alpha1.MicrocksSpec;
import io.github.microcks.operator.model.NamedSecondaryResourceProvider;

import io.fabric8.kubernetes.api.model.ObjectMeta;
import io.fabric8.kubernetes.api.model.apps.Deployment;
import io.fabric8.kubernetes.api.model.apps.DeploymentBuilder;
import io.javaoperatorsdk.operator.ReconcilerUtils;
import io.javaoperatorsdk.operator.api.reconciler.Context;
import io.javaoperatorsdk.operator.processing.dependent.kubernetes.CRUDKubernetesDependentResource;
import io.javaoperatorsdk.operator.processing.dependent.kubernetes.KubernetesDependent;
import org.jboss.logging.Logger;

/**
 * A MongoDB Kubernetes Deployment dependent resource.
 * @author laurent
 */
@KubernetesDependent(labelSelector = MicrocksOperatorConfig.RESOURCE_LABEL_SELECTOR)
public class MongoDBDeploymentDependentResource extends CRUDKubernetesDependentResource
      implements NamedSecondaryResourceProvider {

   /** Get a JBoss logging logger. */
   private final Logger logger = Logger.getLogger(getClass());

   private static final String RESOURCE_SUFFIX = "-mongodb";

   /** Default empty constructor. */
   public MongoDBDeploymentDependentResource() {
      super(Deployment.class);
   }

   /**
    * Get the name of Deployment given the primary Microcks resource.
    * @param microcks The primary resource
    * @return The name of Deployment
    */
   public static final String getDeploymentName(Microcks microcks) {
      return microcks.getMetadata().getName() + RESOURCE_SUFFIX;
   }

   @Override
   public String getSecondaryResourceName(Microcks primary) {
      return getDeploymentName(primary);
   }

   @Override
   protected Deployment desired(Microcks microcks, Context context) {
      logger.debugf("Building desired MongoDB Deployment for '%s'", microcks.getMetadata().getName());

      final ObjectMeta microcksMetadata = microcks.getMetadata();
      final String microcksName = microcksMetadata.getName();
      final MicrocksSpec spec = microcks.getSpec();

      Deployment deployment = ReconcilerUtils.loadYaml(Deployment.class, getClass(), "/k8s/mongodb-deployment.yml");
      DeploymentBuilder builder = new DeploymentBuilder(deployment)
            .editMetadata()
               .withName(getDeploymentName(microcks))
               .withNamespace(microcksMetadata.getNamespace())
               .addToLabels("app", microcksName)
               .addToLabels("app.kubernetes.io/name", getDeploymentName(microcks))
               .addToLabels("app.kubernetes.io/version", microcks.getSpec().getVersion())
               .addToLabels("app.kubernetes.io/part-of", microcksName)
               .addToLabels(microcks.getSpec().getCommonLabels())
               .addToAnnotations(microcks.getSpec().getCommonAnnotations())
            .endMetadata()
            .editSpec()
               .editSelector().addToMatchLabels("app", microcksName).endSelector()
               .editTemplate()
                  // make sure label selector matches label (which has to be matched by service selector too)
                  .editMetadata()
                     .addToLabels("app", microcksName)
                     .addToLabels(microcks.getSpec().getCommonLabels())
                     .addToAnnotations(microcks.getSpec().getCommonAnnotations())
                  .endMetadata()
                  .editSpec()
                     .editFirstContainer()
                        .withImage(spec.getMongoDB().getImage().getCoordinates())
                        .withResources(spec.getMongoDB().getResources())
                        .addNewEnv()
                           .withName("MONGODB_USER")
                           .withNewValueFrom()
                              .withNewSecretKeyRef()
                                 .withName(MongoDBSecretDependentResource.getSecretName(microcks))
                                 .withKey(MongoDBSecretDependentResource.MONGODB_USER_KEY)
                              .endSecretKeyRef()
                           .endValueFrom()
                        .endEnv()
                        .addNewEnv()
                           .withName("MONGODB_PASSWORD")
                           .withNewValueFrom()
                              .withNewSecretKeyRef()
                                 .withName(MongoDBSecretDependentResource.getSecretName(microcks))
                                 .withKey(MongoDBSecretDependentResource.MONGODB_PASSWORD_KEY)
                              .endSecretKeyRef()
                           .endValueFrom()
                        .endEnv()
                        .addNewEnv()
                           .withName("MONGO_INITDB_ROOT_USERNAME")
                           .withNewValueFrom()
                              .withNewSecretKeyRef()
                                 .withName(MongoDBSecretDependentResource.getSecretName(microcks))
                                 .withKey(MongoDBSecretDependentResource.MONGODB_ADMIN_USER_KEY)
                              .endSecretKeyRef()
                           .endValueFrom()
                        .endEnv()
                        .addNewEnv()
                           .withName("MONGO_INITDB_ROOT_PASSWORD")
                           .withNewValueFrom()
                              .withNewSecretKeyRef()
                                 .withName(MongoDBSecretDependentResource.getSecretName(microcks))
                                 .withKey(MongoDBSecretDependentResource.MONGODB_ADMIN_PASSWORD_KEY)
                              .endSecretKeyRef()
                           .endValueFrom()
                        .endEnv()
                        .addNewEnv()
                           .withName("MONGO_INITDB_DATABASE")
                           .withValue(microcksName)
                        .endEnv()
                     .endContainer()
                  .endSpec()
               .endTemplate()
            .endSpec();

      // Add the custom-init-script volume.
      builder.editSpec().editTemplate()
            .editSpec()
               .addNewVolume()
                  .withName("custom-init-scripts")
                  .withNewConfigMap()
                     .withName(MongoDBConfigMapDependantResource.getConfigMapName(microcks))
                  .endConfigMap()
               .endVolume()
            .endSpec().endTemplate().endSpec();

      if (spec.getMongoDB().isPersistent()) {
         builder.editSpec().editTemplate()
               .editSpec()
                  .addNewVolume()
                     .withName("mongodb-data")
                     .withNewPersistentVolumeClaim()
                        .withClaimName(MongoDBPVCDependentResource.getPVCName(microcks))
                     .endPersistentVolumeClaim()
                  .endVolume()
               .endSpec().endTemplate().endSpec();
      } else {
         builder.editSpec().editTemplate()
               .editSpec()
                  .addNewVolume()
                     .withName("mongodb-data")
                     .withNewEmptyDir()
                        .withMedium("")
                     .endEmptyDir()
                  .endVolume()
               .endSpec().endTemplate().endSpec();
      }

      // Complete configuration with optional stuffs.
      if (spec.getMongoDB().getSecurityContext() != null) {
         builder.editSpec().editTemplate().editSpec().withSecurityContext(spec.getMongoDB().getSecurityContext()).endSpec().endTemplate().endSpec();
      }

      if (spec.getCommonAffinities() != null) {
         builder.editSpec().editTemplate().editSpec().withAffinity(spec.getCommonAffinities()).endSpec().endTemplate().endSpec();
      }
      if (spec.getCommonTolerations() != null) {
         builder.editSpec().editTemplate().editSpec().withTolerations(spec.getCommonTolerations()).endSpec().endTemplate().endSpec();
      }

      return builder.build();
   }
}