spade.query.neo4j.GetVertex.java Source code

Introduction

Here is the source code for spade.query.neo4j.GetVertex.java

Source

/*
 --------------------------------------------------------------------------------
 SPADE - Support for Provenance Auditing in Distributed Environments.
 Copyright (C) 2017 SRI International
 This program is free software: you can redistribute it and/or
 modify it under the terms of the GNU General Public License as
 published by the Free Software Foundation, either version 3 of the
 License, or (at your option) any later version.
 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 General Public License for more details.
 You should have received a copy of the GNU General Public License
 along with this program. If not, see <http://www.gnu.org/licenses/>.
 --------------------------------------------------------------------------------
 */
package spade.query.neo4j;

import org.apache.commons.collections.CollectionUtils;
import spade.core.AbstractVertex;
import spade.core.Graph;

import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.regex.Pattern;

/**
 * @author raza
 */
public class GetVertex extends Neo4j<Set<AbstractVertex>> {
    @Override
    public Set<AbstractVertex> execute(String argument_string) {
        Pattern argument_pattern = Pattern.compile(",");
        String[] arguments = argument_pattern.split(argument_string);
        String constraints = arguments[0].trim();
        Map<String, List<String>> parameters = parseConstraints(constraints);
        Integer limit = null;
        if (arguments.length > 1)
            limit = Integer.parseInt(arguments[1].trim());

        return execute(parameters, limit);
    }

    @Override
    public Set<AbstractVertex> execute(Map<String, List<String>> parameters, Integer limit) {
        Set<AbstractVertex> vertexSet = null;
        try {
            StringBuilder vertexQueryBuilder = new StringBuilder(50);
            for (Map.Entry<String, List<String>> entry : parameters.entrySet()) {
                String colName = entry.getKey();
                List<String> values = entry.getValue();
                vertexQueryBuilder.append(colName);
                vertexQueryBuilder.append(":");
                vertexQueryBuilder.append(values.get(COL_VALUE));
                String boolOperator = values.get(BOOLEAN_OPERATOR);
                if (boolOperator != null) {
                    vertexQueryBuilder.append(boolOperator);
                }
            }
            spade.storage.Neo4j neo4jStorage = (spade.storage.Neo4j) currentStorage;
            Graph result = neo4jStorage.getVertices(vertexQueryBuilder.toString());
            vertexSet = result.vertexSet();
            if (!CollectionUtils.isEmpty(vertexSet))
                return vertexSet;
        } catch (Exception ex) {
            Logger.getLogger(GetVertex.class.getName()).log(Level.SEVERE, "Error creating vertex set!", ex);
        }

        return vertexSet;
    }
}