io.nosqlbench.driver.jms.ops.JmsMsgSendMapper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of driver-jms Show documentation

A JMS driver for nosqlbench. This provides the ability to inject synthetic data into a pulsar system via JMS 2.0 compatibile APIs. NOTE: this is JMS compatible driver from DataStax that allows using a Pulsar cluster as the potential JMS Destination

There is a newer version: 4.15.97

Show newest version

package io.nosqlbench.driver.jms.ops;

import io.nosqlbench.driver.jms.JmsActivity;
import io.nosqlbench.driver.jms.util.JmsHeader;
import io.nosqlbench.driver.jms.util.JmsHeaderLongFunc;

import javax.jms.Destination;
import java.util.Map;
import java.util.function.LongFunction;

/**
 * This maps a set of specifier functions to a pulsar operation. The pulsar operation contains
 * enough state to define a pulsar operation such that it can be executed, measured, and possibly
 * retried if needed.
 *
 * This function doesn't act *as* the operation. It merely maps the construction logic into
 * a simple functional type, given the component functions.
 *
 * For additional parameterization, the command template is also provided.
 */
public class JmsMsgSendMapper extends JmsOpMapper {
    private final JmsHeaderLongFunc jmsHeaderLongFunc;
    private final Map jmsMsgProperties;
    private final LongFunction msgBodyFunc;

    public JmsMsgSendMapper(JmsActivity jmsActivity,
                            LongFunction asyncApiFunc,
                            LongFunction jmsDestinationFunc,
                            JmsHeaderLongFunc jmsHeaderLongFunc,
                            Map jmsMsgProperties,
                            LongFunction msgBodyFunc) {
        super(jmsActivity, asyncApiFunc, jmsDestinationFunc);

        this.jmsHeaderLongFunc = jmsHeaderLongFunc;
        this.jmsMsgProperties = jmsMsgProperties;
        this.msgBodyFunc = msgBodyFunc;
    }

    @Override
    public JmsOp apply(long value) {
        boolean asyncApi = asyncApiFunc.apply(value);
        Destination jmsDestination = jmsDestinationFunc.apply(value);
        JmsHeader jmsHeader = (JmsHeader)jmsHeaderLongFunc.apply(value);
        String msgBody = msgBodyFunc.apply(value);

        return new JmsMsgSendOp(
            jmsActivity,
            asyncApi,
            jmsDestination,
            jmsHeader,
            jmsMsgProperties,
            msgBody
        );
    }
}