io.hyperfoil.clustering.RequestBenchmarkData Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hyperfoil-clustering Show documentation

There is a newer version: 0.27.1

package io.hyperfoil.clustering;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;

import io.hyperfoil.api.config.BenchmarkData;
import io.hyperfoil.api.config.BenchmarkDefinitionException;

import org.apache.logging.log4j.Logger;
import org.apache.logging.log4j.LogManager;

public class RequestBenchmarkData implements BenchmarkData {
   private static final Logger log = LogManager.getLogger(RequestBenchmarkData.class);
   private final Map files = new HashMap<>();

   public void addFile(String name, byte[] bytes) {
      files.put(Objects.requireNonNull(name), Objects.requireNonNull(bytes));
   }

   @Override
   public InputStream readFile(String file) {
      byte[] bytes = files.get(file);
      if (bytes == null) {
         throw new BenchmarkDefinitionException("Missing request file " + file + ", available files are: " + files.keySet());
      }
      return new ByteArrayInputStream(bytes);
   }

   @Override
   public Map files() {
      return files;
   }
}