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

    • io.nosqlbench.engine.api.activityconfig.rawyaml.StatementsOwner 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.*;

public class StatementsOwner extends RawStmtFields {

    private List rawStmtDefs = new ArrayList<>();

    public StatementsOwner() {
    }

    public List getRawStmtDefs() {
        return rawStmtDefs;
    }

    public void setRawStmtDefs(List rawStmtDefs) {
        this.rawStmtDefs = rawStmtDefs;
    }

    public void setFieldsByReflection(Map propsmap) {
        Object statementsObject = propsmap.remove("statements");

        if (statementsObject==null) {
            statementsObject = propsmap.remove("statement");
        }

        if (statementsObject!=null) {
            setStatementsFieldByObjectType(statementsObject);
        }
//        if (statementsObject!=null) {
//            if (statementsObject instanceof List) {
//                setByObject(statementsObject);
//            } else {
//                throw new RuntimeException("Invalid type for statements property: " + statementsObject.getClass().getCanonicalName());
//            }
//        }
        super.setFieldsByReflection(propsmap);
    }

    @SuppressWarnings("unchecked")
    public void setStatementsFieldByObjectType(Object object) {
        if (object instanceof List) {
            List stmtList = (List) object;
            List defs = new ArrayList<>(stmtList.size());
            for (int i = 0; i < stmtList.size(); i++) {
                String defaultName = "stmt"+(i+1);
                Object o = stmtList.get(i);
                if (o instanceof String) {
                    defs.add(new RawStmtDef(defaultName,(String)o));
                } else if (o instanceof Map) {
                    defs.add(new RawStmtDef(defaultName,(Map)o));
                } else {
                    throw new RuntimeException("Can not construct stmt def from object type:" + o.getClass());
                }
            }
            this.setRawStmtDefs(defs);
        } else if (object instanceof Map) {
            Map map = (Map) object;
            List> itemizedMaps = new ArrayList<>();
            for (Map.Entry entries : map.entrySet()) {
                Object value = entries.getValue();
                if (value instanceof Map) {
                    Map valueMap = ((Map)value);
                    valueMap.put("name", entries.getKey());
                    itemizedMaps.add(valueMap);
                } else if (value instanceof String) {
                    Map stmtDetails = new HashMap<>() {{
                        put("name", entries.getKey());
                        put("stmt", entries.getValue());
                    }};
                    itemizedMaps.add(stmtDetails);
                } else {
                    throw new RuntimeException("Unknown inner value type on map-based statement definition.");
                }
            }
            setStatementsFieldByObjectType(itemizedMaps);
        } else if (object instanceof String) {
            setStatementsFieldByObjectType(Map.of("stmt1",(String)object));
        } else {
            throw new RuntimeException("Unknown object type: " + object.getClass());
        }
    }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy