hcq/MaxScale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

MXS-1908 Replace HASHTABLE with std::unordered_map

Browse Source 
Just minimal changes, further cleanup and simplification could
be done.
This commit is contained in:
Johan Wikman
2018-06-11 18:39:18 +03:00
parent 52ae0a66d1
commit 9e3b4b46fc
1 changed files with 138 additions and 176 deletions
Download Patch File Download Diff File Expand all files Collapse all files

212
server/core/log_manager.cc
View File

@ -30,11 +30,10 @@
#include <maxscale/atomic.h>
#include <maxscale/config.h>
#include <maxscale/debug.h>
#include <maxscale/hashtable.h>
#include <maxscale/json_api.h>
#include <maxscale/platform.h>
#include <maxscale/session.h>
#include <maxscale/spinlock.h>
#include <maxscale/spinlock.hh>
#include <maxscale/utils.h>
#include "internal/mlist.h"
@ -159,7 +158,13 @@ static logmanager_t* lm;
static bool flushall_flag;
static bool flushall_started_flag;
static bool flushall_done_flag;
static HASHTABLE* message_stats;
namespace
{
class MessageStats;
}
static MessageStats* message_stats;
/** This is used to detect if the initialization of the log manager has failed
* and that it isn't initialized again after a failure has occurred. */
@ -200,9 +205,6 @@ typedef struct lm_message_stats
size_t count; /** How many times the error has been reported within this window. */
} LM_MESSAGE_STATS;
static const int LM_MESSAGE_HASH_SIZE = 293; /** A prime, and roughly a quarter of current
number of MXS_{ERROR|WARNING|NOTICE} calls. */
/**
* Returns the current time.
*
@ -216,21 +218,26 @@ static uint64_t time_monotonic_ms()
return now.tv_sec * 1000 + now.tv_nsec / 1000000;
}
/**
* Hash-function for lm_message_key.
*
namespace std
{
namespace tr1
{
template<>
struct hash<LM_MESSAGE_KEY>
{
typedef LM_MESSAGE_KEY Key;
typedef size_t result_type;
size_t operator()(const LM_MESSAGE_KEY& key) const
{
/*
* This is an implementation of the Jenkin's one-at-a-time hash function.
* https://en.wikipedia.org/wiki/Jenkins_hash_function
*
* @param v A pointer to a LM_MESSAGE_KEY
* @return Hash for the contents of the LM_MESSAGE_KEY structure.
*/
int lm_message_key_hash(const void *v)
{
const LM_MESSAGE_KEY* key = (const LM_MESSAGE_KEY*)v;
uint64_t key1 = (uint64_t)key->filename;
uint16_t key2 = (uint16_t)key->linenumber; // The first 48 bits are likely to be 0.
uint64_t key1 = (uint64_t)key.filename;
uint16_t key2 = (uint16_t)key.linenumber; // The first 48 bits are likely to be 0.
uint32_t hash = 0;
size_t i;
@ -254,73 +261,65 @@ int lm_message_key_hash(const void *v)
hash += (hash << 15);
return hash;
}
};
/**
* Compares two LM_MESSAGE_KEY structures
*
* @param v1 Pointer to a LM_MESSAGE_KEY struct.
* @param v2 Pointer to a LM_MESSAGE_KEY struct.
* @return 0 if the structures are equal.
* -1 if v1 is less than v2
* 1 if v1 is greater than v2
*/
static int lm_message_key_cmp(const void* v1, const void* v2)
{
const LM_MESSAGE_KEY* key1 = (const LM_MESSAGE_KEY*)v1;
const LM_MESSAGE_KEY* key2 = (const LM_MESSAGE_KEY*)v2;
int cmp = (int64_t)key1->filename - (int64_t)key2->filename;
if (cmp == 0)
{
cmp = key1->linenumber - key2->linenumber;
}
return cmp == 0 ? 0 : (cmp < 0 ? -1 : 1);
template<>
struct equal_to<LM_MESSAGE_KEY>
{
typedef bool result_type;
typedef LM_MESSAGE_KEY first_argument_type;
typedef LM_MESSAGE_KEY second_argument_type;
bool operator()(const LM_MESSAGE_KEY& lhs, const LM_MESSAGE_KEY& rhs) const
{
return
(lhs.filename == rhs.filename) &&
(lhs.linenumber == rhs.linenumber);
}
};
}
/**
* Clone a LM_MESSAGE_KEY
*
* @param v Pointer to a LM_MESSAGE_KEY structure
* @return A copy of v or NULL if memory allocation fails.
*/
static void* lm_message_key_clone(const void* v)
namespace
{
const LM_MESSAGE_KEY* src = (const LM_MESSAGE_KEY*)v;
LM_MESSAGE_KEY* dst = (LM_MESSAGE_KEY*)MXS_MALLOC(sizeof(LM_MESSAGE_KEY));
if (dst)
class MessageStats
{
public:
MessageStats(const MessageStats&) = delete;
MessageStats& operator=(const MessageStats&) = delete;
MessageStats()
{
dst->filename = src->filename;
dst->linenumber = src->linenumber;
}
return dst;
LM_MESSAGE_STATS& get(const LM_MESSAGE_KEY& key)
{
mxs::SpinLockGuard guard(m_lock);
auto i = m_registry.find(key);
if (i == m_registry.end())
{
LM_MESSAGE_STATS stats;
spinlock_init(&stats.lock);
stats.first_ms = time_monotonic_ms();
stats.last_ms = 0;
stats.count = 0;
i = m_registry.insert(std::make_pair(key, stats)).first;
}
/**
* Clone a LM_MESSAGE_STATS
*
* @param v Pointer to a LM_MESSAGE_STATS structure
* @return A copy of v or NULL if memory allocation fails.
*/
static void* lm_message_stats_clone(const void* v)
{
const LM_MESSAGE_STATS* src = (const LM_MESSAGE_STATS*)v;
LM_MESSAGE_STATS* dst = (LM_MESSAGE_STATS*)MXS_MALLOC(sizeof(LM_MESSAGE_STATS));
if (dst)
{
// Somewhat questionable to copy a lock, but in this context we
// know that src is stack allocated and will not be used after this.
dst->lock = src->lock;
dst->first_ms = src->first_ms;
dst->last_ms = src->last_ms;
dst->count = src->count;
return i->second;
}
return dst;
private:
mxs::SpinLock m_lock;
std::tr1::unordered_map<LM_MESSAGE_KEY, LM_MESSAGE_STATS> m_registry;
};
}
/**
@ -605,24 +604,15 @@ bool mxs_log_init(const char* ident, const char* logdir, mxs_log_target_t target
{
ss_dassert(!message_stats);
message_stats = hashtable_alloc(LM_MESSAGE_HASH_SIZE,
lm_message_key_hash,
lm_message_key_cmp);
message_stats = new (std::nothrow) MessageStats;
if (message_stats)
{
// As entries are added to the hashtable they will be cloned,
// so stack allocated keys and values are ok.
hashtable_memory_fns(message_stats,
lm_message_key_clone,
lm_message_stats_clone,
hashtable_item_free,
hashtable_item_free);
succ = logmanager_init_nomutex(ident, logdir, target, log_config.do_maxlog);
if (!succ)
{
hashtable_free(message_stats);
delete message_stats;
message_stats = NULL;
}
}
@ -681,7 +671,7 @@ static void logmanager_done_nomutex(void)
MXS_FREE(lm);
lm = NULL;
hashtable_free(message_stats);
delete message_stats;
message_stats = NULL;
}
@ -2776,51 +2766,24 @@ static message_suppression_t message_status(const char* file, int line)
if ((t.count != 0) && (t.window_ms != 0) && (t.suppress_ms != 0))
{
LM_MESSAGE_KEY key = { file, line };
LM_MESSAGE_STATS *value;
LM_MESSAGE_STATS& value = message_stats->get(key);
errno = 0;
// Since the hashtable can not be externally locked, the lookup/creation
// must be done in a loop to ensure that everything works even if two
// threads logs the same message at the very same time.
do
{
value = (LM_MESSAGE_STATS*) hashtable_fetch(message_stats, &key);
if (!value)
{
LM_MESSAGE_STATS stats;
spinlock_init(&stats.lock);
stats.first_ms = time_monotonic_ms();
stats.last_ms = 0;
stats.count = 0;
// hashtable_add may fail for two reasons; the key exists
// already or it runs out of memory. In the latter case
// errno is set.
hashtable_add(message_stats, &key, &stats);
}
}
while (!value && (errno == 0));
if (value)
{
uint64_t now_ms = time_monotonic_ms();
spinlock_acquire(&value->lock);
spinlock_acquire(&value.lock);
++value->count;
++value.count;
// Less that t.window_ms milliseconds since the message was logged
// the last time. May have to be throttled.
if (value->count < t.count)
if (value.count < t.count)
{
// t.count times has not been reached, still ok to log.
}
else if (value->count == t.count)
else if (value.count == t.count)
{
// t.count times has been reached. Was it within the window?
if (now_ms - value->first_ms < t.window_ms)
if (now_ms - value.first_ms < t.window_ms)
{
// Within the window, suppress the message.
rv = MESSAGE_SUPPRESSED;
@ -2839,14 +2802,14 @@ static message_suppression_t message_status(const char* file, int line)
// However, if that's the case, it was a spike so the
// flooding will stop anyway.
value->first_ms = now_ms;
value->count = 1;
value.first_ms = now_ms;
value.count = 1;
}
}
else
{
// In suppression mode.
if (now_ms - value->first_ms < (t.window_ms + t.suppress_ms))
if (now_ms - value.first_ms < (t.window_ms + t.suppress_ms))
{
// Still in the suppression window.
rv = MESSAGE_STILL_SUPPRESSED;
@ -2854,15 +2817,14 @@ static message_suppression_t message_status(const char* file, int line)
else
{
// We have exited the suppression window, reset the situation.
value->first_ms = now_ms;
value->count = 1;
value.first_ms = now_ms;
value.count = 1;
}
}
value->last_ms = now_ms;
value.last_ms = now_ms;
spinlock_release(&value->lock);
}
spinlock_release(&value.lock);
}
return rv;