com.teststeps.thekla4j.websocket.stomp.spp.activities.Messages Maven / Gradle / Ivy

Go to download
package com.teststeps.thekla4j.websocket.stomp.spp.activities;

import com.teststeps.thekla4j.activityLog.annotations.Action;
import com.teststeps.thekla4j.commons.error.ActivityError;
import com.teststeps.thekla4j.core.base.activities.Interaction;
import com.teststeps.thekla4j.core.base.persona.Actor;
import com.teststeps.thekla4j.websocket.stomp.core.Destination;
import com.teststeps.thekla4j.websocket.stomp.core.StompDestination;
import com.teststeps.thekla4j.websocket.stomp.core.StompFrame;
import com.teststeps.thekla4j.websocket.stomp.spp.abilities.UseWebsocketWithStomp;
import io.vavr.collection.List;
import io.vavr.control.Either;
import lombok.AccessLevel;
import lombok.AllArgsConstructor;

import java.util.function.Predicate;

@AllArgsConstructor(access = AccessLevel.PRIVATE)
@Action("getting messages of destination @{destination}")
public class Messages extends Interaction> {

  private Destination destination;

  private Predicate filter;

  @Override
  protected Either> performAs(Actor actor, Void result) {

    return UseWebsocketWithStomp.as(actor)
        .flatMap(ability -> ability.atDestination(destination))
        .flatMap(StompDestination::messages)
        .map(frames -> frames.filter(filter));
  }


  public static Messages of(Destination destination) {
    return new Messages(destination, x -> true);
  }

  public Messages filterBy(Predicate filter) {
    this.filter = filter;
    return this;
  }
}