1 files changed, 106 insertions, 0 deletions
diff --git a/java/perftests/src/main/java/org/apache/qpid/disttest/results/aggregation/TestResultAggregator.java b/java/perftests/src/main/java/org/apache/qpid/disttest/results/aggregation/TestResultAggregator.java
new file mode 100644
index 0000000000..5934e0e997
--- /dev/null
+++ b/java/perftests/src/main/java/org/apache/qpid/disttest/results/aggregation/TestResultAggregator.java
@@ -0,0 +1,106 @@
+/*
+ * 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.qpid.disttest.results.aggregation;
+
+import org.apache.qpid.disttest.message.ConsumerParticipantResult;
+import org.apache.qpid.disttest.message.ParticipantResult;
+import org.apache.qpid.disttest.message.ProducerParticipantResult;
+
+public class TestResultAggregator
+{
+    static final String AGGREGATED_ERROR_MESSAGE = "One or more participants reported errors.";
+    public static final String ALL_PARTICIPANTS_NAME = "All";
+    public static final String ALL_PRODUCER_PARTICIPANTS_NAME = "All Producers";
+    public static final String ALL_CONSUMER_PARTICIPANTS_NAME = "All Consumers";
+
+    public AggregatedTestResult aggregateTestResult(ITestResult originalTestResult)
+    {
+        ParticipantResultAggregator consumerResultAggregator = new ParticipantResultAggregator(ConsumerParticipantResult.class,
+                                                                                               ALL_CONSUMER_PARTICIPANTS_NAME);
+        ParticipantResultAggregator producerResultAggregator = new ParticipantResultAggregator(ProducerParticipantResult.class,
+                                                                                               ALL_PRODUCER_PARTICIPANTS_NAME);
+        ParticipantResultAggregator aggregatedResultsAggregator = new ParticipantResultAggregator(ParticipantResult.class,
+                                                                                               ALL_PARTICIPANTS_NAME);
+
+        boolean hasError = aggregateOriginalResults(originalTestResult,
+                                     consumerResultAggregator,
+                                     producerResultAggregator);
+
+        ParticipantResult aggregatedProducerResult = producerResultAggregator.getAggregatedResult();
+        ParticipantResult aggregaredConsumerResult = consumerResultAggregator.getAggregatedResult();
+
+        ParticipantResult aggregatedAllResult = aggregateAggregatedResults(
+                aggregatedResultsAggregator, aggregatedProducerResult,
+                aggregaredConsumerResult);
+
+        applyNonAggregateablesToAll(aggregatedAllResult,
+                aggregatedProducerResult, aggregaredConsumerResult);
+
+        AggregatedTestResult newTestResult = new AggregatedTestResult(originalTestResult);
+        newTestResult.setAllProducerParticipantResult(aggregatedProducerResult);
+        newTestResult.setAllConsumerParticipantResult(aggregaredConsumerResult);
+        newTestResult.setAllParticipantResult(aggregatedAllResult);
+
+        if (hasError)
+        {
+            aggregatedAllResult.setErrorMessage(TestResultAggregator.AGGREGATED_ERROR_MESSAGE);
+        }
+
+        return newTestResult;
+    }
+
+    private ParticipantResult aggregateAggregatedResults(
+            ParticipantResultAggregator aggregatedResultsAggregator,
+            ParticipantResult aggregatedProducerResult,
+            ParticipantResult aggregaredConsumerResult)
+    {
+        aggregatedResultsAggregator.aggregate(aggregatedProducerResult);
+        aggregatedResultsAggregator.aggregate(aggregaredConsumerResult);
+        ParticipantResult aggregatedAllResult = aggregatedResultsAggregator.getAggregatedResult();
+        return aggregatedAllResult;
+    }
+
+    private boolean aggregateOriginalResults(ITestResult originalTestResult,
+            ParticipantResultAggregator consumerParticipantResultAggregator,
+            ParticipantResultAggregator producerParticipantResultAggregator)
+    {
+        boolean hasError = false;
+        for (ParticipantResult result : originalTestResult.getParticipantResults())
+        {
+            consumerParticipantResultAggregator.aggregate(result);
+            producerParticipantResultAggregator.aggregate(result);
+
+            if (result.hasError())
+            {
+                hasError = true;
+            }
+        }
+        return hasError;
+    }
+
+    private void applyNonAggregateablesToAll(ParticipantResult aggregatedAllResult, ParticipantResult aggregatedProducerResult, ParticipantResult aggregatedConsumerResult)
+    {
+        aggregatedAllResult.setStartDate(aggregatedProducerResult.getStartDate());
+        aggregatedAllResult.setEndDate(aggregatedConsumerResult.getEndDate());
+
+        aggregatedAllResult.setNumberOfMessagesProcessed(aggregatedConsumerResult.getNumberOfMessagesProcessed());
+        aggregatedAllResult.setTotalPayloadProcessed(aggregatedConsumerResult.getTotalPayloadProcessed());
+        aggregatedAllResult.setThroughput(aggregatedConsumerResult.getThroughput());
+    }
+}