com.cisco.trex.stateless.model.Stream Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of trex-java-sdk Show documentation
Java client SDK provides an implementation for TRex RPC APIs
There is a newer version: 1.69
package com.cisco.trex.stateless.model;

import java.util.Base64;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
import org.pcap4j.packet.Packet;

public class Stream {
  private Integer id;
  private Integer flags = 1;
  private Integer action_count = 0;
  private Integer random_seed = 0;
  private Integer core_id = -1;
  private Boolean enabled;
  private Double isg;
  private StreamMode mode;
  private Integer next_stream_id;
  private StreamPacket packet;
  private StreamVM vm;
  private Boolean self_start;
  private Map flow_stats = new HashMap<>();

  public Stream(
      Integer id,
      Boolean enabled,
      int flags,
      Double isg,
      StreamMode mode,
      Integer next_stream_id,
      Packet packet,
      StreamVM vm,
      Boolean self_start,
      boolean use_flow_stats,
      RuleType rule_type,
      int core_id) {
    this.id = id;
    this.core_id = core_id;
    this.flags = flags;
    this.enabled = enabled;
    this.isg = isg;
    this.mode = mode;
    this.next_stream_id = next_stream_id;
    String pkt = Base64.getEncoder().encodeToString(packet.getRawData());
    this.packet = new StreamPacket(pkt);
    this.vm = vm;
    this.self_start = self_start;
    flow_stats.put("enabled", use_flow_stats);
    flow_stats.put("stream_id", id);
    if (rule_type != null) {
      flow_stats.put("rule_type", rule_type.toString());
    }
  }

  public Stream(
      Integer id,
      Boolean enabled,
      int flags,
      Double isg,
      StreamMode mode,
      Integer next_stream_id,
      Packet packet,
      StreamVM vm,
      Boolean self_start,
      boolean use_flow_stats,
      RuleType rule_type) {
    this(
        id,
        enabled,
        flags,
        isg,
        mode,
        next_stream_id,
        packet,
        vm,
        self_start,
        use_flow_stats,
        rule_type,
        -1);
  }

  public Stream(
      Integer id,
      Boolean enabled,
      int flags,
      Double isg,
      StreamMode mode,
      Integer next_stream_id,
      Packet packet,
      StreamVM vm,
      Boolean self_start) {
    this(
        id,
        enabled,
        flags,
        isg,
        mode,
        next_stream_id,
        packet,
        vm,
        self_start,
        true,
        RuleType.STATS,
        -1);
  }

  public Integer getId() {
    return id;
  }

  public Integer getCore_id() {
    return core_id;
  }

  public Boolean getEnabled() {
    return enabled;
  }

  public Double getIsg() {
    return isg;
  }

  public StreamMode getMode() {
    return mode;
  }

  public Integer getNext_stream_id() {
    return next_stream_id;
  }

  public StreamPacket getPacket() {
    return packet;
  }

  public StreamVM getVm() {
    return vm;
  }

  public Boolean getSelf_start() {
    return self_start;
  }

  @Override
  public boolean equals(Object obj) {
    if (this == obj) {
      return true;
    }

    if (obj == null) {
      return false;
    }

    if (!(obj instanceof Stream)) {
      return false;
    }

    Stream s2 = (Stream) obj;

    return Objects.equals(this.id, s2.id)
        && Objects.equals(this.flags, s2.flags)
        && Objects.equals(this.action_count, s2.action_count)
        && Objects.equals(this.random_seed, s2.random_seed)
        && Objects.equals(this.core_id, s2.core_id)
        && Objects.equals(this.enabled, s2.enabled)
        && Objects.equals(this.isg, s2.isg)
        && Objects.equals(this.mode, s2.mode)
        && Objects.equals(this.next_stream_id, s2.next_stream_id)
        && Objects.equals(this.packet, s2.packet)
        && Objects.equals(this.vm, s2.vm)
        && Objects.equals(this.self_start, s2.self_start)
        && Objects.equals(this.flow_stats, s2.flow_stats);
  }

  public static class StreamPacket {
    /** Binary representation encoded as base64 string */
    private String binary;

    private String meta = "";

    public StreamPacket(String binaryBase64) {
      this(binaryBase64, "");
    }

    public StreamPacket(String binaryBase64, String meta) {
      binary = binaryBase64;
      this.meta = meta;
    }

    public String getBinary() {
      return binary;
    }

    public byte[] getBytes() {
      return Base64.getDecoder().decode(binary.getBytes());
    }

    public String getMeta() {
      return meta;
    }

    @Override
    public boolean equals(Object obj) {
      if (this == obj) {
        return true;
      }

      if (obj == null) {
        return false;
      }

      if (!(obj instanceof StreamPacket)) {
        return false;
      }

      StreamPacket rhs = (StreamPacket) obj;

      return Objects.equals(this.binary, rhs.binary) && Objects.equals(this.meta, rhs.meta);
    }
  }

  public enum RuleType {
    STATS("stats"),
    LATENCY("latency");
    String type;

    RuleType(String type) {
      this.type = type;
    }

    @Override
    public String toString() {
      return type;
    }
  }
}