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
  25 import java.util.ArrayList;
  26 import java.util.List;
  27 import java.util.Map;
  28
  29 /**
  30  * This class creates statistics for the event generator's current status.
  31  */
  32 public class EventGeneratorStats {
  33     private final EventBatchStats totalStats;
  34     private final List<EventBatchStats> batchStatsList = new ArrayList<>();
  35
  36     /**
  37      * Create the statistics using the current batch map.
  38      * @param batchMap the batch map to use
  39      */
  40     public EventGeneratorStats(final Map<Integer, EventBatch> batchMap) {
  41         for (EventBatch eventBatch: batchMap.values()) {
  42             batchStatsList.add(new EventBatchStats(eventBatch));
  43         }
  44
  45         totalStats = new EventBatchStats(batchStatsList);
  46     }
  47
  48     /**
  49      * Get the batch statistics as a JSON string.
  50      * @return the statistics as a JSON string
  51      */
  52     public String getStatsAsJsonString() {
  53         return new GsonBuilder().setPrettyPrinting().create().toJson(this);
  54     }
  55
  56     public EventBatchStats getTotalStats() {
  57         return totalStats;
  58     }
  59 }