|
|
|
|
@ -1,10 +1,15 @@
|
|
|
|
|
#ifndef __CLASS_STAT_COUNTERS_H
|
|
|
|
|
#define __CLASS_STAT_COUNTERS_H
|
|
|
|
|
#include "proxysql_atomic.h"
|
|
|
|
|
|
|
|
|
|
#define PROXYSQL_STATSCOUNTERS_NOLOCK
|
|
|
|
|
class StatCounters {
|
|
|
|
|
private:
|
|
|
|
|
#ifdef PROXYSQL_STATSCOUNTERS_NOLOCK
|
|
|
|
|
#else
|
|
|
|
|
bool with_lock;
|
|
|
|
|
rwlock_t _lock;
|
|
|
|
|
#endif
|
|
|
|
|
int last;
|
|
|
|
|
int keep;
|
|
|
|
|
void cleanup() {
|
|
|
|
|
@ -26,10 +31,14 @@ class StatCounters {
|
|
|
|
|
public:
|
|
|
|
|
int len;
|
|
|
|
|
int *val;
|
|
|
|
|
#ifdef PROXYSQL_STATSCOUNTERS_NOLOCK
|
|
|
|
|
StatCounters(int _l, int _k) {
|
|
|
|
|
#else
|
|
|
|
|
StatCounters(int _l, int _k, bool _wl=false) {
|
|
|
|
|
with_lock=_wl;
|
|
|
|
|
if (with_lock)
|
|
|
|
|
spinlock_rwlock_init(&_lock);
|
|
|
|
|
#endif
|
|
|
|
|
last=0;
|
|
|
|
|
keep=_k;
|
|
|
|
|
len=_l;
|
|
|
|
|
@ -43,40 +52,61 @@ class StatCounters {
|
|
|
|
|
free(val);
|
|
|
|
|
}
|
|
|
|
|
void set(int _i, int _v) {
|
|
|
|
|
#ifdef PROXYSQL_STATSCOUNTERS_NOLOCK
|
|
|
|
|
#else
|
|
|
|
|
if (with_lock)
|
|
|
|
|
spin_wrlock(&_lock);
|
|
|
|
|
#endif
|
|
|
|
|
if ( _i > last ) {
|
|
|
|
|
last=_i; cleanup();
|
|
|
|
|
}
|
|
|
|
|
val[_i%len]=_v;
|
|
|
|
|
#ifdef PROXYSQL_STATSCOUNTERS_NOLOCK
|
|
|
|
|
#else
|
|
|
|
|
if (with_lock)
|
|
|
|
|
spin_wrunlock(&_lock);
|
|
|
|
|
#endif
|
|
|
|
|
}
|
|
|
|
|
void incr(int _i) {
|
|
|
|
|
#ifdef PROXYSQL_STATSCOUNTERS_NOLOCK
|
|
|
|
|
#else
|
|
|
|
|
if (with_lock)
|
|
|
|
|
spin_wrlock(&_lock);
|
|
|
|
|
#endif
|
|
|
|
|
if ( _i > last ) {
|
|
|
|
|
if ( _i > last + keep ) val[_i%len]=0;
|
|
|
|
|
last=_i; cleanup();
|
|
|
|
|
}
|
|
|
|
|
val[_i%len]++;
|
|
|
|
|
#ifdef PROXYSQL_STATSCOUNTERS_NOLOCK
|
|
|
|
|
#else
|
|
|
|
|
if (with_lock)
|
|
|
|
|
spin_wrunlock(&_lock);
|
|
|
|
|
#endif
|
|
|
|
|
}
|
|
|
|
|
void decr(int _i) {
|
|
|
|
|
#ifdef PROXYSQL_STATSCOUNTERS_NOLOCK
|
|
|
|
|
#else
|
|
|
|
|
if (with_lock)
|
|
|
|
|
spin_wrlock(&_lock);
|
|
|
|
|
#endif
|
|
|
|
|
if ( _i > last ) {
|
|
|
|
|
if ( _i > last + keep ) val[_i%len]=0;
|
|
|
|
|
last=_i; cleanup();
|
|
|
|
|
}
|
|
|
|
|
val[_i%len]--;
|
|
|
|
|
#ifdef PROXYSQL_STATSCOUNTERS_NOLOCK
|
|
|
|
|
#else
|
|
|
|
|
if (with_lock)
|
|
|
|
|
spin_wrunlock(&_lock);
|
|
|
|
|
#endif
|
|
|
|
|
}
|
|
|
|
|
int sum(int _i, int _k) {
|
|
|
|
|
#ifdef PROXYSQL_STATSCOUNTERS_NOLOCK
|
|
|
|
|
#else
|
|
|
|
|
if (with_lock)
|
|
|
|
|
spin_wrlock(&_lock);
|
|
|
|
|
#endif
|
|
|
|
|
if ( _i > last ) {
|
|
|
|
|
if ( _i > last + keep ) val[_i%len]=0;
|
|
|
|
|
last=_i; cleanup();
|
|
|
|
|
@ -86,8 +116,11 @@ class StatCounters {
|
|
|
|
|
for (i=0; i<_k; i++) {
|
|
|
|
|
ret+=val[(_i-i)%len];
|
|
|
|
|
}
|
|
|
|
|
#ifdef PROXYSQL_STATSCOUNTERS_NOLOCK
|
|
|
|
|
#else
|
|
|
|
|
if (with_lock)
|
|
|
|
|
spin_wrunlock(&_lock);
|
|
|
|
|
#endif
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|