forked from redis/hiredis
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
bcf8354
commit 61277d0
Showing
3 changed files
with
184 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,123 @@ | ||
#ifndef __HIREDIS_LIBUV_H__ | ||
#define __HIREDIS_LIBUV_H__ | ||
#include <uv.h> | ||
#include "../hiredis.h" | ||
#include "../async.h" | ||
//#include <stdlib.h> // malloc(), free() | ||
//#include <string.h> // memset() | ||
//#include <assert.h> // assert() | ||
|
||
|
||
typedef struct redisLibuvEvents { | ||
redisAsyncContext* context; | ||
uv_poll_t handle; | ||
int events; | ||
} redisLibuvEvents; | ||
|
||
|
||
static void redisLibuvPoll(uv_poll_t* handle, int status, int events) { | ||
redisLibuvEvents* p = (redisLibuvEvents*)handle->data; | ||
|
||
if (status != 0) { | ||
return; | ||
} | ||
|
||
if (events & UV_READABLE) { | ||
redisAsyncHandleRead(p->context); | ||
} | ||
if (events & UV_WRITABLE) { | ||
redisAsyncHandleWrite(p->context); | ||
} | ||
} | ||
|
||
|
||
static void redisLibuvAddRead(void *privdata) { | ||
redisLibuvEvents* p = (redisLibuvEvents*)privdata; | ||
|
||
p->events |= UV_READABLE; | ||
|
||
uv_poll_start(&p->handle, p->events, redisLibuvPoll); | ||
} | ||
|
||
|
||
static void redisLibuvDelRead(void *privdata) { | ||
redisLibuvEvents* p = (redisLibuvEvents*)privdata; | ||
|
||
p->events &= ~UV_READABLE; | ||
|
||
if (p->events) { | ||
uv_poll_start(&p->handle, p->events, redisLibuvPoll); | ||
} else { | ||
uv_poll_stop(&p->handle); | ||
} | ||
} | ||
|
||
|
||
static void redisLibuvAddWrite(void *privdata) { | ||
redisLibuvEvents* p = (redisLibuvEvents*)privdata; | ||
|
||
p->events |= UV_WRITABLE; | ||
|
||
uv_poll_start(&p->handle, p->events, redisLibuvPoll); | ||
} | ||
|
||
|
||
static void redisLibuvDelWrite(void *privdata) { | ||
redisLibuvEvents* p = (redisLibuvEvents*)privdata; | ||
|
||
p->events &= ~UV_WRITABLE; | ||
|
||
if (p->events) { | ||
uv_poll_start(&p->handle, p->events, redisLibuvPoll); | ||
} else { | ||
uv_poll_stop(&p->handle); | ||
} | ||
} | ||
|
||
|
||
static void on_close(uv_handle_t* handle) { | ||
redisLibuvEvents* p = (redisLibuvEvents*)handle->data; | ||
|
||
free(p); | ||
} | ||
|
||
|
||
static void redisLibuvCleanup(void *privdata) { | ||
redisLibuvEvents* p = (redisLibuvEvents*)privdata; | ||
|
||
uv_close((uv_handle_t*)&p->handle, on_close); | ||
} | ||
|
||
|
||
int redisLibuvAttach(redisAsyncContext* ac, uv_loop_t* loop) { | ||
redisContext *c = &(ac->c); | ||
|
||
if (ac->ev.data != NULL) { | ||
return REDIS_ERR; | ||
} | ||
|
||
ac->ev.addRead = redisLibuvAddRead; | ||
ac->ev.delRead = redisLibuvDelRead; | ||
ac->ev.addWrite = redisLibuvAddWrite; | ||
ac->ev.delWrite = redisLibuvDelWrite; | ||
ac->ev.cleanup = redisLibuvCleanup; | ||
|
||
redisLibuvEvents* p = malloc(sizeof(*p)); | ||
|
||
if (!p) { | ||
return REDIS_ERR; | ||
} | ||
|
||
memset(p, 0, sizeof(*p)); | ||
|
||
if (uv_poll_init(loop, &p->handle, c->fd) != 0) { | ||
return REDIS_ERR; | ||
} | ||
|
||
ac->ev.data = p; | ||
p->handle.data = p; | ||
p->context = ac; | ||
|
||
return REDIS_OK; | ||
} | ||
#endif |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,52 @@ | ||
#include <stdio.h> | ||
#include <stdlib.h> | ||
#include <string.h> | ||
#include <signal.h> | ||
#include "hiredis.h" | ||
#include "async.h" | ||
#include "adapters/libuv.h" | ||
|
||
void getCallback(redisAsyncContext *c, void *r, void *privdata) { | ||
redisReply *reply = r; | ||
if (reply == NULL) return; | ||
printf("argv[%s]: %s\n", (char*)privdata, reply->str); | ||
|
||
/* Disconnect after receiving the reply to GET */ | ||
redisAsyncDisconnect(c); | ||
} | ||
|
||
void connectCallback(const redisAsyncContext *c, int status) { | ||
if (status != REDIS_OK) { | ||
printf("Error: %s\n", c->errstr); | ||
return; | ||
} | ||
printf("Connected...\n"); | ||
} | ||
|
||
void disconnectCallback(const redisAsyncContext *c, int status) { | ||
if (status != REDIS_OK) { | ||
printf("Error: %s\n", c->errstr); | ||
return; | ||
} | ||
printf("Disconnected...\n"); | ||
} | ||
|
||
int main (int argc, char **argv) { | ||
signal(SIGPIPE, SIG_IGN); | ||
uv_loop_t* loop = uv_default_loop(); | ||
|
||
redisAsyncContext *c = redisAsyncConnect("127.0.0.1", 6379); | ||
if (c->err) { | ||
/* Let *c leak for now... */ | ||
printf("Error: %s\n", c->errstr); | ||
return 1; | ||
} | ||
|
||
redisLibuvAttach(c,loop); | ||
redisAsyncSetConnectCallback(c,connectCallback); | ||
redisAsyncSetDisconnectCallback(c,disconnectCallback); | ||
redisAsyncCommand(c, NULL, NULL, "SET key %b", argv[argc-1], strlen(argv[argc-1])); | ||
redisAsyncCommand(c, getCallback, (char*)"end-1", "GET key"); | ||
uv_run(loop, UV_RUN_DEFAULT); | ||
return 0; | ||
} |