Add example-libevent-pubsub

Co-Authored-By: Tudor Bosman <[email protected]>
redis · Jul 19, 2023 · e90ea9a · e90ea9a
1 parent 1dff7b4
commit e90ea9a
Show file tree

Hide file tree

Showing 3 changed files with 175 additions and 1 deletion.
diff --git a/Makefile b/Makefile
@@ -4,7 +4,7 @@
 # This file is released under the BSD license, see the COPYING file
 
 OBJ=alloc.o net.o hiredis.o sds.o async.o read.o sockcompat.o
-EXAMPLES=hiredis-example hiredis-example-libevent hiredis-example-libev hiredis-example-glib hiredis-example-push hiredis-example-poll
+EXAMPLES=hiredis-example hiredis-example-libevent hiredis-example-libevent-pubsub hiredis-example-libev hiredis-example-glib hiredis-example-push hiredis-example-poll
 TESTS=hiredis-test
 LIBNAME=libhiredis
 PKGCONFNAME=hiredis.pc
@@ -185,6 +185,9 @@ $(SSL_OBJ): ssl.c hiredis.h read.h sds.h alloc.h async.h win32.h async_private.h
 hiredis-example-libevent: examples/example-libevent.c adapters/libevent.h $(STLIBNAME)
 	$(CC) -o examples/$@ $(REAL_CFLAGS) -I. $< -levent $(STLIBNAME) $(REAL_LDFLAGS)
 
+hiredis-example-libevent-pubsub: examples/example-libevent-pubsub.c adapters/libevent.h $(STLIBNAME)
+	$(CC) -o examples/$@ $(REAL_CFLAGS) -I. $< -levent $(STLIBNAME) $(REAL_LDFLAGS)
+
 hiredis-example-libevent-ssl: examples/example-libevent-ssl.c adapters/libevent.h $(STLIBNAME) $(SSL_STLIBNAME)
 	$(CC) -o examples/$@ $(REAL_CFLAGS) -I. $< -levent $(STLIBNAME) $(SSL_STLIBNAME) $(REAL_LDFLAGS) $(SSL_LDFLAGS)
 

diff --git a/examples/CMakeLists.txt b/examples/CMakeLists.txt
@@ -23,6 +23,8 @@ FIND_PATH(LIBEVENT event.h)
 if (LIBEVENT)
     ADD_EXECUTABLE(example-libevent example-libevent.c)
     TARGET_LINK_LIBRARIES(example-libevent hiredis event)
+    ADD_EXECUTABLE(example-libevent-pubsub example-libevent-pubsub.c)
+    TARGET_LINK_LIBRARIES(example-libevent-pubsub hiredis event)
 ENDIF()
 
 FIND_PATH(LIBHV hv/hv.h)

diff --git a/examples/example-libevent-pubsub.c b/examples/example-libevent-pubsub.c
@@ -0,0 +1,169 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <signal.h>
+
+#include <assert.h>
+#include <hiredis.h>
+#include <async.h>
+#include <adapters/libevent.h>
+
+void printReplyInternal(const redisReply* r) {
+    switch (r->type) {
+    case REDIS_REPLY_INTEGER:
+        printf("(integer %lld)", r->integer);
+        break;
+    case REDIS_REPLY_DOUBLE:
+        printf("(double %f)", r->dval);
+        break;
+    case REDIS_REPLY_ERROR:
+        printf("(error %s)", r->str);
+        break;
+    case REDIS_REPLY_STATUS:
+        printf("(status %s)", r->str);
+        break;
+    case REDIS_REPLY_STRING:
+        printf("(string %s)", r->str);
+        break;
+    case REDIS_REPLY_VERB:
+        printf("(verb %s)", r->str);
+        break;
+    case REDIS_REPLY_ARRAY:
+        printf("(array %zu", r->elements);
+        for (size_t i = 0; i < r->elements; ++i) {
+            putchar(' ');
+            printReplyInternal(r->element[i]);
+        }
+        putchar(')');
+        break;
+    default:
+        printf("(?%d)", r->type);
+        break;
+    }
+}
+
+void printReply(const redisReply* r) {
+    printReplyInternal(r);
+    putchar('\n');
+}
+
+void getCallback(redisAsyncContext *c, void *r, void *privdata) {
+    redisReply *reply = r;
+    if (reply == NULL) {
+        if (c->errstr) {
+            printf("errstr: %s\n", c->errstr);
+        }
+        return;
+    }
+    printf("getCallback: privdata='%s' reply='%s'\n", (char*)privdata, reply->str);
+}
+
+void getFinalizer(redisAsyncContext *c, void *privdata) {
+    printf("Get finalizer called\n");
+    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");
+}
+
+typedef struct _SubscribeData {
+    int break_loop;
+    int remaining_message_count;
+} SubscribeData;
+
+void subscribeCallback(redisAsyncContext *c, void *r, void *privdata) {
+    redisReply *reply = r;
+    SubscribeData *sd = privdata;
+    if (reply == NULL) {
+        if (c->errstr) {
+            printf("errstr: %s\n", c->errstr);
+        }
+        return;
+    }
+
+    printf("Subscribe reply: ");
+    printReply(reply);
+
+    assert(reply->type == REDIS_REPLY_ARRAY);
+    assert(reply->elements == 3);
+    assert(reply->element[0]->type == REDIS_REPLY_STRING);
+
+    if (!strcasecmp(reply->element[0]->str, "message")) {
+        if (--sd->remaining_message_count == 0) {
+            redisAsyncCommand(c, NULL, NULL, "UNSUBSCRIBE foo");
+        }
+    }
+
+    if (sd->break_loop) {
+        sd->break_loop = 0;
+        redisLibeventEvents* e = c->ev.data;
+        event_base_loopbreak(e->base);
+    }
+}
+
+void subscribeFinalizer(redisAsyncContext *c, void *privdata) {
+    printf("Subscribe finalizer called\n");
+    redisAsyncDisconnect(c);
+}
+
+int main(void) {
+#ifndef _WIN32
+    signal(SIGPIPE, SIG_IGN);
+#endif
+
+    struct event_base *base = event_base_new();
+    redisOptions options = {0};
+    REDIS_OPTIONS_SET_TCP(&options, "127.0.0.1", 6379);
+    struct timeval tv = {0};
+    tv.tv_sec = 1;
+    options.connect_timeout = &tv;
+
+
+    redisAsyncContext *sub = redisAsyncConnectWithOptions(&options);
+    if (sub->err) {
+        /* Let *c leak for now... */
+        printf("Error: %s\n", sub->errstr);
+        return 1;
+    }
+
+    redisLibeventAttach(sub,base);
+
+    redisAsyncContext *c = redisAsyncConnectWithOptions(&options);
+    if (c->err) {
+        /* Let *c leak for now... */
+        printf("Error: %s\n", c->errstr);
+        return 1;
+    }
+
+    redisLibeventAttach(c,base);
+    redisAsyncSetConnectCallback(c,connectCallback);
+    redisAsyncSetDisconnectCallback(c,disconnectCallback);
+
+    SubscribeData sd;
+    memset(&sd, 0, sizeof(sd));
+    sd.break_loop = 1;
+    sd.remaining_message_count = 3;
+    redisAsyncCommandWithFinalizer(sub, subscribeCallback, subscribeFinalizer, &sd, "SUBSCRIBE foo");
+    event_base_dispatch(base);
+
+    redisAsyncCommand(c, NULL, NULL, "SET key %b", "hello");
+    for (int i = 0; i < 3; ++i) {
+        redisAsyncCommand(c, NULL, NULL, "PUBLISH foo msg%d", i);
+    }
+    redisAsyncCommandWithFinalizer(c, getCallback, getFinalizer, "get-with-finalizer", "GET key");
+    event_base_dispatch(base);
+    return 0;
+}