From 52e051cb6924266003ef21814b5ebaf2b7b45241 Mon Sep 17 00:00:00 2001 From: Determinant Date: Tue, 28 Aug 2018 21:30:03 -0400 Subject: ... --- include/salticidae/conn.h | 2 +- include/salticidae/network.h | 4 ++-- include/salticidae/ref.h | 2 +- include/salticidae/util.h | 20 +++++++++++++++++++- src/util.cpp | 20 -------------------- 5 files changed, 23 insertions(+), 25 deletions(-) diff --git a/include/salticidae/conn.h b/include/salticidae/conn.h index 4cc8b11..d93f787 100644 --- a/include/salticidae/conn.h +++ b/include/salticidae/conn.h @@ -136,7 +136,7 @@ class SegBuffer { } }; -class ConnPoolError: public SalticidaeError { +struct ConnPoolError: public SalticidaeError { using SalticidaeError::SalticidaeError; }; diff --git a/include/salticidae/network.h b/include/salticidae/network.h index ec112ad..4e966d1 100644 --- a/include/salticidae/network.h +++ b/include/salticidae/network.h @@ -207,8 +207,8 @@ class ClientNetwork: public MsgNetwork { void send_msg(const MsgType &msg, const NetAddr &addr); }; -class PeerNetworkError: public SalticidaeError { - using SalticidaeError::SalticidaeError; +class PeerNetworkError: public ConnPoolError { + using ConnPoolError::ConnPoolError; }; /** Peer-to-peer network where any two nodes could hold a bi-diretional message diff --git a/include/salticidae/ref.h b/include/salticidae/ref.h index f97bd43..44eb01e 100644 --- a/include/salticidae/ref.h +++ b/include/salticidae/ref.h @@ -171,7 +171,7 @@ struct _RCCtl { struct _ARCCtl { std::atomic_size_t ref_cnt; std::atomic_size_t weak_cnt; - std::atomic_uint8_t dcnt; + std::atomic dcnt; void add_ref() { ref_cnt++; } void add_weak() { weak_cnt++; } bool release_ref() { diff --git a/include/salticidae/util.h b/include/salticidae/util.h index 39a3683..4e6825a 100644 --- a/include/salticidae/util.h +++ b/include/salticidae/util.h @@ -52,7 +52,25 @@ class SalticidaeError: public std::exception { std::string msg; public: SalticidaeError(); - SalticidaeError(const std::string &fmt, ...); + + template + SalticidaeError(const std::string &fmt, Args... args) { + int guessed_size = 128; + std::string buff; + for (;;) + { + buff.resize(guessed_size); + int nwrote = snprintf((char *)buff.data(), guessed_size, fmt.c_str(), args...); + if (nwrote < 0 || nwrote == guessed_size) + { + guessed_size <<= 1; + continue; + } + buff.resize(nwrote); + msg = std::move(buff); + break; + } + } operator std::string() const { return msg; } const char *what() const throw() override { return msg.c_str(); } diff --git a/src/util.cpp b/src/util.cpp index 07849fc..6c8866d 100644 --- a/src/util.cpp +++ b/src/util.cpp @@ -64,26 +64,6 @@ const std::string get_current_datetime() { SalticidaeError::SalticidaeError() : msg("unknown") {} -SalticidaeError::SalticidaeError(const std::string &fmt, ...) { - int guessed_size = 128; - std::string buff; - va_list ap; - for (;;) - { - buff.resize(guessed_size); - va_start(ap, fmt); - int nwrote = vsnprintf((char *)buff.data(), guessed_size, fmt.c_str(), ap); - if (nwrote < 0 || nwrote == guessed_size) - { - guessed_size <<= 1; - continue; - } - buff.resize(nwrote); - msg = std::move(buff); - break; - } -} - void Logger::write(const char *tag, const char *fmt, va_list ap) { fprintf(output, "%s [%s %s] ", get_current_datetime().c_str(), prefix, tag); vfprintf(output, fmt, ap); -- cgit v1.2.3-70-g09d2