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_LIBEV_H__
32 #define __HIREDIS_LIBEV_H__
33 #include <stdlib.h>
34 #include <sys/types.h>
35 #include <ev.h>
36 #include "../hiredis.h"
37 #include "../async.h"
38
39 typedef struct redisLibevEvents {
40 redisAsyncContext *context;
41 struct ev_loop *loop;
42 int reading, writing;
43 ev_io rev, wev;
44 } redisLibevEvents;
45
redisLibevReadEvent(EV_P_ ev_io * watcher,int revents)46 static void redisLibevReadEvent(EV_P_ ev_io *watcher, int revents) {
47 #if EV_MULTIPLICITY
48 ((void)loop);
49 #endif
50 ((void)revents);
51
52 redisLibevEvents *e = (redisLibevEvents*)watcher->data;
53 redisAsyncHandleRead(e->context);
54 }
55
redisLibevWriteEvent(EV_P_ ev_io * watcher,int revents)56 static void redisLibevWriteEvent(EV_P_ ev_io *watcher, int revents) {
57 #if EV_MULTIPLICITY
58 ((void)loop);
59 #endif
60 ((void)revents);
61
62 redisLibevEvents *e = (redisLibevEvents*)watcher->data;
63 redisAsyncHandleWrite(e->context);
64 }
65
redisLibevAddRead(void * privdata)66 static void redisLibevAddRead(void *privdata) {
67 redisLibevEvents *e = (redisLibevEvents*)privdata;
68 struct ev_loop *loop = e->loop;
69 ((void)loop);
70 if (!e->reading) {
71 e->reading = 1;
72 ev_io_start(EV_A_ &e->rev);
73 }
74 }
75
redisLibevDelRead(void * privdata)76 static void redisLibevDelRead(void *privdata) {
77 redisLibevEvents *e = (redisLibevEvents*)privdata;
78 struct ev_loop *loop = e->loop;
79 ((void)loop);
80 if (e->reading) {
81 e->reading = 0;
82 ev_io_stop(EV_A_ &e->rev);
83 }
84 }
85
redisLibevAddWrite(void * privdata)86 static void redisLibevAddWrite(void *privdata) {
87 redisLibevEvents *e = (redisLibevEvents*)privdata;
88 struct ev_loop *loop = e->loop;
89 ((void)loop);
90 if (!e->writing) {
91 e->writing = 1;
92 ev_io_start(EV_A_ &e->wev);
93 }
94 }
95
redisLibevDelWrite(void * privdata)96 static void redisLibevDelWrite(void *privdata) {
97 redisLibevEvents *e = (redisLibevEvents*)privdata;
98 struct ev_loop *loop = e->loop;
99 ((void)loop);
100 if (e->writing) {
101 e->writing = 0;
102 ev_io_stop(EV_A_ &e->wev);
103 }
104 }
105
redisLibevCleanup(void * privdata)106 static void redisLibevCleanup(void *privdata) {
107 redisLibevEvents *e = (redisLibevEvents*)privdata;
108 redisLibevDelRead(privdata);
109 redisLibevDelWrite(privdata);
110 free(e);
111 }
112
redisLibevAttach(EV_P_ redisAsyncContext * ac)113 static int redisLibevAttach(EV_P_ redisAsyncContext *ac) {
114 redisContext *c = &(ac->c);
115 redisLibevEvents *e;
116
117 /* Nothing should be attached when something is already attached */
118 if (ac->ev.data != NULL)
119 return REDIS_ERR;
120
121 /* Create container for context and r/w events */
122 e = (redisLibevEvents*)malloc(sizeof(*e));
123 e->context = ac;
124 #if EV_MULTIPLICITY
125 e->loop = loop;
126 #else
127 e->loop = NULL;
128 #endif
129 e->reading = e->writing = 0;
130 e->rev.data = e;
131 e->wev.data = e;
132
133 /* Register functions to start/stop listening for events */
134 ac->ev.addRead = redisLibevAddRead;
135 ac->ev.delRead = redisLibevDelRead;
136 ac->ev.addWrite = redisLibevAddWrite;
137 ac->ev.delWrite = redisLibevDelWrite;
138 ac->ev.cleanup = redisLibevCleanup;
139 ac->ev.data = e;
140
141 /* Initialize read/write events */
142 ev_io_init(&e->rev,redisLibevReadEvent,c->fd,EV_READ);
143 ev_io_init(&e->wev,redisLibevWriteEvent,c->fd,EV_WRITE);
144 return REDIS_OK;
145 }
146
147 #endif
148