mirror of
https://github.com/rbock/sqlpp11.git
synced 2026-01-08 14:10:51 -06:00
123 lines
3.9 KiB
C++
123 lines
3.9 KiB
C++
/*
|
|
* Copyright (c) 2013, Roland Bock
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without modification,
|
|
* are permitted provided that the following conditions are met:
|
|
*
|
|
* Redistributions of source code must retain the above copyright notice, this
|
|
* list of conditions and the following disclaimer.
|
|
*
|
|
* Redistributions in binary form must reproduce the above copyright notice, this
|
|
* list of conditions and the following disclaimer in the documentation and/or
|
|
* other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
|
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
|
|
* ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
|
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
|
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
|
|
* ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#ifndef SQLPP_UPDATE_H
|
|
#define SQLPP_UPDATE_H
|
|
|
|
#include <sstream>
|
|
#include <sqlpp11/noop.h>
|
|
#include <sqlpp11/assignment_list.h>
|
|
#include <sqlpp11/where.h>
|
|
#include <sqlpp11/type_traits.h>
|
|
|
|
namespace sqlpp
|
|
{
|
|
template<
|
|
typename Table = noop,
|
|
typename Assignments = noop,
|
|
typename Where = noop
|
|
>
|
|
struct update_t;
|
|
|
|
template<
|
|
typename Table,
|
|
typename Assignments,
|
|
typename Where
|
|
>
|
|
struct update_t
|
|
{
|
|
static_assert(is_noop<Table>::value or is_table_t<Table>::value, "invalid 'Table' argument");
|
|
static_assert(is_noop<Assignments>::value or is_assignment_list_t<Assignments>::value, "invalid 'Assignments' arguments");
|
|
static_assert(is_noop<Where>::value or is_where_t<Where>::value, "invalid 'Where' argument");
|
|
|
|
template<typename... Assignment>
|
|
using add_assignments_t = update_t<Table, assignment_list_t<must_not_update_t, typename std::decay<Assignment>::type...>, Where>;
|
|
template<typename Expr>
|
|
using add_where_t = update_t<Table, Assignments, where_t<typename std::decay<Expr>::type>>;
|
|
|
|
template<typename... Assignment>
|
|
add_assignments_t<Assignment...> set(Assignment&&... assignment)
|
|
{
|
|
static_assert(std::is_same<Assignments, noop>::value, "cannot call set() twice");
|
|
return {
|
|
_table,
|
|
{std::tuple<typename std::decay<Assignment>::type...>{std::forward<Assignment>(assignment)...}},
|
|
_where,
|
|
};
|
|
}
|
|
|
|
template<typename Expr>
|
|
add_where_t<Expr> where(Expr&& where)
|
|
{
|
|
static_assert(not std::is_same<Assignments, noop>::value, "cannot call where() if set() hasn't been called yet");
|
|
static_assert(std::is_same<Where, noop>::value, "cannot call where() twice");
|
|
return {
|
|
_table,
|
|
_assignments,
|
|
{std::forward<Expr>(where)},
|
|
};
|
|
}
|
|
|
|
template<typename Db>
|
|
const update_t& serialize(std::ostream& os, Db& db) const
|
|
{
|
|
os << "UPDATE ";
|
|
_table.serialize(os, db);
|
|
_assignments.serialize(os, db);
|
|
_where.serialize(os, db);
|
|
return *this;
|
|
}
|
|
|
|
template<typename Db>
|
|
update_t& serialize(std::ostream& os, Db& db)
|
|
{
|
|
static_cast<const update_t*>(this)->serialize(os, db);
|
|
return *this;
|
|
}
|
|
|
|
template<typename Db>
|
|
std::size_t run(Db& db) const
|
|
{
|
|
std::ostringstream oss;
|
|
serialize(oss, db);
|
|
return db.update(oss.str());
|
|
}
|
|
|
|
Table _table;
|
|
Assignments _assignments;
|
|
Where _where;
|
|
};
|
|
|
|
template<typename Table>
|
|
constexpr update_t<typename std::decay<Table>::type> update(Table&& table)
|
|
{
|
|
return {std::forward<Table>(table)};
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|