IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2025-02-25 18:55:28 -06:00
Code Issues Packages Projects Releases Wiki Activity

ErrorResponse enum check

Browse Source
This commit is contained in:
phact 2020-05-07 15:51:17 -04:00
parent a4b2a63957
commit e2e75f4dc0
1 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
driver-cql/src/main/java/io/nosqlbench/activitytype/cql/core/CqlActivity.java
View File

@ -41,6 +41,8 @@ import io.nosqlbench.engine.api.util.SimpleConfig;
import io.nosqlbench.engine.api.templating.StrInterpolator; import io.nosqlbench.engine.api.templating.StrInterpolator;
import io.nosqlbench.engine.api.util.TagFilter; import io.nosqlbench.engine.api.util.TagFilter;
import io.nosqlbench.engine.api.util.Unit; import io.nosqlbench.engine.api.util.Unit;
import io.nosqlbench.nb.api.errors.BasicError;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger; import org.slf4j.Logger;
import org.slf4j.LoggerFactory; import org.slf4j.LoggerFactory;
@ -512,9 +514,10 @@ public class CqlActivity extends SimpleActivity implements Activity, ActivityDef
String[] keyval = spec.split("=|->|:", 2); String[] keyval = spec.split("=|->|:", 2);
if (keyval.length == 1) { if (keyval.length == 1) {
String verb = keyval[0]; String verb = keyval[0];
newerrorHandler.setDefaultHandler( ErrorResponse errorResponse = getErrorResponseOrBasicError(verb);
newerrorHandler.setDefaultHandler(
new NBCycleErrorHandler( new NBCycleErrorHandler(
ErrorResponse.valueOf(verb), errorResponse,
exceptionCountMetrics, exceptionCountMetrics,
exceptionHistoMetrics, exceptionHistoMetrics,
!getParams().getOptionalLong("async").isPresent() !getParams().getOptionalLong("async").isPresent()
@ -524,9 +527,10 @@ public class CqlActivity extends SimpleActivity implements Activity, ActivityDef
String pattern = keyval[0]; String pattern = keyval[0];
String verb = keyval[1]; String verb = keyval[1];
if (newerrorHandler.getGroupNames().contains(pattern)) { if (newerrorHandler.getGroupNames().contains(pattern)) {
ErrorResponse errorResponse = getErrorResponseOrBasicError(verb);
NBCycleErrorHandler handler = NBCycleErrorHandler handler =
new NBCycleErrorHandler( new NBCycleErrorHandler(
ErrorResponse.valueOf(verb), errorResponse,
exceptionCountMetrics, exceptionCountMetrics,
exceptionHistoMetrics, exceptionHistoMetrics,
!getParams().getOptionalLong("async").isPresent() !getParams().getOptionalLong("async").isPresent()
@ -534,8 +538,9 @@ public class CqlActivity extends SimpleActivity implements Activity, ActivityDef
logger.info("Handling error group '" + pattern + "' with handler:" + handler); logger.info("Handling error group '" + pattern + "' with handler:" + handler);
newerrorHandler.setHandlerForGroup(pattern, handler); newerrorHandler.setHandlerForGroup(pattern, handler);
} else { } else {
ErrorResponse errorResponse = ErrorResponse.valueOf(keyval[1]);
NBCycleErrorHandler handler = new NBCycleErrorHandler( NBCycleErrorHandler handler = new NBCycleErrorHandler(
ErrorResponse.valueOf(keyval[1]), errorResponse,
exceptionCountMetrics, exceptionCountMetrics,
exceptionHistoMetrics, exceptionHistoMetrics,
!getParams().getOptionalLong("async").isPresent() !getParams().getOptionalLong("async").isPresent()
@ -549,6 +554,14 @@ public class CqlActivity extends SimpleActivity implements Activity, ActivityDef
return newerrorHandler; return newerrorHandler;
} }
private ErrorResponse getErrorResponseOrBasicError(String verb) {
try {
return ErrorResponse.valueOf(verb);
}catch(IllegalArgumentException e){
throw new BasicError("Invalid parameter for errors: '"+ verb + "' should be one of: " + StringUtils.join(ErrorResponse.values(), ", "));
}
}
public int getMaxTries() { public int getMaxTries() {
return maxTries; return maxTries;
} }