com.datastax.ebdrivers.kafkaproducer.KafkaAction Maven / Gradle / Ivy

Go to download

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

A Kafka driver for nosqlbench. This provides the ability to inject synthetic data into a kafka topic.

There is a newer version: 4.15.102

package com.datastax.ebdrivers.kafkaproducer;

import io.nosqlbench.engine.api.activityapi.core.SyncAction;
import io.nosqlbench.engine.api.activityapi.planning.OpSequence;
import org.apache.logging.log4j.Logger;
import org.apache.logging.log4j.LogManager;


public class KafkaAction implements SyncAction {

    private final static Logger logger = LogManager.getLogger(KafkaAction.class);

    private final KafkaProducerActivity activity;
    private final int slot;

    private OpSequence sequencer;

    public KafkaAction(KafkaProducerActivity activity, int slot) {
        this.activity = activity;
        this.slot = slot;
    }

    @Override
    public void init() {
        this.sequencer = activity.getOpSequencer();
    }

    @Override
    public int runCycle(long cycle) {
        sequencer.apply(cycle).write(cycle);
        return 1;
    }

}