insert_query.hpp 3.74 KB
Newer Older
1 2
#pragma once

3
#include <database/statement.hpp>
4 5 6 7 8 9 10 11 12
#include <database/column.hpp>
#include <database/query.hpp>
#include <logger/logger.hpp>

#include <type_traits>
#include <vector>
#include <string>
#include <tuple>

louiz’'s avatar
louiz’ committed
13
template <std::size_t N=0, typename... T>
14 15
typename std::enable_if<N < sizeof...(T), void>::type
update_autoincrement_id(std::tuple<T...>& columns, Statement& statement)
16
{
17 18 19 20
  using ColumnType = typename std::decay<decltype(std::get<N>(columns))>::type;
  if (std::is_same<ColumnType, Id>::value)
    auto&& column = std::get<Id>(columns);
  update_autoincrement_id<N+1>(columns, statement);
21 22
}

23 24 25 26 27
template <std::size_t N=0, typename... T>
typename std::enable_if<N == sizeof...(T), void>::type
update_autoincrement_id(std::tuple<T...>&, Statement&)
{}

louiz’'s avatar
louiz’ committed
28 29 30 31 32 33 34 35 36 37 38 39
template <typename T>
std::string before_value()
{
  return {};
}

template <typename T>
std::string after_value()
{
  return {};
}

40 41
struct InsertQuery: public Query
{
louiz’'s avatar
louiz’ committed
42 43 44
  template <typename... T>
  InsertQuery(const std::string& name, const std::tuple<T...>& columns):
      Query("INSERT INTO ")
45 46
  {
    this->body += name;
louiz’'s avatar
louiz’ committed
47 48
    this->insert_col_names(columns);
    this->insert_values(columns);
49 50 51
  }

  template <typename... T>
louiz’'s avatar
louiz’ committed
52
  void execute(DatabaseEngine& db, std::tuple<T...>& columns)
53
  {
54 55 56 57
#ifdef DEBUG_SQL_QUERIES
    const auto timer = this->log_and_time();
#endif

louiz’'s avatar
louiz’ committed
58 59 60 61 62 63 64
    auto statement = db.prepare(this->body);
    this->bind_param(columns, *statement);

    if (statement->step() != StepResult::Error)
      db.extract_last_insert_rowid(*statement);
    else
      log_error("Failed to extract the rowid from the last INSERT");
65 66
  }

67
  template <int N=0, typename... T>
68 69
  typename std::enable_if<N < sizeof...(T), void>::type
  bind_param(const std::tuple<T...>& columns, Statement& statement, int index=1)
70
  {
71 72
    auto&& column = std::get<N>(columns);
    using ColumnType = std::decay_t<decltype(column)>;
louiz’'s avatar
louiz’ committed
73

74 75
    if (!std::is_same<ColumnType, Id>::value)
      actual_bind(statement, column.value, index++);
76

77
    this->bind_param<N+1>(columns, statement, index);
78 79
  }

80 81 82 83 84
  template <int N=0, typename... T>
  typename std::enable_if<N == sizeof...(T), void>::type
  bind_param(const std::tuple<T...>&, Statement&, int)
  {}

85 86 87
  template <typename... T>
  void insert_values(const std::tuple<T...>& columns)
  {
louiz’'s avatar
louiz’ committed
88
    this->body += " VALUES (";
89
    this->insert_value(columns);
90 91 92
    this->body += ")";
  }

93
  template <int N=0, typename... T>
94 95
  typename std::enable_if<N < sizeof...(T), void>::type
  insert_value(const std::tuple<T...>& columns, int index=1)
96
  {
97 98 99
    using ColumnType = std::decay_t<decltype(std::get<N>(columns))>;

    if (!std::is_same<ColumnType, Id>::value)
louiz’'s avatar
louiz’ committed
100
      {
louiz’'s avatar
louiz’ committed
101
        this->body += before_value<ColumnType>();
102
        this->body += "$" + std::to_string(index++);
louiz’'s avatar
louiz’ committed
103
        this->body += after_value<ColumnType>();
104 105
        if (N != sizeof...(T) - 1)
          this->body += ", ";
louiz’'s avatar
louiz’ committed
106
      }
107
    this->insert_value<N+1>(columns, index);
108
  }
109 110 111 112
  template <int N=0, typename... T>
  typename std::enable_if<N == sizeof...(T), void>::type
  insert_value(const std::tuple<T...>&, const int)
  { }
113 114 115 116 117

  template <typename... T>
  void insert_col_names(const std::tuple<T...>& columns)
  {
    this->body += " (";
118
    this->insert_col_name(columns);
louiz’'s avatar
louiz’ committed
119
    this->body += ")";
120 121
  }

122
  template <int N=0, typename... T>
123 124
  typename std::enable_if<N < sizeof...(T), void>::type
  insert_col_name(const std::tuple<T...>& columns)
125
  {
126
    using ColumnType = std::decay_t<decltype(std::get<N>(columns))>;
127

128 129 130
    if (!std::is_same<ColumnType, Id>::value)
      {
        this->body += ColumnType::name;
louiz’'s avatar
louiz’ committed
131

132 133
        if (N < (sizeof...(T) - 1))
          this->body += ", ";
134
      }
135 136

    this->insert_col_name<N+1>(columns);
137
  }
138 139 140 141 142

  template <int N=0, typename... T>
  typename std::enable_if<N == sizeof...(T), void>::type
  insert_col_name(const std::tuple<T...>&)
  {}
143
};