sql.h

Go to the documentation of this file.

#ifndef DBALLE_SQL_H
#define DBALLE_SQL_H

#include <dballe/core/error.h>
#include <dballe/fwd.h>
#include <dballe/sql/fwd.h>
#include <memory>
#include <string>

#undef USE_REF_INT

// #define TRACE_DB

#ifdef TRACE_DB
#define TRACE(...) fprintf(stderr, __VA_ARGS__)
#define IFTRACE if (1)
#else

#define TRACE(...)                                                             \
    do                                                                         \
    {                                                                          \
    } while (0)

#define IFTRACE if (0)
#endif

namespace dballe {
class Datetime;

namespace sql {

enum class ServerType {
    MYSQL,
    SQLITE,
    ORACLE,
    POSTGRES,
};

const char* format_server_type(ServerType type);

class Connection : public std::enable_shared_from_this<Connection>
{
private:
    static void atfork_prepare_hook();
    static void atfork_parent_hook();
    static void atfork_child_hook();

protected:
    std::string url;

    Connection();

    void register_atfork();

    // Optional pthread_atfork hooks
    virtual void fork_prepare();
    virtual void fork_parent();
    virtual void fork_child();

public:
    ServerType server_type;

    virtual ~Connection();

    const std::string& get_url() const { return url; }

    virtual std::unique_ptr<Transaction> transaction(bool readonly = false) = 0;

    virtual bool has_table(const std::string& name) = 0;

    virtual std::string get_setting(const std::string& key) = 0;

    virtual void set_setting(const std::string& key,
                             const std::string& value) = 0;

    virtual void drop_settings() = 0;

    virtual void add_datetime(Querybuf& qb, const Datetime& dt) const;

    virtual void execute(const std::string& query) = 0;

    virtual void explain(const std::string& query, FILE* out) = 0;

    static std::shared_ptr<Connection> create(const DBConnectOptions& options);
};

class Transaction
{
public:
    Transaction() {}
    Transaction(const Transaction&)            = delete;
    Transaction& operator=(const Transaction&) = delete;
    virtual ~Transaction();

    virtual void commit() = 0;

    virtual void rollback() = 0;

    virtual void rollback_nothrow() noexcept = 0;

    virtual void lock_table(const char* name) = 0;
};

} // namespace sql
} // namespace dballe

#endif