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

3 4
#include <biboumi.h>

5
#include <utils/optional_bool.hpp>
6
#include <database/statement.hpp>
7
#include <database/column.hpp>
8

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

#include <vector>
#include <string>

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

23 24 25 26 27 28 29 30 31 32 33 34 35 36
#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

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

    Query(std::string str):
        body(std::move(str))
    {}
46 47 48 49 50 51 52 53 54 55 56 57

#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
58 59
};

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

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

73
void actual_add_param(Query& query, const std::string& val);
74
void actual_add_param(Query& query, const OptionalBool& val);
75 76

template <typename T>
77
typename std::enable_if<!std::is_integral<T>::value, Query&>::type
78 79 80 81 82 83
operator<<(Query& query, const T&)
{
  query.body += T::name;
  return query;
}

84 85
Query& operator<<(Query& query, const char* str);
Query& operator<<(Query& query, const std::string& str);
86 87 88
template <typename Integer>
typename std::enable_if<std::is_integral<Integer>::value, Query&>::type
operator<<(Query& query, const Integer& i)
89
{
90 91
  query.body += "$" + std::to_string(query.current_param++);
  actual_add_param(query, i);
92 93
  return query;
}
94