/hiredis/adapters/libevent.h

  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 <event.h>
 34#include "../hiredis.h"
 35#include "../async.h"
 36
 37typedef struct redisLibeventEvents {
 38    redisAsyncContext *context;
 39    struct event rev, wev;
 40} redisLibeventEvents;
 41
 42static void redisLibeventReadEvent(int fd, short event, void *arg) {
 43    ((void)fd); ((void)event);
 44    redisLibeventEvents *e = (redisLibeventEvents*)arg;
 45    redisAsyncHandleRead(e->context);
 46}
 47
 48static void redisLibeventWriteEvent(int fd, short event, void *arg) {
 49    ((void)fd); ((void)event);
 50    redisLibeventEvents *e = (redisLibeventEvents*)arg;
 51    redisAsyncHandleWrite(e->context);
 52}
 53
 54static void redisLibeventAddRead(void *privdata) {
 55    redisLibeventEvents *e = (redisLibeventEvents*)privdata;
 56    event_add(&e->rev,NULL);
 57}
 58
 59static void redisLibeventDelRead(void *privdata) {
 60    redisLibeventEvents *e = (redisLibeventEvents*)privdata;
 61    event_del(&e->rev);
 62}
 63
 64static void redisLibeventAddWrite(void *privdata) {
 65    redisLibeventEvents *e = (redisLibeventEvents*)privdata;
 66    event_add(&e->wev,NULL);
 67}
 68
 69static void redisLibeventDelWrite(void *privdata) {
 70    redisLibeventEvents *e = (redisLibeventEvents*)privdata;
 71    event_del(&e->wev);
 72}
 73
 74static void redisLibeventCleanup(void *privdata) {
 75    redisLibeventEvents *e = (redisLibeventEvents*)privdata;
 76    event_del(&e->rev);
 77    event_del(&e->wev);
 78    free(e);
 79}
 80
 81static 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    event_set(&e->rev,c->fd,EV_READ,redisLibeventReadEvent,e);
103    event_set(&e->wev,c->fd,EV_WRITE,redisLibeventWriteEvent,e);
104    event_base_set(base,&e->rev);
105    event_base_set(base,&e->wev);
106    return REDIS_OK;
107}
108#endif