org.neo4j.jdbc.Driver Maven / Gradle / Ivy

Go to download
/*
 *
 * Copyright (c) 2016 LARUS Business Automation [http://www.larus-ba.it]
 * 
 * This file is part of the "LARUS Integration Framework for Neo4j".
 * 

 * The "LARUS Integration Framework for Neo4j" is 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.
 * 
 * Created on 19/4/2016
 *
 */

package org.neo4j.jdbc;

import org.neo4j.jdbc.bolt.BoltDriver;
import org.neo4j.jdbc.boltrouting.BoltRoutingNeo4jDriver;
import org.neo4j.jdbc.http.HttpDriver;

import java.lang.reflect.Constructor;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;

public class Driver extends Neo4jDriver {

	/**
	 * Prefix/class hashMap of all available Driver.
	 */
	@SuppressWarnings({ "rawtypes", "serial" })
	private static final Map DRIVERS = new HashMap<>();

	static {
		DRIVERS.put(BoltRoutingNeo4jDriver.JDBC_BOLT_ROUTING_PREFIX, BoltRoutingNeo4jDriver.class);
		DRIVERS.put(BoltDriver.JDBC_BOLT_PREFIX, BoltDriver.class);
		DRIVERS.put(HttpDriver.JDBC_HTTP_PREFIX, HttpDriver.class);
	}

	/**
	 * Default constructor.
	 * @throws SQLException sqlexception
	 */
	public Driver() throws SQLException {
		super(null);
	}

	@Override public Connection connect(String url, Properties info) throws SQLException {
		Connection connection = null;
		if(null != getDriver(url)) {
			connection = getDriver(url).connect(url, info);
		}
		return connection;
	}

	/**
	 * Retrieve the corresponding driver from the JDBC url.
	 * @param url The JDBC url
	 * @return The driver
	 */
	@SuppressWarnings({ "rawtypes", "unchecked" })
	private Neo4jDriver getDriver(String url) throws SQLException {
		Neo4jDriver driver = null;

		if (url == null) {
			throw new SQLException("null is not a valid url");
		}

		// We search the driver prefix from the url
		if (url.startsWith(JDBC_PREFIX)) {
			String[] pieces = url.split(":");
			if (pieces.length > 3) {
				String prefix = pieces[2];

				// We look into driver map is it known
				driver = getDriverForPrefix(prefix);
			}
		}
		return driver;
	}

	private Neo4jDriver getDriverForPrefix(String prefix) throws SQLException {
		Neo4jDriver driver = null;
		try {
			for (Map.Entry entry : DRIVERS.entrySet()) {
				if (prefix.matches(entry.getKey())) {
					Constructor constructor = entry.getValue().getDeclaredConstructor();
					driver = (Neo4jDriver) constructor.newInstance();
				}
			}
		} catch (Exception e) {
			throw new SQLException(e);
		}
		return driver;
	}
}