org.neo4j.jdbc.bolt.BoltNeo4jUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of neo4j-jdbc-bolt Show documentation
Bolt implementation for the Neo4j JDBC Driver
There is a newer version: 5.0.0
package org.neo4j.jdbc.bolt;

import org.apache.commons.lang3.exception.ExceptionUtils;
import org.neo4j.driver.Result;
import org.neo4j.driver.Transaction;
import org.neo4j.driver.summary.SummaryCounters;

import java.sql.SQLException;
import java.util.Collections;
import java.util.Map;
import java.util.function.Function;
import java.util.logging.Logger;

/**
 * A set of common functions for bolt connector
 */
public class BoltNeo4jUtils {

    /**
     * Calculate, using the summary, how many operations are executed in the statement
     * @param stats
     * @return
     */
    public static int calculateUpdateCount(SummaryCounters stats) {

		/*
		Neo4j has a detailed report to explain which entity is changed.
		But we need to collect all that value to only one.
		So a priority is defined:
		1 - nodes and relationships creation or deletion
		2 - schema changes
		3 - properties changes
		 */

        int objectCount = 0;
        objectCount += stats.nodesCreated();
        objectCount += stats.nodesDeleted();
        objectCount += stats.relationshipsCreated();
        objectCount += stats.relationshipsDeleted();

        int schemaCount = 0;
        schemaCount += stats.constraintsAdded();
        schemaCount += stats.constraintsRemoved();
        schemaCount += stats.indexesAdded();
        schemaCount += stats.indexesRemoved();
        schemaCount += stats.labelsAdded();
        schemaCount += stats.labelsRemoved();

        int updateCount = (objectCount == 0)?schemaCount:objectCount;

        return (updateCount == 0)?stats.propertiesSet():updateCount;
    }

    public static  R executeInTx(BoltNeo4jConnection connection,
                                    String sql,
                                    Map params,
                                    Function body) throws SQLException {
        try {
            Result statementResult = execute(connection, sql, params);
            R result = body.apply(statementResult);
            if (connection.getAutoCommit()) {
                connection.doCommit();
            }
            return result;
        }  catch (Exception e) {
            connection.doRollback();
            throw new SQLException(e);
        }
    }

    public static  R executeInTx(BoltNeo4jConnection connection,
                                    String sql,
                                    Function body) throws SQLException {
        return executeInTx(connection, sql, Collections.emptyMap(), body);
    }

    public static Result execute(BoltNeo4jConnection connection,
                                          String statement,
                                          Map params) {
        Transaction transaction = connection.getTransaction();
        return transaction.run(statement, params);
    }

    public static void closeSafely(AutoCloseable closeable, Logger logger) {
        if (closeable == null) {
            return;
        }
        try {
            closeable.close();
        } catch (Exception e) {
            if (logger == null) {
                return;
            }
            logger.warning("Exception while trying to close an AutoCloseable, because of the following exception: " +
                    ExceptionUtils.getStackTrace(e));
        }
    }

}