summaryrefslogtreecommitdiff
path: root/backend/src/Redis/Connection.zig
blob: 83adb2cd0e8714a83cfc5877e2bc80d99e845ec2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
const std = @import("std");
const redis = @cImport(
    @cInclude("hiredis/hiredis.h")
    );

var connection: ?*redis.redisContext = undefined;
const addr: [*]const u8 = "127.0.0.1";
const port: u16 = 6379;

pub fn Connect() !void {
    connection = redis.redisConnect(addr, port);
    if (connection == null or connection.?.err != 0) {
        if (connection != null) {
            @panic(&connection.?.errstr);
        }
        @panic("The redis connection failed: general error");
    }
}

pub fn Disconnect() void {
    redis.redisFree(connection);
}

pub const Message = struct {
    Key: []const u8,
    Value: []const u8,
    SecondsToLive: u16
};

pub fn WriteToTopic(topic: []const u8, message: Message) !void {
    _ = redis.redisCommand(connection, "SET %s:%s %s EX %d", 
        topic.ptr,
        message.Key.ptr,
        message.Value.ptr,
        message.SecondsToLive); 
}

pub fn ReadFromTopic(topic: []const u8, Key: []const u8) ?[]const u8 {
    const raw = redis.redisCommand(connection, "GET %s:%s", 
            topic.ptr,
            Key.ptr);
    if (raw == null) {
        return null;
    }
    const resp: *redis.redisReply = @alignCast(@ptrCast(raw));
    if (resp.type != redis.REDIS_REPLY_STRING) {
        return null;
    }
    return std.mem.span(resp.str);
}

test "Redis connection" {
    try Connect();
    defer Disconnect();

    try WriteToTopic("test", Message {
        .Key = "key",
        .Value = "value",
        .SecondsToLive = 1,
    });

    try std.testing.expectEqualStrings("value", ReadFromTopic("test", "key") orelse "not found");
    std.Thread.sleep(1_500_000_000);
    // KVP TTL check
    try std.testing.expectEqualStrings("not found", ReadFromTopic("test", "key") orelse "not found");
}