mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-20 16:50:48 +00:00
166 lines
5.4 KiB
C++
166 lines
5.4 KiB
C++
#pragma once
|
|
|
|
#include "demangle.h"
|
|
#include "getThreadId.h"
|
|
#include <type_traits>
|
|
#include <tuple>
|
|
#include <iomanip>
|
|
#include <iostream>
|
|
|
|
/** Usage:
|
|
*
|
|
* DUMP(variable...)
|
|
*/
|
|
|
|
|
|
template <typename Out, typename T>
|
|
Out & dumpValue(Out &, T &&);
|
|
|
|
|
|
/// Catch-all case.
|
|
template <int priority, typename Out, typename T>
|
|
std::enable_if_t<priority == -1, Out> & dumpImpl(Out & out, T &&)
|
|
{
|
|
return out << "{...}";
|
|
}
|
|
|
|
/// An object, that could be output with operator <<.
|
|
template <int priority, typename Out, typename T>
|
|
std::enable_if_t<priority == 0, Out> & dumpImpl(Out & out, T && x, std::decay_t<decltype(std::declval<Out &>() << std::declval<T>())> * = nullptr)
|
|
{
|
|
return out << x;
|
|
}
|
|
|
|
/// A pointer-like object.
|
|
template <int priority, typename Out, typename T>
|
|
std::enable_if_t<priority == 1
|
|
/// Protect from the case when operator * do effectively nothing (function pointer).
|
|
&& !std::is_same_v<std::decay_t<T>, std::decay_t<decltype(*std::declval<T>())>>
|
|
, Out> & dumpImpl(Out & out, T && x, std::decay_t<decltype(*std::declval<T>())> * = nullptr)
|
|
{
|
|
if (!x)
|
|
return out << "nullptr";
|
|
return dumpValue(out, *x);
|
|
}
|
|
|
|
/// Container.
|
|
template <int priority, typename Out, typename T>
|
|
std::enable_if_t<priority == 2, Out> & dumpImpl(Out & out, T && x, std::decay_t<decltype(std::begin(std::declval<T>()))> * = nullptr)
|
|
{
|
|
bool first = true;
|
|
out << "{";
|
|
for (const auto & elem : x)
|
|
{
|
|
if (first)
|
|
first = false;
|
|
else
|
|
out << ", ";
|
|
dumpValue(out, elem);
|
|
}
|
|
return out << "}";
|
|
}
|
|
|
|
|
|
/// string and const char * - output not as container or pointer.
|
|
|
|
template <int priority, typename Out, typename T>
|
|
std::enable_if_t<priority == 3 && (std::is_same_v<std::decay_t<T>, std::string> || std::is_same_v<std::decay_t<T>, const char *>), Out> &
|
|
dumpImpl(Out & out, T && x)
|
|
{
|
|
return out << std::quoted(x);
|
|
}
|
|
|
|
/// UInt8 - output as number, not char.
|
|
|
|
template <int priority, typename Out, typename T>
|
|
std::enable_if_t<priority == 3 && std::is_same_v<std::decay_t<T>, unsigned char>, Out> &
|
|
dumpImpl(Out & out, T && x)
|
|
{
|
|
return out << int(x);
|
|
}
|
|
|
|
|
|
/// Tuple, pair
|
|
template <size_t N, typename Out, typename T>
|
|
Out & dumpTupleImpl(Out & out, T && x)
|
|
{
|
|
if constexpr (N == 0)
|
|
out << "{";
|
|
else
|
|
out << ", ";
|
|
|
|
dumpValue(out, std::get<N>(x));
|
|
|
|
if constexpr (N + 1 == std::tuple_size_v<std::decay_t<T>>)
|
|
out << "}";
|
|
else
|
|
dumpTupleImpl<N + 1>(out, x);
|
|
|
|
return out;
|
|
}
|
|
|
|
template <int priority, typename Out, typename T>
|
|
std::enable_if_t<priority == 4, Out> & dumpImpl(Out & out, T && x, std::decay_t<decltype(std::get<0>(std::declval<T>()))> * = nullptr)
|
|
{
|
|
return dumpTupleImpl<0>(out, x);
|
|
}
|
|
|
|
|
|
template <int priority, typename Out, typename T>
|
|
Out & dumpDispatchPriorities(Out & out, T && x, std::decay_t<decltype(dumpImpl<priority>(std::declval<Out &>(), std::declval<T>()))> *)
|
|
{
|
|
return dumpImpl<priority>(out, x);
|
|
}
|
|
|
|
struct LowPriority { LowPriority(void *) {} };
|
|
|
|
template <int priority, typename Out, typename T>
|
|
Out & dumpDispatchPriorities(Out & out, T && x, LowPriority)
|
|
{
|
|
return dumpDispatchPriorities<priority - 1>(out, x, nullptr);
|
|
}
|
|
|
|
|
|
template <typename Out, typename T>
|
|
Out & dumpValue(Out & out, T && x)
|
|
{
|
|
return dumpDispatchPriorities<5>(out, x, nullptr);
|
|
}
|
|
|
|
|
|
template <typename Out, typename T>
|
|
Out & dump(Out & out, const char * name, T && x)
|
|
{
|
|
out << demangle(typeid(x).name()) << " " << name << " = ";
|
|
return dumpValue(out, x);
|
|
}
|
|
|
|
#ifdef __clang__
|
|
#pragma clang diagnostic ignored "-Wgnu-zero-variadic-macro-arguments"
|
|
#endif
|
|
|
|
#define DUMPVAR(VAR) ::dump(std::cerr, #VAR, (VAR)); std::cerr << "; ";
|
|
#define DUMPHEAD std::cerr << __FILE__ << ':' << __LINE__ << " [ " << getThreadId() << " ] ";
|
|
#define DUMPTAIL std::cerr << '\n';
|
|
|
|
#define DUMP1(V1) do { DUMPHEAD DUMPVAR(V1) DUMPTAIL } while(0)
|
|
#define DUMP2(V1, V2) do { DUMPHEAD DUMPVAR(V1) DUMPVAR(V2) DUMPTAIL } while(0)
|
|
#define DUMP3(V1, V2, V3) do { DUMPHEAD DUMPVAR(V1) DUMPVAR(V2) DUMPVAR(V3) DUMPTAIL } while(0)
|
|
#define DUMP4(V1, V2, V3, V4) do { DUMPHEAD DUMPVAR(V1) DUMPVAR(V2) DUMPVAR(V3) DUMPVAR(V4) DUMPTAIL } while(0)
|
|
#define DUMP5(V1, V2, V3, V4, V5) do { DUMPHEAD DUMPVAR(V1) DUMPVAR(V2) DUMPVAR(V3) DUMPVAR(V4) DUMPVAR(V5) DUMPTAIL } while(0)
|
|
#define DUMP6(V1, V2, V3, V4, V5, V6) do { DUMPHEAD DUMPVAR(V1) DUMPVAR(V2) DUMPVAR(V3) DUMPVAR(V4) DUMPVAR(V5) DUMPVAR(V6) DUMPTAIL } while(0)
|
|
#define DUMP7(V1, V2, V3, V4, V5, V6, V7) do { DUMPHEAD DUMPVAR(V1) DUMPVAR(V2) DUMPVAR(V3) DUMPVAR(V4) DUMPVAR(V5) DUMPVAR(V6) DUMPVAR(V7) DUMPTAIL } while(0)
|
|
#define DUMP8(V1, V2, V3, V4, V5, V6, V7, V8) do { DUMPHEAD DUMPVAR(V1) DUMPVAR(V2) DUMPVAR(V3) DUMPVAR(V4) DUMPVAR(V5) DUMPVAR(V6) DUMPVAR(V7) DUMPVAR(V8) DUMPTAIL } while(0)
|
|
#define DUMP9(V1, V2, V3, V4, V5, V6, V7, V8, V9) do { DUMPHEAD DUMPVAR(V1) DUMPVAR(V2) DUMPVAR(V3) DUMPVAR(V4) DUMPVAR(V5) DUMPVAR(V6) DUMPVAR(V7) DUMPVAR(V8) DUMPVAR(V9) DUMPTAIL } while(0)
|
|
|
|
/// https://groups.google.com/forum/#!searchin/kona-dev/variadic$20macro%7Csort:date/kona-dev/XMA-lDOqtlI/GCzdfZsD41sJ
|
|
|
|
#define VA_NUM_ARGS_IMPL(x1, x2, x3, x4, x5, x6, x7, x8, x9, N, ...) N
|
|
#define VA_NUM_ARGS(...) VA_NUM_ARGS_IMPL(__VA_ARGS__, 9, 8, 7, 6, 5, 4, 3, 2, 1)
|
|
|
|
#define MAKE_VAR_MACRO_IMPL_CONCAT(PREFIX, NUM_ARGS) PREFIX ## NUM_ARGS
|
|
#define MAKE_VAR_MACRO_IMPL(PREFIX, NUM_ARGS) MAKE_VAR_MACRO_IMPL_CONCAT(PREFIX, NUM_ARGS)
|
|
#define MAKE_VAR_MACRO(PREFIX, ...) MAKE_VAR_MACRO_IMPL(PREFIX, VA_NUM_ARGS(__VA_ARGS__))
|
|
|
|
#define DUMP(...) MAKE_VAR_MACRO(DUMP, __VA_ARGS__)(__VA_ARGS__)
|