mirror of
https://codeberg.org/redict/redict.git
synced 2025-01-22 08:08:53 -05:00
03d144a452
Replaces Redis => Redict API References: https://codeberg.org/redict/redict/issues/21 Signed-off-by: Drew DeVault <sir@cmpwn.com>
108 lines
3.3 KiB
C
108 lines
3.3 KiB
C
// SPDX-FileCopyrightText: 2024 Redict Contributors
|
|
// SPDX-FileCopyrightText: 2024 Salvatore Sanfilippo <antirez at gmail dot com>
|
|
//
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
// SPDX-License-Identifier: LGPL-3.0-only
|
|
|
|
#include "redictmodule.h"
|
|
|
|
static void timer_callback(RedictModuleCtx *ctx, void *data)
|
|
{
|
|
RedictModuleString *keyname = data;
|
|
RedictModuleCallReply *reply;
|
|
|
|
reply = RedictModule_Call(ctx, "INCR", "s", keyname);
|
|
if (reply != NULL)
|
|
RedictModule_FreeCallReply(reply);
|
|
RedictModule_FreeString(ctx, keyname);
|
|
}
|
|
|
|
int test_createtimer(RedictModuleCtx *ctx, RedictModuleString **argv, int argc)
|
|
{
|
|
if (argc != 3) {
|
|
RedictModule_WrongArity(ctx);
|
|
return REDICTMODULE_OK;
|
|
}
|
|
|
|
long long period;
|
|
if (RedictModule_StringToLongLong(argv[1], &period) == REDICTMODULE_ERR) {
|
|
RedictModule_ReplyWithError(ctx, "Invalid time specified.");
|
|
return REDICTMODULE_OK;
|
|
}
|
|
|
|
RedictModuleString *keyname = argv[2];
|
|
RedictModule_RetainString(ctx, keyname);
|
|
|
|
RedictModuleTimerID id = RedictModule_CreateTimer(ctx, period, timer_callback, keyname);
|
|
RedictModule_ReplyWithLongLong(ctx, id);
|
|
|
|
return REDICTMODULE_OK;
|
|
}
|
|
|
|
int test_gettimer(RedictModuleCtx *ctx, RedictModuleString **argv, int argc)
|
|
{
|
|
if (argc != 2) {
|
|
RedictModule_WrongArity(ctx);
|
|
return REDICTMODULE_OK;
|
|
}
|
|
|
|
long long id;
|
|
if (RedictModule_StringToLongLong(argv[1], &id) == REDICTMODULE_ERR) {
|
|
RedictModule_ReplyWithError(ctx, "Invalid id specified.");
|
|
return REDICTMODULE_OK;
|
|
}
|
|
|
|
uint64_t remaining;
|
|
RedictModuleString *keyname;
|
|
if (RedictModule_GetTimerInfo(ctx, id, &remaining, (void **)&keyname) == REDICTMODULE_ERR) {
|
|
RedictModule_ReplyWithNull(ctx);
|
|
} else {
|
|
RedictModule_ReplyWithArray(ctx, 2);
|
|
RedictModule_ReplyWithString(ctx, keyname);
|
|
RedictModule_ReplyWithLongLong(ctx, remaining);
|
|
}
|
|
|
|
return REDICTMODULE_OK;
|
|
}
|
|
|
|
int test_stoptimer(RedictModuleCtx *ctx, RedictModuleString **argv, int argc)
|
|
{
|
|
if (argc != 2) {
|
|
RedictModule_WrongArity(ctx);
|
|
return REDICTMODULE_OK;
|
|
}
|
|
|
|
long long id;
|
|
if (RedictModule_StringToLongLong(argv[1], &id) == REDICTMODULE_ERR) {
|
|
RedictModule_ReplyWithError(ctx, "Invalid id specified.");
|
|
return REDICTMODULE_OK;
|
|
}
|
|
|
|
int ret = 0;
|
|
RedictModuleString *keyname;
|
|
if (RedictModule_StopTimer(ctx, id, (void **) &keyname) == REDICTMODULE_OK) {
|
|
RedictModule_FreeString(ctx, keyname);
|
|
ret = 1;
|
|
}
|
|
|
|
RedictModule_ReplyWithLongLong(ctx, ret);
|
|
return REDICTMODULE_OK;
|
|
}
|
|
|
|
|
|
int RedictModule_OnLoad(RedictModuleCtx *ctx, RedictModuleString **argv, int argc) {
|
|
REDICTMODULE_NOT_USED(argv);
|
|
REDICTMODULE_NOT_USED(argc);
|
|
if (RedictModule_Init(ctx,"timer",1,REDICTMODULE_APIVER_1)== REDICTMODULE_ERR)
|
|
return REDICTMODULE_ERR;
|
|
|
|
if (RedictModule_CreateCommand(ctx,"test.createtimer", test_createtimer,"",0,0,0) == REDICTMODULE_ERR)
|
|
return REDICTMODULE_ERR;
|
|
if (RedictModule_CreateCommand(ctx,"test.gettimer", test_gettimer,"",0,0,0) == REDICTMODULE_ERR)
|
|
return REDICTMODULE_ERR;
|
|
if (RedictModule_CreateCommand(ctx,"test.stoptimer", test_stoptimer,"",0,0,0) == REDICTMODULE_ERR)
|
|
return REDICTMODULE_ERR;
|
|
|
|
return REDICTMODULE_OK;
|
|
}
|