1 /*
2 * Copyright (c) 2010-2011, Pieter Noordhuis <pcnoordhuis at gmail dot com>
3 *
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions are met:
8 *
9 * * Redistributions of source code must retain the above copyright notice,
10 * this list of conditions and the following disclaimer.
11 * * Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 * * Neither the name of Redis nor the names of its contributors may be used
15 * to endorse or promote products derived from this software without
16 * specific prior written permission.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
19 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
22 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
23 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
24 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
25 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
26 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28 * POSSIBILITY OF SUCH DAMAGE.
29 */
30
31 #ifndef __HIREDIS_LIBEVENT_H__
32 #define __HIREDIS_LIBEVENT_H__
33 #include <event2/event.h>
34 #include "../hiredis.h"
35 #include "../async.h"
36
37 typedef struct redisLibeventEvents {
38 redisAsyncContext *context;
39 struct event *rev, *wev;
40 } redisLibeventEvents;
41
redisLibeventReadEvent(int fd,short event,void * arg)42 static void redisLibeventReadEvent(int fd, short event, void *arg) {
43 ((void)fd); ((void)event);
44 redisLibeventEvents *e = (redisLibeventEvents*)arg;
45 redisAsyncHandleRead(e->context);
46 }
47
redisLibeventWriteEvent(int fd,short event,void * arg)48 static void redisLibeventWriteEvent(int fd, short event, void *arg) {
49 ((void)fd); ((void)event);
50 redisLibeventEvents *e = (redisLibeventEvents*)arg;
51 redisAsyncHandleWrite(e->context);
52 }
53
redisLibeventAddRead(void * privdata)54 static void redisLibeventAddRead(void *privdata) {
55 redisLibeventEvents *e = (redisLibeventEvents*)privdata;
56 event_add(e->rev,NULL);
57 }
58
redisLibeventDelRead(void * privdata)59 static void redisLibeventDelRead(void *privdata) {
60 redisLibeventEvents *e = (redisLibeventEvents*)privdata;
61 event_del(e->rev);
62 }
63
redisLibeventAddWrite(void * privdata)64 static void redisLibeventAddWrite(void *privdata) {
65 redisLibeventEvents *e = (redisLibeventEvents*)privdata;
66 event_add(e->wev,NULL);
67 }
68
redisLibeventDelWrite(void * privdata)69 static void redisLibeventDelWrite(void *privdata) {
70 redisLibeventEvents *e = (redisLibeventEvents*)privdata;
71 event_del(e->wev);
72 }
73
redisLibeventCleanup(void * privdata)74 static void redisLibeventCleanup(void *privdata) {
75 redisLibeventEvents *e = (redisLibeventEvents*)privdata;
76 event_del(e->rev);
77 event_del(e->wev);
78 free(e);
79 }
80
redisLibeventAttach(redisAsyncContext * ac,struct event_base * base)81 static int redisLibeventAttach(redisAsyncContext *ac, struct event_base *base) {
82 redisContext *c = &(ac->c);
83 redisLibeventEvents *e;
84
85 /* Nothing should be attached when something is already attached */
86 if (ac->ev.data != NULL)
87 return REDIS_ERR;
88
89 /* Create container for context and r/w events */
90 e = (redisLibeventEvents*)malloc(sizeof(*e));
91 e->context = ac;
92
93 /* Register functions to start/stop listening for events */
94 ac->ev.addRead = redisLibeventAddRead;
95 ac->ev.delRead = redisLibeventDelRead;
96 ac->ev.addWrite = redisLibeventAddWrite;
97 ac->ev.delWrite = redisLibeventDelWrite;
98 ac->ev.cleanup = redisLibeventCleanup;
99 ac->ev.data = e;
100
101 /* Initialize and install read/write events */
102 e->rev = event_new(base, c->fd, EV_READ, redisLibeventReadEvent, e);
103 e->wev = event_new(base, c->fd, EV_WRITE, redisLibeventWriteEvent, e);
104 event_add(e->rev, NULL);
105 event_add(e->wev, NULL);
106 return REDIS_OK;
107 }
108 #endif
109