add base Op behaviors

This commit is contained in:
Jonathan Shook
2021-12-21 13:08:27 -06:00
parent 5209f6e7bc
commit 7deb11b9ef
19 changed files with 153 additions and 39 deletions

View File

@@ -2,16 +2,17 @@ package io.nosqlbench.adapter.cqld4.opdispensers;
import com.datastax.oss.driver.api.core.CqlSession;
import io.nosqlbench.adapter.cqld4.Cqld4Op;
import io.nosqlbench.engine.api.activityimpl.OpDispenser;
import io.nosqlbench.engine.api.activityimpl.BaseOpDispenser;
import io.nosqlbench.engine.api.templating.ParsedOp;
import io.nosqlbench.nb.api.errors.BasicError;
public class CqlD4PreparedBatchOpDispenser implements OpDispenser<Cqld4Op> {
public class CqlD4PreparedBatchOpDispenser extends BaseOpDispenser<Cqld4Op> {
private final CqlSession session;
private final ParsedOp cmd;
public CqlD4PreparedBatchOpDispenser(CqlSession session, ParsedOp cmd) {
super(cmd);
this.session = session;
this.cmd = cmd;
}

View File

@@ -2,14 +2,15 @@ package io.nosqlbench.adapter.cqld4.opdispensers;
import com.datastax.oss.driver.api.core.CqlSession;
import io.nosqlbench.adapter.cqld4.Cqld4Op;
import io.nosqlbench.engine.api.activityimpl.OpDispenser;
import io.nosqlbench.engine.api.activityimpl.BaseOpDispenser;
import io.nosqlbench.engine.api.templating.ParsedOp;
public class Cqld4BatchStatementDispenser implements OpDispenser<Cqld4Op> {
public class Cqld4BatchStatementDispenser extends BaseOpDispenser<Cqld4Op> {
private final CqlSession session;
private final ParsedOp cmd;
public Cqld4BatchStatementDispenser(CqlSession session, ParsedOp cmd) {
super(cmd);
this.session = session;
this.cmd = cmd;
}

View File

@@ -5,15 +5,15 @@ import com.datastax.oss.driver.api.core.cql.BoundStatement;
import com.datastax.oss.driver.api.core.cql.PreparedStatement;
import io.nosqlbench.adapter.cqld4.Cqld4Op;
import io.nosqlbench.adapter.cqld4.Cqld4OpMetrics;
import io.nosqlbench.adapter.cqld4.optypes.Cqld4PreparedStatement;
import io.nosqlbench.adapter.cqld4.RSProcessors;
import io.nosqlbench.engine.api.activityimpl.OpDispenser;
import io.nosqlbench.adapter.cqld4.optypes.Cqld4PreparedStatement;
import io.nosqlbench.engine.api.activityimpl.BaseOpDispenser;
import io.nosqlbench.engine.api.templating.ParsedOp;
import io.nosqlbench.virtdata.core.templates.ParsedTemplate;
import java.util.function.LongFunction;
public class Cqld4PreparedStmtDispenser implements OpDispenser<Cqld4Op> {
public class Cqld4PreparedStmtDispenser extends BaseOpDispenser<Cqld4Op> {
private final CqlSession session;
@@ -25,6 +25,7 @@ public class Cqld4PreparedStmtDispenser implements OpDispenser<Cqld4Op> {
private final RSProcessors processors;
public Cqld4PreparedStmtDispenser(CqlSession session, ParsedOp cmd, RSProcessors processors) {
super(cmd);
this.session = session;
this.processors = processors;

View File

@@ -5,10 +5,10 @@ import com.datastax.oss.driver.api.core.cql.SimpleStatement;
import io.nosqlbench.adapter.cqld4.Cqld4Op;
import io.nosqlbench.adapter.cqld4.Cqld4OpMetrics;
import io.nosqlbench.adapter.cqld4.optypes.Cqld4SimpleCqlStatement;
import io.nosqlbench.engine.api.activityimpl.OpDispenser;
import io.nosqlbench.engine.api.activityimpl.BaseOpDispenser;
import io.nosqlbench.engine.api.templating.ParsedOp;
public class Cqld4SimpleCqlStmtDispenser implements OpDispenser<Cqld4Op> {
public class Cqld4SimpleCqlStmtDispenser extends BaseOpDispenser<Cqld4Op> {
private final CqlSession session;
private final ParsedOp cmd;
@@ -17,6 +17,7 @@ public class Cqld4SimpleCqlStmtDispenser implements OpDispenser<Cqld4Op> {
private final Cqld4OpMetrics metrics;
public Cqld4SimpleCqlStmtDispenser(CqlSession session, ParsedOp cmd) {
super(cmd);
this.session = session;
this.cmd = cmd;
this.maxpages = cmd.getStaticConfigOr("maxpages",1);