mirror of
https://github.com/nosqlbench/nosqlbench.git
synced 2025-02-25 18:55:28 -06:00
StandardActivityType absorbs legacy dispenser logic (#2101)
This commit is contained in:
Dave Fisher
GitHub
@@ -1,146 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright (c) nosqlbench
|
|
||||||
*
|
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
* you may not use this file except in compliance with the License.
|
|
||||||
* You may obtain a copy of the License at
|
|
||||||
*
|
|
||||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
*
|
|
||||||
* Unless required by applicable law or agreed to in writing, software
|
|
||||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
* See the License for the specific language governing permissions and
|
|
||||||
* limitations under the License.
|
|
||||||
*/
|
|
||||||
|
|
||||||
package io.nosqlbench.engine.api.activityapi.core;
|
|
||||||
|
|
||||||
import io.nosqlbench.nb.api.components.core.NBComponent;
|
|
||||||
import io.nosqlbench.nb.api.engine.activityimpl.ActivityDef;
|
|
||||||
import io.nosqlbench.engine.api.activityapi.input.InputDispenser;
|
|
||||||
import io.nosqlbench.engine.api.activityapi.output.OutputDispenser;
|
|
||||||
import io.nosqlbench.engine.api.activityimpl.CoreServices;
|
|
||||||
import io.nosqlbench.engine.api.activityimpl.SimpleActivity;
|
|
||||||
import io.nosqlbench.engine.api.activityimpl.action.CoreActionDispenser;
|
|
||||||
import io.nosqlbench.engine.api.activityimpl.motor.CoreMotorDispenser;
|
|
||||||
|
|
||||||
import java.util.Map;
|
|
||||||
import java.util.Optional;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* <p>An ActivityType was the central extension point in NB for new
|
|
||||||
* activity types drivers. Please use DriverAdapter instead.
|
|
||||||
* It is responsible for naming the activity type, as well as providing
|
|
||||||
* the input, activity, and motor instances that will be assembled into an activity.</p>
|
|
||||||
* <p>At the very minimum, a useful implementation of an activity type should provide
|
|
||||||
* an action dispenser. Default implementations of input and motor dispensers are provided,
|
|
||||||
* and by extension, default inputs and motors.</p>
|
|
||||||
*/
|
|
||||||
//@Deprecated(forRemoval = true,since = "5.0")
|
|
||||||
public interface LegacyActivityType<A extends Activity> {
|
|
||||||
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Create an instance of an activity from the activity type.
|
|
||||||
*
|
|
||||||
* @param activityDef the definition that initializes and controls the activity.
|
|
||||||
* @return a distinct Activity instance for each call
|
|
||||||
*/
|
|
||||||
@SuppressWarnings("unchecked")
|
|
||||||
default A getActivity(final ActivityDef activityDef, final NBComponent parent) {
|
|
||||||
final SimpleActivity activity = new SimpleActivity(parent,activityDef);
|
|
||||||
return (A) activity;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Create an instance of an activity that ties together all the components into a usable
|
|
||||||
* activity instance. This is the method that should be called by executor classes.
|
|
||||||
*
|
|
||||||
* @param activityDef the definition that initializez and controlls the activity.
|
|
||||||
* @param activities a map of existing activities
|
|
||||||
* @return a distinct activity instance for each call
|
|
||||||
*/
|
|
||||||
default Activity getAssembledActivity(final ActivityDef activityDef, final Map<String, Activity> activities, final NBComponent parent) {
|
|
||||||
final A activity = this.getActivity(activityDef, parent);
|
|
||||||
|
|
||||||
final InputDispenser inputDispenser = this.getInputDispenser(activity);
|
|
||||||
if (inputDispenser instanceof ActivitiesAware) ((ActivitiesAware) inputDispenser).setActivitiesMap(activities);
|
|
||||||
activity.setInputDispenserDelegate(inputDispenser);
|
|
||||||
|
|
||||||
final ActionDispenser actionDispenser = this.getActionDispenser(activity);
|
|
||||||
if (actionDispenser instanceof ActivitiesAware)
|
|
||||||
((ActivitiesAware) actionDispenser).setActivitiesMap(activities);
|
|
||||||
activity.setActionDispenserDelegate(actionDispenser);
|
|
||||||
|
|
||||||
final OutputDispenser outputDispenser = this.getOutputDispenser(activity).orElse(null);
|
|
||||||
if ((null != outputDispenser) && (outputDispenser instanceof ActivitiesAware))
|
|
||||||
((ActivitiesAware) outputDispenser).setActivitiesMap(activities);
|
|
||||||
activity.setOutputDispenserDelegate(outputDispenser);
|
|
||||||
|
|
||||||
final MotorDispenser motorDispenser = this.getMotorDispenser(activity, inputDispenser, actionDispenser, outputDispenser);
|
|
||||||
if (motorDispenser instanceof ActivitiesAware) ((ActivitiesAware) motorDispenser).setActivitiesMap(activities);
|
|
||||||
activity.setMotorDispenserDelegate(motorDispenser);
|
|
||||||
|
|
||||||
return activity;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* This method will be called <em>once</em> per action instance.
|
|
||||||
*
|
|
||||||
* @param activity The activity instance that will parameterize the returned MarkerDispenser instance.
|
|
||||||
* @return an instance of MarkerDispenser
|
|
||||||
*/
|
|
||||||
default Optional<OutputDispenser> getOutputDispenser(final A activity) {
|
|
||||||
return CoreServices.getOutputDispenser(activity);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* This method will be called <em>once</em> per action instance.
|
|
||||||
*
|
|
||||||
* @param activity The activity instance that will parameterize the returned ActionDispenser instance.
|
|
||||||
* @return an instance of ActionDispenser
|
|
||||||
*/
|
|
||||||
default ActionDispenser getActionDispenser(final A activity) {
|
|
||||||
return new CoreActionDispenser(activity);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Return the InputDispenser instance that will be used by the associated activity to create Input factories
|
|
||||||
* for each thread slot.
|
|
||||||
*
|
|
||||||
* @param activity the Activity instance which will parameterize this InputDispenser
|
|
||||||
* @return the InputDispenser for the associated activity
|
|
||||||
*/
|
|
||||||
default InputDispenser getInputDispenser(final A activity) {
|
|
||||||
return CoreServices.getInputDispenser(activity);
|
|
||||||
}
|
|
||||||
|
|
||||||
default <T> MotorDispenser<T> getMotorDispenser(
|
|
||||||
final A activity,
|
|
||||||
final InputDispenser inputDispenser,
|
|
||||||
final ActionDispenser actionDispenser,
|
|
||||||
final OutputDispenser outputDispenser) {
|
|
||||||
return new CoreMotorDispenser<T>(activity, inputDispenser, actionDispenser, outputDispenser);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* An ActivityType can describe the canonical named types as known within that driver implementation,
|
|
||||||
* and the Java compatible types which can be assigned to them. This map is consulted when users need to select
|
|
||||||
* the name from within that driver in order to see the compatible functions which may produce a valid type for
|
|
||||||
* it. For example, a CQL users may want to know what java type (and thus what binding functions) can support
|
|
||||||
* the CQL timeuuid type.
|
|
||||||
*
|
|
||||||
* Conversely, a user may want to know what types are supported by the java.util.{@link java.util.UUID} class
|
|
||||||
* in CQL. In that case, this map will also be consulted, and multiple keys will match.
|
|
||||||
* When there are multiple answers in this way, but the driver has its own default about which one to use
|
|
||||||
* when the user's intent is ambiguous, the map should be be ordered as in {@link java.util.LinkedHashMap},
|
|
||||||
* and the preferred form should be listed first.
|
|
||||||
*
|
|
||||||
* @return a type map indicating co-compatible associations of driver-specific type names and Java types
|
|
||||||
*/
|
|
||||||
default Map<String,Class<?>> getTypeMap() {
|
|
||||||
return Map.of();
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
@@ -16,18 +16,29 @@
|
|||||||
|
|
||||||
package io.nosqlbench.engine.api.activityimpl.uniform;
|
package io.nosqlbench.engine.api.activityimpl.uniform;
|
||||||
|
|
||||||
|
|
||||||
import io.nosqlbench.adapters.api.activityimpl.uniform.DriverAdapter;
|
import io.nosqlbench.adapters.api.activityimpl.uniform.DriverAdapter;
|
||||||
import io.nosqlbench.nb.api.components.core.NBComponent;
|
import io.nosqlbench.nb.api.components.core.NBComponent;
|
||||||
import io.nosqlbench.nb.api.engine.activityimpl.ActivityDef;
|
import io.nosqlbench.nb.api.engine.activityimpl.ActivityDef;
|
||||||
|
import io.nosqlbench.engine.api.activityapi.core.Activity;
|
||||||
|
import io.nosqlbench.engine.api.activityapi.core.ActivitiesAware;
|
||||||
import io.nosqlbench.engine.api.activityapi.core.ActionDispenser;
|
import io.nosqlbench.engine.api.activityapi.core.ActionDispenser;
|
||||||
import io.nosqlbench.engine.api.activityapi.core.LegacyActivityType;
|
import io.nosqlbench.engine.api.activityapi.core.MotorDispenser;
|
||||||
|
import io.nosqlbench.engine.api.activityapi.input.InputDispenser;
|
||||||
|
import io.nosqlbench.engine.api.activityapi.output.OutputDispenser;
|
||||||
|
import io.nosqlbench.engine.api.activityimpl.CoreServices;
|
||||||
|
import io.nosqlbench.engine.api.activityimpl.SimpleActivity;
|
||||||
|
import io.nosqlbench.engine.api.activityimpl.action.CoreActionDispenser;
|
||||||
|
import io.nosqlbench.engine.api.activityimpl.motor.CoreMotorDispenser;
|
||||||
|
|
||||||
import org.apache.logging.log4j.LogManager;
|
import org.apache.logging.log4j.LogManager;
|
||||||
import org.apache.logging.log4j.Logger;
|
import org.apache.logging.log4j.Logger;
|
||||||
|
|
||||||
import java.util.HashMap;
|
import java.util.HashMap;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
|
import java.util.Optional;
|
||||||
|
|
||||||
public class StandardActivityType<A extends StandardActivity<?,?>> implements LegacyActivityType<A> {
|
public class StandardActivityType<A extends StandardActivity<?,?>> {
|
||||||
|
|
||||||
private static final Logger logger = LogManager.getLogger("ACTIVITY");
|
private static final Logger logger = LogManager.getLogger("ACTIVITY");
|
||||||
private final Map<String, DriverAdapter> adapters = new HashMap<>();
|
private final Map<String, DriverAdapter> adapters = new HashMap<>();
|
||||||
@@ -54,7 +65,13 @@ public class StandardActivityType<A extends StandardActivity<?,?>> implements Le
|
|||||||
// super(parent,activityDef);
|
// super(parent,activityDef);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
/**
|
||||||
|
* Create an instance of an activity from the activity type.
|
||||||
|
*
|
||||||
|
* @param activityDef the definition that initializes and controls the activity.
|
||||||
|
* @return a distinct Activity instance for each call
|
||||||
|
*/
|
||||||
|
@SuppressWarnings("unchecked")
|
||||||
public A getActivity(final ActivityDef activityDef, final NBComponent parent) {
|
public A getActivity(final ActivityDef activityDef, final NBComponent parent) {
|
||||||
if (activityDef.getParams().getOptionalString("async").isPresent())
|
if (activityDef.getParams().getOptionalString("async").isPresent())
|
||||||
throw new RuntimeException("This driver does not support async mode yet.");
|
throw new RuntimeException("This driver does not support async mode yet.");
|
||||||
@@ -62,10 +79,94 @@ public class StandardActivityType<A extends StandardActivity<?,?>> implements Le
|
|||||||
return (A) new StandardActivity(parent, activityDef);
|
return (A) new StandardActivity(parent, activityDef);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
@Override
|
* This method will be called <em>once</em> per action instance.
|
||||||
|
*
|
||||||
|
* @param activity The activity instance that will parameterize the returned ActionDispenser instance.
|
||||||
|
* @return an instance of ActionDispenser
|
||||||
|
*/
|
||||||
public ActionDispenser getActionDispenser(final A activity) {
|
public ActionDispenser getActionDispenser(final A activity) {
|
||||||
return new StandardActionDispenser(activity);
|
return new StandardActionDispenser(activity);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Create an instance of an activity that ties together all the components into a usable
|
||||||
|
* activity instance. This is the method that should be called by executor classes.
|
||||||
|
*
|
||||||
|
* @param activityDef the definition that initializez and controlls the activity.
|
||||||
|
* @param activities a map of existing activities
|
||||||
|
* @return a distinct activity instance for each call
|
||||||
|
*/
|
||||||
|
public Activity getAssembledActivity(final ActivityDef activityDef, final Map<String, Activity> activities, final NBComponent parent) {
|
||||||
|
final A activity = this.getActivity(activityDef, parent);
|
||||||
|
|
||||||
|
final InputDispenser inputDispenser = this.getInputDispenser(activity);
|
||||||
|
if (inputDispenser instanceof ActivitiesAware) ((ActivitiesAware) inputDispenser).setActivitiesMap(activities);
|
||||||
|
activity.setInputDispenserDelegate(inputDispenser);
|
||||||
|
|
||||||
|
final ActionDispenser actionDispenser = this.getActionDispenser(activity);
|
||||||
|
if (actionDispenser instanceof ActivitiesAware)
|
||||||
|
((ActivitiesAware) actionDispenser).setActivitiesMap(activities);
|
||||||
|
activity.setActionDispenserDelegate(actionDispenser);
|
||||||
|
|
||||||
|
final OutputDispenser outputDispenser = this.getOutputDispenser(activity).orElse(null);
|
||||||
|
if ((null != outputDispenser) && (outputDispenser instanceof ActivitiesAware))
|
||||||
|
((ActivitiesAware) outputDispenser).setActivitiesMap(activities);
|
||||||
|
activity.setOutputDispenserDelegate(outputDispenser);
|
||||||
|
|
||||||
|
final MotorDispenser motorDispenser = this.getMotorDispenser(activity, inputDispenser, actionDispenser, outputDispenser);
|
||||||
|
if (motorDispenser instanceof ActivitiesAware) ((ActivitiesAware) motorDispenser).setActivitiesMap(activities);
|
||||||
|
activity.setMotorDispenserDelegate(motorDispenser);
|
||||||
|
|
||||||
|
return activity;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* This method will be called <em>once</em> per action instance.
|
||||||
|
*
|
||||||
|
* @param activity The activity instance that will parameterize the returned MarkerDispenser instance.
|
||||||
|
* @return an instance of MarkerDispenser
|
||||||
|
*/
|
||||||
|
public Optional<OutputDispenser> getOutputDispenser(final A activity) {
|
||||||
|
return CoreServices.getOutputDispenser(activity);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Return the InputDispenser instance that will be used by the associated activity to create Input factories
|
||||||
|
* for each thread slot.
|
||||||
|
*
|
||||||
|
* @param activity the Activity instance which will parameterize this InputDispenser
|
||||||
|
* @return the InputDispenser for the associated activity
|
||||||
|
*/
|
||||||
|
public InputDispenser getInputDispenser(final A activity) {
|
||||||
|
return CoreServices.getInputDispenser(activity);
|
||||||
|
}
|
||||||
|
|
||||||
|
public <T> MotorDispenser<T> getMotorDispenser(
|
||||||
|
final A activity,
|
||||||
|
final InputDispenser inputDispenser,
|
||||||
|
final ActionDispenser actionDispenser,
|
||||||
|
final OutputDispenser outputDispenser) {
|
||||||
|
return new CoreMotorDispenser<T>(activity, inputDispenser, actionDispenser, outputDispenser);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* An ActivityType can describe the canonical named types as known within that driver implementation,
|
||||||
|
* and the Java compatible types which can be assigned to them. This map is consulted when users need to select
|
||||||
|
* the name from within that driver in order to see the compatible functions which may produce a valid type for
|
||||||
|
* it. For example, a CQL users may want to know what java type (and thus what binding functions) can support
|
||||||
|
* the CQL timeuuid type.
|
||||||
|
*
|
||||||
|
* Conversely, a user may want to know what types are supported by the java.util.{@link java.util.UUID} class
|
||||||
|
* in CQL. In that case, this map will also be consulted, and multiple keys will match.
|
||||||
|
* When there are multiple answers in this way, but the driver has its own default about which one to use
|
||||||
|
* when the user's intent is ambiguous, the map should be be ordered as in {@link java.util.LinkedHashMap},
|
||||||
|
* and the preferred form should be listed first.
|
||||||
|
*
|
||||||
|
* @return a type map indicating co-compatible associations of driver-specific type names and Java types
|
||||||
|
*/
|
||||||
|
public Map<String,Class<?>> getTypeMap() {
|
||||||
|
return Map.of();
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user