SqlConnection.hpp

// SPDX-License-Identifier: Apache-2.0
 
#pragma once
 
#if defined(_WIN32) || defined(_WIN64)
    #include <Windows.h>
#endif
 
#include "Api.hpp"
#include "Async/Fwd.hpp"
#include "SqlConnectInfo.hpp"
#include "SqlError.hpp"
#include "SqlLogger.hpp"
#include "SqlServerType.hpp"
 
#include <atomic>
#include <chrono>
#include <expected>
#include <format>
#include <functional>
#include <memory>
#include <optional>
#include <string>
#include <string_view>
#include <system_error>
 
#include <sql.h>
#include <sqlext.h>
#include <sqlspi.h>
#include <sqltypes.h>
 
namespace Lightweight
{
 
class SqlQueryBuilder;
class SqlMigrationQueryBuilder;
class SqlQueryFormatter;
 
/// @brief Represents a connection to a SQL database.
class SqlConnection final
{
  public:
    /// @brief Constructs a new SQL connection to the default connection.
    ///
    /// The default connection is set via SetDefaultConnectInfo.
    /// In case the default connection is not set, the connection will fail.
    /// And in case the connection fails, the last error will be set.
    LIGHTWEIGHT_API SqlConnection();
 
    /// @brief Constructs a new SQL connection to the given connect informaton.
    ///
    /// @param connectInfo The connection information to use. If `std::nullopt`,
    ///                    no connection is established and the object is left
    ///                    in an unconnected state — use `Connect(...)` later.
    ///                    If a value is provided and the connection fails,
    ///                    `SqlException` is thrown carrying the ODBC diagnostic
    ///                    read from the DBC handle.
    LIGHTWEIGHT_API explicit SqlConnection(std::optional<SqlConnectionString> connectInfo);
 
    /// Move constructor.
    LIGHTWEIGHT_API SqlConnection(SqlConnection&& /*other*/) noexcept;
    /// Move assignment operator.
    LIGHTWEIGHT_API SqlConnection& operator=(SqlConnection&& /*other*/) noexcept;
    SqlConnection(SqlConnection const& /*other*/) = delete;
    SqlConnection& operator=(SqlConnection const& /*other*/) = delete;
 
    /// Destructs this SQL connection object,
    LIGHTWEIGHT_API ~SqlConnection() noexcept;
 
    /// Retrieves the default connection information.
    LIGHTWEIGHT_API static SqlConnectionString const& DefaultConnectionString() noexcept;
 
    /// Sets the default connection information.
    ///
    /// @param connectionString The connection information to use.
    LIGHTWEIGHT_API static void SetDefaultConnectionString(SqlConnectionString const& connectionString) noexcept;
 
    /// Sets the default connection information as SqlConnectionDataSource.
    LIGHTWEIGHT_API static void SetDefaultDataSource(SqlConnectionDataSource const& dataSource) noexcept;
 
    /// Sets a callback to be called after each connection being established.
    LIGHTWEIGHT_API static void SetPostConnectedHook(std::function<void(SqlConnection&)> hook);
 
    /// Resets the post connected hook.
    LIGHTWEIGHT_API static void ResetPostConnectedHook();
 
    /// @brief Retrieves the connection ID.
    ///
    /// This is a unique identifier for the connection, which is useful for debugging purposes.
    /// Note, this ID will not change if the connection is moved nor when it is reused via the connection pool.
    [[nodiscard]] uint64_t ConnectionId() const noexcept
    {
        return m_connectionId;
    }
 
    /// Closes the connection (attempting to put it back into the connect[[ion pool).
    LIGHTWEIGHT_API void Close() noexcept;
 
    /// Connects to the given database with the given username and password.
    ///
    /// This method can be called on a connection that has been closed via Close().
    /// If the ODBC handles have been freed, they will be automatically reallocated.
    ///
    /// @retval true if the connection was successful.
    /// @retval false if the connection failed. Use LastError() to retrieve the error information.
    LIGHTWEIGHT_API bool Connect(SqlConnectionDataSource const& info) noexcept;
 
    /// Connects to the given database with the given connection string.
    ///
    /// This method can be called on a connection that has been closed via Close().
    /// If the ODBC handles have been freed, they will be automatically reallocated.
    ///
    /// @retval true if the connection was successful.
    /// @retval false if the connection failed. Use LastError() to retrieve the error information.
    LIGHTWEIGHT_API bool Connect(SqlConnectionString sqlConnectionString) noexcept;
 
    /// Retrieves the last error information with respect to this SQL connection handle.
    [[nodiscard]] LIGHTWEIGHT_API SqlErrorInfo LastError() const;
 
    /// Retrieves the name of the database in use.
    [[nodiscard]] LIGHTWEIGHT_API std::string DatabaseName() const;
 
    /// Retrieves the name of the user.
    [[nodiscard]] LIGHTWEIGHT_API std::string UserName() const;
 
    /// Retrieves the name of the server.
    [[nodiscard]] LIGHTWEIGHT_API std::string ServerName() const;
 
    /// Retrieves the reported server version.
    [[nodiscard]] LIGHTWEIGHT_API std::string ServerVersion() const;
 
    /// Retrieves the type of the server.
    [[nodiscard]] SqlServerType ServerType() const noexcept;
 
    /// Retrieves the name of the driver used for this connection.
    [[nodiscard]] std::string const& DriverName() const noexcept;
 
    /// Retrieves a query formatter suitable for the SQL server being connected.
    [[nodiscard]] SqlQueryFormatter const& QueryFormatter() const noexcept;
 
    /// @brief Whether this connection's ODBC driver supports native parameter-array binding
    /// (`SQL_ATTR_PARAMSET_SIZE` > 1) for batched row-wise execution.
    ///
    /// The batched `SqlStatement::ExecuteBatch(rows, accessors...)` consults this to decide whether it
    /// may submit the whole batch in a single row-wise `SQLExecute`, or must fall back to a single
    /// prepare followed by consecutive per-row executes. This is a driver/backend capability — not a
    /// SQL-dialect concern — so it belongs to the connection, which knows both the server type and the
    /// driver name (either of which a future carve-out can branch on).
    ///
    /// @return `true` if the driver honours parameter arrays.
    [[nodiscard]] bool SupportsNativeRowBatch() const noexcept;
 
    /// @brief Whether this connection's ODBC driver supports native row-array fetching
    /// (`SQL_ATTR_ROW_ARRAY_SIZE` > 1 with `SQLFetchScroll`) for block result retrieval.
    ///
    /// The fast retrieval path in @c SqlStatement::FetchAllRowWise consults this to decide whether
    /// it may bind the result columns row-wise over a record block and materialize whole row blocks per
    /// `SQLFetchScroll` round-trip, or must fall back to the per-row `SQLFetch` path. Like
    /// @ref SupportsNativeRowBatch this is a driver/backend capability — not a SQL-dialect concern — so
    /// it lives on the connection. Kept distinct from the parameter-array flag so a backend that honours
    /// one but not the other can be carved out independently.
    ///
    /// @return `true` if the driver honours row-array fetching.
    [[nodiscard]] bool SupportsNativeRowArrayFetch() const noexcept;
 
    /// @brief Server-type overload of @ref SupportsNativeRowArrayFetch, for callers that hold only the
    /// server type (e.g. the DataMapper result reader) and not the connection. Keeps the single source of
    /// truth for this capability on the connection rather than scattering a `switch (serverType)` into
    /// business logic.
    /// @param serverType The backend server type to test.
    /// @return `true` if that backend honours row-array fetching.
    [[nodiscard]] static bool SupportsNativeRowArrayFetch(SqlServerType serverType) noexcept;
 
    /// @brief Whether @p serverType's driver round-trips narrow (@c SQL_C_CHAR) character data
    /// byte-exact, so a fixed-capacity char string may be array-bound narrow on the row-wise fetch path.
    ///
    /// PostgreSQL's psqlODBC transcodes @c SQL_C_CHAR through the client codepage (cp1252 on Windows),
    /// mangling non-ASCII bytes — its single-row binder therefore reads narrow strings via @c SQL_C_WCHAR.
    /// That wide round-trip needs an external per-cell buffer + conversion, which cannot be expressed as an
    /// in-place row-wise array bind, so a record carrying a fixed-capacity string falls back to the per-row
    /// path on PostgreSQL. MS SQL Server and SQLite read @c SQL_C_CHAR verbatim and stay on the fast path.
    ///
    /// @param serverType The backend server type to test.
    /// @return `true` if narrow character data round-trips byte-exact on that backend.
    [[nodiscard]] static bool RoundTripsNarrowTextByteExact(SqlServerType serverType) noexcept;
 
    /// @brief The default block-prefetch depth applied to statements created on this connection.
    ///
    /// Classic per-row fetch loops (`while (cursor.FetchRow()) ...`, @c SqlRowIterator,
    /// @c SqlVariantRowCursor) transparently request this many rows per @c SQLFetchScroll round-trip
    /// instead of issuing one @c SQLFetch per row. A value <= 1 disables prefetch. Effective only when
    /// @ref SupportsNativeRowArrayFetch is true; otherwise statements fall back to per-row fetching.
    ///
    /// @return The configured default prefetch depth (defaults to @c PrefetchDepthDefault).
    [[nodiscard]] LIGHTWEIGHT_API std::size_t DefaultPrefetchDepth() const noexcept;
 
    /// @brief Sets the default block-prefetch depth for statements created on this connection.
    ///
    /// @param depth Rows to request per @c SQLFetchScroll round-trip on the transparent prefetch path;
    ///              a value <= 1 disables prefetch (restoring one @c SQLFetch per row).
    LIGHTWEIGHT_API void SetDefaultPrefetchDepth(std::size_t depth) noexcept;
 
    /// Creates a new query builder for the given table, compatible with the current connection.
    ///
    /// @param table The table to query.
    ///              If not provided, the query will be a generic query builder.
    [[nodiscard]] LIGHTWEIGHT_API SqlQueryBuilder Query(std::string_view const& table = {}) const;
 
    /// Creates a new query builder for the given table with an alias, compatible with the current connection.
    ///
    /// @param table The table to query.
    /// @param tableAlias The alias to use for the table.
    [[nodiscard]] LIGHTWEIGHT_API SqlQueryBuilder QueryAs(std::string_view const& table,
                                                          std::string_view const& tableAlias) const;
 
    /// Creates a new migration query builder, compatible the current connection.
    [[nodiscard]] LIGHTWEIGHT_API SqlMigrationQueryBuilder Migration() const;
 
    /// Tests if a transaction is active.
    [[nodiscard]] LIGHTWEIGHT_API bool TransactionActive() const noexcept;
 
    /// Tests if transactions are allowed.
    [[nodiscard]] LIGHTWEIGHT_API bool TransactionsAllowed() const noexcept;
 
    /// Tests if the connection is still active.
    [[nodiscard]] LIGHTWEIGHT_API bool IsAlive() const noexcept;
 
    /// Retrieves the connection information.
    [[nodiscard]] LIGHTWEIGHT_API SqlConnectionString const& ConnectionString() const noexcept;
 
    /// Retrieves the native handle.
    [[nodiscard]] SQLHDBC NativeHandle() const noexcept
    {
        return m_hDbc;
    }
 
    /// Retrieves the last time the connection was used.
    [[nodiscard]] LIGHTWEIGHT_API std::chrono::steady_clock::time_point LastUsed() const noexcept;
 
    /// Sets the last time the connection was used.
    LIGHTWEIGHT_API void SetLastUsed(std::chrono::steady_clock::time_point lastUsed) noexcept;
 
    /// Checks the result of an SQL operation, and throws an exception if it is not successful.
    LIGHTWEIGHT_API void RequireSuccess(SQLRETURN sqlResult,
                                        std::source_location sourceLocation = std::source_location::current()) const;
 
    /// Enables coroutine-based asynchronous methods on this connection.
    ///
    /// Wires the connection to an injected execution context: blocking ODBC work is offloaded to
    /// @p dbWorkers (serialized per connection so the ODBC handle is only ever used by one thread
    /// at a time) and the awaiting coroutine is resumed via @p resume (typically the application's
    /// run loop). A native driver-async backend is selected when the driver advertises support;
    /// otherwise the portable thread-offload backend is used.
    ///
    /// Both executors must outlive this connection and every coroutine driven through it.
    /// Must not be called while asynchronous operations are in flight on this connection (it
    /// replaces the backend); the connection pool calls @ref DisableAsync on return so each
    /// re-acquire is a fresh enable rather than a live replacement.
    ///
    /// @param dbWorkers The worker-thread pool used to run blocking ODBC calls.
    /// @param resume The scheduler used to resume coroutines after a blocking step completes.
    LIGHTWEIGHT_API void EnableAsync(Async::IExecutor& dbWorkers, Async::IResumeScheduler& resume);
 
    /// Enables the asynchronous API on this connection using an explicitly provided backend.
    ///
    /// This is the dependency-injection seam behind the convenience overload above: callers and
    /// tests can supply any @ref Async::IAsyncBackend implementation (a fake/inline backend for
    /// tests, or a future native event backend) instead of the default thread-offload backend.
    /// The same in-flight / lifetime constraints as the convenience overload apply.
    ///
    /// @param backend The async execution backend to install (consumed); must not be null.
    LIGHTWEIGHT_API void EnableAsync(std::unique_ptr<Async::IAsyncBackend> backend);
 
    /// Tears down the asynchronous backend, returning the connection to a non-async state.
    ///
    /// Called by the connection pool when a mapper is returned, so a recycled connection never
    /// carries a stale backend that references executors which may have been destroyed. Safe to
    /// call when async was never enabled.
    ///
    /// @warning Must not be called while async work is in flight on this connection: it destroys the
    /// backend (and the strand/executors an outstanding offloaded step still references), which would
    /// race the worker still touching the ODBC handle. Await every async operation on this connection
    /// to completion before disabling (or before returning the owning pooled @c DataMapper).
    LIGHTWEIGHT_API void DisableAsync() noexcept;
 
    /// @return true if @ref EnableAsync has been called on this connection (and not yet disabled).
    [[nodiscard]] LIGHTWEIGHT_API bool IsAsyncEnabled() const noexcept;
 
    /// Retrieves the connection's asynchronous execution backend.
    ///
    /// Non-const because using the backend schedules and serializes real ODBC work (via its strand and
    /// resume scheduler), which is an observable mutation of the connection's execution state.
    ///
    /// @pre @ref IsAsyncEnabled returns true.
    /// @throws std::logic_error if async has not been enabled (programmer error, fail-fast).
    /// @return The async backend used by this connection's async methods.
    [[nodiscard]] LIGHTWEIGHT_API Async::IAsyncBackend& AsyncBackend();
 
  private:
    /// Ensures ODBC handles are allocated. Called by Connect() methods.
    /// If handles were freed by Close(), this method reallocates them.
    void EnsureHandlesAllocated();
 
    void PostConnect();
 
    // Private data members
    // Note: move/move assignment operators implemented manually
    // if adding new data members, make sure to update them accordingly.
    SQLHENV m_hEnv {};
    SQLHDBC m_hDbc {};
    uint64_t m_connectionId;
    SqlServerType m_serverType = SqlServerType::UNKNOWN;
    SqlQueryFormatter const* m_queryFormatter {};
    std::string m_driverName;
 
    struct Data;
    Data* m_data {};
};
 
inline SqlServerType SqlConnection::ServerType() const noexcept
{
    return m_serverType;
}
 
inline std::string const& SqlConnection::DriverName() const noexcept
{
    return m_driverName;
}
 
inline SqlQueryFormatter const& SqlConnection::QueryFormatter() const noexcept
{
    return *m_queryFormatter;
}
 
inline bool SqlConnection::SupportsNativeRowBatch() const noexcept
{
    // Native ODBC parameter-array binding (SQL_ATTR_PARAMSET_SIZE > 1) is a per-driver capability.
    // Every backend Lightweight supports and tests against honours it; an unverified backend takes the
    // always-correct per-row path rather than risk a driver that silently ignores the parameter array.
    switch (ServerType())
    {
        case SqlServerType::MICROSOFT_SQL:
        case SqlServerType::POSTGRESQL:
        case SqlServerType::SQLITE:
            return true;
        case SqlServerType::MYSQL:
        case SqlServerType::UNKNOWN:
            return false;
    }
    return false;
}
 
inline bool SqlConnection::SupportsNativeRowArrayFetch(SqlServerType serverType) noexcept
{
    // Native ODBC row-array fetching (SQL_ATTR_ROW_ARRAY_SIZE > 1 + SQLFetchScroll) is a per-driver
    // capability. Every backend Lightweight supports and tests against honours it; an unverified backend
    // takes the always-correct per-row SQLFetch path rather than risk a driver that mis-handles the array.
    switch (serverType)
    {
        case SqlServerType::MICROSOFT_SQL:
        case SqlServerType::POSTGRESQL:
        case SqlServerType::SQLITE:
            return true;
        case SqlServerType::MYSQL:
        case SqlServerType::UNKNOWN:
            return false;
    }
    return false;
}
 
inline bool SqlConnection::SupportsNativeRowArrayFetch() const noexcept
{
    return SupportsNativeRowArrayFetch(ServerType());
}
 
inline bool SqlConnection::RoundTripsNarrowTextByteExact(SqlServerType serverType) noexcept
{
    switch (serverType)
    {
        case SqlServerType::MICROSOFT_SQL:
        case SqlServerType::SQLITE:
            return true;
        case SqlServerType::POSTGRESQL: // psqlODBC transcodes SQL_C_CHAR through the client codepage
        case SqlServerType::MYSQL:
        case SqlServerType::UNKNOWN:
            return false;
    }
    return false;
}
 
} // namespace Lightweight