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