forked from naver/arcus-memcached
-
Notifications
You must be signed in to change notification settings - Fork 0
/
thread.h
231 lines (220 loc) · 8.33 KB
/
thread.h
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
/* -*- Mode: C; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
/*
* arcus-memcached - Arcus memory cache server
* Copyright 2010-2014 NAVER Corp.
* Copyright 2014-2020 JaM2in Co., Ltd.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef THREAD_H
#define THREAD_H
#include <event.h>
#include "cache.h"
#include "mc_util.h"
/**
* Stats stored per-thread.
*/
struct thread_stats {
pthread_mutex_t mutex;
uint64_t cmd_get;
uint64_t cmd_set;
uint64_t cmd_incr;
uint64_t cmd_decr;
uint64_t cmd_delete;
uint64_t get_hits;
uint64_t get_misses;
uint64_t incr_hits;
uint64_t incr_misses;
uint64_t decr_hits;
uint64_t decr_misses;
uint64_t delete_hits;
uint64_t delete_misses;
uint64_t cmd_cas;
uint64_t cas_hits;
uint64_t cas_badval;
uint64_t cas_misses;
uint64_t cmd_flush;
uint64_t cmd_flush_prefix;
uint64_t cmd_auth;
uint64_t auth_errors;
uint64_t bytes_read;
uint64_t bytes_written;
uint64_t conn_yields; /* # of yields for connections (-R option)*/
/* list command stats */
uint64_t cmd_lop_create;
uint64_t cmd_lop_insert;
uint64_t cmd_lop_delete;
uint64_t cmd_lop_get;
uint64_t lop_create_oks;
uint64_t lop_insert_hits;
uint64_t lop_insert_misses;
uint64_t lop_delete_elem_hits;
uint64_t lop_delete_none_hits;
uint64_t lop_delete_misses;
uint64_t lop_get_elem_hits;
uint64_t lop_get_none_hits;
uint64_t lop_get_misses;
/* set command stats */
uint64_t cmd_sop_create;
uint64_t cmd_sop_insert;
uint64_t cmd_sop_delete;
uint64_t cmd_sop_get;
uint64_t cmd_sop_exist;
uint64_t sop_create_oks;
uint64_t sop_insert_hits;
uint64_t sop_insert_misses;
uint64_t sop_delete_elem_hits;
uint64_t sop_delete_none_hits;
uint64_t sop_delete_misses;
uint64_t sop_get_elem_hits;
uint64_t sop_get_none_hits;
uint64_t sop_get_misses;
uint64_t sop_exist_hits;
uint64_t sop_exist_misses;
/* map command stats */
uint64_t cmd_mop_create;
uint64_t cmd_mop_insert;
uint64_t cmd_mop_update;
uint64_t cmd_mop_delete;
uint64_t cmd_mop_get;
uint64_t mop_create_oks;
uint64_t mop_insert_hits;
uint64_t mop_insert_misses;
uint64_t mop_update_elem_hits;
uint64_t mop_update_none_hits;
uint64_t mop_update_misses;
uint64_t mop_delete_elem_hits;
uint64_t mop_delete_none_hits;
uint64_t mop_delete_misses;
uint64_t mop_get_elem_hits;
uint64_t mop_get_none_hits;
uint64_t mop_get_misses;
/* btree command stats */
uint64_t cmd_bop_create;
uint64_t cmd_bop_insert;
uint64_t cmd_bop_update;
uint64_t cmd_bop_delete;
uint64_t cmd_bop_get;
uint64_t cmd_bop_count;
uint64_t cmd_bop_position;
uint64_t cmd_bop_pwg;
uint64_t cmd_bop_gbp;
#ifdef SUPPORT_BOP_MGET
uint64_t cmd_bop_mget;
#endif
#ifdef SUPPORT_BOP_SMGET
uint64_t cmd_bop_smget;
#endif
uint64_t cmd_bop_incr;
uint64_t cmd_bop_decr;
uint64_t bop_create_oks;
uint64_t bop_insert_hits;
uint64_t bop_insert_misses;
uint64_t bop_update_elem_hits;
uint64_t bop_update_none_hits;
uint64_t bop_update_misses;
uint64_t bop_delete_elem_hits;
uint64_t bop_delete_none_hits;
uint64_t bop_delete_misses;
uint64_t bop_get_elem_hits;
uint64_t bop_get_none_hits;
uint64_t bop_get_misses;
uint64_t bop_count_hits;
uint64_t bop_count_misses;
uint64_t bop_position_elem_hits;
uint64_t bop_position_none_hits;
uint64_t bop_position_misses;
uint64_t bop_pwg_elem_hits;
uint64_t bop_pwg_none_hits;
uint64_t bop_pwg_misses;
uint64_t bop_gbp_elem_hits;
uint64_t bop_gbp_none_hits;
uint64_t bop_gbp_misses;
#ifdef SUPPORT_BOP_MGET
uint64_t bop_mget_oks;
#endif
#ifdef SUPPORT_BOP_SMGET
uint64_t bop_smget_oks;
#endif
uint64_t bop_incr_elem_hits;
uint64_t bop_incr_none_hits;
uint64_t bop_incr_misses;
uint64_t bop_decr_elem_hits;
uint64_t bop_decr_none_hits;
uint64_t bop_decr_misses;
/* attribute command stats */
uint64_t cmd_getattr;
uint64_t cmd_setattr;
uint64_t getattr_hits;
uint64_t getattr_misses;
uint64_t setattr_hits;
uint64_t setattr_misses;
};
/*
* Macros for incrementing thread_stats
*/
#define THREAD_STATS_INCR_ONE(thread_stats, op) { \
pthread_mutex_lock(&thread_stats->mutex); \
thread_stats->op++; \
pthread_mutex_unlock(&thread_stats->mutex); \
}
#define THREAD_STATS_INCR_TWO(thread_stats, op1, op2) { \
pthread_mutex_lock(&thread_stats->mutex); \
thread_stats->op1++; \
thread_stats->op2++; \
pthread_mutex_unlock(&thread_stats->mutex); \
}
#define THREAD_STATS_INCR_AMT(thread_stats, op, amt) { \
pthread_mutex_lock(&thread_stats->mutex); \
thread_stats->op += (amt); \
pthread_mutex_unlock(&thread_stats->mutex); \
}
enum thread_type {
GENERAL = 11
};
typedef struct {
pthread_t thread_id; /* unique ID of this thread */
struct event_base *base; /* libevent handle this thread uses */
struct event notify_event; /* listen event for notify pipe */
int notify_receive_fd; /* receiving end of notify pipe */
int notify_send_fd; /* sending end of notify pipe */
struct conn_queue *new_conn_queue; /* queue of new connections to handle */
cache_t *suffix_cache; /* suffix cache */
pthread_mutex_t mutex; /* Mutex to lock protect access to the pending_io */
bool is_locked;
struct conn *pending_io; /* List of connection with pending async io ops */
struct conn *conn_list; /* connection list managed by this thread */
int index; /* index of this thread in the threads array */
enum thread_type type; /* Type of IO this thread processes */
token_buff_t token_buff; /* token buffer */
mblck_pool_t mblck_pool; /* memory block pool */
} LIBEVENT_THREAD;
bool has_cycle(struct conn *c);
size_t list_to_array(struct conn **dest, size_t max_items, struct conn **l);
bool should_io_blocked(const void *cookie);
#ifdef MULTI_NOTIFY_IO_COMPLETE
void waitfor_io_complete(const void *cookie);
#endif
void notify_io_complete(const void *cookie, ENGINE_ERROR_CODE status);
void remove_io_pending(const void *cookie);
void dispatch_conn_new(int sfd, STATE_FUNC init_state, int event_flags,
int read_buffer_size, enum network_transport transport);
int is_listen_thread(void);
void *threadlocal_stats_create(int num_threads);
void threadlocal_stats_destroy(void *stats);
void threadlocal_stats_clear(struct thread_stats *stats);
void threadlocal_stats_reset(struct thread_stats *thread_stats);
void threadlocal_stats_aggregate(struct thread_stats *thread_stats, struct thread_stats *stats);
void thread_init(int nthreads, struct event_base *main_base);
void threads_shutdown(void);
#endif