io.nosqlbench.engine.api.activityconfig.rawyaml.RawScenarios Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of adapters-api Show documentation

The DriverAdapter API for nosqlbench; Provides the interfaces needed to build drivers that can be loaded by nosqlbench core, using a minimal and direct API for op mapping.

There is a newer version: 5.17.0

Show newest version

/*
 * Copyright (c) 2022 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.activityconfig.rawyaml;

import io.nosqlbench.api.errors.BasicError;

import java.util.*;

public class RawScenarios {

    public static String STEPNAME = "%03d";
    private final Map> scenarios = new LinkedHashMap<>();

    public List getScenarioNames() {
        return new ArrayList<>(scenarios.keySet());
    }

    public void setPropertiesByReflection(Object scenariosObject) {
        scenarios.clear();

        Objects.requireNonNull(scenariosObject);
        if (scenariosObject instanceof Map) {
            Map rawNamedScenarios = (Map) scenariosObject;
            for (Map.Entry namedEntry : rawNamedScenarios.entrySet()) {
                String scenarioName = namedEntry.getKey();
                Object scenarioObj = namedEntry.getValue();
                if (scenarioObj == null) {
                    throw new BasicError("Unable to use a null value for scenario named " + scenarioName + " in yaml.");
                }
                if (scenarioObj instanceof CharSequence) {
                    scenarios.put(scenarioName, Map.of(String.format(STEPNAME, 1), scenarioObj.toString()));
                } else if (scenarioObj instanceof List) {
                    List list = (List) scenarioObj;
                    Map scenarioMap = new LinkedHashMap<>();
                    for (int i = 0; i < list.size(); i++) {
                        scenarioMap.put(String.format(STEPNAME, i), list.get(i));
                    }
                    scenarios.put(scenarioName, scenarioMap);
                } else if (scenarioObj instanceof Map) {
                    scenarios.put(scenarioName, (Map)scenarioObj);
                }
            }
        } else {
            throw new RuntimeException("Named scenarios must be a map at the top level, instead found '" + scenariosObject.getClass().getCanonicalName() + "'");
        }
    }

    public Map getNamedScenario(String scenarioName) {
        return scenarios.get(scenarioName);
    }
}