edu.iu.dsc.tws.comms.tcp

Class TWSTCPChannel

java.lang.Object

edu.iu.dsc.tws.comms.tcp.TWSTCPChannel

All Implemented Interfaces:

TWSChannel

public class TWSTCPChannel
extends java.lang.Object
implements TWSChannel

Constructor Summary

Constructors

Constructor and Description
TWSTCPChannel(Config config, IWorkerController wController) Create the TCP channel

Method Summary

Modifier and Type	Method and Description
void	close() Close the channel
java.nio.ByteBuffer	createBuffer(int capacity) Create a buffer
boolean	isComplete() Check weather we have any pending sends
void	progress() Progress both sends and receives
void	progressReceives(int group) Progress only receives belonging to a group
void	progressSends() Progress only the sends
boolean	receiveMessage(int group, int rank, int stream, ChannelListener callback, java.util.Queue<DataBuffer> receiveBuffers) Register our interest to receive messages from particular rank using a stream
void	reInit(java.util.List<JobMasterAPI.WorkerInfo> restartedWorkers) re-initialize the channel with restarted workers
void	releaseBuffers(int workerId, int e) Close a worker id with edge
boolean	sendMessage(int id, ChannelMessage message, ChannelListener callback) Send messages to the particular id

Methods inherited from class java.lang.Object

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

Constructor Detail

TWSTCPChannel

public TWSTCPChannel(Config config,
                     IWorkerController wController)

Create the TCP channel

Parameters:

config - configuration

wController - controller

Method Detail

reInit

public void reInit(java.util.List<JobMasterAPI.WorkerInfo> restartedWorkers)

Description copied from interface: TWSChannel

re-initialize the channel with restarted workers

Specified by:

reInit in interface TWSChannel

sendMessage

public boolean sendMessage(int id,
                           ChannelMessage message,
                           ChannelListener callback)

Send messages to the particular id

Specified by:

sendMessage in interface TWSChannel

Parameters:

id - id to be used for sending messages

message - the message

callback - callback for message completions

Returns:

true if the message is accepted to be sent

receiveMessage

public boolean receiveMessage(int group,
                              int rank,
                              int stream,
                              ChannelListener callback,
                              java.util.Queue<DataBuffer> receiveBuffers)

Register our interest to receive messages from particular rank using a stream

Specified by:

receiveMessage in interface TWSChannel

rank - worker id

stream - the graph edge to receive from

callback - callback for message completions

receiveBuffers - the list of receive buffers

Returns:

true if sending is accepted

close

public void close()

Description copied from interface: TWSChannel

Close the channel

Specified by:

close in interface TWSChannel

isComplete

public boolean isComplete()

Description copied from interface: TWSChannel

Check weather we have any pending sends

Specified by:

isComplete in interface TWSChannel

progress

public void progress()

Description copied from interface: TWSChannel

Progress both sends and receives

Specified by:

progress in interface TWSChannel

progressSends

public void progressSends()

Description copied from interface: TWSChannel

Progress only the sends

Specified by:

progressSends in interface TWSChannel

progressReceives

public void progressReceives(int group)

Description copied from interface: TWSChannel

Progress only receives belonging to a group

Specified by:

progressReceives in interface TWSChannel

createBuffer

public java.nio.ByteBuffer createBuffer(int capacity)

Description copied from interface: TWSChannel

Create a buffer

Specified by:

createBuffer in interface TWSChannel

Parameters:

capacity - capacity

Returns:

the byte buffer

releaseBuffers

public void releaseBuffers(int workerId,
                           int e)

Close a worker id with edge

Specified by:

releaseBuffers in interface TWSChannel

Parameters:

workerId - worker id

e - edge