hu.bme.mit.trainbenchmark.benchmark.neo4j.checkers.Neo4jCypherChecker.java Source code

Introduction

Here is the source code for hu.bme.mit.trainbenchmark.benchmark.neo4j.checkers.Neo4jCypherChecker.java

Source

/*******************************************************************************
 * Copyright (c) 2010-2015, Benedek Izso, Gabor Szarnyas, Istvan Rath and Daniel Varro
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *   Benedek Izso - initial API and implementation
 *   Gabor Szarnyas - initial API and implementation
 *******************************************************************************/
package hu.bme.mit.trainbenchmark.benchmark.neo4j.checkers;

import hu.bme.mit.trainbenchmark.benchmark.checker.Checker;
import hu.bme.mit.trainbenchmark.benchmark.config.BenchmarkConfig;
import hu.bme.mit.trainbenchmark.benchmark.neo4j.driver.Neo4jDriver;
import hu.bme.mit.trainbenchmark.benchmark.neo4j.matches.Neo4jMatch;
import hu.bme.mit.trainbenchmark.constants.Query;

import java.io.File;
import java.io.IOException;
import java.util.Collection;

import org.apache.commons.io.FileUtils;

public class Neo4jCypherChecker extends Checker<Neo4jMatch> {

    protected final Neo4jDriver driver;
    protected final Query query;
    protected final String queryDefinition;

    protected Neo4jCypherChecker(final Neo4jDriver driver, final BenchmarkConfig bc) throws IOException {
        super();
        this.driver = driver;

        query = bc.getQuery();
        queryDefinition = FileUtils.readFileToString(new File(
                bc.getWorkspacePath() + "/hu.bme.mit.trainbenchmark.benchmark.neo4j/src/main/resources/queries/"
                        + bc.getQuery() + ".cypher"));
    }

    @Override
    public Collection<Neo4jMatch> check() throws IOException {
        return driver.runQuery(query, queryDefinition);
    }

    public static Checker<Neo4jMatch> newInstance(final Neo4jDriver neoDriver, final BenchmarkConfig bc)
            throws IOException {
        return new Neo4jCypherChecker(neoDriver, bc);
    }

}