io.gatling.javaapi.http.Sse Maven / Gradle / Ivy
/*
* Copyright 2011-2024 GatlingCorp (https://gatling.io)
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.gatling.javaapi.http;
import static io.gatling.javaapi.core.internal.Expressions.*;
import static io.gatling.javaapi.http.internal.SseFunctions.*;
import edu.umd.cs.findbugs.annotations.NonNull;
import io.gatling.http.action.sse.SseInboundMessage;
import io.gatling.javaapi.core.ActionBuilder;
import io.gatling.javaapi.core.Session;
import java.util.List;
import java.util.function.BiFunction;
import java.util.function.Function;
/**
* DSL for building SSE
* configurations
*
* Immutable, so all methods return a new occurrence and leave the original unmodified.
*/
public final class Sse {
private final io.gatling.http.action.sse.Sse wrapped;
Sse(final io.gatling.http.action.sse.Sse wrapped) {
this.wrapped = wrapped;
}
/**
* Define a custom stream name so multiple SSEs for the same virtual users don't conflict
*
* @param sseName the name, expressed as a Gatling Expression Language String
* @return a new Sse instance
*/
@NonNull
public Sse sseName(@NonNull String sseName) {
return new Sse(wrapped.sseName(toStringExpression(sseName)));
}
/**
* Define a custom stream name so multiple SSEs for the same virtual users don't conflict
*
* @param sseName the name, expressed as a function
* @return a new Sse instance
*/
@NonNull
public Sse sseName(@NonNull Function sseName) {
return new Sse(wrapped.sseName(javaFunctionToExpression(sseName)));
}
/**
* Boostrap an action to connect the SSE with a GET request
*
* @param url the url to connect to, expressed as a Gatling Expression Language String
* @return the next DSL step
*/
@NonNull
public SseConnectActionBuilder get(@NonNull String url) {
return new SseConnectActionBuilder(wrapped.get(toStringExpression(url)));
}
/**
* Boostrap an action to connect the SSE with a POST request
*
* @param url the url to connect to, expressed as a Gatling Expression Language String
* @return the next DSL step
*/
@NonNull
public SseConnectActionBuilder post(@NonNull String url) {
return new SseConnectActionBuilder(wrapped.post(toStringExpression(url)));
}
/**
* Boostrap an action to connect the SSE with a GET request
*
* @param url the url to connect to, expressed as a Gatling Expression Language String
* @return the next DSL step
*/
@NonNull
public SseConnectActionBuilder get(@NonNull Function url) {
return new SseConnectActionBuilder(wrapped.get(javaFunctionToExpression(url)));
}
/**
* Boostrap an action to connect the SSE with a POST request
*
* @param url the url to connect to, expressed as a Gatling Expression Language String
* @return the next DSL step
*/
@NonNull
public SseConnectActionBuilder post(@NonNull Function url) {
return new SseConnectActionBuilder(wrapped.post(javaFunctionToExpression(url)));
}
/**
* Boostrap an action to set a check
*
* @return the next DSL step
*/
@NonNull
public SseSetCheckActionBuilder setCheck() {
return new SseSetCheckActionBuilder(wrapped.setCheck());
}
/**
* Create an action to close the stream
*
* @return an ActionBuilder
*/
@NonNull
public ActionBuilder close() {
return wrapped::close;
}
public static final class Prefix {
public static final Prefix INSTANCE = new Prefix();
private Prefix() {}
/**
* Boostrap a SSE check
*
* @param name the name of the check
* @return the next DSL step
*/
@NonNull
public SseMessageCheck checkMessage(@NonNull String name) {
return new SseMessageCheck(io.gatling.http.Predef.sse().checkMessage(name));
}
/**
* Process the currently buffered inbound SSE messages and empty the buffer
*
* @param f the function to process the buffered messages
* @return an ActionBuilder
*/
public ActionBuilder processUnmatchedMessages(
BiFunction, Session, Session> f) {
return () ->
io.gatling.http.Predef.sse()
.processUnmatchedMessages(javaProcessUnmatchedMessagesBiFunctionToExpression(f));
}
/**
* Process the currently buffered inbound SSE messages and empty the buffer
*
* @param sseName the name of the SSE stream, expressed as a Gatling Expression Language String
* @param f the function to process the buffered messages
* @return an ActionBuilder
*/
public ActionBuilder processUnmatchedMessages(
String sseName, BiFunction, Session, Session> f) {
return () ->
io.gatling.http.Predef.sse()
.processUnmatchedMessages(
toStringExpression(sseName),
javaProcessUnmatchedMessagesBiFunctionToExpression(f));
}
/**
* Process the currently buffered inbound SSE messages and empty the buffer
*
* @param sseName the name of the SSE stream, expressed as a function
* @param f the function to process the buffered messages
* @return an ActionBuilder
*/
public ActionBuilder processUnmatchedMessages(
Function sseName,
BiFunction, Session, Session> f) {
return () ->
io.gatling.http.Predef.sse()
.processUnmatchedMessages(
javaFunctionToExpression(sseName),
javaProcessUnmatchedMessagesBiFunctionToExpression(f));
}
}
}