From 34cd75f77d0ca94dbdba4e6cc9451fe4d33e78b3 Mon Sep 17 00:00:00 2001 From: lichao <lichao@aiotlink.com> Date: 星期三, 19 五月 2021 19:14:13 +0800 Subject: [PATCH] add api BHQueryProcs. --- src/bh_util.h | 40 ++++++++++++++++++++++++++++++---------- 1 files changed, 30 insertions(+), 10 deletions(-) diff --git a/src/bh_util.h b/src/bh_util.h index bc48578..15ffeb0 100644 --- a/src/bh_util.h +++ b/src/bh_util.h @@ -18,6 +18,7 @@ #ifndef BH_UTIL_SOXWOK67 #define BH_UTIL_SOXWOK67 +#include "macro.h" #include <functional> #include <mutex> #include <stdint.h> @@ -91,6 +92,8 @@ inline void PutInt(void *p, uint32_t u) { Put32(p, u); } inline void PutInt(void *p, uint64_t u) { Put64(p, u); } +constexpr uint64_t MaskBits(int nbits) { return (uint64_t(1) << nbits) - 1; } + class ExitCall { typedef std::function<void(void)> func_t; @@ -123,13 +126,13 @@ D &operator*() const { return *p_; } }; -template <class T, class Mutex = std::mutex, class Lock = std::unique_lock<Mutex>> +template <class T, class TMutex = std::mutex, class Lock = std::unique_lock<TMutex>> class Synced { typedef T Data; - Mutex mutex_; + TMutex mutex_; Data data_; - typedef SyncedPtr<Data, Mutex, Lock> Ptr; + typedef SyncedPtr<Data, TMutex, Lock> Ptr; public: template <class... P> @@ -143,12 +146,29 @@ } }; -// macro helper -#define JOIN_IMPL(a, b) a##b -#define JOIN(a, b) JOIN_IMPL(a, b) -// defer function / lambda. -#define DEFERF(func) ExitCall JOIN(defer_, __LINE__)(func) -// defer simple expression -#define DEFER1(expr) DEFERF([&]() { expr; }) +template <class T, class Tag> +class StaticDataRef +{ + typedef T *Ptr; + static inline Ptr &ptr() + { + static Ptr sp(nullptr); + return sp; + } + +protected: + static inline T &GetData(const std::string &msg = "Must set data before use!") + { + if (!ptr()) { throw std::logic_error(msg); } + return *ptr(); + } + +public: + static bool SetData(T &t) + { + auto Bind = [&]() { ptr() = &t; return true; }; + return ptr() ? false : Bind(); + } +}; #endif /* end of include guard: BH_UTIL_SOXWOK67 */ -- Gitblit v1.8.0