public void TestReduceOperator()
{
string groupName = "group1";
string operatorName = "reduce";
int numTasks = 4;
string driverId = "driverid";
string masterTaskId = "task0";
int fanOut = 2;
IGroupCommDriver groupCommDriver = GetInstanceOfGroupCommDriver(driverId, masterTaskId, groupName, fanOut, numTasks);
var commGroup = groupCommDriver.DefaultGroup
.AddReduce <int>(operatorName, "task0", TopologyTypes.Flat, GetDefaultDataConverterConfig(), GetDefaultReduceFuncConfig())
.Build();
List <ICommunicationGroupClient> commGroups = CommGroupClients(groupName, numTasks, groupCommDriver, commGroup, GetDefaultCodecConfig());
IReduceReceiver <int> receiver = commGroups[0].GetReduceReceiver <int>(operatorName);
IReduceSender <int> sender1 = commGroups[1].GetReduceSender <int>(operatorName);
IReduceSender <int> sender2 = commGroups[2].GetReduceSender <int>(operatorName);
IReduceSender <int> sender3 = commGroups[3].GetReduceSender <int>(operatorName);
Assert.NotNull(receiver);
Assert.NotNull(sender1);
Assert.NotNull(sender2);
Assert.NotNull(sender3);
sender3.Send(5);
sender1.Send(1);
sender2.Send(3);
Assert.Equal(9, receiver.Reduce());
}
private static void ScatterReceiveReduce(IScatterReceiver <int> receiver, IReduceSender <int> sumSender)
{
List <int> data1 = receiver.Receive();
int sum1 = data1.Sum();
sumSender.Send(sum1);
}
public SlaveTask(IGroupCommClient groupCommClient)
{
_logger.Log(Level.Info, "Hello from slave task");
_groupCommClient = groupCommClient;
_commGroup = _groupCommClient.GetCommunicationGroup(GroupTestConstants.GroupName);
_scatterReceiver = _commGroup.GetScatterReceiver <int>(GroupTestConstants.ScatterOperatorName);
_sumSender = _commGroup.GetReduceSender <int>(GroupTestConstants.ReduceOperatorName);
}
public SlaveTask(IGroupCommClient groupCommClient)
{
Logger.Log(Level.Info, "Hello from slave task");
_groupCommClient = groupCommClient;
_commGroup = _groupCommClient.GetCommunicationGroup(GroupTestConstants.GroupName);
_scatterReceiver = _commGroup.GetScatterReceiver<int>(GroupTestConstants.ScatterOperatorName);
_sumSender = _commGroup.GetReduceSender<int>(GroupTestConstants.ReduceOperatorName);
}
public void TestBroadcastReduceOperators()
{
string groupName = "group1";
string broadcastOperatorName = "broadcast";
string reduceOperatorName = "reduce";
string masterTaskId = "task0";
string driverId = "Driver Id";
int numTasks = 3;
int fanOut = 2;
var groupCommunicationDriver = GetInstanceOfGroupCommDriver(driverId, masterTaskId, groupName, fanOut, numTasks);
ICommunicationGroupDriver commGroup = groupCommunicationDriver.DefaultGroup
.AddBroadcast <int>(
broadcastOperatorName,
masterTaskId,
TopologyTypes.Flat,
GetDefaultDataConverterConfig())
.AddReduce <int>(
reduceOperatorName,
masterTaskId,
TopologyTypes.Flat,
GetDefaultDataConverterConfig(),
GetDefaultReduceFuncConfig())
.Build();
var commGroups = CommGroupClients(groupName, numTasks, groupCommunicationDriver, commGroup, GetDefaultCodecConfig());
// for master task
IBroadcastSender <int> broadcastSender = commGroups[0].GetBroadcastSender <int>(broadcastOperatorName);
IReduceReceiver <int> sumReducer = commGroups[0].GetReduceReceiver <int>(reduceOperatorName);
IBroadcastReceiver <int> broadcastReceiver1 = commGroups[1].GetBroadcastReceiver <int>(broadcastOperatorName);
IReduceSender <int> triangleNumberSender1 = commGroups[1].GetReduceSender <int>(reduceOperatorName);
IBroadcastReceiver <int> broadcastReceiver2 = commGroups[2].GetBroadcastReceiver <int>(broadcastOperatorName);
IReduceSender <int> triangleNumberSender2 = commGroups[2].GetReduceSender <int>(reduceOperatorName);
for (int j = 1; j <= 10; j++)
{
broadcastSender.Send(j);
int n1 = broadcastReceiver1.Receive();
int n2 = broadcastReceiver2.Receive();
Assert.Equal(j, n1);
Assert.Equal(j, n2);
int triangleNum1 = TriangleNumber(n1);
triangleNumberSender1.Send(triangleNum1);
int triangleNum2 = TriangleNumber(n2);
triangleNumberSender2.Send(triangleNum2);
int sum = sumReducer.Reduce();
int expected = TriangleNumber(j) * (numTasks - 1);
Assert.Equal(sum, expected);
}
}
public SlaveTask(
[Parameter(typeof(GroupTestConfig.NumIterations))] int numIters,
IGroupCommClient groupCommClient)
{
Logger.Log(Level.Info, "Hello from slave task");
_numIterations = numIters;
_groupCommClient = groupCommClient;
_commGroup = _groupCommClient.GetCommunicationGroup(GroupTestConstants.GroupName);
_broadcastReceiver = _commGroup.GetBroadcastReceiver <int>(GroupTestConstants.BroadcastOperatorName);
_triangleNumberSender = _commGroup.GetReduceSender <int>(GroupTestConstants.ReduceOperatorName);
}
public SlaveTask(
[Parameter(typeof(GroupTestConfig.NumIterations))] int numIters,
IGroupCommClient groupCommClient)
{
_logger.Log(Level.Info, "Hello from slave task");
_numIterations = numIters;
_groupCommClient = groupCommClient;
_commGroup = _groupCommClient.GetCommunicationGroup(GroupTestConstants.GroupName);
_broadcastReceiver = _commGroup.GetBroadcastReceiver<int>(GroupTestConstants.BroadcastOperatorName);
_triangleNumberSender = _commGroup.GetReduceSender<int>(GroupTestConstants.ReduceOperatorName);
}
private MapTaskHost(
IMapFunction <TMapInput, TMapOutput> mapTask,
IGroupCommClient groupCommunicationsClient,
[Parameter(typeof(InvokeGC))] bool invokeGC)
{
_mapTask = mapTask;
var cg = groupCommunicationsClient.GetCommunicationGroup(IMRUConstants.CommunicationGroupName);
_dataAndMessageReceiver =
cg.GetBroadcastReceiver <MapInputWithControlMessage <TMapInput> >(IMRUConstants.BroadcastOperatorName);
_dataReducer = cg.GetReduceSender <TMapOutput>(IMRUConstants.ReduceOperatorName);
_invokeGC = invokeGC;
}
public void TestReduceOperator()
{
string groupName = "group1";
string operatorName = "reduce";
int numTasks = 10;
string driverId = "driverId";
string masterTaskId = "task0";
int fanOut = 3;
var groupCommDriver = GroupCommunicationTests.GetInstanceOfGroupCommDriver(driverId, masterTaskId, groupName, fanOut, numTasks);
ICommunicationGroupDriver commGroup = groupCommDriver.DefaultGroup
.AddReduce <int>(operatorName, masterTaskId, TopologyTypes.Tree, GetDefaultDataConverterConfig(), GetDefaultReduceFuncConfig())
.Build();
var commGroups = GroupCommunicationTests.CommGroupClients(groupName, numTasks, groupCommDriver, commGroup, GetDefaultCodecConfig());
IReduceReceiver <int> receiver = commGroups[0].GetReduceReceiver <int>(operatorName);
IReduceSender <int> sender1 = commGroups[1].GetReduceSender <int>(operatorName);
IReduceSender <int> sender2 = commGroups[2].GetReduceSender <int>(operatorName);
IReduceSender <int> sender3 = commGroups[3].GetReduceSender <int>(operatorName);
IReduceSender <int> sender4 = commGroups[4].GetReduceSender <int>(operatorName);
IReduceSender <int> sender5 = commGroups[5].GetReduceSender <int>(operatorName);
IReduceSender <int> sender6 = commGroups[6].GetReduceSender <int>(operatorName);
IReduceSender <int> sender7 = commGroups[7].GetReduceSender <int>(operatorName);
IReduceSender <int> sender8 = commGroups[8].GetReduceSender <int>(operatorName);
IReduceSender <int> sender9 = commGroups[9].GetReduceSender <int>(operatorName);
Assert.NotNull(receiver);
Assert.NotNull(sender1);
Assert.NotNull(sender2);
Assert.NotNull(sender3);
Assert.NotNull(sender4);
Assert.NotNull(sender5);
Assert.NotNull(sender6);
Assert.NotNull(sender7);
Assert.NotNull(sender8);
Assert.NotNull(sender9);
sender9.Send(9);
sender8.Send(8);
sender7.Send(7);
sender6.Send(6);
sender5.Send(5);
sender4.Send(4);
sender3.Send(3);
sender2.Send(2);
sender1.Send(1);
Assert.Equal(45, receiver.Reduce());
}
private MapTaskHost(
IMapFunction <TMapInput, TMapOutput> mapTask,
IGroupCommClient groupCommunicationsClient,
TaskCloseCoordinator taskCloseCoordinator,
[Parameter(typeof(InvokeGC))] bool invokeGc,
[Parameter(typeof(TaskConfigurationOptions.Identifier))] string taskId) :
base(groupCommunicationsClient, taskCloseCoordinator, invokeGc)
{
Logger.Log(Level.Info, "Entering constructor of MapTaskHost for task id {0}", taskId);
_mapTask = mapTask;
_dataAndMessageReceiver =
_communicationGroupClient.GetBroadcastReceiver <MapInputWithControlMessage <TMapInput> >(IMRUConstants.BroadcastOperatorName);
_dataReducer = _communicationGroupClient.GetReduceSender <TMapOutput>(IMRUConstants.ReduceOperatorName);
Logger.Log(Level.Info, "MapTaskHost initialized.");
}
public KMeansSlaveTask(
DataPartitionCache dataPartition,
[Parameter(typeof(KMeansConfiguratioinOptions.TotalNumEvaluators))] int clustersNumber,
IGroupCommClient groupCommClient)
{
using (Logger.LogFunction("KMeansSlaveTask::KMeansSlaveTask"))
{
_dataPartition = dataPartition;
_groupCommClient = groupCommClient;
_clustersNum = clustersNumber;
_commGroup = _groupCommClient.GetCommunicationGroup(Constants.KMeansCommunicationGroupName);
_dataBroadcastReceiver = _commGroup.GetBroadcastReceiver <Centroids>(Constants.CentroidsBroadcastOperatorName);
_partialMeansSender = _commGroup.GetReduceSender <ProcessedResults>(Constants.MeansReduceOperatorName);
_controlBroadcastReceiver = _commGroup.GetBroadcastReceiver <ControlMessage>(Constants.ControlMessageBroadcastOperatorName);
}
}
public KMeansSlaveTask(
DataPartitionCache dataPartition,
[Parameter(typeof(KMeansConfiguratioinOptions.TotalNumEvaluators))] int clustersNumber,
IGroupCommClient groupCommClient)
{
using (Logger.LogFunction("KMeansSlaveTask::KMeansSlaveTask"))
{
_dataPartition = dataPartition;
_groupCommClient = groupCommClient;
_clustersNum = clustersNumber;
_commGroup = _groupCommClient.GetCommunicationGroup(Constants.KMeansCommunicationGroupName);
_dataBroadcastReceiver = _commGroup.GetBroadcastReceiver<Centroids>(Constants.CentroidsBroadcastOperatorName);
_partialMeansSender = _commGroup.GetReduceSender<ProcessedResults>(Constants.MeansReduceOperatorName);
_controlBroadcastReceiver = _commGroup.GetBroadcastReceiver<ControlMessage>(Constants.ControlMessageBroadcastOperatorName);
}
}
private MapTaskHost(
IMapFunction <TMapInput, TMapOutput> mapTask,
IGroupCommClient groupCommunicationsClient,
TaskCloseCoordinator taskCloseCoordinator,
[Parameter(typeof(InvokeGC))] bool invokeGC)
{
_mapTask = mapTask;
_groupCommunicationsClient = groupCommunicationsClient;
var cg = groupCommunicationsClient.GetCommunicationGroup(IMRUConstants.CommunicationGroupName);
_dataAndMessageReceiver =
cg.GetBroadcastReceiver <MapInputWithControlMessage <TMapInput> >(IMRUConstants.BroadcastOperatorName);
_dataReducer = cg.GetReduceSender <TMapOutput>(IMRUConstants.ReduceOperatorName);
_invokeGC = invokeGC;
_taskCloseCoordinator = taskCloseCoordinator;
_cancellationSource = new CancellationTokenSource();
}
public void TestBroadcastReduceOperators()
{
string groupName = "group1";
string broadcastOperatorName = "broadcast";
string reduceOperatorName = "reduce";
string masterTaskId = "task0";
string driverId = "Driver Id";
int numTasks = 10;
int fanOut = 3;
var groupCommDriver = GroupCommunicationTests.GetInstanceOfGroupCommDriver(driverId, masterTaskId, groupName, fanOut, numTasks);
ICommunicationGroupDriver commGroup = groupCommDriver.DefaultGroup
.AddBroadcast <int>(
broadcastOperatorName,
masterTaskId,
TopologyTypes.Tree,
GetDefaultDataConverterConfig())
.AddReduce <int>(
reduceOperatorName,
masterTaskId,
TopologyTypes.Tree,
GetDefaultDataConverterConfig(),
GetDefaultReduceFuncConfig())
.Build();
var commGroups = GroupCommunicationTests.CommGroupClients(groupName, numTasks, groupCommDriver, commGroup, GetDefaultCodecConfig());
// for master task
IBroadcastSender <int> broadcastSender = commGroups[0].GetBroadcastSender <int>(broadcastOperatorName);
IReduceReceiver <int> sumReducer = commGroups[0].GetReduceReceiver <int>(reduceOperatorName);
IBroadcastReceiver <int> broadcastReceiver1 = commGroups[1].GetBroadcastReceiver <int>(broadcastOperatorName);
IReduceSender <int> triangleNumberSender1 = commGroups[1].GetReduceSender <int>(reduceOperatorName);
IBroadcastReceiver <int> broadcastReceiver2 = commGroups[2].GetBroadcastReceiver <int>(broadcastOperatorName);
IReduceSender <int> triangleNumberSender2 = commGroups[2].GetReduceSender <int>(reduceOperatorName);
IBroadcastReceiver <int> broadcastReceiver3 = commGroups[3].GetBroadcastReceiver <int>(broadcastOperatorName);
IReduceSender <int> triangleNumberSender3 = commGroups[3].GetReduceSender <int>(reduceOperatorName);
IBroadcastReceiver <int> broadcastReceiver4 = commGroups[4].GetBroadcastReceiver <int>(broadcastOperatorName);
IReduceSender <int> triangleNumberSender4 = commGroups[4].GetReduceSender <int>(reduceOperatorName);
IBroadcastReceiver <int> broadcastReceiver5 = commGroups[5].GetBroadcastReceiver <int>(broadcastOperatorName);
IReduceSender <int> triangleNumberSender5 = commGroups[5].GetReduceSender <int>(reduceOperatorName);
IBroadcastReceiver <int> broadcastReceiver6 = commGroups[6].GetBroadcastReceiver <int>(broadcastOperatorName);
IReduceSender <int> triangleNumberSender6 = commGroups[6].GetReduceSender <int>(reduceOperatorName);
IBroadcastReceiver <int> broadcastReceiver7 = commGroups[7].GetBroadcastReceiver <int>(broadcastOperatorName);
IReduceSender <int> triangleNumberSender7 = commGroups[7].GetReduceSender <int>(reduceOperatorName);
IBroadcastReceiver <int> broadcastReceiver8 = commGroups[8].GetBroadcastReceiver <int>(broadcastOperatorName);
IReduceSender <int> triangleNumberSender8 = commGroups[8].GetReduceSender <int>(reduceOperatorName);
IBroadcastReceiver <int> broadcastReceiver9 = commGroups[9].GetBroadcastReceiver <int>(broadcastOperatorName);
IReduceSender <int> triangleNumberSender9 = commGroups[9].GetReduceSender <int>(reduceOperatorName);
for (int i = 1; i <= 10; i++)
{
broadcastSender.Send(i);
int n1 = broadcastReceiver1.Receive();
int n2 = broadcastReceiver2.Receive();
int n3 = broadcastReceiver3.Receive();
int n4 = broadcastReceiver4.Receive();
int n5 = broadcastReceiver5.Receive();
int n6 = broadcastReceiver6.Receive();
int n7 = broadcastReceiver7.Receive();
int n8 = broadcastReceiver8.Receive();
int n9 = broadcastReceiver9.Receive();
Assert.Equal(i, n1);
Assert.Equal(i, n2);
Assert.Equal(i, n3);
Assert.Equal(i, n4);
Assert.Equal(i, n5);
Assert.Equal(i, n6);
Assert.Equal(i, n7);
Assert.Equal(i, n8);
Assert.Equal(i, n9);
int triangleNum9 = GroupCommunicationTests.TriangleNumber(n9);
triangleNumberSender9.Send(triangleNum9);
int triangleNum8 = GroupCommunicationTests.TriangleNumber(n8);
triangleNumberSender8.Send(triangleNum8);
int triangleNum7 = GroupCommunicationTests.TriangleNumber(n7);
triangleNumberSender7.Send(triangleNum7);
int triangleNum6 = GroupCommunicationTests.TriangleNumber(n6);
triangleNumberSender6.Send(triangleNum6);
int triangleNum5 = GroupCommunicationTests.TriangleNumber(n5);
triangleNumberSender5.Send(triangleNum5);
int triangleNum4 = GroupCommunicationTests.TriangleNumber(n4);
triangleNumberSender4.Send(triangleNum4);
int triangleNum3 = GroupCommunicationTests.TriangleNumber(n3);
triangleNumberSender3.Send(triangleNum3);
int triangleNum2 = GroupCommunicationTests.TriangleNumber(n2);
triangleNumberSender2.Send(triangleNum2);
int triangleNum1 = GroupCommunicationTests.TriangleNumber(n1);
triangleNumberSender1.Send(triangleNum1);
int sum = sumReducer.Reduce();
int expected = GroupCommunicationTests.TriangleNumber(i) * (numTasks - 1);
Assert.Equal(sum, expected);
}
}
public void TestScatterReduceOperators()
{
string groupName = "group1";
string scatterOperatorName = "scatter";
string reduceOperatorName = "reduce";
string masterTaskId = "task0";
string driverId = "Driver Id";
int numTasks = 5;
int fanOut = 2;
IGroupCommDriver groupCommDriver = GetInstanceOfGroupCommDriver(driverId, masterTaskId, groupName, fanOut, numTasks);
ICommunicationGroupDriver commGroup = groupCommDriver.DefaultGroup
.AddScatter <int>(
scatterOperatorName,
masterTaskId,
TopologyTypes.Flat,
GetDefaultDataConverterConfig())
.AddReduce <int>(
reduceOperatorName,
masterTaskId,
TopologyTypes.Flat,
GetDefaultReduceFuncConfig(),
GetDefaultDataConverterConfig())
.Build();
List <ICommunicationGroupClient> commGroups = CommGroupClients(groupName, numTasks, groupCommDriver, commGroup, GetDefaultCodecConfig());
IScatterSender <int> sender = commGroups[0].GetScatterSender <int>(scatterOperatorName);
IReduceReceiver <int> sumReducer = commGroups[0].GetReduceReceiver <int>(reduceOperatorName);
IScatterReceiver <int> receiver1 = commGroups[1].GetScatterReceiver <int>(scatterOperatorName);
IReduceSender <int> sumSender1 = commGroups[1].GetReduceSender <int>(reduceOperatorName);
IScatterReceiver <int> receiver2 = commGroups[2].GetScatterReceiver <int>(scatterOperatorName);
IReduceSender <int> sumSender2 = commGroups[2].GetReduceSender <int>(reduceOperatorName);
IScatterReceiver <int> receiver3 = commGroups[3].GetScatterReceiver <int>(scatterOperatorName);
IReduceSender <int> sumSender3 = commGroups[3].GetReduceSender <int>(reduceOperatorName);
IScatterReceiver <int> receiver4 = commGroups[4].GetScatterReceiver <int>(scatterOperatorName);
IReduceSender <int> sumSender4 = commGroups[4].GetReduceSender <int>(reduceOperatorName);
Assert.NotNull(sender);
Assert.NotNull(receiver1);
Assert.NotNull(receiver2);
Assert.NotNull(receiver3);
Assert.NotNull(receiver4);
List <int> data = Enumerable.Range(1, 100).ToList();
List <string> order = new List <string> {
"task4", "task3", "task2", "task1"
};
sender.Send(data, order);
ScatterReceiveReduce(receiver4, sumSender4);
ScatterReceiveReduce(receiver3, sumSender3);
ScatterReceiveReduce(receiver2, sumSender2);
ScatterReceiveReduce(receiver1, sumSender1);
int sum = sumReducer.Reduce();
Assert.Equal(sum, data.Sum());
}
public void TestGetBroadcastReduceOperatorsForIntArrayMessageType()
{
const string groupName = "group1";
const string broadcastOperatorName = "broadcast";
const string reduceOperatorName = "reduce";
const string masterTaskId = "task0";
const string driverId = "Driver Id";
const int numTasks = 3;
const int fanOut = 2;
IConfiguration codecConfig = StreamingCodecConfiguration <int[]> .Conf
.Set(StreamingCodecConfiguration <int[]> .Codec, GenericType <IntArrayStreamingCodec> .Class)
.Build();
IConfiguration reduceFunctionConfig = ReduceFunctionConfiguration <int[]> .Conf
.Set(ReduceFunctionConfiguration <int[]> .ReduceFunction, GenericType <ArraySumFunction> .Class)
.Build();
IConfiguration dataConverterConfig = TangFactory.GetTang().NewConfigurationBuilder(
PipelineDataConverterConfiguration <int[]> .Conf
.Set(PipelineDataConverterConfiguration <int[]> .DataConverter,
GenericType <PipelineIntDataConverter> .Class)
.Build())
.BindNamedParameter <GroupTestConfig.ChunkSize, int>(
GenericType <GroupTestConfig.ChunkSize> .Class,
GroupTestConstants.ChunkSize.ToString(CultureInfo.InvariantCulture))
.Build();
var groupCommunicationDriver = GetInstanceOfGroupCommDriver(driverId, masterTaskId, groupName, fanOut, numTasks);
ICommunicationGroupDriver commGroup = groupCommunicationDriver.DefaultGroup
.AddBroadcast <int[]>(
broadcastOperatorName,
masterTaskId,
TopologyTypes.Flat,
dataConverterConfig)
.AddReduce <int[]>(
reduceOperatorName,
masterTaskId,
TopologyTypes.Flat,
dataConverterConfig,
reduceFunctionConfig)
.Build();
var commGroups = CommGroupClients(groupName, numTasks, groupCommunicationDriver, commGroup, codecConfig);
// for master task
IBroadcastSender <int[]> broadcastSender = commGroups[0].GetBroadcastSender <int[]>(broadcastOperatorName);
IReduceReceiver <int[]> sumReducer = commGroups[0].GetReduceReceiver <int[]>(reduceOperatorName);
IBroadcastReceiver <int[]> broadcastReceiver1 = commGroups[1].GetBroadcastReceiver <int[]>(broadcastOperatorName);
IReduceSender <int[]> triangleNumberSender1 = commGroups[1].GetReduceSender <int[]>(reduceOperatorName);
IBroadcastReceiver <int[]> broadcastReceiver2 = commGroups[2].GetBroadcastReceiver <int[]>(broadcastOperatorName);
IReduceSender <int[]> triangleNumberSender2 = commGroups[2].GetReduceSender <int[]>(reduceOperatorName);
Assert.NotNull(broadcastSender);
Assert.NotNull(sumReducer);
Assert.NotNull(broadcastReceiver1);
Assert.NotNull(triangleNumberSender1);
Assert.NotNull(broadcastReceiver2);
Assert.NotNull(triangleNumberSender2);
}
public void TestScatterReduceOperators()
{
string groupName = "group1";
string scatterOperatorName = "scatter";
string reduceOperatorName = "reduce";
string masterTaskId = "task0";
string driverId = "Driver Id";
int numTasks = 5;
int fanOut = 2;
var groupCommDriver = GroupCommunicationTests.GetInstanceOfGroupCommDriver(driverId, masterTaskId, groupName, fanOut, numTasks);
ICommunicationGroupDriver commGroup = groupCommDriver.DefaultGroup
.AddScatter <int>(
scatterOperatorName,
masterTaskId,
TopologyTypes.Tree,
GetDefaultDataConverterConfig())
.AddReduce <int>(
reduceOperatorName,
masterTaskId,
TopologyTypes.Tree,
GetDefaultDataConverterConfig(),
GetDefaultReduceFuncConfig())
.Build();
var commGroups = GroupCommunicationTests.CommGroupClients(groupName, numTasks, groupCommDriver, commGroup, GetDefaultCodecConfig());
IScatterSender <int> sender = commGroups[0].GetScatterSender <int>(scatterOperatorName);
IReduceReceiver <int> sumReducer = commGroups[0].GetReduceReceiver <int>(reduceOperatorName);
IScatterReceiver <int> receiver1 = commGroups[1].GetScatterReceiver <int>(scatterOperatorName);
IReduceSender <int> sumSender1 = commGroups[1].GetReduceSender <int>(reduceOperatorName);
IScatterReceiver <int> receiver2 = commGroups[2].GetScatterReceiver <int>(scatterOperatorName);
IReduceSender <int> sumSender2 = commGroups[2].GetReduceSender <int>(reduceOperatorName);
IScatterReceiver <int> receiver3 = commGroups[3].GetScatterReceiver <int>(scatterOperatorName);
IReduceSender <int> sumSender3 = commGroups[3].GetReduceSender <int>(reduceOperatorName);
IScatterReceiver <int> receiver4 = commGroups[4].GetScatterReceiver <int>(scatterOperatorName);
IReduceSender <int> sumSender4 = commGroups[4].GetReduceSender <int>(reduceOperatorName);
Assert.IsNotNull(sender);
Assert.IsNotNull(receiver1);
Assert.IsNotNull(receiver2);
Assert.IsNotNull(receiver3);
Assert.IsNotNull(receiver4);
List <int> data = Enumerable.Range(1, 100).ToList();
sender.Send(data);
List <int> data1 = receiver1.Receive();
List <int> data2 = receiver2.Receive();
List <int> data3 = receiver3.Receive();
List <int> data4 = receiver4.Receive();
int sum3 = data3.Sum();
sumSender3.Send(sum3);
int sum4 = data4.Sum();
sumSender4.Send(sum4);
int sum2 = data2.Sum();
sumSender2.Send(sum2);
int sum1 = data1.Sum();
sumSender1.Send(sum1);
int sum = sumReducer.Reduce();
Assert.AreEqual(sum, 6325);
}
public void TestScatterReduceOperatorsWithCancelation()
{
string groupName = "group1";
string scatterOperatorName = "scatter";
string reduceOperatorName = "reduce";
string masterTaskId = "task0";
string driverId = "Driver Id";
int numTasks = 5;
int fanOut = 2;
var groupCommDriver = GroupCommunicationTests.GetInstanceOfGroupCommDriver(driverId, masterTaskId, groupName, fanOut, numTasks);
ICommunicationGroupDriver commGroup = groupCommDriver.DefaultGroup
.AddScatter <int>(
scatterOperatorName,
masterTaskId,
TopologyTypes.Tree,
GetDefaultDataConverterConfig())
.AddReduce <int>(
reduceOperatorName,
masterTaskId,
TopologyTypes.Tree,
GetDefaultDataConverterConfig(),
GetDefaultReduceFuncConfig())
.Build();
var commGroups = GroupCommunicationTests.CommGroupClients(groupName, numTasks, groupCommDriver, commGroup, GetDefaultCodecConfig());
IScatterSender <int> sender = commGroups[0].GetScatterSender <int>(scatterOperatorName);
IReduceReceiver <int> sumReducer = commGroups[0].GetReduceReceiver <int>(reduceOperatorName);
IScatterReceiver <int> receiver1 = commGroups[1].GetScatterReceiver <int>(scatterOperatorName);
IReduceSender <int> sumSender1 = commGroups[1].GetReduceSender <int>(reduceOperatorName);
IScatterReceiver <int> receiver2 = commGroups[2].GetScatterReceiver <int>(scatterOperatorName);
IReduceSender <int> sumSender2 = commGroups[2].GetReduceSender <int>(reduceOperatorName);
IScatterReceiver <int> receiver3 = commGroups[3].GetScatterReceiver <int>(scatterOperatorName);
IReduceSender <int> sumSender3 = commGroups[3].GetReduceSender <int>(reduceOperatorName);
IScatterReceiver <int> receiver4 = commGroups[4].GetScatterReceiver <int>(scatterOperatorName);
IReduceSender <int> sumSender4 = commGroups[4].GetReduceSender <int>(reduceOperatorName);
Assert.NotNull(sender);
Assert.NotNull(receiver1);
Assert.NotNull(receiver2);
Assert.NotNull(receiver3);
Assert.NotNull(receiver4);
List <int> data = Enumerable.Range(1, 100).ToList();
sender.Send(data);
List <int> data1 = receiver1.Receive();
List <int> data2 = receiver2.Receive();
List <int> data3 = receiver3.Receive();
List <int> data4 = receiver4.Receive();
int sum3 = data3.Sum();
sumSender3.Send(sum3);
int sum4 = data4.Sum();
sumSender4.Send(sum4);
int sum2 = data2.Sum();
sumSender2.Send(sum2);
int sum1 = data1.Sum();
var token = new CancellationTokenSource();
token.Cancel();
Action send = () => sumSender1.Send(sum1, token);
Assert.Throws <OperationCanceledException>(send);
var taskThread = new Thread(() =>
{
Action reduce = () => sumReducer.Reduce(token);
Assert.Throws <OperationCanceledException>(reduce);
});
taskThread.Start();
token.Cancel();
}
