forked from mirror/ledisdb
109 lines
3.9 KiB
C
109 lines
3.9 KiB
C
/*
|
|
* 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_LIBEVENT_H__
|
|
#define __HIREDIS_LIBEVENT_H__
|
|
#include <event.h>
|
|
#include "../hiredis.h"
|
|
#include "../async.h"
|
|
|
|
typedef struct redisLibeventEvents {
|
|
redisAsyncContext *context;
|
|
struct event rev, wev;
|
|
} redisLibeventEvents;
|
|
|
|
static void redisLibeventReadEvent(int fd, short event, void *arg) {
|
|
((void)fd); ((void)event);
|
|
redisLibeventEvents *e = (redisLibeventEvents*)arg;
|
|
redisAsyncHandleRead(e->context);
|
|
}
|
|
|
|
static void redisLibeventWriteEvent(int fd, short event, void *arg) {
|
|
((void)fd); ((void)event);
|
|
redisLibeventEvents *e = (redisLibeventEvents*)arg;
|
|
redisAsyncHandleWrite(e->context);
|
|
}
|
|
|
|
static void redisLibeventAddRead(void *privdata) {
|
|
redisLibeventEvents *e = (redisLibeventEvents*)privdata;
|
|
event_add(&e->rev,NULL);
|
|
}
|
|
|
|
static void redisLibeventDelRead(void *privdata) {
|
|
redisLibeventEvents *e = (redisLibeventEvents*)privdata;
|
|
event_del(&e->rev);
|
|
}
|
|
|
|
static void redisLibeventAddWrite(void *privdata) {
|
|
redisLibeventEvents *e = (redisLibeventEvents*)privdata;
|
|
event_add(&e->wev,NULL);
|
|
}
|
|
|
|
static void redisLibeventDelWrite(void *privdata) {
|
|
redisLibeventEvents *e = (redisLibeventEvents*)privdata;
|
|
event_del(&e->wev);
|
|
}
|
|
|
|
static void redisLibeventCleanup(void *privdata) {
|
|
redisLibeventEvents *e = (redisLibeventEvents*)privdata;
|
|
event_del(&e->rev);
|
|
event_del(&e->wev);
|
|
free(e);
|
|
}
|
|
|
|
static int redisLibeventAttach(redisAsyncContext *ac, struct event_base *base) {
|
|
redisContext *c = &(ac->c);
|
|
redisLibeventEvents *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 = (redisLibeventEvents*)malloc(sizeof(*e));
|
|
e->context = ac;
|
|
|
|
/* Register functions to start/stop listening for events */
|
|
ac->ev.addRead = redisLibeventAddRead;
|
|
ac->ev.delRead = redisLibeventDelRead;
|
|
ac->ev.addWrite = redisLibeventAddWrite;
|
|
ac->ev.delWrite = redisLibeventDelWrite;
|
|
ac->ev.cleanup = redisLibeventCleanup;
|
|
ac->ev.data = e;
|
|
|
|
/* Initialize and install read/write events */
|
|
event_set(&e->rev,c->fd,EV_READ,redisLibeventReadEvent,e);
|
|
event_set(&e->wev,c->fd,EV_WRITE,redisLibeventWriteEvent,e);
|
|
event_base_set(base,&e->rev);
|
|
event_base_set(base,&e->wev);
|
|
return REDIS_OK;
|
|
}
|
|
#endif
|