All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.rabbitmq.perf.VaryingScenarioStats Maven / Gradle / Ivy

The newest version!
// Copyright (c) 2007-2023 Broadcom. All Rights Reserved.
// The term "Broadcom" refers to Broadcom Inc. and/or its subsidiaries.
//
// This software, the RabbitMQ Java client library, is triple-licensed under the
// Mozilla Public License 2.0 ("MPL"), the GNU General Public License version 2
// ("GPL") and the Apache License version 2 ("ASL"). For the MPL, please see
// LICENSE-MPL-RabbitMQ. For the GPL, please see LICENSE-GPL2.  For the ASL,
// please see LICENSE-APACHE2.
//
// This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND,
// either express or implied. See the LICENSE file for specific language governing
// rights and limitations of this software.
//
// If you have any questions regarding licensing, please contact us at
// [email protected].
package com.rabbitmq.perf;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class VaryingScenarioStats implements ScenarioStats {
  private final Map, SimpleScenarioStats> stats =
      new HashMap, SimpleScenarioStats>();
  private final List> keys = new ArrayList>();

  public VaryingScenarioStats() {}

  public SimpleScenarioStats next(List value) {
    SimpleScenarioStats stats = new SimpleScenarioStats(1000L);
    keys.add(value);
    this.stats.put(value, stats);
    return stats;
  }

  @SuppressWarnings("unchecked")
  public Map results() {
    Map map = new HashMap();

    List dimensions = new ArrayList();
    for (VariableValue keyElem : keys.get(0)) {
      dimensions.add(keyElem.getName());
    }
    map.put("dimensions", dimensions);

    Map> dimensionValues = new HashMap>();
    for (List key : keys) {
      for (VariableValue elem : key) {
        List values = get(elem.getName(), dimensionValues, new ArrayList());
        String value = elem.getValue().toString();
        if (!values.contains(value)) {
          values.add(value);
        }
      }
    }
    map.put("dimension-values", dimensionValues);

    Map data = new HashMap();
    for (List key : keys) {
      Map results = stats.get(key).results();
      Map node = data;
      for (int i = 0; i < key.size(); i++) {
        VariableValue elem = key.get(i);
        if (i == key.size() - 1) {
          node.put(elem.getValue().toString(), results);
        } else {
          node =
              (Map)
                  get(elem.getValue().toString(), node, new HashMap());
        }
      }
    }
    map.put("data", data);

    return map;
  }

  private  V get(K key, Map map, V def) {
    V val = map.get(key);
    if (val == null) {
      val = def;
      map.put(key, val);
    }
    return val;
  }
}