org.apache.ignite.tensorflow.cluster

Class TensorFlowClusterGatewayManager

java.lang.Object

org.apache.ignite.tensorflow.cluster.TensorFlowClusterGatewayManager

public class TensorFlowClusterGatewayManager
extends Object

TensorFlow cluster manager that allows to start, maintain and stop TensorFlow cluster using TensorFlowClusterManager and Ignite Service Grid.

Constructor Summary

Constructors

Constructor and Description
TensorFlowClusterGatewayManager(Ignite ignite) Constructs a new instance of TensorFlow cluster manager with maintenance.

Method Summary

Modifier and Type	Method and Description
TensorFlowClusterGateway	createCluster(UUID clusterId, TensorFlowJobArchive jobArchive) Creates and starts a new TensorFlow cluster for the specified cache.
TensorFlowClusterGateway	getCluster(UUID clusterId) Subscribes on changes of the specified cluster.
void	listenToClusterUserScript(UUID clusterId, Consumer<String> out, Consumer<String> err) Listens to TensorFlow cluster user script.
void	stopClusterIfExists(UUID clusterId) Stops TensorFlow cluster.

Methods inherited from class java.lang.Object

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

Constructor Detail

TensorFlowClusterGatewayManager

public TensorFlowClusterGatewayManager(Ignite ignite)

Constructs a new instance of TensorFlow cluster manager with maintenance.

Parameters:

ignite - Ignite instance.

Method Detail

getCluster

public TensorFlowClusterGateway getCluster(UUID clusterId)

Subscribes on changes of the specified cluster.

Parameters:

clusterId - Cluster identifier.

Returns:

TensorFlow cluster gateway that allows to subscribe on cluster changes.

createCluster

public TensorFlowClusterGateway createCluster(UUID clusterId,
                                              TensorFlowJobArchive jobArchive)

Creates and starts a new TensorFlow cluster for the specified cache.

Parameters:

clusterId - Cluster identifier.

jobArchive - Job archive.

Returns:

TensorFlow cluster gateway that allows to subscribe on cluster changes.

listenToClusterUserScript

public void listenToClusterUserScript(UUID clusterId,
                                      Consumer<String> out,
                                      Consumer<String> err)

Listens to TensorFlow cluster user script.

Parameters:

clusterId - Cluster identifier.

out - Output stream consumer.

err - Error stream consumer.

stopClusterIfExists

public void stopClusterIfExists(UUID clusterId)

Stops TensorFlow cluster.

Parameters:

clusterId - Cluster identifier.