mirror of
https://github.com/rbock/sqlpp11.git
synced 2024-11-16 04:47:18 +08:00
125 lines
4.0 KiB
C++
125 lines
4.0 KiB
C++
/*
|
|
* Copyright (c) 2013-2014, 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_COUNT_H
|
|
#define SQLPP_COUNT_H
|
|
|
|
#include <sqlpp11/char_sequence.h>
|
|
#include <sqlpp11/select_flags.h>
|
|
#include <sqlpp11/integral.h>
|
|
|
|
namespace sqlpp
|
|
{
|
|
template<typename Flag, typename Expr>
|
|
struct count_t:
|
|
public expression_operators<count_t<Flag, Expr>, integral>,
|
|
public alias_operators<count_t<Flag, Expr>>
|
|
{
|
|
using _traits = make_traits<integral, tag::is_expression, tag::is_selectable>;
|
|
struct _recursive_traits
|
|
{
|
|
using _required_ctes = required_ctes_of<Expr>;
|
|
using _provided_ctes = detail::type_set<>;
|
|
using _required_tables = required_tables_of<Expr>;
|
|
using _provided_tables = provided_tables_of<Expr>;
|
|
using _provided_outer_tables = provided_outer_tables_of<Expr>;
|
|
using _extra_tables = extra_tables_of<Expr>;
|
|
using _parameters = parameters_of<Expr>;
|
|
using _tags = detail::make_difference_set_t<detail::joined_set_t<recursive_tags_of<Expr>, recursive_tags_of<aggregate_function>>,
|
|
detail::type_set<tag::can_be_null>>;
|
|
};
|
|
|
|
|
|
static_assert(is_noop<Flag>::value or std::is_same<distinct_t, Flag>::value, "count() used with flag other than 'distinct'");
|
|
|
|
struct _alias_t
|
|
{
|
|
static constexpr const char _literal[] = "count_";
|
|
using _name_t = sqlpp::make_char_sequence<sizeof(_literal), _literal>;
|
|
template<typename T>
|
|
struct _member_t
|
|
{
|
|
T count;
|
|
T& operator()() { return count; }
|
|
const T& operator()() const { return count; }
|
|
};
|
|
};
|
|
|
|
count_t(const Expr expr):
|
|
_expr(expr)
|
|
{}
|
|
|
|
count_t(const count_t&) = default;
|
|
count_t(count_t&&) = default;
|
|
count_t& operator=(const count_t&) = default;
|
|
count_t& operator=(count_t&&) = default;
|
|
~count_t() = default;
|
|
|
|
Expr _expr;
|
|
};
|
|
|
|
template<typename Context, typename Flag, typename Expr>
|
|
struct serializer_t<Context, count_t<Flag, Expr>>
|
|
{
|
|
using _serialize_check = serialize_check_of<Context, Flag, Expr>;
|
|
using T = count_t<Flag, Expr>;
|
|
|
|
static Context& _(const T& t, Context& context)
|
|
{
|
|
context << "COUNT(";
|
|
if (std::is_same<distinct_t, Flag>::value)
|
|
{
|
|
serialize(Flag(), context);
|
|
context << ' ';
|
|
serialize_operand(t._expr, context);
|
|
}
|
|
else
|
|
{
|
|
serialize(t._expr, context);
|
|
}
|
|
context << ")";
|
|
return context;
|
|
}
|
|
};
|
|
|
|
template<typename T>
|
|
auto count(T t) -> count_t<noop, wrap_operand_t<T>>
|
|
{
|
|
static_assert(is_expression_t<wrap_operand_t<T>>::value, "count() requires an expression as argument");
|
|
return { t };
|
|
}
|
|
|
|
template<typename T>
|
|
auto count(const distinct_t&, T t) -> count_t<distinct_t, wrap_operand_t<T>>
|
|
{
|
|
static_assert(is_expression_t<wrap_operand_t<T>>::value, "count() requires an expression as argument");
|
|
return { t };
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|