org.neo4j.jdbc.bolt.BoltDatabaseMetaData 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 01/03/2016
 */
package org.neo4j.jdbc.bolt;

import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.List;

import org.neo4j.driver.v1.Record;
import org.neo4j.driver.v1.Session;
import org.neo4j.driver.v1.StatementResult;
import org.neo4j.jdbc.DatabaseMetaData;
import org.neo4j.jdbc.metadata.Column;
import org.neo4j.jdbc.metadata.Table;

/**
 * Provides metadata
 *
 * @author AgileLARUS	
 * @since 3.0.0
 */
class BoltDatabaseMetaData extends DatabaseMetaData {

	public BoltDatabaseMetaData(BoltConnection connection) {
		this(connection, false);
	}

	private BoltDatabaseMetaData(BoltConnection connection, boolean debug) {
		super(connection, debug);

		// compute database metadata: version, tables == labels, columns = properties (by label)   
		if (connection != null) {
			try {
				BoltConnection conn = (BoltConnection) DriverManager.getConnection(connection.getUrl(), connection.getProperties());
				Session session = conn.getSession();
				getDatabaseVersion(session);
				getDatabaseLabels(session);
				getDatabaseProperties(session);
				conn.close();
			} catch (SQLException e) {
				//e.printStackTrace();
			}
		}
	}

	private void getDatabaseVersion(Session session) {
		StatementResult rs = session.run("CALL dbms.components() yield name,versions WITH * WHERE name=\"Neo4j Kernel\" RETURN versions[0] AS version");
		if (rs != null && rs.hasNext()) {
			Record record = rs.next();
			if (record.containsKey("version")) {
				databaseVersion = record.get("version").asString();
			}
		}
	}
	
	private void getDatabaseLabels(Session session) {
		StatementResult rs = session.run("CALL db.labels() yield label return label");
		if (rs != null) {
			while (rs.hasNext()) {
				Record record = rs.next();
				this.databaseLabels.add(new Table(record.get("label").asString()));
		  }
		}
	}
	
	private void getDatabaseProperties(Session session) {
		if (this.databaseLabels != null) {
			for (Table databaseLabel : this.databaseLabels) {
				StatementResult rs = session.run("MATCH (n:" + databaseLabel.getTableName() + ") WITH n LIMIT " + DatabaseMetaData.PROPERTY_SAMPLE_SIZE + " UNWIND keys(n) as key RETURN collect(distinct key) as keys");
				if (rs != null) {
					while (rs.hasNext()) {
						Record record = rs.next();
						List