io.nosqlbench.driver.pulsar.ops.PulsarAdminTopicMapper Maven / Gradle / Ivy

Go to download

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

A Pulsar driver for nosqlbench. This provides the ability to inject synthetic data into a pulsar system.

There is a newer version: 4.15.102

package io.nosqlbench.driver.pulsar.ops;

import io.nosqlbench.driver.pulsar.PulsarActivity;
import io.nosqlbench.driver.pulsar.PulsarSpace;
import io.nosqlbench.engine.api.templating.CommandTemplate;
import org.apache.commons.lang3.BooleanUtils;
import org.apache.commons.lang3.StringUtils;

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 PulsarAdminTopicMapper extends PulsarAdminMapper {
    private final LongFunction topicUriFunc;
    private final LongFunction enablePartionFunc;
    private final LongFunction partitionNumFunc;

    public PulsarAdminTopicMapper(CommandTemplate cmdTpl,
                                  PulsarSpace clientSpace,
                                  PulsarActivity pulsarActivity,
                                  LongFunction asyncApiFunc,
                                  LongFunction adminDelOpFunc,
                                  LongFunction topicUriFunc,
                                  LongFunction enablePartionFunc,
                                  LongFunction partitionNumFunc)
    {
        super(cmdTpl, clientSpace, pulsarActivity, asyncApiFunc, adminDelOpFunc);
        this.topicUriFunc = topicUriFunc;
        this.enablePartionFunc = enablePartionFunc;
        this.partitionNumFunc = partitionNumFunc;
    }

    @Override
    public PulsarOp apply(long value) {
        String topicUri = topicUriFunc.apply(value);
        String enablePartitionStr = enablePartionFunc.apply(value);
        String partitionNumStr = partitionNumFunc.apply(value);
        boolean asyncApi = asyncApiFunc.apply(value);
        boolean adminDelOp = adminDelOpFunc.apply(value);

        if ( StringUtils.isBlank(topicUri) ) {
            throw new RuntimeException("\"topic_uri\" parameter can't be empty when creating a Pulsar topic!");
        }

        boolean partitionTopic = BooleanUtils.toBoolean(enablePartitionStr);

        boolean invalidPartStr;
        int partitionNum = 0;
        if ( StringUtils.isBlank(partitionNumStr) || !StringUtils.isNumeric(partitionNumStr) ) {
            invalidPartStr = true;
        } else {
            partitionNum = Integer.parseInt(partitionNumStr);
            invalidPartStr = (partitionNum <= 0);
        }
        if (partitionTopic && invalidPartStr) {
            throw new RuntimeException("Invalid specified value for \"partition_num\" parameter when creating partitioned topic!");
        }

        return new PulsarAdminTopicOp(
            clientSpace,
            topicUri,
            partitionTopic,
            partitionNum,
            asyncApi,
            adminDelOp);
    }
}