io.hyperfoil.cli.commands.Sessions Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hyperfoil-cli Show documentation
There is a newer version: 0.27.1
package io.hyperfoil.cli.commands;

import java.util.Map;
import java.util.TreeMap;
import java.util.function.BinaryOperator;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import org.aesh.command.CommandDefinition;
import org.aesh.command.CommandException;
import org.aesh.command.CommandResult;

import io.hyperfoil.cli.Table;
import io.hyperfoil.cli.context.HyperfoilCommandInvocation;
import io.hyperfoil.client.Client;
import io.hyperfoil.client.RestClientException;
import io.hyperfoil.core.util.Util;

@CommandDefinition(name = "sessions", description = "Show sessions statistics")
public class Sessions extends BaseRunIdCommand {
   Table> SESSION_STATS = new Table>()
         .column("AGENT", Map.Entry::getKey)
         .column("MIN", e -> String.valueOf(e.getValue().min), Table.Align.RIGHT)
         .column("MAX", e -> String.valueOf(e.getValue().max), Table.Align.RIGHT);

   @Override
   public CommandResult execute(HyperfoilCommandInvocation invocation) throws CommandException {
      Client.RunRef runRef = getRunRef(invocation);
      Map> sessionStats = null;
      for (; ; ) {
         try {
            int numLines = sessionStats == null ? 0 : sessionStats.values().stream().mapToInt(Map::size).sum() + 2;
            sessionStats = runRef.sessionStatsRecent();
            clearLines(invocation, numLines);
            if (sessionStats == null || sessionStats.isEmpty()) {
               Client.Run run = runRef.get();
               if (run.terminated != null) {
                  invocation.println("Run " + run.id + " has terminated.");
                  invocation.print(SESSION_STATS.print("PHASE", toMapOfStreams(runRef.sessionStatsTotal())));
                  return CommandResult.SUCCESS;
               }
            }
            invocation.print(SESSION_STATS.print("PHASE", toMapOfStreams(sessionStats)));
            if (interruptibleDelay(invocation)) {
               return CommandResult.SUCCESS;
            }
         } catch (RestClientException e) {
            if (e.getCause() instanceof InterruptedException) {
               clearLines(invocation, 1);
               invocation.println("");
               return CommandResult.SUCCESS;
            }
            invocation.println("ERROR: " + Util.explainCauses(e));
            throw new CommandException("Cannot display session stats.", e);
         }
      }
   }

   private Map>> toMapOfStreams(Map> sessionStats) {
      return sessionStats.entrySet().stream()
            .collect(Collectors.toMap(Map.Entry::getKey, Sessions::soretdEntries, throwingMerger(), TreeMap::new));
   }

   private static Stream> soretdEntries(Map.Entry> e) {
      return e.getValue().entrySet().stream().sorted(Map.Entry.comparingByKey());
   }

   private static BinaryOperator>> throwingMerger() {
      return (u, v) -> { throw new IllegalStateException(); };
   }
}