From a66916719b89f30c92afeb6e9ce5a4c3a4d4c552 Mon Sep 17 00:00:00 2001
From: "hewei.it" <hewei.it@bytedance.com>
Date: Wed, 23 Dec 2020 10:39:12 +0800
Subject: [PATCH] Add adapters/libhv

---
 adapters/libhv.h         | 69 ++++++++++++++++++++++++++++++++++++++++
 examples/CMakeLists.txt  |  6 ++++
 examples/example-libhv.c | 55 ++++++++++++++++++++++++++++++++
 3 files changed, 130 insertions(+)
 create mode 100644 adapters/libhv.h
 create mode 100644 examples/example-libhv.c

diff --git a/adapters/libhv.h b/adapters/libhv.h
new file mode 100644
index 000000000..368a37935
--- /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 49cd8d440..23b6a925b 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 000000000..23c8bc12e
--- /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;
+}