/*
 * Copyright 2015-2022 Transmogrify LLC, 2022-2026 Revetware LLC.
 *
 * 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.
 */

package com.pyranid;

import org.jspecify.annotations.NonNull;
import org.jspecify.annotations.Nullable;

import javax.annotation.concurrent.NotThreadSafe;
import javax.annotation.concurrent.ThreadSafe;
import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.ParameterMetaData;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.SQLFeatureNotSupportedException;
import java.sql.Types;
import java.time.Duration;
import java.time.ZoneId;
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Deque;
import java.util.HashSet;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Optional;
import java.util.OptionalDouble;
import java.util.OptionalInt;
import java.util.OptionalLong;
import java.util.Queue;
import java.util.Set;
import java.util.Spliterator;
import java.util.Spliterators;
import java.util.concurrent.atomic.AtomicLong;
import java.util.concurrent.locks.ReentrantLock;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.logging.Logger;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;

import static java.lang.String.format;
import static java.lang.System.nanoTime;
import static java.util.Objects.requireNonNull;
import static java.util.logging.Level.WARNING;

/**
 * Main class for performing database access operations.
 *
 * @author <a href="https://www.revetkn.com">Mark Allen</a>
 * @since 1.0.0
 */
@ThreadSafe
public final class Database {
        @NonNull
        private static final ThreadLocal<Deque<Transaction>> TRANSACTION_STACK_HOLDER;
        private static final int DEFAULT_PARSED_SQL_CACHE_CAPACITY = 1024;

        static {
                TRANSACTION_STACK_HOLDER = ThreadLocal.withInitial(() -> new ArrayDeque<>());
        }

        @NonNull
        private final DataSource dataSource;
        @NonNull
        private final DatabaseType databaseType;
        @NonNull
        private final ZoneId timeZone;
        @NonNull
        private final InstanceProvider instanceProvider;
        @NonNull
        private final PreparedStatementBinder preparedStatementBinder;
        @NonNull
        private final ResultSetMapper resultSetMapper;
        @NonNull
        private final StatementLogger statementLogger;
        @Nullable
        private final Map<String, ParsedSql> parsedSqlCache;
        @NonNull
        private final AtomicLong defaultIdGenerator;
        @NonNull
        private final Logger logger;

        @NonNull
        private volatile DatabaseOperationSupportStatus executeLargeBatchSupported;
        @NonNull
        private volatile DatabaseOperationSupportStatus executeLargeUpdateSupported;

        protected Database(@NonNull Builder builder) {
                requireNonNull(builder);

                this.dataSource = requireNonNull(builder.dataSource);
                this.databaseType = builder.databaseType == null ? DatabaseType.fromDataSource(builder.dataSource) : builder.databaseType;
                this.timeZone = builder.timeZone == null ? ZoneId.systemDefault() : builder.timeZone;
                this.instanceProvider = builder.instanceProvider == null ? new InstanceProvider() {} : builder.instanceProvider;
                this.preparedStatementBinder = builder.preparedStatementBinder == null ? PreparedStatementBinder.withDefaultConfiguration() : builder.preparedStatementBinder;
                this.resultSetMapper = builder.resultSetMapper == null ? ResultSetMapper.withDefaultConfiguration() : builder.resultSetMapper;
                this.statementLogger = builder.statementLogger == null ? (statementLog) -> {} : builder.statementLogger;
                if (builder.parsedSqlCacheCapacity != null && builder.parsedSqlCacheCapacity < 0)
                        throw new IllegalArgumentException("parsedSqlCacheCapacity must be >= 0");

                int parsedSqlCacheCapacity = builder.parsedSqlCacheCapacity == null
                                ? DEFAULT_PARSED_SQL_CACHE_CAPACITY
                                : builder.parsedSqlCacheCapacity;
                this.parsedSqlCache = parsedSqlCacheCapacity == 0 ? null : new ConcurrentLruMap<>(parsedSqlCacheCapacity);
                this.defaultIdGenerator = new AtomicLong();
                this.logger = Logger.getLogger(getClass().getName());
                this.executeLargeBatchSupported = DatabaseOperationSupportStatus.UNKNOWN;
                this.executeLargeUpdateSupported = DatabaseOperationSupportStatus.UNKNOWN;
        }

        /**
         * Provides a {@link Database} builder for the given {@link DataSource}.
         *
         * @param dataSource data source used to create the {@link Database} builder
         * @return a {@link Database} builder
         */
        @NonNull
        public static Builder withDataSource(@NonNull DataSource dataSource) {
                requireNonNull(dataSource);
                return new Builder(dataSource);
        }

        /**
         * Gets a reference to the current transaction, if any.
         *
         * @return the current transaction
         */
        @NonNull
        public Optional<Transaction> currentTransaction() {
                Deque<Transaction> transactionStack = TRANSACTION_STACK_HOLDER.get();
                return Optional.ofNullable(transactionStack.isEmpty() ? null : transactionStack.peek());
        }

        /**
         * Performs an operation transactionally.
         * <p>
         * The transaction will be automatically rolled back if an exception bubbles out of {@code transactionalOperation}.
         *
         * @param transactionalOperation the operation to perform transactionally
         */
        public void transaction(@NonNull TransactionalOperation transactionalOperation) {
                requireNonNull(transactionalOperation);

                transaction(() -> {
                        transactionalOperation.perform();
                        return Optional.empty();
                });
        }

        /**
         * Performs an operation transactionally with the given isolation level.
         * <p>
         * The transaction will be automatically rolled back if an exception bubbles out of {@code transactionalOperation}.
         *
         * @param transactionIsolation   the desired database transaction isolation level
         * @param transactionalOperation the operation to perform transactionally
         */
        public void transaction(@NonNull TransactionIsolation transactionIsolation,
                                                                                                        @NonNull TransactionalOperation transactionalOperation) {
                requireNonNull(transactionIsolation);
                requireNonNull(transactionalOperation);

                transaction(transactionIsolation, () -> {
                        transactionalOperation.perform();
                        return Optional.empty();
                });
        }

        /**
         * Performs an operation transactionally and optionally returns a value.
         * <p>
         * The transaction will be automatically rolled back if an exception bubbles out of {@code transactionalOperation}.
         *
         * @param transactionalOperation the operation to perform transactionally
         * @param <T>                    the type to be returned
         * @return the result of the transactional operation
         */
        @NonNull
        public <T> Optional<T> transaction(@NonNull ReturningTransactionalOperation<T> transactionalOperation) {
                requireNonNull(transactionalOperation);
                return transaction(TransactionIsolation.DEFAULT, transactionalOperation);
        }

        /**
         * Performs an operation transactionally with the given isolation level, optionally returning a value.
         * <p>
         * The transaction will be automatically rolled back if an exception bubbles out of {@code transactionalOperation}.
         *
         * @param transactionIsolation   the desired database transaction isolation level
         * @param transactionalOperation the operation to perform transactionally
         * @param <T>                    the type to be returned
         * @return the result of the transactional operation
         */
        @NonNull
        public <T> Optional<T> transaction(@NonNull TransactionIsolation transactionIsolation,
                                                                                                                                                 @NonNull ReturningTransactionalOperation<T> transactionalOperation) {
                requireNonNull(transactionIsolation);
                requireNonNull(transactionalOperation);

                Transaction transaction = new Transaction(dataSource, transactionIsolation);
                TRANSACTION_STACK_HOLDER.get().push(transaction);
                boolean committed = false;
                Throwable thrown = null;

                try {
                        Optional<T> returnValue = transactionalOperation.perform();

                        // Safeguard in case user code accidentally returns null instead of Optional.empty()
                        if (returnValue == null)
                                returnValue = Optional.empty();

                        if (transaction.isRollbackOnly()) {
                                transaction.rollback();
                        } else {
                                transaction.commit();
                                committed = true;
                        }

                        return returnValue;
                } catch (RuntimeException e) {
                        thrown = e;
                        try {
                                transaction.rollback();
                        } catch (Exception rollbackException) {
                                logger.log(WARNING, "Unable to roll back transaction", rollbackException);
                        }

                        restoreInterruptIfNeeded(e);
                        throw e;
                } catch (Error e) {
                        thrown = e;
                        try {
                                transaction.rollback();
                        } catch (Exception rollbackException) {
                                logger.log(WARNING, "Unable to roll back transaction", rollbackException);
                        }

                        restoreInterruptIfNeeded(e);
                        throw e;
                } catch (Throwable t) {
                        try {
                                transaction.rollback();
                        } catch (Exception rollbackException) {
                                logger.log(WARNING, "Unable to roll back transaction", rollbackException);
                        }

                        restoreInterruptIfNeeded(t);
                        RuntimeException wrapped = new RuntimeException(t);
                        thrown = wrapped;
                        throw wrapped;
                } finally {
                        Deque<Transaction> transactionStack = TRANSACTION_STACK_HOLDER.get();

                        transactionStack.pop();

                        // Ensure txn stack is fully cleaned up
                        if (transactionStack.isEmpty())
                                TRANSACTION_STACK_HOLDER.remove();

                        Throwable cleanupFailure = null;

                        try {
                                try {
                                        transaction.restoreTransactionIsolationIfNeeded();

                                        if (transaction.getInitialAutoCommit().isPresent() && transaction.getInitialAutoCommit().get())
                                                // Autocommit was true initially, so restoring to true now that transaction has completed
                                                transaction.setAutoCommit(true);
                                } catch (Throwable cleanupException) {
                                        cleanupFailure = cleanupException;
                                } finally {
                                        if (transaction.hasConnection()) {
                                                try {
                                                        closeConnection(transaction.getConnection());
                                                } catch (Throwable cleanupException) {
                                                        if (cleanupFailure == null)
                                                                cleanupFailure = cleanupException;
                                                        else
                                                                cleanupFailure.addSuppressed(cleanupException);
                                                }
                                        }
                                }
                        } finally {
                                // Execute any user-supplied post-execution hooks
                                for (Consumer<TransactionResult> postTransactionOperation : transaction.getPostTransactionOperations()) {
                                        try {
                                                postTransactionOperation.accept(committed ? TransactionResult.COMMITTED : TransactionResult.ROLLED_BACK);
                                        } catch (Throwable cleanupException) {
                                                if (cleanupFailure == null)
                                                        cleanupFailure = cleanupException;
                                                else
                                                        cleanupFailure.addSuppressed(cleanupException);
                                        }
                                }
                        }

                        if (cleanupFailure != null) {
                                if (thrown != null) {
                                        thrown.addSuppressed(cleanupFailure);
                                } else if (cleanupFailure instanceof RuntimeException) {
                                        throw (RuntimeException) cleanupFailure;
                                } else if (cleanupFailure instanceof Error) {
                                        throw (Error) cleanupFailure;
                                } else {
                                        throw new RuntimeException(cleanupFailure);
                                }
                        }
                }
        }

        protected void closeConnection(@NonNull Connection connection) {
                requireNonNull(connection);

                try {
                        connection.close();
                } catch (SQLException e) {
                        throw new DatabaseException("Unable to close database connection", e);
                }
        }

        /**
         * Performs an operation in the context of a pre-existing transaction.
         * <p>
         * No commit or rollback on the transaction will occur when {@code transactionalOperation} completes.
         * <p>
         * However, if an exception bubbles out of {@code transactionalOperation}, the transaction will be marked as rollback-only.
         *
         * @param transaction            the transaction in which to participate
         * @param transactionalOperation the operation that should participate in the transaction
         */
        public void participate(@NonNull Transaction transaction,
                                                                                                        @NonNull TransactionalOperation transactionalOperation) {
                requireNonNull(transaction);
                requireNonNull(transactionalOperation);

                participate(transaction, () -> {
                        transactionalOperation.perform();
                        return Optional.empty();
                });
        }

        /**
         * Performs an operation in the context of a pre-existing transaction, optionall returning a value.
         * <p>
         * No commit or rollback on the transaction will occur when {@code transactionalOperation} completes.
         * <p>
         * However, if an exception bubbles out of {@code transactionalOperation}, the transaction will be marked as rollback-only.
         *
         * @param transaction            the transaction in which to participate
         * @param transactionalOperation the operation that should participate in the transaction
         * @param <T>                    the type to be returned
         * @return the result of the transactional operation
         */
        @NonNull
        public <T> Optional<T> participate(@NonNull Transaction transaction,
                                                                                                                                                 @NonNull ReturningTransactionalOperation<T> transactionalOperation) {
                requireNonNull(transaction);
                requireNonNull(transactionalOperation);

                Deque<Transaction> transactionStack = TRANSACTION_STACK_HOLDER.get();
                transactionStack.push(transaction);

                try {
                        Optional<T> returnValue = transactionalOperation.perform();
                        return returnValue == null ? Optional.empty() : returnValue;
                } catch (RuntimeException e) {
                        if (!(e instanceof StatementLoggerFailureException))
                                transaction.setRollbackOnly(true);
                        restoreInterruptIfNeeded(e);
                        throw e;
                } catch (Error e) {
                        transaction.setRollbackOnly(true);
                        restoreInterruptIfNeeded(e);
                        throw e;
                } catch (Throwable t) {
                        transaction.setRollbackOnly(true);
                        restoreInterruptIfNeeded(t);
                        throw new RuntimeException(t);
                } finally {
                        try {
                                transactionStack.pop();
                        } finally {
                                if (transactionStack.isEmpty())
                                        TRANSACTION_STACK_HOLDER.remove();
                        }
                }
        }

        /**
         * Creates a fluent builder for executing SQL.
         * <p>
         * Named parameters use the {@code :paramName} syntax and are bound via {@link Query#bind(String, Object)}.
         * Positional parameters via {@code ?} are not supported.
         * <p>
         * Example:
         * <pre>{@code
         * Optional<Employee> employee = database.query("SELECT * FROM employee WHERE id = :id")
         *   .bind("id", 42)
         *   .fetchObject(Employee.class);
         * }</pre>
         *
         * @param sql SQL containing {@code :paramName} placeholders
         * @return a fluent builder for binding parameters and executing
         * @since 4.0.0
         */
        @NonNull
        public Query query(@NonNull String sql) {
                requireNonNull(sql);
                return new DefaultQuery(this, sql);
        }

        private static void restoreInterruptIfNeeded(@NonNull Throwable throwable) {
                requireNonNull(throwable);

                Throwable current = throwable;

                while (current != null) {
                        if (current instanceof InterruptedException) {
                                Thread.currentThread().interrupt();
                                return;
                        }

                        current = current.getCause();
                }
        }

        @Nullable
        private static Object unwrapOptionalValue(@Nullable Object value) {
                if (value == null)
                        return null;

                if (value instanceof Optional<?> optional)
                        return optional.orElse(null);
                if (value instanceof OptionalInt optionalInt)
                        return optionalInt.isPresent() ? optionalInt.getAsInt() : null;
                if (value instanceof OptionalLong optionalLong)
                        return optionalLong.isPresent() ? optionalLong.getAsLong() : null;
                if (value instanceof OptionalDouble optionalDouble)
                        return optionalDouble.isPresent() ? optionalDouble.getAsDouble() : null;

                return value;
        }

        @Nullable
        private static Throwable closeStatementContextResources(@NonNull StatementContext<?> statementContext,
                                                                                                                                                                                                                                        @Nullable Throwable cleanupFailure) {
                requireNonNull(statementContext);

                Queue<AutoCloseable> cleanupOperations = statementContext.getCleanupOperations();
                AutoCloseable cleanupOperation;

                while ((cleanupOperation = cleanupOperations.poll()) != null) {
                        try {
                                cleanupOperation.close();
                        } catch (Throwable cleanupException) {
                                if (cleanupFailure == null)
                                        cleanupFailure = cleanupException;
                                else
                                        cleanupFailure.addSuppressed(cleanupException);
                        }
                }

                return cleanupFailure;
        }

        private static boolean isUnsupportedSqlFeature(@NonNull SQLException e) {
                requireNonNull(e);

                String sqlState = e.getSQLState();
                if (sqlState != null) {
                        if (sqlState.startsWith("0A") || "HYC00".equals(sqlState))
                                return true;
                }

                Throwable cause = e.getCause();
                if (cause instanceof SQLFeatureNotSupportedException
                                || cause instanceof UnsupportedOperationException
                                || cause instanceof AbstractMethodError)
                        return true;

                String message = e.getMessage();
                if (message == null)
                        return false;

                String lower = message.toLowerCase(Locale.ROOT);
                return lower.contains("not supported")
                                || lower.contains("unsupported")
                                || lower.contains("not implemented")
                                || lower.contains("feature not supported");
        }

        @NonNull
        private ParsedSql getParsedSql(@NonNull String sql) {
                requireNonNull(sql);

                if (this.parsedSqlCache == null)
                        return parseNamedParameterSql(sql);

                return this.parsedSqlCache.computeIfAbsent(sql, Database::parseNamedParameterSql);
        }

        /**
         * Default internal implementation of {@link Query}.
         * <p>
         * This class is intended for use by a single thread.
         */
        @NotThreadSafe
        private static final class DefaultQuery implements Query {
                @NonNull
                private final Database database;
                @NonNull
                private final String originalSql;
                @NonNull
                private final List<String> sqlFragments;
                @NonNull
                private final List<String> parameterNames;
                @NonNull
                private final Set<String> distinctParameterNames;
                @NonNull
                private final Map<String, Object> bindings;
                @Nullable
                private PreparedStatementCustomizer preparedStatementCustomizer;
                @Nullable
                private Object id;

                private DefaultQuery(@NonNull Database database,
                                                                                                 @NonNull String sql) {
                        requireNonNull(database);
                        requireNonNull(sql);

                        this.database = database;
                        this.originalSql = sql;

                        ParsedSql parsedSql = database.getParsedSql(sql);
                        this.sqlFragments = parsedSql.sqlFragments;
                        this.parameterNames = parsedSql.parameterNames;
                        this.distinctParameterNames = parsedSql.distinctParameterNames;

                        this.bindings = new LinkedHashMap<>(Math.max(8, this.distinctParameterNames.size()));
                        this.preparedStatementCustomizer = null;
                }

                @NonNull
                @Override
                public Query bind(@NonNull String name,
                                                                                        @Nullable Object value) {
                        requireNonNull(name);

                        if (!this.distinctParameterNames.contains(name))
                                throw new IllegalArgumentException(format("Unknown named parameter '%s' for SQL: %s", name, this.originalSql));

                        this.bindings.put(name, value);
                        return this;
                }

                @NonNull
                @Override
                public Query bindAll(@NonNull Map<@NonNull String, @Nullable Object> parameters) {
                        requireNonNull(parameters);

                        for (Map.Entry<@NonNull String, @Nullable Object> entry : parameters.entrySet())
                                bind(entry.getKey(), entry.getValue());

                        return this;
                }

                @NonNull
                @Override
                public Query id(@Nullable Object id) {
                        this.id = id;
                        return this;
                }

                @NonNull
                @Override
                public Query customize(@NonNull PreparedStatementCustomizer preparedStatementCustomizer) {
                        requireNonNull(preparedStatementCustomizer);
                        this.preparedStatementCustomizer = preparedStatementCustomizer;

                        return this;
                }

                @NonNull
                @Override
                public <T> Optional<T> fetchObject(@NonNull Class<T> resultType) {
                        requireNonNull(resultType);
                        PreparedQuery preparedQuery = prepare(this.bindings);
                        return this.database.queryForObject(preparedQuery.statement, resultType, this.preparedStatementCustomizer, preparedQuery.parameters);
                }

                @NonNull
                @Override
                public <T> List<@Nullable T> fetchList(@NonNull Class<T> resultType) {
                        requireNonNull(resultType);
                        PreparedQuery preparedQuery = prepare(this.bindings);
                        return this.database.queryForList(preparedQuery.statement, resultType, this.preparedStatementCustomizer, preparedQuery.parameters);
                }

                @Nullable
                @Override
                public <T, R> R fetchStream(@NonNull Class<T> resultType,
                                                                                                                                @NonNull Function<Stream<@Nullable T>, R> streamFunction) {
                        requireNonNull(resultType);
                        requireNonNull(streamFunction);
                        PreparedQuery preparedQuery = prepare(this.bindings);
                        return this.database.queryForStream(preparedQuery.statement, resultType, this.preparedStatementCustomizer, streamFunction, preparedQuery.parameters);
                }


                @NonNull
                @Override
                public Long execute() {
                        PreparedQuery preparedQuery = prepare(this.bindings);
                        return this.database.execute(preparedQuery.statement, this.preparedStatementCustomizer, preparedQuery.parameters);
                }

                @NonNull
                @Override
                public List<Long> executeBatch(@NonNull List<@NonNull Map<@NonNull String, @Nullable Object>> parameterGroups) {
                        requireNonNull(parameterGroups);
                        if (parameterGroups.isEmpty())
                                return List.of();

                        List<List<Object>> parametersAsList = new ArrayList<>(parameterGroups.size());
                        Object statementId = this.id == null ? this.database.generateId() : this.id;
                        Statement statement = null;
                        String expandedSql = null;

                        for (Map<@NonNull String, @Nullable Object> parameterGroup : parameterGroups) {
                                requireNonNull(parameterGroup);

                                for (String parameterName : parameterGroup.keySet())
                                        if (!this.distinctParameterNames.contains(parameterName))
                                                throw new IllegalArgumentException(format("Unknown named parameter '%s' for SQL: %s", parameterName, this.originalSql));

                                Map<String, Object> mergedBindings;
                                if (this.bindings.isEmpty()) {
                                        mergedBindings = parameterGroup;
                                } else if (parameterGroup.isEmpty()) {
                                        mergedBindings = this.bindings;
                                } else {
                                        Map<String, Object> combinedBindings = new LinkedHashMap<>(this.bindings);
                                        combinedBindings.putAll(parameterGroup);
                                        mergedBindings = combinedBindings;
                                }

                                PreparedQuery preparedQuery = prepare(mergedBindings, statementId);

                                if (expandedSql == null) {
                                        expandedSql = preparedQuery.statement.getSql();
                                        statement = preparedQuery.statement;
                                } else if (!expandedSql.equals(preparedQuery.statement.getSql())) {
                                        throw new IllegalArgumentException(format(
                                                        "Inconsistent SQL after expanding parameters for batch execution; ensure collection sizes are consistent. SQL: %s",
                                                        this.originalSql));
                                }

                                parametersAsList.add(Arrays.asList(preparedQuery.parameters));
                        }

                        if (statement == null)
                                statement = Statement.of(statementId, buildPlaceholderSql());

                        return this.database.executeBatch(statement, parametersAsList, this.preparedStatementCustomizer);
                }

                @NonNull
                @Override
                public <T> Optional<T> executeForObject(@NonNull Class<T> resultType) {
                        requireNonNull(resultType);
                        PreparedQuery preparedQuery = prepare(this.bindings);
                        return this.database.executeForObject(preparedQuery.statement, resultType, this.preparedStatementCustomizer, preparedQuery.parameters);
                }

                @NonNull
                @Override
                public <T> List<@Nullable T> executeForList(@NonNull Class<T> resultType) {
                        requireNonNull(resultType);
                        PreparedQuery preparedQuery = prepare(this.bindings);
                        return this.database.executeForList(preparedQuery.statement, resultType, this.preparedStatementCustomizer, preparedQuery.parameters);
                }

                @NonNull
                private PreparedQuery prepare(@NonNull Map<String, Object> bindings) {
                        Object statementId = this.id == null ? this.database.generateId() : this.id;
                        return prepare(bindings, statementId);
                }

                @NonNull
                private PreparedQuery prepare(@NonNull Map<String, Object> bindings,
                                                                                                                                        @NonNull Object statementId) {
                        requireNonNull(bindings);
                        requireNonNull(statementId);

                        if (this.parameterNames.isEmpty())
                                return new PreparedQuery(Statement.of(statementId, this.originalSql), new Object[0]);

                        StringBuilder sql = new StringBuilder(this.originalSql.length() + this.parameterNames.size() * 2);
                        List<String> missingParameterNames = null;
                        List<Object> parameters = new ArrayList<>(this.parameterNames.size());

                        for (int i = 0; i < this.parameterNames.size(); ++i) {
                                String parameterName = this.parameterNames.get(i);
                                sql.append(this.sqlFragments.get(i));

                                if (!bindings.containsKey(parameterName)) {
                                        if (missingParameterNames == null)
                                                missingParameterNames = new ArrayList<>();

                                        missingParameterNames.add(parameterName);
                                        sql.append('?');
                                        continue;
                                }

                                Object value = unwrapOptionalValue(bindings.get(parameterName));

                                if (value instanceof InListParameter inListParameter) {
                                        Object[] elements = inListParameter.getElements();

                                        if (elements.length == 0)
                                                throw new IllegalArgumentException(format("IN-list parameter '%s' for SQL: %s is empty", parameterName, this.originalSql));

                                        appendPlaceholders(sql, elements.length);
                                        parameters.addAll(Arrays.asList(elements));
                                } else if (value instanceof Collection<?>) {
                                        throw new IllegalArgumentException(format(
                                                        "Collection parameter '%s' for SQL: %s must be wrapped with %s.inList(...) or %s.listOf/%s.setOf(...)",
                                                        parameterName, this.originalSql,
                                                        Parameters.class.getSimpleName(),
                                                        Parameters.class.getSimpleName(), Parameters.class.getSimpleName()));
                                } else if (value != null && value.getClass().isArray() && !(value instanceof byte[])) {
                                        throw new IllegalArgumentException(format(
                                                        "Array parameter '%s' for SQL: %s must be wrapped with %s.inList(...), %s.sqlArrayOf(...), or %s.arrayOf(Class, ...)",
                                                        parameterName, this.originalSql,
                                                        Parameters.class.getSimpleName(), Parameters.class.getSimpleName(), Parameters.class.getSimpleName()));
                                } else {
                                        sql.append('?');
                                        parameters.add(value);
                                }
                        }

                        sql.append(this.sqlFragments.get(this.sqlFragments.size() - 1));

                        if (missingParameterNames != null)
                                throw new IllegalArgumentException(format("Missing required named parameters %s for SQL: %s", missingParameterNames, this.originalSql));

                        return new PreparedQuery(Statement.of(statementId, sql.toString()), parameters.toArray());
                }

                @NonNull
                private String buildPlaceholderSql() {
                        if (this.parameterNames.isEmpty())
                                return this.originalSql;

                        StringBuilder sql = new StringBuilder(this.originalSql.length() + this.parameterNames.size() * 2);

                        for (int i = 0; i < this.parameterNames.size(); ++i)
                                sql.append(this.sqlFragments.get(i)).append('?');

                        sql.append(this.sqlFragments.get(this.sqlFragments.size() - 1));
                        return sql.toString();
                }

                private void appendPlaceholders(@NonNull StringBuilder sql,
                                                                                                                                                int count) {
                        requireNonNull(sql);

                        for (int i = 0; i < count; ++i) {
                                if (i > 0)
                                        sql.append(", ");
                                sql.append('?');
                        }
                }

                private static final class PreparedQuery {
                        @NonNull
                        private final Statement statement;
                        @NonNull
                        private final Object @NonNull [] parameters;

                        private PreparedQuery(@NonNull Statement statement,
                                                                                                                Object @NonNull [] parameters) {
                                this.statement = requireNonNull(statement);
                                this.parameters = requireNonNull(parameters);
                        }
                }

        }

        static final class ParsedSql {
                @NonNull
                private final List<String> sqlFragments;
                @NonNull
                private final List<String> parameterNames;
                @NonNull
                private final Set<String> distinctParameterNames;

                private ParsedSql(@NonNull List<String> sqlFragments,
                                                                                        @NonNull List<String> parameterNames,
                                                                                        @NonNull Set<String> distinctParameterNames) {
                        requireNonNull(sqlFragments);
                        requireNonNull(parameterNames);
                        requireNonNull(distinctParameterNames);

                        this.sqlFragments = sqlFragments;
                        this.parameterNames = parameterNames;
                        this.distinctParameterNames = distinctParameterNames;
                }
        }

        @NonNull
        static ParsedSql parseNamedParameterSql(@NonNull String sql) {
                requireNonNull(sql);

                List<String> sqlFragments = new ArrayList<>();
                StringBuilder sqlFragment = new StringBuilder(sql.length());
                List<String> parameterNames = new ArrayList<>();
                Set<String> distinctParameterNames = new HashSet<>();

                boolean inSingleQuote = false;
                boolean inSingleQuoteEscapesBackslash = false;
                boolean inDoubleQuote = false;
                boolean inBacktickQuote = false;
                boolean inBracketQuote = false;
                boolean inLineComment = false;
                boolean inBlockComment = false;
                String dollarQuoteDelimiter = null;

                for (int i = 0; i < sql.length(); ) {
                        if (dollarQuoteDelimiter != null) {
                                if (sql.startsWith(dollarQuoteDelimiter, i)) {
                                        sqlFragment.append(dollarQuoteDelimiter);
                                        i += dollarQuoteDelimiter.length();
                                        dollarQuoteDelimiter = null;
                                } else {
                                        sqlFragment.append(sql.charAt(i));
                                        ++i;
                                }

                                continue;
                        }

                        char c = sql.charAt(i);

                        if (inLineComment) {
                                sqlFragment.append(c);
                                ++i;

                                if (c == '\n' || c == '\r')
                                        inLineComment = false;

                                continue;
                        }

                        if (inBlockComment) {
                                sqlFragment.append(c);

                                if (c == '*' && i + 1 < sql.length() && sql.charAt(i + 1) == '/') {
                                        sqlFragment.append('/');
                                        i += 2;
                                        inBlockComment = false;
                                } else {
                                        ++i;
                                }

                                continue;
                        }

                        if (inSingleQuote) {
                                sqlFragment.append(c);

                                if (inSingleQuoteEscapesBackslash && c == '\\' && i + 1 < sql.length()) {
                                        sqlFragment.append(sql.charAt(i + 1));
                                        i += 2;
                                        continue;
                                }

                                if (c == '\'') {
                                        // Escaped quote: ''
                                        if (i + 1 < sql.length() && sql.charAt(i + 1) == '\'') {
                                                sqlFragment.append('\'');
                                                i += 2;
                                                continue;
                                        }

                                        inSingleQuote = false;
                                        inSingleQuoteEscapesBackslash = false;
                                }

                                ++i;
                                continue;
                        }

                        if (inDoubleQuote) {
                                sqlFragment.append(c);

                                if (c == '"') {
                                        // Escaped quote: ""
                                        if (i + 1 < sql.length() && sql.charAt(i + 1) == '"') {
                                                sqlFragment.append('"');
                                                i += 2;
                                                continue;
                                        }

                                        inDoubleQuote = false;
                                }

                                ++i;
                                continue;
                        }

                        if (inBacktickQuote) {
                                sqlFragment.append(c);

                                if (c == '`')
                                        inBacktickQuote = false;

                                ++i;
                                continue;
                        }

                        if (inBracketQuote) {
                                sqlFragment.append(c);

                                if (c == ']')
                                        inBracketQuote = false;

                                ++i;
                                continue;
                        }

                        // Not inside string/comment
                        if (c == '-' && i + 1 < sql.length() && sql.charAt(i + 1) == '-') {
                                sqlFragment.append("--");
                                i += 2;
                                inLineComment = true;
                                continue;
                        }

                        if (c == '/' && i + 1 < sql.length() && sql.charAt(i + 1) == '*') {
                                sqlFragment.append("/*");
                                i += 2;
                                inBlockComment = true;
                                continue;
                        }

                        if ((c == 'U' || c == 'u') && i + 2 < sql.length() && sql.charAt(i + 1) == '&' && sql.charAt(i + 2) == '\'') {
                                inSingleQuote = true;
                                inSingleQuoteEscapesBackslash = true;
                                sqlFragment.append(c).append("&'");
                                i += 3;
                                continue;
                        }

                        if ((c == 'E' || c == 'e') && i + 1 < sql.length() && sql.charAt(i + 1) == '\'') {
                                inSingleQuote = true;
                                inSingleQuoteEscapesBackslash = true;
                                sqlFragment.append(c).append('\'');
                                i += 2;
                                continue;
                        }

                        if (c == '\'') {
                                inSingleQuote = true;
                                inSingleQuoteEscapesBackslash = false;
                                sqlFragment.append(c);
                                ++i;
                                continue;
                        }

                        if (c == '"') {
                                inDoubleQuote = true;
                                sqlFragment.append(c);
                                ++i;
                                continue;
                        }

                        if (c == '`') {
                                inBacktickQuote = true;
                                sqlFragment.append(c);
                                ++i;
                                continue;
                        }

                        if (c == '[') {
                                inBracketQuote = true;
                                sqlFragment.append(c);
                                ++i;
                                continue;
                        }

                        if (c == '$') {
                                String delimiter = parseDollarQuoteDelimiter(sql, i);

                                if (delimiter != null) {
                                        sqlFragment.append(delimiter);
                                        i += delimiter.length();
                                        dollarQuoteDelimiter = delimiter;
                                        continue;
                                }
                        }

                        if (c == '?') {
                                if (isAllowedQuestionMarkOperator(sql, i)) {
                                        sqlFragment.append(c);
                                        ++i;
                                        continue;
                                }

                                throw new IllegalArgumentException(format("Positional parameters ('?') are not supported. Use named parameters (e.g. ':id') and %s#bind. SQL: %s",
                                                Query.class.getSimpleName(), sql));
                        }

                        if (c == ':' && i + 1 < sql.length() && sql.charAt(i + 1) == ':') {
                                // Postgres type-cast operator (::), do not treat second ':' as a parameter prefix.
                                sqlFragment.append("::");
                                i += 2;
                                continue;
                        }

                        if (c == ':' && i + 1 < sql.length() && Character.isJavaIdentifierStart(sql.charAt(i + 1))) {
                                int nameStartIndex = i + 1;
                                int nameEndIndex = nameStartIndex + 1;

                                while (nameEndIndex < sql.length() && Character.isJavaIdentifierPart(sql.charAt(nameEndIndex)))
                                        ++nameEndIndex;

                                String parameterName = sql.substring(nameStartIndex, nameEndIndex);
                                parameterNames.add(parameterName);
                                distinctParameterNames.add(parameterName);
                                sqlFragments.add(sqlFragment.toString());
                                sqlFragment.setLength(0);
                                i = nameEndIndex;
                                continue;
                        }

                        sqlFragment.append(c);
                        ++i;
                }

                sqlFragments.add(sqlFragment.toString());

                return new ParsedSql(List.copyOf(sqlFragments), List.copyOf(parameterNames), Set.copyOf(distinctParameterNames));
        }

        @Nullable
        private static String parseDollarQuoteDelimiter(@NonNull String sql,
                                                                                                                                                                                                        int startIndex) {
                requireNonNull(sql);

                if (startIndex < 0 || startIndex >= sql.length())
                        return null;

                if (sql.charAt(startIndex) != '$')
                        return null;

                int i = startIndex + 1;

                while (i < sql.length()) {
                        char c = sql.charAt(i);

                        if (c == '$')
                                return sql.substring(startIndex, i + 1);

                        if (Character.isWhitespace(c))
                                return null;

                        ++i;
                }

                return null;
        }

        @NonNull
        private static final Set<@NonNull String> QUESTION_MARK_PREFIX_KEYWORDS = Set.of(
                        "SELECT", "WHERE", "AND", "OR", "ON", "HAVING", "WHEN", "THEN", "ELSE", "IN",
                        "VALUES", "SET", "RETURNING", "USING", "LIKE", "BETWEEN", "IS", "NOT", "NULL",
                        "JOIN", "FROM"
        );

        @NonNull
        private static final Set<@NonNull String> QUESTION_MARK_SUFFIX_KEYWORDS = Set.of(
                        "FROM", "WHERE", "AND", "OR", "GROUP", "ORDER", "HAVING", "LIMIT", "OFFSET",
                        "UNION", "EXCEPT", "INTERSECT", "RETURNING", "JOIN", "ON"
        );

        private static boolean isAllowedQuestionMarkOperator(@NonNull String sql,
                                                                                                                                                                                                                         int questionMarkIndex) {
                requireNonNull(sql);

                if (questionMarkIndex + 1 < sql.length()) {
                        char nextChar = sql.charAt(questionMarkIndex + 1);
                        if (nextChar == '|' || nextChar == '&')
                                return true;
                }

                int previousIndex = previousNonWhitespaceIndex(sql, questionMarkIndex - 1);
                int nextIndex = nextNonWhitespaceIndex(sql, questionMarkIndex + 1);

                if (previousIndex < 0 || nextIndex < 0)
                        return false;

                char previousChar = sql.charAt(previousIndex);
                char nextChar = sql.charAt(nextIndex);

                if (isOperatorBeforeQuestionMark(previousChar))
                        return false;

                if (isTerminatorAfterQuestionMark(nextChar))
                        return false;

                String previousKeyword = keywordBefore(sql, previousIndex);
                if (previousKeyword != null && QUESTION_MARK_PREFIX_KEYWORDS.contains(previousKeyword))
                        return false;

                String nextKeyword = keywordAfter(sql, nextIndex);
                if (nextKeyword != null && QUESTION_MARK_SUFFIX_KEYWORDS.contains(nextKeyword))
                        return false;

                return true;
        }

        private static boolean isOperatorBeforeQuestionMark(char c) {
                return switch (c) {
                        case '=', '<', '>', '!', '+', '-', '*', '/', '%', ',', '(' -> true;
                        default -> false;
                };
        }

        private static boolean isTerminatorAfterQuestionMark(char c) {
                return switch (c) {
                        case ')', ',', ';' -> true;
                        default -> false;
                };
        }

        private static int previousNonWhitespaceIndex(@NonNull String sql,
                                                                                                                                                                                                int startIndex) {
                for (int i = startIndex; i >= 0; i--)
                        if (!Character.isWhitespace(sql.charAt(i)))
                                return i;
                return -1;
        }

        private static int nextNonWhitespaceIndex(@NonNull String sql,
                                                                                                                                                                                int startIndex) {
                for (int i = startIndex; i < sql.length(); i++)
                        if (!Character.isWhitespace(sql.charAt(i)))
                                return i;
                return -1;
        }

        @Nullable
        private static String keywordBefore(@NonNull String sql,
                                                                                                                                                        int index) {
                char c = sql.charAt(index);
                if (!Character.isJavaIdentifierPart(c))
                        return null;

                int endIndex = index + 1;
                int startIndex = index;
                while (startIndex >= 0 && Character.isJavaIdentifierPart(sql.charAt(startIndex)))
                        --startIndex;

                return sql.substring(startIndex + 1, endIndex).toUpperCase(Locale.ROOT);
        }

        @Nullable
        private static String keywordAfter(@NonNull String sql,
                                                                                                                                                 int index) {
                char c = sql.charAt(index);
                if (!Character.isJavaIdentifierPart(c))
                        return null;

                int endIndex = index + 1;
                while (endIndex < sql.length() && Character.isJavaIdentifierPart(sql.charAt(endIndex)))
                        ++endIndex;

                return sql.substring(index, endIndex).toUpperCase(Locale.ROOT);
        }

        /**
         * Performs a SQL query that is expected to return 0 or 1 result rows.
         *
         * @param sql              the SQL query to execute
         * @param resultSetRowType the type to which {@link ResultSet} rows should be marshaled
         * @param parameters       {@link PreparedStatement} parameters, if any
         * @param <T>              the type to be returned
         * @return a single result (or no result)
         * @throws DatabaseException if > 1 row is returned
         */
        @NonNull
        private <T> Optional<T> queryForObject(@NonNull String sql,
                                                                                                                                                                 @NonNull Class<T> resultSetRowType,
                                                                                                                                                                 Object @Nullable ... parameters) {
                requireNonNull(sql);
                requireNonNull(resultSetRowType);

                return queryForObject(Statement.of(generateId(), sql), resultSetRowType, parameters);
        }

        /**
         * Performs a SQL query that is expected to return 0 or 1 result rows.
         *
         * @param statement        the SQL statement to execute
         * @param resultSetRowType the type to which {@link ResultSet} rows should be marshaled
         * @param parameters       {@link PreparedStatement} parameters, if any
         * @param <T>              the type to be returned
         * @return a single result (or no result)
         * @throws DatabaseException if > 1 row is returned
         */
        private <T> Optional<T> queryForObject(@NonNull Statement statement,
                                                                                                                                                                 @NonNull Class<T> resultSetRowType,
                                                                                                                                                                 Object @Nullable ... parameters) {
                requireNonNull(statement);
                requireNonNull(resultSetRowType);

                return queryForObject(statement, resultSetRowType, null, parameters);
        }

        private <T> Optional<T> queryForObject(@NonNull Statement statement,
                                                                                                                                                                 @NonNull Class<T> resultSetRowType,
                                                                                                                                                                 @Nullable PreparedStatementCustomizer preparedStatementCustomizer,
                                                                                                                                                                 Object @Nullable ... parameters) {
                requireNonNull(statement);
                requireNonNull(resultSetRowType);

                ResultHolder<Optional<T>> resultHolder = new ResultHolder<>();
                StatementContext<T> statementContext = StatementContext.<T>with(statement, this)
                                .resultSetRowType(resultSetRowType)
                                .parameters(parameters)
                                .build();

                List<Object> parametersAsList = parameters == null ? List.of() : Arrays.asList(parameters);

                performDatabaseOperation(statementContext, parametersAsList, preparedStatementCustomizer, (PreparedStatement preparedStatement) -> {
                        long startTime = nanoTime();

                        try (ResultSet resultSet = preparedStatement.executeQuery()) {
                                Duration executionDuration = Duration.ofNanos(nanoTime() - startTime);
                                startTime = nanoTime();

                                Optional<T> result = Optional.empty();

                                if (resultSet.next()) {
                                        try {
                                                T value = getResultSetMapper().map(statementContext, resultSet, statementContext.getResultSetRowType().get(), getInstanceProvider()).orElse(null);
                                                result = Optional.ofNullable(value);
                                        } catch (SQLException e) {
                                                throw new DatabaseException(format("Unable to map JDBC %s row to %s", ResultSet.class.getSimpleName(), statementContext.getResultSetRowType().get()), e);
                                        }

                                        if (resultSet.next())
                                                throw new DatabaseException("Expected 1 row in resultset but got more than 1 instead");
                                }

                                resultHolder.value = result;
                                Duration resultSetMappingDuration = Duration.ofNanos(nanoTime() - startTime);
                                return new DatabaseOperationResult(executionDuration, resultSetMappingDuration);
                        }
                });

                return resultHolder.value;
        }

        /**
         * Performs a SQL query that is expected to return any number of result rows.
         *
         * @param sql              the SQL query to execute
         * @param resultSetRowType the type to which {@link ResultSet} rows should be marshaled
         * @param parameters       {@link PreparedStatement} parameters, if any
         * @param <T>              the type to be returned
         * @return a list of results
         */
        @NonNull
        private <T> List<@Nullable T> queryForList(@NonNull String sql,
                                                                                                                                                                                 @NonNull Class<T> resultSetRowType,
                                                                                                                                                                                 Object @Nullable ... parameters) {
                requireNonNull(sql);
                requireNonNull(resultSetRowType);

                return queryForList(Statement.of(generateId(), sql), resultSetRowType, parameters);
        }

        /**
         * Performs a SQL query that is expected to return any number of result rows.
         *
         * @param statement        the SQL statement to execute
         * @param resultSetRowType the type to which {@link ResultSet} rows should be marshaled
         * @param parameters       {@link PreparedStatement} parameters, if any
         * @param <T>              the type to be returned
         * @return a list of results
         */
        @NonNull
        private <T> List<@Nullable T> queryForList(@NonNull Statement statement,
                                                                                                                                                                                 @NonNull Class<T> resultSetRowType,
                                                                                                                                                                                 Object @Nullable ... parameters) {
                requireNonNull(statement);
                requireNonNull(resultSetRowType);

                return queryForList(statement, resultSetRowType, null, parameters);
        }

        private <T> List<@Nullable T> queryForList(@NonNull Statement statement,
                                                                                                                                                                                 @NonNull Class<T> resultSetRowType,
                                                                                                                                                                                 @Nullable PreparedStatementCustomizer preparedStatementCustomizer,
                                                                                                                                                                                 Object @Nullable ... parameters) {
                requireNonNull(statement);
                requireNonNull(resultSetRowType);

                List<T> list = new ArrayList<>();
                StatementContext<T> statementContext = StatementContext.<T>with(statement, this)
                                .resultSetRowType(resultSetRowType)
                                .parameters(parameters)
                                .build();

                List<Object> parametersAsList = parameters == null ? List.of() : Arrays.asList(parameters);

                performDatabaseOperation(statementContext, parametersAsList, preparedStatementCustomizer, (PreparedStatement preparedStatement) -> {
                        long startTime = nanoTime();

                        try (ResultSet resultSet = preparedStatement.executeQuery()) {
                                Duration executionDuration = Duration.ofNanos(nanoTime() - startTime);
                                startTime = nanoTime();

                                while (resultSet.next()) {
                                        try {
                                                T listElement = getResultSetMapper().map(statementContext, resultSet, statementContext.getResultSetRowType().get(), getInstanceProvider()).orElse(null);
                                                list.add(listElement);
                                        } catch (SQLException e) {
                                                throw new DatabaseException(format("Unable to map JDBC %s row to %s", ResultSet.class.getSimpleName(), statementContext.getResultSetRowType().get()), e);
                                        }
                                }

                                Duration resultSetMappingDuration = Duration.ofNanos(nanoTime() - startTime);
                                return new DatabaseOperationResult(executionDuration, resultSetMappingDuration);
                        }
                });

                return list;
        }

        @Nullable
        private <T, R> R queryForStream(@NonNull Statement statement,
                                                                                                                                        @NonNull Class<T> resultSetRowType,
                                                                                                                                        @Nullable PreparedStatementCustomizer preparedStatementCustomizer,
                                                                                                                                        @NonNull Function<Stream<@Nullable T>, R> streamFunction,
                                                                                                                                        Object @Nullable ... parameters) {
                requireNonNull(statement);
                requireNonNull(resultSetRowType);
                requireNonNull(streamFunction);

                StatementContext<T> statementContext = StatementContext.<T>with(statement, this)
                                .resultSetRowType(resultSetRowType)
                                .parameters(parameters)
                                .build();

                List<Object> parametersAsList = parameters == null ? List.of() : Arrays.asList(parameters);
                StreamingResultSet<T> iterator = new StreamingResultSet<>(this, statementContext, parametersAsList, preparedStatementCustomizer);

                try (Stream<@Nullable T> stream = StreamSupport.stream(Spliterators.spliteratorUnknownSize(iterator, Spliterator.ORDERED), false)
                                .onClose(iterator::close)) {
                        return streamFunction.apply(stream);
                }
        }

        /**
         * Executes a SQL Data Manipulation Language (DML) statement, such as {@code INSERT}, {@code UPDATE}, or {@code DELETE};
         * or a SQL statement that returns nothing, such as a DDL statement.
         *
         * @param sql        the SQL to execute
         * @param parameters {@link PreparedStatement} parameters, if any
         * @return the number of rows affected by the SQL statement
         */
        @NonNull
        private Long execute(@NonNull String sql,
                                                                                         Object @Nullable ... parameters) {
                requireNonNull(sql);
                return execute(Statement.of(generateId(), sql), parameters);
        }

        /**
         * Executes a SQL Data Manipulation Language (DML) statement, such as {@code INSERT}, {@code UPDATE}, or {@code DELETE};
         * or a SQL statement that returns nothing, such as a DDL statement.
         *
         * @param statement  the SQL statement to execute
         * @param parameters {@link PreparedStatement} parameters, if any
         * @return the number of rows affected by the SQL statement
         */
        @NonNull
        private Long execute(@NonNull Statement statement,
                                                                                         Object @Nullable ... parameters) {
                requireNonNull(statement);

                return execute(statement, null, parameters);
        }

        private Long execute(@NonNull Statement statement,
                                                                                         @Nullable PreparedStatementCustomizer preparedStatementCustomizer,
                                                                                         Object @Nullable ... parameters) {
                requireNonNull(statement);

                ResultHolder<Long> resultHolder = new ResultHolder<>();
                StatementContext<Void> statementContext = StatementContext.with(statement, this)
                                .parameters(parameters)
                                .build();

                List<Object> parametersAsList = parameters == null ? List.of() : Arrays.asList(parameters);

                performDatabaseOperation(statementContext, parametersAsList, preparedStatementCustomizer, (PreparedStatement preparedStatement) -> {
                        long startTime = nanoTime();

                        DatabaseOperationSupportStatus executeLargeUpdateSupported = getExecuteLargeUpdateSupported();

                        // Use the appropriate "large" value if we know it.
                        // If we don't know it, detect it and store it.
                        if (executeLargeUpdateSupported == DatabaseOperationSupportStatus.YES) {
                                resultHolder.value = preparedStatement.executeLargeUpdate();
                        } else if (executeLargeUpdateSupported == DatabaseOperationSupportStatus.NO) {
                                resultHolder.value = (long) preparedStatement.executeUpdate();
                        } else {
                                // If the driver doesn't support executeLargeUpdate, then UnsupportedOperationException is thrown.
                                try {
                                        resultHolder.value = preparedStatement.executeLargeUpdate();
                                        setExecuteLargeUpdateSupported(DatabaseOperationSupportStatus.YES);
                                } catch (SQLFeatureNotSupportedException | UnsupportedOperationException | AbstractMethodError e) {
                                        setExecuteLargeUpdateSupported(DatabaseOperationSupportStatus.NO);
                                        resultHolder.value = (long) preparedStatement.executeUpdate();
                                } catch (SQLException e) {
                                        if (isUnsupportedSqlFeature(e)) {
                                                setExecuteLargeUpdateSupported(DatabaseOperationSupportStatus.NO);
                                                resultHolder.value = (long) preparedStatement.executeUpdate();
                                        } else {
                                                throw e;
                                        }
                                }
                        }

                        Duration executionDuration = Duration.ofNanos(nanoTime() - startTime);
                        return new DatabaseOperationResult(executionDuration, null);
                });

                return resultHolder.value;
        }

        /**
         * Executes a SQL Data Manipulation Language (DML) statement, such as {@code INSERT}, {@code UPDATE}, or {@code DELETE},
         * which returns 0 or 1 rows, e.g. with Postgres/Oracle's {@code RETURNING} clause.
         *
         * @param sql              the SQL query to execute
         * @param resultSetRowType the type to which the {@link ResultSet} row should be marshaled
         * @param parameters       {@link PreparedStatement} parameters, if any
         * @param <T>              the type to be returned
         * @return a single result (or no result)
         * @throws DatabaseException if > 1 row is returned
         */
        @NonNull
        private <T> Optional<T> executeForObject(@NonNull String sql,
                                                                                                                                                                         @NonNull Class<T> resultSetRowType,
                                                                                                                                                                         Object @Nullable ... parameters) {
                requireNonNull(sql);
                requireNonNull(resultSetRowType);

                return executeForObject(Statement.of(generateId(), sql), resultSetRowType, parameters);
        }

        /**
         * Executes a SQL Data Manipulation Language (DML) statement, such as {@code INSERT}, {@code UPDATE}, or {@code DELETE},
         * which returns 0 or 1 rows, e.g. with Postgres/Oracle's {@code RETURNING} clause.
         *
         * @param statement        the SQL statement to execute
         * @param resultSetRowType the type to which {@link ResultSet} rows should be marshaled
         * @param parameters       {@link PreparedStatement} parameters, if any
         * @param <T>              the type to be returned
         * @return a single result (or no result)
         * @throws DatabaseException if > 1 row is returned
         */
        private <T> Optional<T> executeForObject(@NonNull Statement statement,
                                                                                                                                                                         @NonNull Class<T> resultSetRowType,
                                                                                                                                                                         Object @Nullable ... parameters) {
                requireNonNull(statement);
                requireNonNull(resultSetRowType);

                return executeForObject(statement, resultSetRowType, null, parameters);
        }

        private <T> Optional<T> executeForObject(@NonNull Statement statement,
                                                                                                                                                                         @NonNull Class<T> resultSetRowType,
                                                                                                                                                                         @Nullable PreparedStatementCustomizer preparedStatementCustomizer,
                                                                                                                                                                         Object @Nullable ... parameters) {
                requireNonNull(statement);
                requireNonNull(resultSetRowType);

                // Ultimately we just delegate to queryForObject.
                // Having `executeForList` is to allow for users to explicitly express intent
                // and make static analysis of code easier (e.g. maybe you'd like to hook all of your "execute" statements for
                // logging, or delegation to a writable master as opposed to a read replica)
                return queryForObject(statement, resultSetRowType, preparedStatementCustomizer, parameters);
        }

        /**
         * Executes a SQL Data Manipulation Language (DML) statement, such as {@code INSERT}, {@code UPDATE}, or {@code DELETE},
         * which returns any number of rows, e.g. with Postgres/Oracle's {@code RETURNING} clause.
         *
         * @param sql              the SQL to execute
         * @param resultSetRowType the type to which {@link ResultSet} rows should be marshaled
         * @param parameters       {@link PreparedStatement} parameters, if any
         * @param <T>              the type to be returned
         * @return a list of results
         */
        @NonNull
        private <T> List<@Nullable T> executeForList(@NonNull String sql,
                                                                                                                                                                                         @NonNull Class<T> resultSetRowType,
                                                                                                                                                                                         Object @Nullable ... parameters) {
                requireNonNull(sql);
                requireNonNull(resultSetRowType);

                return executeForList(Statement.of(generateId(), sql), resultSetRowType, parameters);
        }

        /**
         * Executes a SQL Data Manipulation Language (DML) statement, such as {@code INSERT}, {@code UPDATE}, or {@code DELETE},
         * which returns any number of rows, e.g. with Postgres/Oracle's {@code RETURNING} clause.
         *
         * @param statement        the SQL statement to execute
         * @param resultSetRowType the type to which {@link ResultSet} rows should be marshaled
         * @param parameters       {@link PreparedStatement} parameters, if any
         * @param <T>              the type to be returned
         * @return a list of results
         */
        @NonNull
        private <T> List<@Nullable T> executeForList(@NonNull Statement statement,
                                                                                                                                                                                         @NonNull Class<T> resultSetRowType,
                                                                                                                                                                                         Object @Nullable ... parameters) {
                requireNonNull(statement);
                requireNonNull(resultSetRowType);

                return executeForList(statement, resultSetRowType, null, parameters);
        }

        private <T> List<@Nullable T> executeForList(@NonNull Statement statement,
                                                                                                                                                                                         @NonNull Class<T> resultSetRowType,
                                                                                                                                                                                         @Nullable PreparedStatementCustomizer preparedStatementCustomizer,
                                                                                                                                                                                         Object @Nullable ... parameters) {
                requireNonNull(statement);
                requireNonNull(resultSetRowType);

                // Ultimately we just delegate to queryForList.
                // Having `executeForList` is to allow for users to explicitly express intent
                // and make static analysis of code easier (e.g. maybe you'd like to hook all of your "execute" statements for
                // logging, or delegation to a writable master as opposed to a read replica)
                return queryForList(statement, resultSetRowType, preparedStatementCustomizer, parameters);
        }

        /**
         * Executes a SQL Data Manipulation Language (DML) statement, such as {@code INSERT}, {@code UPDATE}, or {@code DELETE}
         * in "batch" over a set of parameter groups.
         * <p>
         * Useful for bulk-inserting or updating large amounts of data.
         *
         * @param sql             the SQL to execute
         * @param parameterGroups Groups of {@link PreparedStatement} parameters
         * @return the number of rows affected by the SQL statement per-group
         */
        @NonNull
        private List<Long> executeBatch(@NonNull String sql,
                                                                                                                                        @NonNull List<List<Object>> parameterGroups) {
                requireNonNull(sql);
                requireNonNull(parameterGroups);

                return executeBatch(Statement.of(generateId(), sql), parameterGroups);
        }

        /**
         * Executes a SQL Data Manipulation Language (DML) statement, such as {@code INSERT}, {@code UPDATE}, or {@code DELETE}
         * in "batch" over a set of parameter groups.
         * <p>
         * Useful for bulk-inserting or updating large amounts of data.
         *
         * @param statement       the SQL statement to execute
         * @param parameterGroups Groups of {@link PreparedStatement} parameters
         * @return the number of rows affected by the SQL statement per-group
         */
        @NonNull
        private List<Long> executeBatch(@NonNull Statement statement,
                                                                                                                                        @NonNull List<List<Object>> parameterGroups) {
                requireNonNull(statement);
                requireNonNull(parameterGroups);

                return executeBatch(statement, parameterGroups, null);
        }

        private List<Long> executeBatch(@NonNull Statement statement,
                                                                                                                                        @NonNull List<List<Object>> parameterGroups,
                                                                                                                                        @Nullable PreparedStatementCustomizer preparedStatementCustomizer) {
                requireNonNull(statement);
                requireNonNull(parameterGroups);
                if (parameterGroups.isEmpty())
                        return List.of();

                Integer expectedParameterCount = null;

                for (int i = 0; i < parameterGroups.size(); i++) {
                        List<Object> parameterGroup = parameterGroups.get(i);

                        if (parameterGroup == null)
                                throw new IllegalArgumentException(format("Parameter group at index %s is null", i));

                        int parameterCount = parameterGroup.size();
                        if (expectedParameterCount == null) {
                                expectedParameterCount = parameterCount;
                        } else if (parameterCount != expectedParameterCount) {
                                throw new IllegalArgumentException(format(
                                                "Inconsistent parameter group size at index %s: expected %s but found %s",
                                                i, expectedParameterCount, parameterCount));
                        }
                }

                ResultHolder<List<Long>> resultHolder = new ResultHolder<>();
                StatementContext<List<Long>> statementContext = StatementContext.with(statement, this)
                                .parameters((List) parameterGroups)
                                .resultSetRowType(List.class)
                                .build();

                performDatabaseOperation(statementContext, (preparedStatement) -> {
                        applyPreparedStatementCustomizer(statementContext, preparedStatement, preparedStatementCustomizer);

                        for (List<Object> parameterGroup : parameterGroups) {
                                if (parameterGroup.size() > 0)
                                        performPreparedStatementBinding(statementContext, preparedStatement, parameterGroup);

                                preparedStatement.addBatch();
                        }
                }, (PreparedStatement preparedStatement) -> {
                        long startTime = nanoTime();
                        List<Long> result;

                        DatabaseOperationSupportStatus executeLargeBatchSupported = getExecuteLargeBatchSupported();

                        // Use the appropriate "large" value if we know it.
                        // If we don't know it, detect it and store it.
                        if (executeLargeBatchSupported == DatabaseOperationSupportStatus.YES) {
                                long[] resultArray = preparedStatement.executeLargeBatch();
                                result = Arrays.stream(resultArray).boxed().collect(Collectors.toList());
                        } else if (executeLargeBatchSupported == DatabaseOperationSupportStatus.NO) {
                                int[] resultArray = preparedStatement.executeBatch();
                                result = Arrays.stream(resultArray).asLongStream().boxed().collect(Collectors.toList());
                        } else {
                                // If the driver doesn't support executeLargeBatch, then UnsupportedOperationException is thrown.
                                try {
                                        long[] resultArray = preparedStatement.executeLargeBatch();
                                        result = Arrays.stream(resultArray).boxed().collect(Collectors.toList());
                                        setExecuteLargeBatchSupported(DatabaseOperationSupportStatus.YES);
                                } catch (SQLFeatureNotSupportedException | UnsupportedOperationException | AbstractMethodError e) {
                                        setExecuteLargeBatchSupported(DatabaseOperationSupportStatus.NO);
                                        int[] resultArray = preparedStatement.executeBatch();
                                        result = Arrays.stream(resultArray).asLongStream().boxed().collect(Collectors.toList());
                                } catch (SQLException e) {
                                        if (isUnsupportedSqlFeature(e)) {
                                                setExecuteLargeBatchSupported(DatabaseOperationSupportStatus.NO);
                                                int[] resultArray = preparedStatement.executeBatch();
                                                result = Arrays.stream(resultArray).asLongStream().boxed().collect(Collectors.toList());
                                        } else {
                                                throw e;
                                        }
                                }
                        }

                        resultHolder.value = result;
                        Duration executionDuration = Duration.ofNanos(nanoTime() - startTime);
                        return new DatabaseOperationResult(executionDuration, null);
                }, parameterGroups.size());

                return resultHolder.value;
        }

        /**
         * Exposes a temporary handle to JDBC {@link DatabaseMetaData}, which provides comprehensive vendor-specific information about this database as a whole.
         * <p>
         * This method acquires {@link DatabaseMetaData} on its own newly-borrowed connection, which it manages internally.
         * <p>
         * It does <strong>not</strong> participate in the active transaction, if one exists.
         * <p>
         * The connection is closed as soon as {@link DatabaseMetaDataReader#read(DatabaseMetaData)} completes.
         * <p>
         * See <a href="https://docs.oracle.com/en/java/javase/24/docs/api/java.sql/java/sql/DatabaseMetaData.html">{@code DatabaseMetaData} Javadoc</a> for details.
         */
        public void readDatabaseMetaData(@NonNull DatabaseMetaDataReader databaseMetaDataReader) {
                requireNonNull(databaseMetaDataReader);

                performRawConnectionOperation((connection -> {
                        databaseMetaDataReader.read(connection.getMetaData());
                        return Optional.empty();
                }), false);
        }

        protected <T> void performDatabaseOperation(@NonNull StatementContext<T> statementContext,
                                                                                                                                                                                        @NonNull List<Object> parameters,
                                                                                                                                                                                        @NonNull DatabaseOperation databaseOperation) {
                requireNonNull(statementContext);
                requireNonNull(parameters);
                requireNonNull(databaseOperation);

                performDatabaseOperation(statementContext, parameters, null, databaseOperation);
        }

        protected <T> void performDatabaseOperation(@NonNull StatementContext<T> statementContext,
                                                                                                                                                                                        @NonNull List<Object> parameters,
                                                                                                                                                                                        @Nullable PreparedStatementCustomizer preparedStatementCustomizer,
                                                                                                                                                                                        @NonNull DatabaseOperation databaseOperation) {
                requireNonNull(statementContext);
                requireNonNull(parameters);
                requireNonNull(databaseOperation);

                performDatabaseOperation(statementContext, (preparedStatement) -> {
                        applyPreparedStatementCustomizer(statementContext, preparedStatement, preparedStatementCustomizer);
                        if (parameters.size() > 0)
                                performPreparedStatementBinding(statementContext, preparedStatement, parameters);
                }, databaseOperation);
        }

        protected <T> void performPreparedStatementBinding(@NonNull StatementContext<T> statementContext,
                                                                                                                                                                                                                 @NonNull PreparedStatement preparedStatement,
                                                                                                                                                                                                                 @NonNull List<Object> parameters) {
                requireNonNull(statementContext);
                requireNonNull(preparedStatement);
                requireNonNull(parameters);

                try {
                        for (int i = 0; i < parameters.size(); ++i) {
                                Object parameter = parameters.get(i);

                                if (parameter != null) {
                                        getPreparedStatementBinder().bindParameter(statementContext, preparedStatement, i + 1, parameter);
                                } else {
                                        try {
                                                ParameterMetaData parameterMetaData = preparedStatement.getParameterMetaData();

                                                if (parameterMetaData != null) {
                                                        preparedStatement.setNull(i + 1, parameterMetaData.getParameterType(i + 1));
                                                } else {
                                                        preparedStatement.setNull(i + 1, Types.NULL);
                                                }
                                        } catch (SQLException | AbstractMethodError e) {
                                                preparedStatement.setNull(i + 1, Types.NULL);
                                        }
                                }
                        }
                } catch (Exception e) {
                        throw new DatabaseException(e);
                }
        }

        protected void applyPreparedStatementCustomizer(@NonNull StatementContext<?> statementContext,
                                                                                                                                                                                                        @NonNull PreparedStatement preparedStatement,
                                                                                                                                                                                                        @Nullable PreparedStatementCustomizer preparedStatementCustomizer) throws SQLException {
                requireNonNull(statementContext);
                requireNonNull(preparedStatement);

                if (preparedStatementCustomizer == null)
                        return;

                preparedStatementCustomizer.customize(statementContext, preparedStatement);
        }

        @FunctionalInterface
        protected interface RawConnectionOperation<R> {
                @NonNull
                Optional<R> perform(@NonNull Connection connection) throws Exception;
        }

        /**
         * @since 3.0.0
         */
        @NonNull
        public DatabaseType getDatabaseType() {
                return this.databaseType;
        }

        /**
         * @since 3.0.0
         */
        @NonNull
        public ZoneId getTimeZone() {
                return this.timeZone;
        }

        /**
         * Useful for single-shot "utility" calls that operate outside of normal query operations, e.g. pulling DB metadata.
         * <p>
         * Example: {@link #readDatabaseMetaData(DatabaseMetaDataReader)}.
         */
        @NonNull
        protected <R> Optional<R> performRawConnectionOperation(@NonNull RawConnectionOperation<R> rawConnectionOperation,
                                                                                                                                                                                                                                        @NonNull Boolean shouldParticipateInExistingTransactionIfPossible) {
                requireNonNull(rawConnectionOperation);
                requireNonNull(shouldParticipateInExistingTransactionIfPossible);

                if (shouldParticipateInExistingTransactionIfPossible) {
                        Optional<Transaction> transaction = currentTransaction();
                        ReentrantLock connectionLock = transaction.isPresent() ? transaction.get().getConnectionLock() : null;
                        // Try to participate in txn if it's available
                        Connection connection = null;
                        Throwable thrown = null;

                        if (connectionLock != null)
                                connectionLock.lock();

                        try {
                                connection = transaction.isPresent() ? transaction.get().getConnection() : acquireConnection();
                                return rawConnectionOperation.perform(connection);
                        } catch (DatabaseException e) {
                                thrown = e;
                                throw e;
                        } catch (Exception e) {
                                DatabaseException wrapped = new DatabaseException(e);
                                thrown = wrapped;
                                throw wrapped;
                        } finally {
                                Throwable cleanupFailure = null;

                                try {
                                        // If this was a single-shot operation (not in a transaction), close the connection
                                        if (connection != null && !transaction.isPresent()) {
                                                try {
                                                        closeConnection(connection);
                                                } catch (Throwable cleanupException) {
                                                        cleanupFailure = cleanupException;
                                                }
                                        }
                                } finally {
                                        if (connectionLock != null)
                                                connectionLock.unlock();

                                        if (cleanupFailure != null) {
                                                if (thrown != null) {
                                                        thrown.addSuppressed(cleanupFailure);
                                                } else if (cleanupFailure instanceof RuntimeException) {
                                                        throw (RuntimeException) cleanupFailure;
                                                } else if (cleanupFailure instanceof Error) {
                                                        throw (Error) cleanupFailure;
                                                } else {
                                                        throw new RuntimeException(cleanupFailure);
                                                }
                                        }
                                }
                        }
                } else {
                        boolean acquiredConnection = false;
                        Connection connection = null;
                        Throwable thrown = null;

                        // Always get a fresh connection no matter what and close it afterwards
                        try {
                                connection = getDataSource().getConnection();
                                acquiredConnection = true;
                                return rawConnectionOperation.perform(connection);
                        } catch (DatabaseException e) {
                                thrown = e;
                                throw e;
                        } catch (Exception e) {
                                DatabaseException wrapped = acquiredConnection
                                                ? new DatabaseException(e)
                                                : new DatabaseException("Unable to acquire database connection", e);
                                thrown = wrapped;
                                throw wrapped;
                        } finally {
                                if (connection != null) {
                                        try {
                                                closeConnection(connection);
                                        } catch (Throwable cleanupException) {
                                                if (thrown != null) {
                                                        thrown.addSuppressed(cleanupException);
                                                } else if (cleanupException instanceof RuntimeException) {
                                                        throw (RuntimeException) cleanupException;
                                                } else if (cleanupException instanceof Error) {
                                                        throw (Error) cleanupException;
                                                } else {
                                                        throw new RuntimeException(cleanupException);
                                                }
                                        }
                                }
                        }
                }
        }

        protected <T> void performDatabaseOperation(@NonNull StatementContext<T> statementContext,
                                                                                                                                                                                        @NonNull PreparedStatementBindingOperation preparedStatementBindingOperation,
                                                                                                                                                                                        @NonNull DatabaseOperation databaseOperation) {
                performDatabaseOperation(statementContext, preparedStatementBindingOperation, databaseOperation, null);
        }

        protected <T> void performDatabaseOperation(@NonNull StatementContext<T> statementContext,
                                                                                                                                                                                        @NonNull PreparedStatementBindingOperation preparedStatementBindingOperation,
                                                                                                                                                                                        @NonNull DatabaseOperation databaseOperation,
                                                                                                                                                                                        @Nullable Integer batchSize) {
                requireNonNull(statementContext);
                requireNonNull(preparedStatementBindingOperation);
                requireNonNull(databaseOperation);

                long startTime = nanoTime();
                Duration connectionAcquisitionDuration = null;
                Duration preparationDuration = null;
                Duration executionDuration = null;
                Duration resultSetMappingDuration = null;
                Exception exception = null;
                Throwable thrown = null;
                Connection connection = null;
                Optional<Transaction> transaction = currentTransaction();
                ReentrantLock connectionLock = transaction.isPresent() ? transaction.get().getConnectionLock() : null;

                if (connectionLock != null)
                        connectionLock.lock();

                try {
                        boolean alreadyHasConnection = transaction.isPresent() && transaction.get().hasConnection();
                        connection = transaction.isPresent() ? transaction.get().getConnection() : acquireConnection();
                        connectionAcquisitionDuration = alreadyHasConnection ? null : Duration.ofNanos(nanoTime() - startTime);
                        startTime = nanoTime();

                        try (PreparedStatement preparedStatement = connection.prepareStatement(statementContext.getStatement().getSql())) {
                                preparedStatementBindingOperation.perform(preparedStatement);
                                preparationDuration = Duration.ofNanos(nanoTime() - startTime);

                                DatabaseOperationResult databaseOperationResult = databaseOperation.perform(preparedStatement);
                                executionDuration = databaseOperationResult.getExecutionDuration().orElse(null);
                                resultSetMappingDuration = databaseOperationResult.getResultSetMappingDuration().orElse(null);
                        }
                } catch (DatabaseException e) {
                        exception = e;
                        thrown = e;
                        throw e;
                } catch (Error e) {
                        exception = new DatabaseException(e);
                        thrown = e;
                        throw e;
                } catch (Exception e) {
                        exception = e;
                        DatabaseException wrapped = new DatabaseException(e);
                        thrown = wrapped;
                        throw wrapped;
                } finally {
                        Throwable cleanupFailure = null;

                        try {
                                cleanupFailure = closeStatementContextResources(statementContext, cleanupFailure);

                                // If this was a single-shot operation (not in a transaction), close the connection
                                if (connection != null && !transaction.isPresent()) {
                                        try {
                                                closeConnection(connection);
                                        } catch (Throwable cleanupException) {
                                                cleanupFailure = cleanupException;
                                        }
                                }
                        } finally {
                                if (connectionLock != null)
                                        connectionLock.unlock();

                                StatementLog statementLog =
                                                StatementLog.withStatementContext(statementContext)
                                                                .connectionAcquisitionDuration(connectionAcquisitionDuration)
                                                                .preparationDuration(preparationDuration)
                                                                .executionDuration(executionDuration)
                                                                .resultSetMappingDuration(resultSetMappingDuration)
                                                                .batchSize(batchSize)
                                                                .exception(exception)
                                                                .build();

                                try {
                                        getStatementLogger().log(statementLog);
                                } catch (Throwable cleanupException) {
                                        if (transaction.isPresent() && thrown == null && cleanupFailure == null) {
                                                Throwable loggerFailure = cleanupException;
                                                Transaction currentTransaction = transaction.get();

                                                if (!currentTransaction.isOwnedByCurrentThread()) {
                                                        cleanupFailure = new StatementLoggerFailureException(loggerFailure);
                                                } else {
                                                        currentTransaction.addPostTransactionOperation(result -> {
                                                                if (loggerFailure instanceof RuntimeException runtimeException)
                                                                        throw runtimeException;
                                                                if (loggerFailure instanceof Error error)
                                                                        throw error;
                                                                throw new RuntimeException(loggerFailure);
                                                        });
                                                }
                                        } else {
                                                if (cleanupFailure == null)
                                                        cleanupFailure = cleanupException;
                                                else
                                                        cleanupFailure.addSuppressed(cleanupException);
                                        }
                                }
                        }

                        if (cleanupFailure != null) {
                                if (thrown != null) {
                                        thrown.addSuppressed(cleanupFailure);
                                } else if (cleanupFailure instanceof RuntimeException) {
                                        throw (RuntimeException) cleanupFailure;
                                } else if (cleanupFailure instanceof Error) {
                                        throw (Error) cleanupFailure;
                                } else {
                                        throw new RuntimeException(cleanupFailure);
                                }
                        }
                }
        }

        @NonNull
        protected Connection acquireConnection() {
                Optional<Transaction> transaction = currentTransaction();

                if (transaction.isPresent())
                        return transaction.get().getConnection();

                try {
                        return getDataSource().getConnection();
                } catch (SQLException e) {
                        throw new DatabaseException("Unable to acquire database connection", e);
                }
        }

        @NonNull
        protected DataSource getDataSource() {
                return this.dataSource;
        }

        @NonNull
        protected InstanceProvider getInstanceProvider() {
                return this.instanceProvider;
        }

        @NonNull
        protected PreparedStatementBinder getPreparedStatementBinder() {
                return this.preparedStatementBinder;
        }

        @NonNull
        protected ResultSetMapper getResultSetMapper() {
                return this.resultSetMapper;
        }

        @NonNull
        protected StatementLogger getStatementLogger() {
                return this.statementLogger;
        }

        @NonNull
        protected DatabaseOperationSupportStatus getExecuteLargeBatchSupported() {
                return this.executeLargeBatchSupported;
        }

        protected void setExecuteLargeBatchSupported(@NonNull DatabaseOperationSupportStatus executeLargeBatchSupported) {
                requireNonNull(executeLargeBatchSupported);
                this.executeLargeBatchSupported = executeLargeBatchSupported;
        }

        @NonNull
        protected DatabaseOperationSupportStatus getExecuteLargeUpdateSupported() {
                return this.executeLargeUpdateSupported;
        }

        protected void setExecuteLargeUpdateSupported(@NonNull DatabaseOperationSupportStatus executeLargeUpdateSupported) {
                requireNonNull(executeLargeUpdateSupported);
                this.executeLargeUpdateSupported = executeLargeUpdateSupported;
        }

        @NonNull
        protected Object generateId() {
                // "Unique" keys
                return format("com.pyranid.%s", this.defaultIdGenerator.incrementAndGet());
        }

        @FunctionalInterface
        protected interface DatabaseOperation {
                @NonNull
                DatabaseOperationResult perform(@NonNull PreparedStatement preparedStatement) throws Exception;
        }

        @FunctionalInterface
        protected interface PreparedStatementBindingOperation {
                void perform(@NonNull PreparedStatement preparedStatement) throws Exception;
        }

        @NotThreadSafe
        private static final class StreamingResultSet<T> implements java.util.Iterator<T>, AutoCloseable {
                private final Database database;
                private final StatementContext<T> statementContext;
                private final List<Object> parameters;
                @Nullable
                private final PreparedStatementCustomizer preparedStatementCustomizer;
                @NonNull
                private final Optional<Transaction> transaction;
                @Nullable
                private final ReentrantLock connectionLock;
                @Nullable
                private Connection connection;
                @Nullable
                private PreparedStatement preparedStatement;
                @Nullable
                private ResultSet resultSet;
                private boolean closed;
                private boolean hasNextEvaluated;
                private boolean hasNext;
                @Nullable
                private Duration connectionAcquisitionDuration;
                @Nullable
                private Duration preparationDuration;
                @Nullable
                private Duration executionDuration;
                private long resultSetMappingNanos;
                @Nullable
                private Exception exception;
                @Nullable
                private Throwable thrown;

                private StreamingResultSet(@NonNull Database database,
                                                                                                                         @NonNull StatementContext<T> statementContext,
                                                                                                                         @NonNull List<Object> parameters,
                                                                                                                         @Nullable PreparedStatementCustomizer preparedStatementCustomizer) {
                        this.database = requireNonNull(database);
                        this.statementContext = requireNonNull(statementContext);
                        this.parameters = requireNonNull(parameters);
                        this.preparedStatementCustomizer = preparedStatementCustomizer;
                        this.transaction = database.currentTransaction();
                        this.connectionLock = this.transaction.isPresent() ? this.transaction.get().getConnectionLock() : null;

                        open();
                }

                private void open() {
                        long startTime = nanoTime();

                        if (this.connectionLock != null)
                                this.connectionLock.lock();

                        try {
                                boolean alreadyHasConnection = this.transaction.isPresent() && this.transaction.get().hasConnection();
                                this.connection = this.transaction.isPresent() ? this.transaction.get().getConnection() : this.database.acquireConnection();
                                this.connectionAcquisitionDuration = alreadyHasConnection ? null : Duration.ofNanos(nanoTime() - startTime);
                                startTime = nanoTime();

                                this.preparedStatement = this.connection.prepareStatement(this.statementContext.getStatement().getSql());
                                this.database.applyPreparedStatementCustomizer(this.statementContext, this.preparedStatement, this.preparedStatementCustomizer);
                                if (this.parameters.size() > 0)
                                        this.database.performPreparedStatementBinding(this.statementContext, this.preparedStatement, this.parameters);
                                this.preparationDuration = Duration.ofNanos(nanoTime() - startTime);

                                startTime = nanoTime();
                                this.resultSet = this.preparedStatement.executeQuery();
                                this.executionDuration = Duration.ofNanos(nanoTime() - startTime);
                        } catch (DatabaseException e) {
                                this.exception = e;
                                this.thrown = e;
                                close();
                                throw e;
                        } catch (Exception e) {
                                this.exception = e;
                                DatabaseException wrapped = new DatabaseException(e);
                                this.thrown = wrapped;
                                close();
                                throw wrapped;
                        }
                }

                @Override
                public boolean hasNext() {
                        if (this.closed)
                                return false;

                        if (!this.hasNextEvaluated) {
                                try {
                                        this.hasNext = this.resultSet != null && this.resultSet.next();
                                        this.hasNextEvaluated = true;
                                        if (!this.hasNext)
                                                close();
                                } catch (SQLException e) {
                                        this.exception = e;
                                        this.thrown = new DatabaseException(e);
                                        close();
                                        throw (DatabaseException) this.thrown;
                                }
                        }

                        return this.hasNext;
                }

                @Override
                public T next() {
                        if (!hasNext())
                                throw new java.util.NoSuchElementException();

                        this.hasNextEvaluated = false;
                        long startTime = nanoTime();

                        try {
                                T value = this.database.getResultSetMapper()
                                                .map(this.statementContext, requireNonNull(this.resultSet), this.statementContext.getResultSetRowType().get(), this.database.getInstanceProvider())
                                                .orElse(null);
                                this.resultSetMappingNanos += nanoTime() - startTime;
                                return value;
                        } catch (SQLException e) {
                                this.exception = e;
                                this.thrown = new DatabaseException(format("Unable to map JDBC %s row to %s", ResultSet.class.getSimpleName(), this.statementContext.getResultSetRowType().get()), e);
                                close();
                                throw (DatabaseException) this.thrown;
                        } catch (DatabaseException e) {
                                this.exception = e;
                                this.thrown = e;
                                close();
                                throw e;
                        }
                }

                @Override
                public void close() {
                        if (this.closed)
                                return;

                        this.closed = true;
                        Throwable cleanupFailure = null;

                        try {
                                cleanupFailure = closeStatementContextResources(this.statementContext, cleanupFailure);

                                if (this.resultSet != null) {
                                        try {
                                                this.resultSet.close();
                                        } catch (Throwable cleanupException) {
                                                cleanupFailure = cleanupFailure == null ? cleanupException : addSuppressed(cleanupFailure, cleanupException);
                                        }
                                }

                                if (this.preparedStatement != null) {
                                        try {
                                                this.preparedStatement.close();
                                        } catch (Throwable cleanupException) {
                                                cleanupFailure = cleanupFailure == null ? cleanupException : addSuppressed(cleanupFailure, cleanupException);
                                        }
                                }

                                if (this.connection != null && this.transaction.isEmpty()) {
                                        try {
                                                this.database.closeConnection(this.connection);
                                        } catch (Throwable cleanupException) {
                                                cleanupFailure = cleanupFailure == null ? cleanupException : addSuppressed(cleanupFailure, cleanupException);
                                        }
                                }
                        } finally {
                                if (this.connectionLock != null)
                                        this.connectionLock.unlock();

                                Duration mappingDuration = this.resultSetMappingNanos == 0L ? null : Duration.ofNanos(this.resultSetMappingNanos);

                                StatementLog statementLog =
                                                StatementLog.withStatementContext(this.statementContext)
                                                                .connectionAcquisitionDuration(this.connectionAcquisitionDuration)
                                                                .preparationDuration(this.preparationDuration)
                                                                .executionDuration(this.executionDuration)
                                                                .resultSetMappingDuration(mappingDuration)
                                                                .exception(this.exception)
                                                                .build();

                                try {
                                        this.database.getStatementLogger().log(statementLog);
                                } catch (Throwable cleanupException) {
                                        if (this.transaction.isPresent() && this.thrown == null && cleanupFailure == null) {
                                                Throwable loggerFailure = cleanupException;
                                                Transaction currentTransaction = this.transaction.get();

                                                if (!currentTransaction.isOwnedByCurrentThread()) {
                                                        cleanupFailure = new StatementLoggerFailureException(loggerFailure);
                                                } else {
                                                        currentTransaction.addPostTransactionOperation(result -> {
                                                                if (loggerFailure instanceof RuntimeException runtimeException)
                                                                        throw runtimeException;
                                                                if (loggerFailure instanceof Error error)
                                                                        throw error;
                                                                throw new RuntimeException(loggerFailure);
                                                        });
                                                }
                                        } else {
                                                cleanupFailure = cleanupFailure == null ? cleanupException : addSuppressed(cleanupFailure, cleanupException);
                                        }
                                }
                        }

                        if (cleanupFailure != null) {
                                if (this.thrown != null) {
                                        this.thrown.addSuppressed(cleanupFailure);
                                } else if (cleanupFailure instanceof RuntimeException) {
                                        throw (RuntimeException) cleanupFailure;
                                } else if (cleanupFailure instanceof Error) {
                                        throw (Error) cleanupFailure;
                                } else {
                                        throw new RuntimeException(cleanupFailure);
                                }
                        }
                }

                @NonNull
                private static Throwable addSuppressed(@NonNull Throwable existing,
                                                                                                                                                                         @NonNull Throwable additional) {
                        existing.addSuppressed(additional);
                        return existing;
                }
        }

        /**
         * Builder used to construct instances of {@link Database}.
         * <p>
         * This class is intended for use by a single thread.
         *
         * @author <a href="https://www.revetkn.com">Mark Allen</a>
         * @since 1.0.0
         */
        @NotThreadSafe
        public static class Builder {
                @NonNull
                private final DataSource dataSource;
                @Nullable
                private DatabaseType databaseType;
                @Nullable
                private ZoneId timeZone;
                @Nullable
                private InstanceProvider instanceProvider;
                @Nullable
                private PreparedStatementBinder preparedStatementBinder;
                @Nullable
                private ResultSetMapper resultSetMapper;
                @Nullable
                private StatementLogger statementLogger;
                @Nullable
                private Integer parsedSqlCacheCapacity;

                private Builder(@NonNull DataSource dataSource) {
                        this.dataSource = requireNonNull(dataSource);
                        this.databaseType = null;
                        this.parsedSqlCacheCapacity = null;
                }

                /**
                 * Overrides automatic database type detection.
                 *
                 * @param databaseType the database type to use (null to enable auto-detection)
                 * @return this {@code Builder}, for chaining
                 * @since 4.0.0
                 */
                @NonNull
                public Builder databaseType(@Nullable DatabaseType databaseType) {
                        this.databaseType = databaseType;
                        return this;
                }

                @NonNull
                public Builder timeZone(@Nullable ZoneId timeZone) {
                        this.timeZone = timeZone;
                        return this;
                }

                @NonNull
                public Builder instanceProvider(@Nullable InstanceProvider instanceProvider) {
                        this.instanceProvider = instanceProvider;
                        return this;
                }

                @NonNull
                public Builder preparedStatementBinder(@Nullable PreparedStatementBinder preparedStatementBinder) {
                        this.preparedStatementBinder = preparedStatementBinder;
                        return this;
                }

                @NonNull
                public Builder resultSetMapper(@Nullable ResultSetMapper resultSetMapper) {
                        this.resultSetMapper = resultSetMapper;
                        return this;
                }

                @NonNull
                public Builder statementLogger(@Nullable StatementLogger statementLogger) {
                        this.statementLogger = statementLogger;
                        return this;
                }

                /**
                 * Configures the size of the parsed SQL cache.
                 * <p>
                 * A value of {@code 0} disables caching. A value of {@code null} uses the default size.
                 *
                 * @param parsedSqlCacheCapacity cache size (0 disables caching, null uses default)
                 * @return this {@code Builder}, for chaining
                 */
                @NonNull
                public Builder parsedSqlCacheCapacity(@Nullable Integer parsedSqlCacheCapacity) {
                        if (parsedSqlCacheCapacity != null && parsedSqlCacheCapacity < 0)
                                throw new IllegalArgumentException("parsedSqlCacheCapacity must be >= 0");

                        this.parsedSqlCacheCapacity = parsedSqlCacheCapacity;
                        return this;
                }

                @NonNull
                public Database build() {
                        return new Database(this);
                }
        }

        @ThreadSafe
        static class DatabaseOperationResult {
                @Nullable
                private final Duration executionDuration;
                @Nullable
                private final Duration resultSetMappingDuration;

                public DatabaseOperationResult(@Nullable Duration executionDuration,
                                                                                                                                         @Nullable Duration resultSetMappingDuration) {
                        this.executionDuration = executionDuration;
                        this.resultSetMappingDuration = resultSetMappingDuration;
                }

                @NonNull
                public Optional<Duration> getExecutionDuration() {
                        return Optional.ofNullable(this.executionDuration);
                }

                @NonNull
                public Optional<Duration> getResultSetMappingDuration() {
                        return Optional.ofNullable(this.resultSetMappingDuration);
                }
        }

        @NotThreadSafe
        static class ResultHolder<T> {
                T value;
        }

        enum DatabaseOperationSupportStatus {
                UNKNOWN,
                YES,
                NO
        }

        private static final class StatementLoggerFailureException extends RuntimeException {
                private StatementLoggerFailureException(@NonNull Throwable cause) {
                        super("Statement logger failed", cause);
                }
        }
}