All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • io.nosqlbench.engine.api.activityconfig.rawyaml.RawStmtsDoc Maven / Gradle / Ivy

    Go to download

    The engine API for nosqlbench; Provides the interfaces needed to build internal modules for the nosqlbench core engine

    There is a newer version: 5.17.0
    Show newest version
    /*
 *
 *    Copyright 2016 jshook
 *    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 java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * A statements doc can have both a list of statement blocks and/or a
 * list of statements. It can also have all the block parameters
 * assignable to {@link RawStmtFields}.
 * 
    
 * The reason for having support both statements or statement blocks
 * is merely convenience. If you do not need or want to deal with the
 * full blocks format, the extra structure gets in the way.
 */
public class RawStmtsDoc extends StatementsOwner {

    private RawScenarios scenarios = new RawScenarios();
    private final List blocks = new ArrayList<>();

    // no-args ctor is required
    public RawStmtsDoc() {
    }

    public static RawStmtsDoc forSingleStatement(String statement) {
        RawStmtsDoc rawStmtsDoc = new RawStmtsDoc();
        rawStmtsDoc.setStatementsFieldByType(statement);
        return rawStmtsDoc;
    }

    public void setFieldsByReflection(Map properties) {
        Object blocksObjects = properties.remove("blocks");
        if (blocksObjects instanceof List) {
            List blockList = ((List) blocksObjects);
            for (Object blockData : blockList) {
                if (blockData instanceof Map) {
                    Map blockDataMap = (Map) blockData;
                    RawStmtsBlock rawStmtsBlock = new RawStmtsBlock();
                    rawStmtsBlock.setFieldsByReflection(blockDataMap);
                    blocks.add(rawStmtsBlock);
                } else {
                    throw new RuntimeException("Invalid object type for block data: " + blockData.getClass().getCanonicalName());
                }
            }
        } else if (blocksObjects instanceof Map) {
            Map blockDataAsMap = (Map) blocksObjects;
            for (Map.Entry entry : blockDataAsMap.entrySet()) {
                String blockName = entry.getKey();
                Object blockData = entry.getValue();
                if (blockData instanceof Map) {
                    Map blockDataMap = (Map) blockData;
                    RawStmtsBlock rawStmtsBlock = new RawStmtsBlock();
                    rawStmtsBlock.setName(blockName);
                    rawStmtsBlock.setFieldsByReflection(blockDataMap);
                    blocks.add(rawStmtsBlock);
                } else {
                    throw new RuntimeException("Invalid object type for block data: " + blockData.getClass().getCanonicalName());
                }

            }
        } else if (blocksObjects != null) {
            throw new RuntimeException("Type of blocks interior data type not recognized:" + blocksObjects.getClass().getCanonicalName());
        }

        Object scenariosData = properties.remove("scenarios");

        if (scenariosData != null) {
            scenarios.setPropertiesByReflection(scenariosData);
        }

        super.setFieldsByReflection(properties);

    }

    /**
     * Return the list of statement blocks in this RawStmtsDoc.
     * If raw statements are defined on this RawStmtsDoc, then a single
     * StmtBlock containing those statements is prepended to the block list.
     * Otherwise, the list of StmtBlocks is returned as-is.
     *
     * @return all logical statement blocks containing statements
     */
    public List getBlocks() {
        List stmtBlocks = new ArrayList<>();
        if (!getRawStmtDefs().isEmpty()) {
            RawStmtsBlock rawStmtsBlock = new RawStmtsBlock();
            rawStmtsBlock.setName("block0");
            rawStmtsBlock.setRawStmtDefs(getRawStmtDefs());
            stmtBlocks.add(rawStmtsBlock);
        }
        stmtBlocks.addAll(this.blocks);
        return stmtBlocks;
    }

    public void setBlocks(List blocks) {
        this.blocks.clear();
        this.blocks.addAll(blocks);
    }

    public RawScenarios getRawScenarios() {
        return this.scenarios;
    }

    public void setScenarios(RawScenarios scenarios) {
        this.scenarios = scenarios;
    }

}
    
    
    
    

    




    
    
    © 2015 - 2024 Weber Informatics LLC | Privacy Policy