testsuites/performance/performance-benchmark-test/src/main/java/org/onap/policy/apex/testsuites/performance/benchmark/eventgenerator/EventGeneratorStats.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2018 Ericsson. All rights reserved.
   4  * ================================================================================
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  * SPDX-License-Identifier: Apache-2.0
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.policy.apex.testsuites.performance.benchmark.eventgenerator;
  22
  23 import com.google.gson.GsonBuilder;
  24 import java.util.ArrayList;
  25 import java.util.List;
  26 import java.util.Map;
  27
  28 /**
  29  * This class creates statistics for the event generator's current status.
  30  */
  31 public class EventGeneratorStats {
  32     private final EventBatchStats totalStats;
  33     private final List<EventBatchStats> batchStatsList = new ArrayList<>();
  34
  35     /**
  36      * Create the statistics using the current batch map.
  37      * @param batchMap the batch map to use
  38      */
  39     public EventGeneratorStats(final Map<Integer, EventBatch> batchMap) {
  40         for (EventBatch eventBatch: batchMap.values()) {
  41             batchStatsList.add(new EventBatchStats(eventBatch));
  42         }
  43
  44         totalStats = new EventBatchStats(batchStatsList);
  45     }
  46
  47     /**
  48      * Get the batch statistics as a JSON string.
  49      * @return the statistics as a JSON string
  50      */
  51     public String getStatsAsJsonString() {
  52         return new GsonBuilder().setPrettyPrinting().create().toJson(this);
  53     }
  54
  55     public EventBatchStats getTotalStats() {
  56         return totalStats;
  57     }
  58 }