Skip to content

Commit

Permalink
chore(citrus-kubernetes): Enhance Kubernetes support
Browse files Browse the repository at this point in the history
- Add several test actions to manage Kubernetes resources
- Add XML DSL support
- Add YAML DSL support
  • Loading branch information
christophd committed Oct 7, 2024
1 parent 397827c commit bbedb8b
Show file tree
Hide file tree
Showing 145 changed files with 11,817 additions and 2 deletions.
12 changes: 12 additions & 0 deletions connectors/citrus-kubernetes/pom.xml
Original file line number Diff line number Diff line change
Expand Up @@ -142,6 +142,18 @@
<version>${project.version}</version>
<scope>test</scope>
</dependency>
<dependency>
<groupId>org.citrusframework</groupId>
<artifactId>citrus-xml</artifactId>
<version>${project.version}</version>
<scope>test</scope>
</dependency>
<dependency>
<groupId>org.citrusframework</groupId>
<artifactId>citrus-yaml</artifactId>
<version>${project.version}</version>
<scope>test</scope>
</dependency>
<dependency>
<groupId>io.fabric8</groupId>
<artifactId>kubernetes-server-mock</artifactId>
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,75 @@
/*
* Copyright the original author or 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 org.citrusframework.kubernetes;

import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;

import io.fabric8.kubernetes.client.KubernetesClient;
import io.fabric8.kubernetes.client.KubernetesClientBuilder;
import org.citrusframework.TestActor;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
* Test actor disabled when running a local test where no Kubernetes and Openshift is involved.
*/
public class KubernetesActor extends TestActor {

/** Logger */
private static final Logger logger = LoggerFactory.getLogger(KubernetesActor.class);

/** Kubernetes' connection state, checks connectivity to Kubernetes cluster */
private static AtomicBoolean connected;

private final KubernetesClient kubernetesClient;

public KubernetesActor(KubernetesClient kubernetesClient) {
setName("k8s");

if (kubernetesClient != null) {
this.kubernetesClient = kubernetesClient;
} else {
this.kubernetesClient = new KubernetesClientBuilder().build();
}
}

@Override
public boolean isDisabled() {
synchronized (logger) {
if (connected == null) {
if (KubernetesSettings.isEnabled()) {
try {
Future<Boolean> future = Executors.newSingleThreadExecutor().submit(() -> {
kubernetesClient.pods().list();
return true;
});

connected = new AtomicBoolean((future.get(KubernetesSettings.getConnectTimeout(), TimeUnit.MILLISECONDS)));
} catch (Exception e) {
logger.warn("Skipping Kubernetes action as no proper Kubernetes environment is available on host system!", e);
connected = new AtomicBoolean(false);
}
}
}

return !connected.get();
}
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,231 @@
/*
* Copyright the original author or 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 org.citrusframework.kubernetes;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.util.Map;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.util.StringUtils;

public class KubernetesSettings {

/** Logger */
private static final Logger logger = LoggerFactory.getLogger(KubernetesSettings.class);

private static final String KUBERNETES_PROPERTY_PREFIX = "citrus.kubernetes.";
private static final String KUBERNETES_ENV_PREFIX = "CITRUS_KUBERNETES_";

private static final String AUTO_REMOVE_RESOURCES_PROPERTY = KUBERNETES_PROPERTY_PREFIX + "auto.remove.resources";
private static final String AUTO_REMOVE_RESOURCES_ENV = KUBERNETES_ENV_PREFIX + "AUTO_REMOVE_RESOURCES";
private static final String AUTO_REMOVE_RESOURCES_DEFAULT = "true";

private static final String ENABLED_PROPERTY = KUBERNETES_PROPERTY_PREFIX + "enabled";
private static final String ENABLED_ENV = KUBERNETES_ENV_PREFIX + "ENABLED";
private static final String ENABLED_DEFAULT = "true";

private static final String SERVICE_TIMEOUT_PROPERTY = KUBERNETES_PROPERTY_PREFIX + "service.timeout";
private static final String SERVICE_TIMEOUT_ENV = KUBERNETES_ENV_PREFIX + "SERVICE_TIMEOUT";
private static final String SERVICE_TIMEOUT_DEFAULT = "2000";

private static final String CONNECT_TIMEOUT_PROPERTY = KUBERNETES_PROPERTY_PREFIX + "connect.timeout";
private static final String CONNECT_TIMEOUT_ENV = KUBERNETES_ENV_PREFIX + "CONNECT_TIMEOUT";
private static final String CONNECT_TIMEOUT_DEFAULT = "5000";

private static final String NAMESPACE_PROPERTY = KUBERNETES_PROPERTY_PREFIX + "namespace";
private static final String NAMESPACE_ENV = KUBERNETES_ENV_PREFIX + "NAMESPACE";
private static final String NAMESPACE_DEFAULT = "default";

private static final String API_VERSION_PROPERTY = KUBERNETES_PROPERTY_PREFIX + "api.version";
private static final String API_VERSION_ENV = KUBERNETES_ENV_PREFIX + "API_VERSION";
private static final String API_VERSION_DEFAULT = "v1";

private static final String SERVICE_NAME_PROPERTY = KUBERNETES_PROPERTY_PREFIX + "service.name";
private static final String SERVICE_NAME_ENV = KUBERNETES_ENV_PREFIX + "SERVICE_NAME";
private static final String SERVICE_NAME_DEFAULT = "citrus-k8s-service";

private static final String SERVICE_PORT_PROPERTY = KUBERNETES_PROPERTY_PREFIX + "service.port";
private static final String SERVICE_PORT_ENV = KUBERNETES_ENV_PREFIX + "SERVICE_PORT";
private static final String SERVICE_PORT_DEFAULT = "8080";

private static final String DEFAULT_LABELS_PROPERTY = KUBERNETES_PROPERTY_PREFIX + "default.labels";
private static final String DEFAULT_LABELS_ENV = KUBERNETES_ENV_PREFIX + "DEFAULT_LABELS";
private static final String DEFAULT_LABELS_DEFAULT = "app=citrus";

private static final String MAX_ATTEMPTS_PROPERTY = KUBERNETES_PROPERTY_PREFIX + "max.attempts";
private static final String MAX_ATTEMPTS_ENV = KUBERNETES_ENV_PREFIX + "MAX_ATTEMPTS";
private static final String MAX_ATTEMPTS_DEFAULT = "150";

private static final String DELAY_BETWEEN_ATTEMPTS_PROPERTY = KUBERNETES_PROPERTY_PREFIX + "delay.between.attempts";
private static final String DELAY_BETWEEN_ATTEMPTS_ENV = KUBERNETES_ENV_PREFIX + "DELAY_BETWEEN_ATTEMPTS";
private static final String DELAY_BETWEEN_ATTEMPTS_DEFAULT = "2000";

private static final String PRINT_POD_LOGS_PROPERTY = KUBERNETES_PROPERTY_PREFIX + "print.pod.logs";
private static final String PRINT_POD_LOGS_ENV = KUBERNETES_ENV_PREFIX + "PRINT_POD_LOGS";
private static final String PRINT_POD_LOGS_DEFAULT = "true";

private static final String WATCH_LOGS_TIMEOUT_PROPERTY = KUBERNETES_PROPERTY_PREFIX + "watch.logs.timeout";
private static final String WATCH_LOGS_TIMEOUT_ENV = KUBERNETES_ENV_PREFIX + "WATCH_LOGS_TIMEOUT";
private static final String WATCH_LOGS_TIMEOUT_DEFAULT = "60000";

private KubernetesSettings() {
// prevent instantiation of utility class
}

/**
* Request timeout when receiving cloud events.
* @return
*/
public static long getServiceTimeout() {
return Long.parseLong(System.getProperty(SERVICE_TIMEOUT_PROPERTY,
System.getenv(SERVICE_TIMEOUT_ENV) != null ? System.getenv(SERVICE_TIMEOUT_ENV) : SERVICE_TIMEOUT_DEFAULT));
}

/**
* Timeout when connecting to Kubernetes cluster.
* @return
*/
public static long getConnectTimeout() {
return Long.parseLong(System.getProperty(CONNECT_TIMEOUT_PROPERTY,
System.getenv(CONNECT_TIMEOUT_ENV) != null ? System.getenv(CONNECT_TIMEOUT_ENV) : CONNECT_TIMEOUT_DEFAULT));
}

/**
* Namespace to work on when performing Kubernetes client operations such as creating triggers, services and so on.
* @return
*/
public static String getNamespace() {
String systemNamespace = System.getProperty(NAMESPACE_PROPERTY, System.getenv(NAMESPACE_ENV));

if (systemNamespace != null) {
return systemNamespace;
}

final File namespace = new File("/var/run/secrets/kubernetes.io/serviceaccount/namespace");
if (namespace.exists()){
try {
return Files.readString(namespace.toPath());
} catch (IOException e) {
logger.warn("Failed to read Kubernetes namespace from filesystem {}", namespace, e);
}
}

return NAMESPACE_DEFAULT;
}

/**
* Api version for current Kubernetes installation.
* @return
*/
public static String getApiVersion() {
return System.getProperty(API_VERSION_PROPERTY,
System.getenv(API_VERSION_ENV) != null ? System.getenv(API_VERSION_ENV) : API_VERSION_DEFAULT);
}

/**
* Service name to use when creating a new service for cloud event subscriptions.
* @return
*/
public static String getServiceName() {
return System.getProperty(SERVICE_NAME_PROPERTY,
System.getenv(SERVICE_NAME_ENV) != null ? System.getenv(SERVICE_NAME_ENV) : SERVICE_NAME_DEFAULT);
}

/**
* Service port used when consuming cloud events via Http.
* @return
*/
public static String getServicePort() {
return System.getProperty(SERVICE_PORT_PROPERTY,
System.getenv(SERVICE_PORT_ENV) != null ? System.getenv(SERVICE_PORT_ENV) : SERVICE_PORT_DEFAULT);
}

/**
* Read labels for Kubernetes resources created by the test. The environment setting should be a
* comma delimited list of key-value pairs.
* @return
*/
public static Map<String, String> getDefaultLabels() {
String labelsConfig = System.getProperty(DEFAULT_LABELS_PROPERTY,
System.getenv(DEFAULT_LABELS_ENV) != null ? System.getenv(DEFAULT_LABELS_ENV) : DEFAULT_LABELS_DEFAULT);

return Stream.of(StringUtils.commaDelimitedListToStringArray(labelsConfig))
.map(item -> StringUtils.delimitedListToStringArray(item, "="))
.filter(keyValue -> keyValue.length == 2)
.collect(Collectors.toMap(item -> item[0], item -> item[1]));
}

/**
* Kubernetes may be disabled by default.
* @return
*/
public static boolean isEnabled() {
return Boolean.parseBoolean(System.getProperty(ENABLED_PROPERTY,
System.getenv(ENABLED_ENV) != null ? System.getenv(ENABLED_ENV) : ENABLED_DEFAULT));
}

/**
* When set to true Kubernetes resources (e.g. services) created during the test are
* automatically removed after the test.
* @return
*/
public static boolean isAutoRemoveResources() {
return Boolean.parseBoolean(System.getProperty(AUTO_REMOVE_RESOURCES_PROPERTY,
System.getenv(AUTO_REMOVE_RESOURCES_ENV) != null ? System.getenv(AUTO_REMOVE_RESOURCES_ENV) : AUTO_REMOVE_RESOURCES_DEFAULT));
}

/**
* When set to true test will print pod logs e.g. while waiting for a pod log message.
* @return
*/
public static boolean isPrintPodLogs() {
return Boolean.parseBoolean(System.getProperty(PRINT_POD_LOGS_PROPERTY,
System.getenv(PRINT_POD_LOGS_ENV) != null ? System.getenv(PRINT_POD_LOGS_ENV) : PRINT_POD_LOGS_DEFAULT));
}

/**
* Maximum number of attempts when polling for running state and log messages.
* @return
*/
public static int getMaxAttempts() {
return Integer.parseInt(System.getProperty(MAX_ATTEMPTS_PROPERTY,
System.getenv(MAX_ATTEMPTS_ENV) != null ? System.getenv(MAX_ATTEMPTS_ENV) : MAX_ATTEMPTS_DEFAULT));
}

/**
* Delay in milliseconds to wait after polling attempt.
* @return
*/
public static long getDelayBetweenAttempts() {
return Long.parseLong(System.getProperty(DELAY_BETWEEN_ATTEMPTS_PROPERTY,
System.getenv(DELAY_BETWEEN_ATTEMPTS_ENV) != null ? System.getenv(DELAY_BETWEEN_ATTEMPTS_ENV) : DELAY_BETWEEN_ATTEMPTS_DEFAULT));
}

/**
* Duration in milliseconds to watch pod logs.
* @return
*/
public static long getWatchLogsTimeout() {
return Long.parseLong(System.getProperty(WATCH_LOGS_TIMEOUT_PROPERTY,
System.getenv(WATCH_LOGS_TIMEOUT_ENV) != null ? System.getenv(WATCH_LOGS_TIMEOUT_ENV) : WATCH_LOGS_TIMEOUT_DEFAULT));
}
}
Loading

0 comments on commit bbedb8b

Please sign in to comment.