org.apache.beam.runners.fnexecution.control

Class DockerJobBundleFactory

java.lang.Object

org.apache.beam.runners.fnexecution.control.DockerJobBundleFactory

All Implemented Interfaces:

java.lang.AutoCloseable, JobBundleFactory

@ThreadSafe
public class DockerJobBundleFactory
extends java.lang.Object
implements JobBundleFactory

A JobBundleFactory that uses a DockerEnvironmentFactory for environment management. Note that returned stage bundle factories are not thread-safe. Instead, a new stage factory should be created for each client.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static interface	DockerJobBundleFactory.JobBundleFactoryFactory Factory that creates JobBundleFactory for the given JobInfo.

Field Summary

Fields

Modifier and Type	Field and Description
static java.util.concurrent.atomic.AtomicReference<DockerJobBundleFactory.JobBundleFactoryFactory>	FACTORY

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	DockerJobBundleFactory(JobInfo jobInfo)

Method Summary

Modifier and Type	Method and Description
void	close()
static DockerJobBundleFactory	create(JobInfo jobInfo)
StageBundleFactory	forStage(org.apache.beam.runners.core.construction.graph.ExecutableStage executableStage)
protected EnvironmentFactory	getEnvironmentFactory(GrpcFnServer<FnApiControlClientPoolService> controlServiceServer, GrpcFnServer<GrpcLoggingService> loggingServiceServer, GrpcFnServer<ArtifactRetrievalService> retrievalServiceServer, GrpcFnServer<StaticGrpcProvisionService> provisioningServiceServer, ControlClientPool.Source clientSource, IdGenerator idGenerator) Create EnvironmentFactory for the given services.
protected ServerFactory	getServerFactory()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

FACTORY

public static final java.util.concurrent.atomic.AtomicReference<DockerJobBundleFactory.JobBundleFactoryFactory> FACTORY

Constructor Detail

DockerJobBundleFactory

protected DockerJobBundleFactory(JobInfo jobInfo)
                          throws java.lang.Exception

Throws:

java.lang.Exception

Method Detail

create

public static DockerJobBundleFactory create(JobInfo jobInfo)
                                     throws java.lang.Exception

Throws:

java.lang.Exception

forStage

public StageBundleFactory forStage(org.apache.beam.runners.core.construction.graph.ExecutableStage executableStage)

Specified by:

forStage in interface JobBundleFactory

close

public void close()
           throws java.lang.Exception

Specified by:

close in interface java.lang.AutoCloseable

Throws:

java.lang.Exception

getServerFactory

protected ServerFactory getServerFactory()

getEnvironmentFactory

protected EnvironmentFactory getEnvironmentFactory(GrpcFnServer<FnApiControlClientPoolService> controlServiceServer,
                                                   GrpcFnServer<GrpcLoggingService> loggingServiceServer,
                                                   GrpcFnServer<ArtifactRetrievalService> retrievalServiceServer,
                                                   GrpcFnServer<StaticGrpcProvisionService> provisioningServiceServer,
                                                   ControlClientPool.Source clientSource,
                                                   IdGenerator idGenerator)

Create EnvironmentFactory for the given services.