| /* |
| * Copyright (c) 2010-2011, Pieter Noordhuis <pcnoordhuis at gmail dot com> |
| * |
| * All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions are met: |
| * |
| * * Redistributions of source code must retain the above copyright notice, |
| * this list of conditions and the following disclaimer. |
| * * Redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in the |
| * documentation and/or other materials provided with the distribution. |
| * * Neither the name of Redis nor the names of its contributors may be used |
| * to endorse or promote products derived from this software without |
| * specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
| * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
| * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE |
| * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
| * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
| * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
| * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
| * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
| * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
| * POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| #ifndef __HIREDIS_AE_H__ |
| #define __HIREDIS_AE_H__ |
| #include <sys/types.h> |
| #include <ae.h> |
| #include "../hiredis.h" |
| #include "../async.h" |
| |
| typedef struct redisAeEvents { |
| redisAsyncContext *context; |
| aeEventLoop *loop; |
| int fd; |
| int reading, writing; |
| } redisAeEvents; |
| |
| static void redisAeReadEvent(aeEventLoop *el, int fd, void *privdata, int mask) { |
| ((void)el); ((void)fd); ((void)mask); |
| |
| redisAeEvents *e = (redisAeEvents*)privdata; |
| redisAsyncHandleRead(e->context); |
| } |
| |
| static void redisAeWriteEvent(aeEventLoop *el, int fd, void *privdata, int mask) { |
| ((void)el); ((void)fd); ((void)mask); |
| |
| redisAeEvents *e = (redisAeEvents*)privdata; |
| redisAsyncHandleWrite(e->context); |
| } |
| |
| static void redisAeAddRead(void *privdata) { |
| redisAeEvents *e = (redisAeEvents*)privdata; |
| aeEventLoop *loop = e->loop; |
| if (!e->reading) { |
| e->reading = 1; |
| aeCreateFileEvent(loop,e->fd,AE_READABLE,redisAeReadEvent,e); |
| } |
| } |
| |
| static void redisAeDelRead(void *privdata) { |
| redisAeEvents *e = (redisAeEvents*)privdata; |
| aeEventLoop *loop = e->loop; |
| if (e->reading) { |
| e->reading = 0; |
| aeDeleteFileEvent(loop,e->fd,AE_READABLE); |
| } |
| } |
| |
| static void redisAeAddWrite(void *privdata) { |
| redisAeEvents *e = (redisAeEvents*)privdata; |
| aeEventLoop *loop = e->loop; |
| if (!e->writing) { |
| e->writing = 1; |
| aeCreateFileEvent(loop,e->fd,AE_WRITABLE,redisAeWriteEvent,e); |
| } |
| } |
| |
| static void redisAeDelWrite(void *privdata) { |
| redisAeEvents *e = (redisAeEvents*)privdata; |
| aeEventLoop *loop = e->loop; |
| if (e->writing) { |
| e->writing = 0; |
| aeDeleteFileEvent(loop,e->fd,AE_WRITABLE); |
| } |
| } |
| |
| static void redisAeCleanup(void *privdata) { |
| redisAeEvents *e = (redisAeEvents*)privdata; |
| redisAeDelRead(privdata); |
| redisAeDelWrite(privdata); |
| free(e); |
| } |
| |
| static int redisAeAttach(aeEventLoop *loop, redisAsyncContext *ac) { |
| redisContext *c = &(ac->c); |
| redisAeEvents *e; |
| |
| /* Nothing should be attached when something is already attached */ |
| if (ac->ev.data != NULL) |
| return REDIS_ERR; |
| |
| /* Create container for context and r/w events */ |
| e = (redisAeEvents*)malloc(sizeof(*e)); |
| e->context = ac; |
| e->loop = loop; |
| e->fd = c->fd; |
| e->reading = e->writing = 0; |
| |
| /* Register functions to start/stop listening for events */ |
| ac->ev.addRead = redisAeAddRead; |
| ac->ev.delRead = redisAeDelRead; |
| ac->ev.addWrite = redisAeAddWrite; |
| ac->ev.delWrite = redisAeDelWrite; |
| ac->ev.cleanup = redisAeCleanup; |
| ac->ev.data = e; |
| |
| return REDIS_OK; |
| } |
| #endif |