mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-17 21:24:28 +00:00
132 lines
4.1 KiB
C++
132 lines
4.1 KiB
C++
#pragma once
|
|
|
|
#include <cerrno>
|
|
#include <vector>
|
|
#include <memory>
|
|
|
|
#include <Poco/Exception.h>
|
|
|
|
#include <Common/StackTrace.h>
|
|
|
|
namespace Poco { class Logger; }
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class Exception : public Poco::Exception
|
|
{
|
|
public:
|
|
Exception() {} /// For deferred initialization.
|
|
Exception(const std::string & msg, int code = 0) : Poco::Exception(msg, code) {}
|
|
Exception(const std::string & msg, const std::string & arg, int code = 0) : Poco::Exception(msg, arg, code) {}
|
|
Exception(const std::string & msg, const Exception & exc, int code = 0) : Poco::Exception(msg, exc, code), trace(exc.trace) {}
|
|
explicit Exception(const Poco::Exception & exc) : Poco::Exception(exc.displayText()) {}
|
|
|
|
const char * name() const throw() override { return "DB::Exception"; }
|
|
const char * className() const throw() override { return "DB::Exception"; }
|
|
DB::Exception * clone() const override { return new DB::Exception(*this); }
|
|
void rethrow() const override { throw *this; }
|
|
|
|
/// Add something to the existing message.
|
|
void addMessage(const std::string & arg) { extendedMessage(arg); }
|
|
|
|
const StackTrace & getStackTrace() const { return trace; }
|
|
|
|
private:
|
|
StackTrace trace;
|
|
};
|
|
|
|
|
|
/// Contains an additional member `saved_errno`. See the throwFromErrno function.
|
|
class ErrnoException : public Exception
|
|
{
|
|
public:
|
|
ErrnoException(const std::string & msg, int code = 0, int saved_errno_ = 0)
|
|
: Exception(msg, code), saved_errno(saved_errno_) {}
|
|
ErrnoException(const std::string & msg, const std::string & arg, int code = 0, int saved_errno_ = 0)
|
|
: Exception(msg, arg, code), saved_errno(saved_errno_) {}
|
|
ErrnoException(const std::string & msg, const Exception & exc, int code = 0, int saved_errno_ = 0)
|
|
: Exception(msg, exc, code), saved_errno(saved_errno_) {}
|
|
|
|
int getErrno() const { return saved_errno; }
|
|
|
|
private:
|
|
int saved_errno;
|
|
};
|
|
|
|
|
|
using Exceptions = std::vector<std::exception_ptr>;
|
|
|
|
|
|
[[noreturn]] void throwFromErrno(const std::string & s, int code = 0, int the_errno = errno);
|
|
|
|
|
|
/** Try to write an exception to the log (and forget about it).
|
|
* Can be used in destructors in the catch-all block.
|
|
*/
|
|
void tryLogCurrentException(const char * log_name, const std::string & start_of_message = "");
|
|
void tryLogCurrentException(Poco::Logger * logger, const std::string & start_of_message = "");
|
|
|
|
|
|
/** Prints current exception in canonical format.
|
|
* with_stacktrace - prints stack trace for DB::Exception.
|
|
* check_embedded_stacktrace - if DB::Exception has embedded stacktrace then
|
|
* only this stack trace will be printed.
|
|
*/
|
|
std::string getCurrentExceptionMessage(bool with_stacktrace, bool check_embedded_stacktrace = false);
|
|
|
|
/// Returns error code from ErrorCodes
|
|
int getCurrentExceptionCode();
|
|
|
|
|
|
/// An execution status of any piece of code, contains return code and optional error
|
|
struct ExecutionStatus
|
|
{
|
|
int code = 0;
|
|
std::string message;
|
|
|
|
ExecutionStatus() = default;
|
|
|
|
explicit ExecutionStatus(int return_code, const std::string & exception_message = "")
|
|
: code(return_code), message(exception_message) {}
|
|
|
|
static ExecutionStatus fromCurrentException(const std::string & start_of_message = "");
|
|
|
|
std::string serializeText() const;
|
|
|
|
void deserializeText(const std::string & data);
|
|
|
|
bool tryDeserializeText(const std::string & data);
|
|
};
|
|
|
|
|
|
void tryLogException(std::exception_ptr e, const char * log_name, const std::string & start_of_message = "");
|
|
void tryLogException(std::exception_ptr e, Poco::Logger * logger, const std::string & start_of_message = "");
|
|
|
|
std::string getExceptionMessage(const Exception & e, bool with_stacktrace, bool check_embedded_stacktrace = false);
|
|
std::string getExceptionMessage(std::exception_ptr e, bool with_stacktrace);
|
|
|
|
|
|
void rethrowFirstException(const Exceptions & exceptions);
|
|
|
|
|
|
template <typename T>
|
|
std::enable_if_t<std::is_pointer_v<T>, T> exception_cast(std::exception_ptr e)
|
|
{
|
|
try
|
|
{
|
|
std::rethrow_exception(std::move(e));
|
|
}
|
|
catch (std::remove_pointer_t<T> & concrete)
|
|
{
|
|
return &concrete;
|
|
}
|
|
catch (...)
|
|
{
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
}
|