QueryExecutorTemplate.java

/*
 *    Copyright (c) 2022, VRAI Labs and/or its affiliates. All rights reserved.
 *
 *    This software is licensed under the Apache License, Version 2.0 (the
 *    "License") as published by the Apache Software Foundation.
 *
 *    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 io.supertokens.storage.postgresql;

import io.supertokens.pluginInterface.exceptions.StorageQueryException;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

public interface QueryExecutorTemplate {

    static <T> T execute(Start start, String QUERY, PreparedStatementValueSetter setter,
            ResultSetValueExtractor<T> mapper) throws SQLException, StorageQueryException {
        try (Connection con = ConnectionPool.getConnection(start)) {
            return execute(con, QUERY, setter, mapper);
        }
    }

    static <T> T execute(Connection con, String QUERY, PreparedStatementValueSetter setter,
            ResultSetValueExtractor<T> mapper) throws SQLException, StorageQueryException {
        if (setter == null)
            setter = PreparedStatementValueSetter.NO_OP_SETTER;
        try (PreparedStatement pst = con.prepareStatement(QUERY)) {
            setter.setValues(pst);
            try (ResultSet result = pst.executeQuery()) {
                return mapper.extract(result);
            }
        }
    }

    static int update(Start start, String QUERY, PreparedStatementValueSetter setter)
            throws SQLException, StorageQueryException {
        try (Connection con = ConnectionPool.getConnection(start)) {
            return update(con, QUERY, setter);
        }
    }

    static <T> T update(Start start, String QUERY, PreparedStatementValueSetter setter, ResultSetValueExtractor<T> mapper)
            throws SQLException, StorageQueryException {
        try (Connection con = ConnectionPool.getConnection(start)) {
            try (PreparedStatement pst = con.prepareStatement(QUERY)) {
                setter.setValues(pst);
                try (ResultSet result = pst.executeQuery()) {
                    return mapper.extract(result);
                }
            }
        }
    }

    static int update(Connection con, String QUERY, PreparedStatementValueSetter setter)
            throws SQLException, StorageQueryException {
        try (PreparedStatement pst = con.prepareStatement(QUERY)) {
            setter.setValues(pst);
            return pst.executeUpdate();
        }
    }

}