com.netflix.spinnaker.clouddriver.kubernetes.v2.op.artifact.KubernetesCleanupArtifactsOperation.java Source code

Introduction

Here is the source code for com.netflix.spinnaker.clouddriver.kubernetes.v2.op.artifact.KubernetesCleanupArtifactsOperation.java

Source

/*
 * Copyright 2018 Google, Inc.
 *
 * 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 com.netflix.spinnaker.clouddriver.kubernetes.v2.op.artifact;

import com.netflix.spinnaker.clouddriver.data.task.Task;
import com.netflix.spinnaker.clouddriver.data.task.TaskRepository;
import com.netflix.spinnaker.clouddriver.kubernetes.v2.description.KubernetesResourceProperties;
import com.netflix.spinnaker.clouddriver.kubernetes.v2.description.KubernetesResourcePropertyRegistry;
import com.netflix.spinnaker.clouddriver.kubernetes.v2.description.artifact.KubernetesCleanupArtifactsDescription;
import com.netflix.spinnaker.clouddriver.kubernetes.v2.description.manifest.KubernetesKind;
import com.netflix.spinnaker.clouddriver.kubernetes.v2.description.manifest.KubernetesManifest;
import com.netflix.spinnaker.clouddriver.kubernetes.v2.description.manifest.KubernetesManifestAnnotater;
import com.netflix.spinnaker.clouddriver.kubernetes.v2.description.manifest.KubernetesManifestStrategy;
import com.netflix.spinnaker.clouddriver.kubernetes.v2.op.handler.KubernetesHandler;
import com.netflix.spinnaker.clouddriver.kubernetes.v2.security.KubernetesV2Credentials;
import com.netflix.spinnaker.clouddriver.model.ArtifactProvider;
import com.netflix.spinnaker.clouddriver.orchestration.AtomicOperation;
import com.netflix.spinnaker.kork.artifacts.model.Artifact;
import io.kubernetes.client.models.V1DeleteOptions;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors;

@Slf4j
public class KubernetesCleanupArtifactsOperation implements AtomicOperation<Void> {
    private final KubernetesCleanupArtifactsDescription description;
    private final KubernetesV2Credentials credentials;
    private final String accountName;
    private final ArtifactProvider artifactProvider;
    private final KubernetesResourcePropertyRegistry registry;
    private static final String OP_NAME = "CLEANUP_KUBERNETES_ARTIFACTS";

    public KubernetesCleanupArtifactsOperation(KubernetesCleanupArtifactsDescription description,
            ArtifactProvider artifactProvider, KubernetesResourcePropertyRegistry registry) {
        this.description = description;
        this.credentials = (KubernetesV2Credentials) description.getCredentials().getCredentials();
        this.accountName = description.getCredentials().getName();
        this.artifactProvider = artifactProvider;
        this.registry = registry;
    }

    private static Task getTask() {
        return TaskRepository.threadLocalTask.get();
    }

    @Override
    public Void operate(List priorOutputs) {
        List<Artifact> artifacts = description.getManifests().stream().map(this::artifactsToDelete)
                .flatMap(Collection::stream).collect(Collectors.toList());

        artifacts.forEach(a -> {
            String type = a.getType();
            if (!type.startsWith("kubernetes/")) {
                log.warn("Non-kubernetes type deletion requested...");
                return;
            }
            String kind = type.substring("kubernetes/".length());
            KubernetesResourceProperties properties = registry.get(accountName, KubernetesKind.fromString(kind));
            if (properties == null) {
                log.warn("No properties for artifact {}, ignoring", a);
                return;
            }

            getTask().updateStatus(OP_NAME, "Deleting artifact '" + a + '"');
            KubernetesHandler handler = properties.getHandler();
            String name = a.getName();
            if (StringUtils.isNotEmpty(a.getVersion())) {
                name = String.join("-", name, a.getVersion());
            }
            // todo add to outputs
            handler.delete(credentials, a.getLocation(), name, null, new V1DeleteOptions());
        });

        return null;
    }

    private List<Artifact> artifactsToDelete(KubernetesManifest manifest) {
        KubernetesManifestStrategy strategy = KubernetesManifestAnnotater.getStrategy(manifest);
        if (strategy.getMaxVersionHistory() == null) {
            return new ArrayList<>();
        }

        int maxVersionHistory = strategy.getMaxVersionHistory();
        Optional<Artifact> optional = KubernetesManifestAnnotater.getArtifact(manifest);
        if (!optional.isPresent()) {
            return new ArrayList<>();
        }

        Artifact artifact = optional.get();

        List<Artifact> artifacts = artifactProvider
                .getArtifacts(artifact.getType(), artifact.getName(), artifact.getLocation()).stream()
                .filter(a -> a.getMetadata() != null && accountName.equals(a.getMetadata().get("account")))
                .collect(Collectors.toList());

        if (maxVersionHistory >= artifacts.size()) {
            return new ArrayList<>();
        } else {
            return artifacts.subList(0, artifacts.size() - maxVersionHistory);
        }
    }
}