memcached_client/worker.h at master · drouyang/memcached_client · GitHub

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
#ifndef WORKER_H
#define WORKER_H

#include <pthread.h>
#include <malloc.h>
#include <event2/event.h>
#include <sys/time.h>
#include <errno.h>
#include "config.h"
#include "conn.h"
#include "request.h"
#include "response.h"
#include "generate.h"

#include "mt19937p.h"

#define QUEUE_SIZE 1000000
#define INCR_FIX_QUEUE_SIZE 1000

struct worker {

  struct config* config;

  pthread_t thread;
  struct event_base* event_base;
  struct conn** connections;
  int nConnections;
  int cpu_num;
  struct timeval last_write_time;
  int interarrival_time;

  //Circular queue
  struct request* request_queue[QUEUE_SIZE];
  int head;
  int tail;
  int n_requests;
  int current_request_id;

  struct request* incr_fix_queue[INCR_FIX_QUEUE_SIZE];
  int incr_fix_queue_head;
  int incr_fix_queue_tail;
  struct mt19937p myMT19937p;
  int warmup_key;
  int warmup_key_check;
  int received_warmup_keys;

};


void sendCallback(int fd, short eventType, void* args);
void receiveCallback(int fd, short eventType, void* args);
void* workerFunction(void* arg);
void workerLoop(struct worker* worker);
void createWorkers(struct config* config);
struct worker* createWorker(struct config* config, int cpuNum);
int pushRequest(struct worker* worker, struct request* request);


#endif