如何利用Redis和C++实现发布-订阅功能

#include <iOStream>
#include <hiredis/hiredis.h>

redisContext* connectToRedis(const char* hostname, int port) {
    redisContext* conn = redisConnect(hostname, port);
    if (conn == NULL || conn->err) {
        if (conn) {
            std::cout << "Error: " << conn->errstr << std::endl;
        } else {
            std::cout << "Unable to allocate redis context." << std::endl;
        }
        return NULL;
    }
    return conn;
}

bool publishMessage(redisContext* conn, const char* channel, const char* message) {
    redisReply* reply = (redisReply*)redisCommand(conn, "PUBLISH %s %s", channel, message);
    if (reply && reply->type == REDIS_REPLY_INTEGER && reply->integer > 0) {
        freeReplyObject(reply);
        return true;
    }
    freeReplyObject(reply);
    return false;
}

void subscribeChannel(redisContext* conn, const char* channel) {
    redisReply* reply = (redisReply*)redisCommand(conn, "SUBSCRIBE %s", channel);
    freeReplyObject(reply);
    while (redisGetReply(conn, (void**)&reply) == REDIS_OK) {
        if (reply->type == REDIS_REPLY_ARRAY && reply->elements >= 3 && strcmp(reply->element[0]->str, "message") == 0) {
            std::cout << "Received message: " << reply->element[2]->str << std::endl;
        }
        freeReplyObject(reply);
    }
}

int main() {
    // 连接Redis服务器
    redisContext* conn = connectToRedis("localhost", 6379);
    if (conn == NULL) {
        return 1;
    }

    // 发布消息
    std::string channel = "test_channel";
    std::string message = "Hello, Redis!";
    if (publishMessage(conn, channel.c_str(), message.c_str())) {
        std::cout << "Message published successfully." << std::endl;
    } else {
        std::cout << "Failed to publish message." << std::endl;
    }

    // 订阅消息
    subscribeChannel(conn, channel.c_str());

    // 关闭Redis连接
    redisFree(conn);

    return 0;
}