blob: 34935a1544c752da2829d8d011ad4d32f8bca924 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
#include <string.h>
#include <stdlib.h>
#include "radius_p.h"
#include "mempool.h"
#include "memdebug.h"
struct item_t
{
struct list_head entry;
unsigned int val;
time_t ts;
};
struct stat_accm_t
{
pthread_mutex_t lock;
struct list_head items;
unsigned int items_cnt;
unsigned int time;
unsigned long total;
};
static mempool_t item_pool;
struct stat_accm_t *stat_accm_create(unsigned int time)
{
struct stat_accm_t *s = _malloc(sizeof(*s));
memset(s, 0, sizeof(*s));
pthread_mutex_init(&s->lock, NULL);
INIT_LIST_HEAD(&s->items);
s->time = time;
return s;
}
static void stat_accm_clean(struct stat_accm_t *s)
{
struct item_t *it;
struct timespec ts;
clock_gettime(CLOCK_MONOTONIC, &ts);
while (!list_empty(&s->items)) {
it = list_entry(s->items.next, typeof(*it), entry);
if (ts.tv_sec - it->ts > s->time) {
list_del(&it->entry);
--s->items_cnt;
s->total -= it->val;
mempool_free(it);
} else
break;
}
}
void stat_accm_add(struct stat_accm_t *s, unsigned int val)
{
struct item_t *it;
struct timespec ts;
clock_gettime(CLOCK_MONOTONIC, &ts);
pthread_mutex_lock(&s->lock);
stat_accm_clean(s);
it = mempool_alloc(item_pool);
it->ts = ts.tv_sec;
it->val = val;
list_add_tail(&it->entry, &s->items);
++s->items_cnt;
s->total += val;
pthread_mutex_unlock(&s->lock);
}
unsigned long stat_accm_get_cnt(struct stat_accm_t *s)
{
pthread_mutex_lock(&s->lock);
stat_accm_clean(s);
pthread_mutex_unlock(&s->lock);
return s->items_cnt;
}
unsigned long stat_accm_get_avg(struct stat_accm_t *s)
{
unsigned long val;
pthread_mutex_lock(&s->lock);
stat_accm_clean(s);
val = s->items_cnt ? s->total/s->items_cnt : 0;
pthread_mutex_unlock(&s->lock);
return val;
}
static void init(void)
{
item_pool = mempool_create(sizeof(struct item_t));
}
DEFINE_INIT(50, init);
|
