Examples with PolyglotMetricRegistryBindings io.nosqlbench.engine.core.metrics.PolyglotMetricRegistryBindings used on opensource projects

Example 1 with PolyglotMetricRegistryBindings

use of io.nosqlbench.engine.core.metrics.PolyglotMetricRegistryBindings in project nosqlbench by nosqlbench.

the class MetricsMapper method metricsDetail.

public static String metricsDetail(String activitySpec) {
    // StringBuilder metricsDetail = new StringBuilder();
    List<String> metricsDetails = new ArrayList<>();
    ActivityDef activityDef = ActivityDef.parseActivityDef(activitySpec);
    logger.info("introspecting metric names for " + activitySpec);
    Optional<ActivityType> activityType = new ActivityTypeLoader().load(activityDef);
    if (!activityType.isPresent()) {
        throw new RuntimeException("Activity type '" + activityDef.getActivityType() + "' does not exist in this runtime.");
    }
    Activity activity = activityType.get().getAssembledActivity(activityDef, new HashMap<>());
    PolyglotMetricRegistryBindings nashornMetricRegistryBindings = new PolyglotMetricRegistryBindings(ActivityMetrics.getMetricRegistry());
    activity.initActivity();
    activity.getInputDispenserDelegate().getInput(0);
    activity.getActionDispenserDelegate().getAction(0);
    activity.getMotorDispenserDelegate().getMotor(activityDef, 0);
    Map<String, Metric> metricMap = nashornMetricRegistryBindings.getMetrics();
    for (Map.Entry<String, Metric> metricEntry : metricMap.entrySet()) {
        String metricName = metricEntry.getKey();
        Metric metricValue = metricEntry.getValue();
        Map<String, String> getterSummary = getGetterSummary(metricValue);
        // details.put(metricName,getterSummary);
        List<String> methodDetails = getterSummary.entrySet().stream().map(es -> metricName + es.getKey() + "  " + es.getValue()).collect(Collectors.toList());
        methodDetails.sort(String::compareTo);
        String getterText = methodDetails.stream().collect(Collectors.joining("\n"));
        metricsDetails.add(metricName + "\n" + getterText);
    }
    return metricsDetails.stream().collect(Collectors.joining("\n"));
}

Example 2 with PolyglotMetricRegistryBindings

use of io.nosqlbench.engine.core.metrics.PolyglotMetricRegistryBindings in project nosqlbench by nosqlbench.

the class Scenario method init.

private void init() {
    logger.debug("Using engine " + engine.toString());
    MetricRegistry metricRegistry = ActivityMetrics.getMetricRegistry();
    switch(engine) {
        case Nashorn:
            throw new RuntimeException("The nashorn engine has been deprecated in this version of NoSQLBench.");
        case Graalvm:
            Context.Builder contextSettings = Context.newBuilder("js").allowHostAccess(HostAccess.ALL).allowNativeAccess(true).allowCreateThread(true).allowIO(true).allowHostClassLookup(s -> true).allowHostClassLoading(true).allowCreateProcess(true).allowAllAccess(true).allowEnvironmentAccess(EnvironmentAccess.INHERIT).allowPolyglotAccess(PolyglotAccess.ALL).option("js.ecmascript-version", "2020").option("js.nashorn-compat", "true");
            // TODO: add in, out, err for this scenario
            this.scriptEngine = GraalJSScriptEngine.create(null, contextSettings);
            break;
    }
    scenarioController = new ScenarioController(this.scenarioName, minMaturity);
    if (!progressInterval.equals("disabled")) {
        activityProgressIndicator = new ActivityProgressIndicator(scenarioController, progressInterval);
    }
    scriptEnv = new ScenarioContext(scenarioController);
    scriptEngine.setContext(scriptEnv);
    scriptEngine.put("params", scenarioScriptParams);
    if (engine == Engine.Graalvm) {
        // https://github.com/graalvm/graaljs/blob/master/docs/user/JavaInterop.md
        if (wantsGraaljsCompatMode) {
            scriptEngine.put("scenario", scenarioController);
            scriptEngine.put("metrics", new PolyglotMetricRegistryBindings(metricRegistry));
            scriptEngine.put("activities", new NashornActivityBindings(scenarioController));
        } else {
            scriptEngine.put("scenario", new PolyglotScenarioController(scenarioController));
            scriptEngine.put("metrics", new PolyglotMetricRegistryBindings(metricRegistry));
            scriptEngine.put("activities", new NashornActivityBindings(scenarioController));
        }
    } else if (engine == Engine.Nashorn) {
        throw new RuntimeException("The Nashorn engine has been deprecated in this version of NoSQLBench.");
    } else {
        throw new RuntimeException("Unsupported engine: " + engine);
    }
    for (ScriptingPluginInfo<?> extensionDescriptor : SandboxExtensionFinder.findAll()) {
        if (!extensionDescriptor.isAutoLoading()) {
            logger.info("Not loading " + extensionDescriptor + ", autoloading is false");
            continue;
        }
        Logger extensionLogger = LogManager.getLogger("extensions." + extensionDescriptor.getBaseVariableName());
        Object extensionObject = extensionDescriptor.getExtensionObject(extensionLogger, metricRegistry, scriptEnv);
        ScenarioMetadataAware.apply(extensionObject, getScenarioMetadata());
        logger.trace("Adding extension object:  name=" + extensionDescriptor.getBaseVariableName() + " class=" + extensionObject.getClass().getSimpleName());
        scriptEngine.put(extensionDescriptor.getBaseVariableName(), extensionObject);
    }
}