diff options
author | hewei.it <hewei.it@bytedance.com> | 2020-12-23 10:39:12 +0800 |
---|---|---|
committer | Michael Grunder <michael.grunder@gmail.com> | 2022-09-07 11:14:45 -0700 |
commit | a66916719b89f30c92afeb6e9ce5a4c3a4d4c552 (patch) | |
tree | b350f050f4a6dfb91036ed7cb0d00a4cf68490ae | |
parent | 855b48a8191cd1fab865091cf4760017f85594a7 (diff) |
Add adapters/libhv
-rw-r--r-- | adapters/libhv.h | 69 | ||||
-rw-r--r-- | examples/CMakeLists.txt | 6 | ||||
-rw-r--r-- | examples/example-libhv.c | 55 |
3 files changed, 130 insertions, 0 deletions
diff --git a/adapters/libhv.h b/adapters/libhv.h new file mode 100644 index 0000000..368a379 --- /dev/null +++ b/adapters/libhv.h @@ -0,0 +1,69 @@ +#ifndef __HIREDIS_LIBHV_H__ +#define __HIREDIS_LIBHV_H__ + +#include <hv/hloop.h> +#include "../hiredis.h" +#include "../async.h" + +static void redisLibhvHandleEvents(hio_t* io) { + redisAsyncContext* context = (redisAsyncContext*)hevent_userdata(io); + int events = hio_events(io); + int revents = hio_revents(io); + if (context && (events & HV_READ) && (revents & HV_READ)) { + redisAsyncHandleRead(context); + } + if (context && (events & HV_WRITE) && (revents & HV_WRITE)) { + redisAsyncHandleWrite(context); + } +} + +static void redisLibhvAddRead(void *privdata) { + hio_t* io = (hio_t*)privdata; + hio_add(io, redisLibhvHandleEvents, HV_READ); +} + +static void redisLibhvDelRead(void *privdata) { + hio_t* io = (hio_t*)privdata; + hio_del(io, HV_READ); +} + +static void redisLibhvAddWrite(void *privdata) { + hio_t* io = (hio_t*)privdata; + hio_add(io, redisLibhvHandleEvents, HV_WRITE); +} + +static void redisLibhvDelWrite(void *privdata) { + hio_t* io = (hio_t*)privdata; + hio_del(io, HV_WRITE); +} + +static void redisLibhvCleanup(void *privdata) { + hio_t* io = (hio_t*)privdata; + hio_close(io); + hevent_set_userdata(io, NULL); +} + +static int redisLibhvAttach(redisAsyncContext* ac, hloop_t* loop) { + redisContext *c = &(ac->c); + hio_t* io = NULL; + + if (ac->ev.data != NULL) { + return REDIS_ERR; + } + + io = hio_get(loop, c->fd); + if (io == NULL) { + return REDIS_ERR; + } + hevent_set_userdata(io, ac); + + ac->ev.addRead = redisLibhvAddRead; + ac->ev.delRead = redisLibhvDelRead; + ac->ev.addWrite = redisLibhvAddWrite; + ac->ev.delWrite = redisLibhvDelWrite; + ac->ev.cleanup = redisLibhvCleanup; + ac->ev.data = io; + + return REDIS_OK; +} +#endif diff --git a/examples/CMakeLists.txt b/examples/CMakeLists.txt index 49cd8d4..23b6a92 100644 --- a/examples/CMakeLists.txt +++ b/examples/CMakeLists.txt @@ -25,6 +25,12 @@ if (LIBEVENT) TARGET_LINK_LIBRARIES(example-libevent hiredis event) ENDIF() +FIND_PATH(LIBHV hv/hv.h) +IF (LIBHV) + ADD_EXECUTABLE(example-libhv example-libhv.c) + TARGET_LINK_LIBRARIES(example-libhv hiredis hv) +ENDIF() + FIND_PATH(LIBUV uv.h) IF (LIBUV) ADD_EXECUTABLE(example-libuv example-libuv.c) diff --git a/examples/example-libhv.c b/examples/example-libhv.c new file mode 100644 index 0000000..23c8bc1 --- /dev/null +++ b/examples/example-libhv.c @@ -0,0 +1,55 @@ +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <signal.h> + +#include <hiredis.h> +#include <async.h> +#include <adapters/libhv.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) { +#ifndef _WIN32 + signal(SIGPIPE, SIG_IGN); +#endif + + redisAsyncContext *c = redisAsyncConnect("127.0.0.1", 6379); + if (c->err) { + /* Let *c leak for now... */ + printf("Error: %s\n", c->errstr); + return 1; + } + + hloop_t* loop = hloop_new(HLOOP_FLAG_QUIT_WHEN_NO_ACTIVE_EVENTS); + redisLibhvAttach(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"); + hloop_run(loop); + return 0; +} |