query.hpp 2.74 KB
Newer Older
1 2
#pragma once

3 4
#include <biboumi.h>

5
#include <utils/optional_bool.hpp>
louiz’'s avatar
louiz’ committed
6
#include <utils/datetime.hpp>
7
#include <database/statement.hpp>
8
#include <database/column.hpp>
9

10 11 12 13
#include <logger/logger.hpp>

#include <vector>
#include <string>
louiz’'s avatar
louiz’ committed
14
#include <database/datetime_writer.hpp>
15

louiz’'s avatar
louiz’ committed
16
void actual_bind(Statement& statement, const std::string& value, int index);
17
void actual_bind(Statement& statement, const std::int64_t& value, int index);
18
template <typename T, typename std::enable_if_t<std::is_integral<T>::value>* = 0>
19 20 21 22
void actual_bind(Statement& statement, const T& value, int index)
{
  actual_bind(statement, static_cast<std::int64_t>(value), index);
}
23
void actual_bind(Statement& statement, const OptionalBool& value, int index);
louiz’'s avatar
louiz’ committed
24
void actual_bind(Statement& statement, const DateTime& value, int index);
louiz’'s avatar
louiz’ committed
25

26 27 28 29 30 31 32 33 34 35 36 37 38 39
#ifdef DEBUG_SQL_QUERIES
#include <utils/scopetimer.hpp>

inline auto make_sql_timer()
{
  return make_scope_timer([](const std::chrono::steady_clock::duration& elapsed)
                          {
                            const auto seconds = std::chrono::duration_cast<std::chrono::seconds>(elapsed);
                            const auto rest = elapsed - seconds;
                            log_debug("Query executed in ", seconds.count(), ".", rest.count(), "s.");
                          });
}
#endif

40 41 42 43
struct Query
{
    std::string body;
    std::vector<std::string> params;
louiz’'s avatar
louiz’ committed
44
    int current_param{1};
45 46 47 48

    Query(std::string str):
        body(std::move(str))
    {}
49 50 51 52 53 54 55 56 57 58 59 60

#ifdef DEBUG_SQL_QUERIES
    auto log_and_time()
    {
       std::ostringstream os;
       os << this->body << "; ";
       for (const auto& param: this->params)
         os << "'" << param << "' ";
       log_debug("SQL QUERY: ", os.str());
       return make_sql_timer();
    }
#endif
61 62
};

63 64 65 66 67 68 69
template <typename ColumnType>
void add_param(Query& query, const ColumnType& column)
{
  std::cout << "add_param<ColumnType>" << std::endl;
  actual_add_param(query, column.value);
}

70 71 72 73 74 75
template <typename T>
void actual_add_param(Query& query, const T& val)
{
  query.params.push_back(std::to_string(val));
}

76
void actual_add_param(Query& query, const std::string& val);
77
void actual_add_param(Query& query, const OptionalBool& val);
78 79

template <typename T>
louiz’'s avatar
louiz’ committed
80
typename std::enable_if<!std::is_arithmetic<T>::value, Query&>::type
81 82 83 84 85 86 87 88 89
operator<<(Query& query, const T&)
{
  query.body += T::name;
  return query;
}

template <typename Integer>
typename std::enable_if<std::is_integral<Integer>::value, Query&>::type
operator<<(Query& query, const Integer& i)
90
{
91 92
  query.body += "$" + std::to_string(query.current_param++);
  actual_add_param(query, i);
93 94
  return query;
}
95

louiz’'s avatar
louiz’ committed
96 97 98
Query& operator<<(Query& query, const char* str);
Query& operator<<(Query& query, const std::string& str);
Query& operator<<(Query& query, const DatetimeWriter& datetime);