org.apache.flink.kubernetes.kubeclient.parameters.KubernetesTaskManagerParameters Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of flink-kubernetes_2.11 Show documentation
The newest version!
/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.apache.flink.kubernetes.kubeclient.parameters;

import org.apache.flink.configuration.Configuration;
import org.apache.flink.configuration.TaskManagerOptions;
import org.apache.flink.kubernetes.configuration.KubernetesConfigOptions;
import org.apache.flink.kubernetes.utils.KubernetesUtils;
import org.apache.flink.runtime.clusterframework.ContaineredTaskManagerParameters;

import java.util.Collections;
import java.util.Map;

import static org.apache.flink.util.Preconditions.checkArgument;
import static org.apache.flink.util.Preconditions.checkNotNull;

/**
 * A utility class helps parse, verify, and manage the Kubernetes side parameters
 * that are used for constructing the TaskManager Pod.
 */
public class KubernetesTaskManagerParameters extends AbstractKubernetesParameters {

	public static final String TASK_MANAGER_MAIN_CONTAINER_NAME = "flink-task-manager";

	private final String podName;

	private final int taskManagerMemoryMB;

	private final String dynamicProperties;

	private final ContaineredTaskManagerParameters containeredTaskManagerParameters;

	public KubernetesTaskManagerParameters(
			Configuration flinkConfig,
			String podName,
			int taskManagerMemoryMB,
			String dynamicProperties,
			ContaineredTaskManagerParameters containeredTaskManagerParameters) {
		super(flinkConfig);
		this.podName = checkNotNull(podName);
		this.taskManagerMemoryMB = taskManagerMemoryMB;
		this.dynamicProperties = checkNotNull(dynamicProperties);
		this.containeredTaskManagerParameters = checkNotNull(containeredTaskManagerParameters);
	}

	@Override
	public Map getLabels() {
		return KubernetesUtils.getTaskManagerLabels(getClusterId(), flinkConfig);
	}

	@Override
	public Map getEnvironments() {
		return this.containeredTaskManagerParameters.taskManagerEnv();
	}

	@Override
	public Map getAnnotations() {
		return flinkConfig.getOptional(KubernetesConfigOptions.TASK_MANAGER_ANNOTATIONS).orElse(Collections.emptyMap());
	}

	@Override
	public double getCpuRequest() {
		return containeredTaskManagerParameters.getTaskExecutorProcessSpec().getCpuCores().getValue().doubleValue();
	}

	@Override
	public Map getNodeSelectors() {
		return flinkConfig.getOptional(KubernetesConfigOptions.TASK_MANAGER_NODE_SELECTOR).orElse(Collections.emptyMap());
	}

	public String getTaskManagerMainContainerName() {
		return TASK_MANAGER_MAIN_CONTAINER_NAME;
	}

	public String getPodName() {
		return podName;
	}

	public int getTaskManagerMemoryMB() {
		return taskManagerMemoryMB;
	}

	public int getRPCPort() {
		final int taskManagerRpcPort = KubernetesUtils.parsePort(flinkConfig, TaskManagerOptions.RPC_PORT);
		checkArgument(taskManagerRpcPort > 0, "%s should not be 0.", TaskManagerOptions.RPC_PORT.key());
		return taskManagerRpcPort;
	}

	public String getDynamicProperties() {
		return dynamicProperties;
	}

	public ContaineredTaskManagerParameters getContaineredTaskManagerParameters() {
		return containeredTaskManagerParameters;
	}
}