ru.tinkoff.gatling.kafka.javaapi.request.builder.KafkaRequestBuilderBase Maven / Gradle / Ivy

Go to download
package ru.tinkoff.gatling.kafka.javaapi.request.builder;

import org.apache.kafka.common.header.Headers;
import org.apache.kafka.common.header.internals.RecordHeaders;

import static io.gatling.javaapi.core.internal.Expressions.*;

public class KafkaRequestBuilderBase {

    private final ru.tinkoff.gatling.kafka.request.builder.KafkaRequestBuilderBase wrapped;
    private final String requestName;

    public KafkaRequestBuilderBase(ru.tinkoff.gatling.kafka.request.builder.KafkaRequestBuilderBase wrapped, String requestName){
        this.wrapped = wrapped;
        this.requestName = requestName;
    }

    public  RequestBuilder send(K key, V payload) {
        return new RequestBuilder<>(
                wrapped.send(
                        toStaticValueExpression(key),
                        toStaticValueExpression(payload),
                        toStaticValueExpression(new RecordHeaders()),
                        ru.tinkoff.gatling.kafka.request.builder.Sender.noSchemaSender()
                ));
    }

    public  RequestBuilder send(K key, V payload, Headers headers) {
        return new RequestBuilder<>(
                wrapped.send(
                        toStaticValueExpression(key),
                        toStaticValueExpression(payload),
                        toStaticValueExpression(headers),
                        ru.tinkoff.gatling.kafka.request.builder.Sender.noSchemaSender()
                ));
    }

    public  RequestBuilder send(V payload) {
        return new RequestBuilder<>(wrapped.send(
                toStaticValueExpression(payload),
                ru.tinkoff.gatling.kafka.request.builder.Sender.noSchemaSender()));
    }

    public ReqRepBase requestReply() {
        return new ReqRepBase(requestName);
    }

}