org.eclipse.jkube.kit.enricher.handler.HandlerHub Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jkube-kit-enricher-api Show documentation
The newest version!
/*
 * Copyright (c) 2019 Red Hat, Inc.
 * This program and the accompanying materials are made
 * available under the terms of the Eclipse Public License 2.0
 * which is available at:
 *
 *     https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *   Red Hat, Inc. - initial API and implementation
 */
package org.eclipse.jkube.kit.enricher.handler;


import java.util.Arrays;
import java.util.List;
import java.util.Properties;

import lombok.Getter;
import org.eclipse.jkube.kit.common.util.LazyBuilder;
import org.eclipse.jkube.kit.config.resource.GroupArtifactVersion;

import io.fabric8.kubernetes.api.model.HasMetadata;
import io.fabric8.kubernetes.api.model.ReplicationController;
import io.fabric8.kubernetes.api.model.apps.DaemonSet;
import io.fabric8.kubernetes.api.model.apps.Deployment;
import io.fabric8.kubernetes.api.model.apps.ReplicaSet;
import io.fabric8.kubernetes.api.model.apps.StatefulSet;
import io.fabric8.kubernetes.api.model.batch.v1.Job;
import io.fabric8.kubernetes.api.model.batch.v1.CronJob;
import io.fabric8.openshift.api.model.DeploymentConfig;

/**
 * @author roland
 * @since 08/04/16
 */
public class HandlerHub {

    private final PodTemplateHandler podTemplateHandler;
    @Getter
    private final List> controllerHandlers;
    private final LazyBuilder.VoidLazyBuilder namespaceHandler;
    private final LazyBuilder.VoidLazyBuilder projectHandler;
    private final LazyBuilder.VoidLazyBuilder serviceHandler;

    public HandlerHub(GroupArtifactVersion groupArtifactVersion, Properties configuration) {
        ProbeHandler probeHandler = new ProbeHandler();
        ContainerHandler containerHandler = new ContainerHandler(configuration, groupArtifactVersion, probeHandler);
        podTemplateHandler = new PodTemplateHandler(containerHandler);
        controllerHandlers = Arrays.asList(
            new ControllerHandlerLazyBuilder<>(Deployment.class, () -> new DeploymentHandler(podTemplateHandler)),
            new ControllerHandlerLazyBuilder<>(DeploymentConfig.class, () ->
                new DeploymentConfigHandler(podTemplateHandler)),
            new ControllerHandlerLazyBuilder<>(ReplicaSet.class,() -> new ReplicaSetHandler(podTemplateHandler)),
            new ControllerHandlerLazyBuilder<>(ReplicationController.class,() ->
                new ReplicationControllerHandler(podTemplateHandler)),
            new ControllerHandlerLazyBuilder<>(StatefulSet.class,() -> new StatefulSetHandler(podTemplateHandler)),
            new ControllerHandlerLazyBuilder<>(DaemonSet.class,() -> new DaemonSetHandler(podTemplateHandler)),
            new ControllerHandlerLazyBuilder<>(Job.class,() -> new JobHandler(podTemplateHandler)),
            new ControllerHandlerLazyBuilder<>(CronJob.class, () -> new CronJobHandler(podTemplateHandler))
        );
        namespaceHandler = new LazyBuilder.VoidLazyBuilder<>(NamespaceHandler::new);
        projectHandler = new LazyBuilder.VoidLazyBuilder<>(ProjectHandler::new);
        serviceHandler = new LazyBuilder.VoidLazyBuilder<>(ServiceHandler::new);
    }

    public NamespaceHandler getNamespaceHandler() { return namespaceHandler.get(); }

    public ProjectHandler getProjectHandler() { return projectHandler.get(); }

    public ServiceHandler getServiceHandler() {
        return serviceHandler.get();
    }


    @SuppressWarnings("unchecked")
    public  ControllerHandler getHandlerFor(T item) {
        if (item == null) {
            return null;
        }
        return (ControllerHandler) getHandlerFor(item.getClass());
    }

    @SuppressWarnings("unchecked")
    public  ControllerHandler getHandlerFor(Class controllerType) {
        return (ControllerHandler) controllerHandlers.stream()
            .filter(handler -> handler.getControllerHandlerType().isAssignableFrom(controllerType))
            .findAny().map(LazyBuilder.VoidLazyBuilder::get).orElse(null);
    }
}