org.apache.flink.runtime.dispatcher.DispatcherServices Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of flink-runtime_2.11 Show documentation
There is a newer version: 1.13.6
/*
 * 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.runtime.dispatcher;

import org.apache.flink.configuration.Configuration;
import org.apache.flink.runtime.blob.BlobServer;
import org.apache.flink.runtime.heartbeat.HeartbeatServices;
import org.apache.flink.runtime.highavailability.HighAvailabilityServices;
import org.apache.flink.runtime.jobmanager.JobGraphWriter;
import org.apache.flink.runtime.metrics.groups.JobManagerMetricGroup;
import org.apache.flink.runtime.resourcemanager.ResourceManagerGateway;
import org.apache.flink.runtime.rpc.FatalErrorHandler;
import org.apache.flink.runtime.webmonitor.retriever.GatewayRetriever;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;

/**
 * {@link Dispatcher} services container.
 */
public class DispatcherServices {

	@Nonnull
	private final Configuration configuration;

	@Nonnull
	private final HighAvailabilityServices highAvailabilityServices;

	@Nonnull
	private final GatewayRetriever resourceManagerGatewayRetriever;

	@Nonnull
	private final BlobServer blobServer;

	@Nonnull
	private final HeartbeatServices heartbeatServices;

	@Nonnull
	private final JobManagerMetricGroup jobManagerMetricGroup;

	@Nonnull
	private final ArchivedExecutionGraphStore archivedExecutionGraphStore;

	@Nonnull
	private final FatalErrorHandler fatalErrorHandler;

	@Nonnull
	private final HistoryServerArchivist historyServerArchivist;

	@Nullable
	private final String metricQueryServiceAddress;

	@Nonnull
	private final JobGraphWriter jobGraphWriter;

	@Nonnull
	private final JobManagerRunnerFactory jobManagerRunnerFactory;

	public DispatcherServices(
			@Nonnull Configuration configuration,
			@Nonnull HighAvailabilityServices highAvailabilityServices,
			@Nonnull GatewayRetriever resourceManagerGatewayRetriever,
			@Nonnull BlobServer blobServer,
			@Nonnull HeartbeatServices heartbeatServices,
			@Nonnull ArchivedExecutionGraphStore archivedExecutionGraphStore,
			@Nonnull FatalErrorHandler fatalErrorHandler,
			@Nonnull HistoryServerArchivist historyServerArchivist,
			@Nullable String metricQueryServiceAddress,
			@Nonnull JobManagerMetricGroup jobManagerMetricGroup,
			@Nonnull JobGraphWriter jobGraphWriter,
			@Nonnull JobManagerRunnerFactory jobManagerRunnerFactory) {
		this.configuration = configuration;
		this.highAvailabilityServices = highAvailabilityServices;
		this.resourceManagerGatewayRetriever = resourceManagerGatewayRetriever;
		this.blobServer = blobServer;
		this.heartbeatServices = heartbeatServices;
		this.archivedExecutionGraphStore = archivedExecutionGraphStore;
		this.fatalErrorHandler = fatalErrorHandler;
		this.historyServerArchivist = historyServerArchivist;
		this.metricQueryServiceAddress = metricQueryServiceAddress;
		this.jobManagerMetricGroup = jobManagerMetricGroup;
		this.jobGraphWriter = jobGraphWriter;
		this.jobManagerRunnerFactory = jobManagerRunnerFactory;
	}

	@Nonnull
	public Configuration getConfiguration() {
		return configuration;
	}

	@Nonnull
	public HighAvailabilityServices getHighAvailabilityServices() {
		return highAvailabilityServices;
	}

	@Nonnull
	public GatewayRetriever getResourceManagerGatewayRetriever() {
		return resourceManagerGatewayRetriever;
	}

	@Nonnull
	public BlobServer getBlobServer() {
		return blobServer;
	}

	@Nonnull
	public HeartbeatServices getHeartbeatServices() {
		return heartbeatServices;
	}

	@Nonnull
	public JobManagerMetricGroup getJobManagerMetricGroup() {
		return jobManagerMetricGroup;
	}

	@Nonnull
	public ArchivedExecutionGraphStore getArchivedExecutionGraphStore() {
		return archivedExecutionGraphStore;
	}

	@Nonnull
	public FatalErrorHandler getFatalErrorHandler() {
		return fatalErrorHandler;
	}

	@Nonnull
	public HistoryServerArchivist getHistoryServerArchivist() {
		return historyServerArchivist;
	}

	@Nullable
	public String getMetricQueryServiceAddress() {
		return metricQueryServiceAddress;
	}

	@Nonnull
	public JobGraphWriter getJobGraphWriter() {
		return jobGraphWriter;
	}

	@Nonnull
	JobManagerRunnerFactory getJobManagerRunnerFactory() {
		return jobManagerRunnerFactory;
	}

	public static DispatcherServices from(
			@Nonnull PartialDispatcherServicesWithJobGraphStore partialDispatcherServicesWithJobGraphStore,
			@Nonnull JobManagerRunnerFactory jobManagerRunnerFactory) {
		return new DispatcherServices(
			partialDispatcherServicesWithJobGraphStore.getConfiguration(),
			partialDispatcherServicesWithJobGraphStore.getHighAvailabilityServices(),
			partialDispatcherServicesWithJobGraphStore.getResourceManagerGatewayRetriever(),
			partialDispatcherServicesWithJobGraphStore.getBlobServer(),
			partialDispatcherServicesWithJobGraphStore.getHeartbeatServices(),
			partialDispatcherServicesWithJobGraphStore.getArchivedExecutionGraphStore(),
			partialDispatcherServicesWithJobGraphStore.getFatalErrorHandler(),
			partialDispatcherServicesWithJobGraphStore.getHistoryServerArchivist(),
			partialDispatcherServicesWithJobGraphStore.getMetricQueryServiceAddress(),
			partialDispatcherServicesWithJobGraphStore.getJobManagerMetricGroupFactory().create(),
			partialDispatcherServicesWithJobGraphStore.getJobGraphWriter(),
			jobManagerRunnerFactory);
	}
}