0
0
mirror of https://github.com/rbock/sqlpp11.git synced 2024-11-16 04:47:18 +08:00

Add support for dynamic insert_set and corresponding tests

This commit is contained in:
Roland Bock 2024-09-15 15:16:48 +02:00
parent 57dd15143d
commit 2198f95dab
8 changed files with 178 additions and 61 deletions

View File

@ -27,6 +27,7 @@
*/ */
#include <sqlpp11/core/operator/assign_expression.h> #include <sqlpp11/core/operator/assign_expression.h>
#include <sqlpp11/core/query/dynamic.h>
#include <sqlpp11/core/basic/column_fwd.h> #include <sqlpp11/core/basic/column_fwd.h>
#include <sqlpp11/core/clause/insert_value.h> #include <sqlpp11/core/clause/insert_value.h>
#include <sqlpp11/core/tuple_to_sql_string.h> #include <sqlpp11/core/tuple_to_sql_string.h>
@ -81,6 +82,7 @@ namespace sqlpp
_data_t _data; _data_t _data;
#warning: Need to check that all columns actually have a default value!
using _consistency_check = consistent_t; using _consistency_check = consistent_t;
}; };
}; };
@ -89,7 +91,7 @@ namespace sqlpp
struct insert_list_data_t struct insert_list_data_t
{ {
insert_list_data_t(std::tuple<Assignments...> assignments) insert_list_data_t(std::tuple<Assignments...> assignments)
: _assignments(assignments), _columns(columns_from_tuple(assignments)), _values(values_from_tuple(assignments)) : _assignments(std::move(assignments))
{ {
} }
@ -99,39 +101,7 @@ namespace sqlpp
insert_list_data_t& operator=(insert_list_data_t&&) = default; insert_list_data_t& operator=(insert_list_data_t&&) = default;
~insert_list_data_t() = default; ~insert_list_data_t() = default;
std::tuple<Assignments...> _assignments; // FIXME: Need to replace _columns and _values by _assignments std::tuple<Assignments...> _assignments;
// (connector-container requires assignments)
std::tuple<simple_column_t<lhs_t<Assignments>>...> _columns;
std::tuple<rhs_t<Assignments>...> _values;
private:
template <size_t... Indexes>
auto columns_from_tuple(::sqlpp::index_sequence<Indexes...>, std::tuple<Assignments...> assignments)
-> decltype(_columns)
{
(void)assignments;
return decltype(_columns)(std::get<Indexes>(assignments)._l...);
}
auto columns_from_tuple(std::tuple<Assignments...> assignments) -> decltype(_columns)
{
const auto seq = ::sqlpp::make_index_sequence<sizeof...(Assignments)>{};
return columns_from_tuple(seq, assignments);
}
template <size_t... Indexes>
auto values_from_tuple(::sqlpp::index_sequence<Indexes...>, std::tuple<Assignments...> assignments)
-> decltype(_values)
{
(void)assignments;
return decltype(_values)(std::get<Indexes>(assignments)._r...);
}
auto values_from_tuple(std::tuple<Assignments...> assignments) -> decltype(_values)
{
const auto seq = ::sqlpp::make_index_sequence<sizeof...(Assignments)>{};
return values_from_tuple(seq, assignments);
}
}; };
SQLPP_PORTABLE_STATIC_ASSERT(assert_insert_set_assignments_t, "at least one argument is not an assignment in set()"); SQLPP_PORTABLE_STATIC_ASSERT(assert_insert_set_assignments_t, "at least one argument is not an assignment in set()");
@ -145,35 +115,29 @@ namespace sqlpp
SQLPP_PORTABLE_STATIC_ASSERT(assert_insert_dynamic_set_statement_dynamic_t, SQLPP_PORTABLE_STATIC_ASSERT(assert_insert_dynamic_set_statement_dynamic_t,
"dynamic_set must not be called in a static statement"); "dynamic_set must not be called in a static statement");
template <typename... Assignments>
using check_insert_set_t = static_combined_check_t<
static_check_t<logic::all<is_assignment<Assignments>::value...>::value,
assert_insert_set_assignments_t>,
static_check_t<not detail::has_duplicates<typename lhs<Assignments>::type...>::value,
assert_insert_set_no_duplicates_t>,
static_check_t<sizeof...(Assignments) == 0 or detail::type_vector_cat_t<required_tables_of_t<
typename lhs<Assignments>::type>...>::are_same(),
assert_insert_set_single_table_t>>;
// workaround for msvc bug https://connect.microsoft.com/VisualStudio/Feedback/Details/2173269 // workaround for msvc bug https://connect.microsoft.com/VisualStudio/Feedback/Details/2173269
// template <typename... Assignments> // template <typename... Assignments>
// using check_insert_static_set_t = // using check_insert_set_t =
// static_combined_check_t<check_insert_set_t<Assignments...>, // static_combined_check_t<check_insert_set_t<Assignments...>,
// static_check_t<sizeof...(Assignments) != 0, assert_insert_static_set_count_args_t>, // static_check_t<sizeof...(Assignments) != 0, assert_insert_static_set_count_args_t>,
// static_check_t<detail::have_all_required_columns<lhs_t<Assignments>...>::value, // static_check_t<detail::have_all_required_columns<lhs_t<Assignments>...>::value,
// assert_insert_static_set_all_required_t>>; // assert_insert_static_set_all_required_t>>;
template <typename... Assignments> template <typename... Assignments>
struct check_insert_static_set struct check_insert_set
{ {
using type = static_combined_check_t< using type = static_combined_check_t<
check_insert_set_t<Assignments...>, static_check_t<logic::all<is_assignment<Assignments>::value...>::value, assert_insert_set_assignments_t>,
static_check_t<not detail::has_duplicates<typename lhs<Assignments>::type...>::value,
assert_insert_set_no_duplicates_t>,
static_check_t<detail::type_vector_cat_t<required_tables_of_t<typename lhs<Assignments>::type>...>::are_same(),
assert_insert_set_single_table_t>,
static_check_t<sizeof...(Assignments) != 0, assert_insert_static_set_count_args_t>, static_check_t<sizeof...(Assignments) != 0, assert_insert_static_set_count_args_t>,
static_check_t<detail::have_all_required_columns<typename lhs<Assignments>::type...>::value, static_check_t<detail::have_all_required_columns<typename lhs<Assignments>::type...>::value,
assert_insert_static_set_all_required_t>>; assert_insert_static_set_all_required_t>>;
}; };
template <typename... Assignments> template <typename... Assignments>
using check_insert_static_set_t = typename check_insert_static_set<Assignments...>::type; using check_insert_set_t = typename check_insert_set<remove_dynamic_t<Assignments>...>::type;
SQLPP_PORTABLE_STATIC_ASSERT( SQLPP_PORTABLE_STATIC_ASSERT(
assert_no_unknown_tables_in_insert_assignments_t, assert_no_unknown_tables_in_insert_assignments_t,
@ -230,7 +194,7 @@ namespace sqlpp
~column_list_data_t() = default; ~column_list_data_t() = default;
using _value_tuple_t = std::tuple<insert_value_t<Columns>...>; using _value_tuple_t = std::tuple<insert_value_t<Columns>...>;
std::tuple<simple_column_t<Columns>...> _columns; std::tuple<Columns...> _columns;
std::vector<_value_tuple_t> _insert_values; std::vector<_value_tuple_t> _insert_values;
}; };
@ -347,17 +311,17 @@ namespace sqlpp
template <typename... Assignments> template <typename... Assignments>
auto set(Assignments... assignments) const auto set(Assignments... assignments) const
-> _new_statement_t<check_insert_static_set_t<Assignments...>, insert_list_t<Assignments...>> -> _new_statement_t<check_insert_set_t<Assignments...>, insert_list_t<Assignments...>>
{ {
using Check = check_insert_static_set_t<Assignments...>; using Check = check_insert_set_t<Assignments...>;
return _set_impl(Check{}, std::make_tuple(assignments...)); return _set_impl(Check{}, std::make_tuple(assignments...));
} }
template <typename... Assignments> template <typename... Assignments>
auto set(std::tuple<Assignments...> assignments) const auto set(std::tuple<Assignments...> assignments) const
-> _new_statement_t<check_insert_static_set_t<Assignments...>, insert_list_t<Assignments...>> -> _new_statement_t<check_insert_set_t<Assignments...>, insert_list_t<Assignments...>>
{ {
using Check = check_insert_static_set_t<Assignments...>; using Check = check_insert_set_t<Assignments...>;
return _set_impl(Check{}, assignments); return _set_impl(Check{}, assignments);
} }
@ -426,17 +390,74 @@ namespace sqlpp
return result; return result;
} }
// Used to serialize left hand side of assignment tuple that should ignore dynamic elements.
struct tuple_lhs_assignment_operand_no_dynamic
{
template <typename Context, typename Lhs, typename Op, typename Rhs>
auto operator()(Context& context, const assign_expression<Lhs, Op, Rhs>&, size_t ) const -> std::string
{
const auto prefix = need_prefix ? std::string{separator} : std::string{};
need_prefix = true;
return prefix + name_to_sql_string(context, name_tag_of_t<Lhs>::name);
}
template <typename Context, typename T>
auto operator()(Context& context, const sqlpp::dynamic_t<T>& t, size_t index) const -> std::string
{
if (t._condition)
{
return operator()(context, t._expr, index);
}
return "";
}
sqlpp::string_view separator;
mutable bool need_prefix = false;
};
// Used to serialize right hand side of assignment tuple that should ignore dynamic elements.
struct tuple_rhs_assignment_operand_no_dynamic
{
template <typename Context, typename Lhs, typename Op, typename Rhs>
auto operator()(Context& context, const assign_expression<Lhs, Op, Rhs>& t, size_t ) const -> std::string
{
const auto prefix = need_prefix ? std::string{separator} : std::string{};
need_prefix = true;
return prefix + operand_to_sql_string(context, t._r);
}
template <typename Context, typename T>
auto operator()(Context& context, const sqlpp::dynamic_t<T>& t, size_t index) const -> std::string
{
if (t._condition)
{
return operator()(context, t._expr, index);
}
return "";
}
sqlpp::string_view separator;
mutable bool need_prefix = false;
};
template <typename Context, typename... Assignments> template <typename Context, typename... Assignments>
auto to_sql_string(Context& context, const insert_list_data_t<Assignments...>& t) -> std::string auto to_sql_string(Context& context, const insert_list_data_t<Assignments...>& t) -> std::string
{ {
auto result = std::string{" ("}; auto result = std::string{" ("};
result += tuple_to_sql_string(context, t._columns, tuple_operand{", "}); result += tuple_to_sql_string(context, t._assignments, tuple_lhs_assignment_operand_no_dynamic{", "});
result += ") VALUES("; result += ") VALUES(";
result += tuple_to_sql_string(context, t._values, tuple_operand{", "}); result += tuple_to_sql_string(context, t._assignments, tuple_rhs_assignment_operand_no_dynamic{", "});
result += ")"; result += ")";
return result; return result;
} }
template <typename... Assignments>
auto insert_default_values()
-> decltype(statement_t<no_insert_value_list_t>().default_values())
{
return statement_t<no_insert_value_list_t>().default_values();
}
template <typename... Assignments> template <typename... Assignments>
auto insert_set(Assignments... assignments) auto insert_set(Assignments... assignments)
-> decltype(statement_t<no_insert_value_list_t>().set(assignments...)) -> decltype(statement_t<no_insert_value_list_t>().set(assignments...))

View File

@ -26,6 +26,7 @@
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/ */
#include <sqlpp11/core/query/dynamic.h>
#include <sqlpp11/core/detail/type_set.h> #include <sqlpp11/core/detail/type_set.h>
#include <sqlpp11/core/tuple_to_sql_string.h> #include <sqlpp11/core/tuple_to_sql_string.h>
#include <sqlpp11/core/type_traits.h> #include <sqlpp11/core/type_traits.h>

View File

@ -45,7 +45,7 @@ namespace sqlpp
sqlpp::string_view separator; sqlpp::string_view separator;
}; };
// Used to serialize tuple tuples that should ignore dynamic elements. // Used to serialize tuple that should ignore dynamic elements.
struct tuple_operand_no_dynamic struct tuple_operand_no_dynamic
{ {
template <typename Context, typename T> template <typename Context, typename T>

View File

@ -299,6 +299,12 @@ namespace sqlpp
using type = void; using type = void;
}; };
template <typename T>
struct lhs<dynamic_t<T>>
{
using type = dynamic_t<typename lhs<T>::type>;
};
template <typename T> template <typename T>
using lhs_t = typename lhs<T>::type; using lhs_t = typename lhs<T>::type;
@ -308,6 +314,12 @@ namespace sqlpp
using type = void; using type = void;
}; };
template <typename T>
struct rhs<dynamic_t<T>>
{
using type = dynamic_t<typename rhs<T>::type>;
};
template <typename T> template <typename T>
using rhs_t = typename rhs<T>::type; using rhs_t = typename rhs<T>::type;

View File

@ -30,6 +30,8 @@ function(create_test name)
endfunction() endfunction()
create_test(group_by) create_test(group_by)
create_test(insert_default_values)
create_test(insert_set)
create_test(select_columns) create_test(select_columns)
create_test(select_flags) create_test(select_flags)
create_test(update_set) create_test(update_set)

View File

@ -0,0 +1,36 @@
/*
* Copyright (c) 2024, 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.
*/
#include "../compare.h"
#include <sqlpp11/sqlpp11.h>
int main(int, char* [])
{
#warning: This looks different for MySQL
SQLPP_COMPARE(sqlpp::insert_default_values(), " DEFAULT VALUES");
return 0;
}

View File

@ -0,0 +1,50 @@
/*
* Copyright (c) 2024, 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.
*/
#include "Sample.h"
#include "../compare.h"
#include <sqlpp11/sqlpp11.h>
SQLPP_CREATE_NAME_TAG(v);
int main(int, char* [])
{
const auto val = sqlpp::value(17);
const auto expr = sqlpp::value(17) + 4;
const auto foo = test::TabFoo{};
// Plain assignments.
SQLPP_COMPARE(insert_set(foo.id = 7), " (id) VALUES(7)");
SQLPP_COMPARE(insert_set(foo.id = 7, foo.textNnD = "cheesecake"), " (id, text_nn_d) VALUES(7, 'cheesecake')");
// Dynamic assignments.
SQLPP_COMPARE(insert_set(sqlpp::dynamic(true, foo.id = 7), sqlpp::dynamic(false, foo.textNnD = "cheesecake")),
" (id) VALUES(7)");
SQLPP_COMPARE(insert_set(sqlpp::dynamic(false, foo.id = 7), sqlpp::dynamic(true, foo.textNnD = "cheesecake")),
" (text_nn_d) VALUES('cheesecake')");
return 0;
}

View File

@ -27,13 +27,8 @@
#include "../compare.h" #include "../compare.h"
#include <sqlpp11/sqlpp11.h> #include <sqlpp11/sqlpp11.h>
SQLPP_CREATE_NAME_TAG(v);
int main(int, char* []) int main(int, char* [])
{ {
const auto val = sqlpp::value(17);
const auto expr = sqlpp::value(17) + 4;
const auto foo = test::TabFoo{}; const auto foo = test::TabFoo{};
// Plain assignments. // Plain assignments.