diff --git a/src/modules/hellodict.c b/src/modules/hellodict.c deleted file mode 100644 index db2fd17e8..000000000 --- a/src/modules/hellodict.c +++ /dev/null @@ -1,125 +0,0 @@ -/* Hellodict -- An example of modules dictionary API - * - * This module implements a volatile key-value store on top of the - * dictionary exported by the modules API. - * - * ----------------------------------------------------------------------------- - * - * Copyright (c) 2018, Redis Ltd. - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions are met: - * - * * Redistributions of source code must retain the above copyright notice, - * this list of conditions and the following disclaimer. - * * Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in the - * documentation and/or other materials provided with the distribution. - * * Neither the name of Redis nor the names of its contributors may be used - * to endorse or promote products derived from this software without - * specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" - * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE - * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE - * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE - * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR - * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF - * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS - * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN - * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) - * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE - * POSSIBILITY OF SUCH DAMAGE. - */ - -#include "../valkeymodule.h" -#include -#include -#include -#include - -static ValkeyModuleDict *Keyspace; - -/* HELLODICT.SET - * - * Set the specified key to the specified value. */ -int cmd_SET(ValkeyModuleCtx *ctx, ValkeyModuleString **argv, int argc) { - if (argc != 3) return ValkeyModule_WrongArity(ctx); - ValkeyModule_DictSet(Keyspace, argv[1], argv[2]); - /* We need to keep a reference to the value stored at the key, otherwise - * it would be freed when this callback returns. */ - ValkeyModule_RetainString(NULL, argv[2]); - return ValkeyModule_ReplyWithSimpleString(ctx, "OK"); -} - -/* HELLODICT.GET - * - * Return the value of the specified key, or a null reply if the key - * is not defined. */ -int cmd_GET(ValkeyModuleCtx *ctx, ValkeyModuleString **argv, int argc) { - if (argc != 2) return ValkeyModule_WrongArity(ctx); - ValkeyModuleString *val = ValkeyModule_DictGet(Keyspace, argv[1], NULL); - if (val == NULL) { - return ValkeyModule_ReplyWithNull(ctx); - } else { - return ValkeyModule_ReplyWithString(ctx, val); - } -} - -/* HELLODICT.KEYRANGE - * - * Return a list of matching keys, lexicographically between startkey - * and endkey. No more than 'count' items are emitted. */ -int cmd_KEYRANGE(ValkeyModuleCtx *ctx, ValkeyModuleString **argv, int argc) { - if (argc != 4) return ValkeyModule_WrongArity(ctx); - - /* Parse the count argument. */ - long long count; - if (ValkeyModule_StringToLongLong(argv[3], &count) != VALKEYMODULE_OK) { - return ValkeyModule_ReplyWithError(ctx, "ERR invalid count"); - } - - /* Seek the iterator. */ - ValkeyModuleDictIter *iter = ValkeyModule_DictIteratorStart(Keyspace, ">=", argv[1]); - - /* Reply with the matching items. */ - char *key; - size_t keylen; - long long replylen = 0; /* Keep track of the emitted array len. */ - ValkeyModule_ReplyWithArray(ctx, VALKEYMODULE_POSTPONED_LEN); - while ((key = ValkeyModule_DictNextC(iter, &keylen, NULL)) != NULL) { - if (replylen >= count) break; - if (ValkeyModule_DictCompare(iter, "<=", argv[2]) == VALKEYMODULE_ERR) break; - ValkeyModule_ReplyWithStringBuffer(ctx, key, keylen); - replylen++; - } - ValkeyModule_ReplySetArrayLength(ctx, replylen); - - /* Cleanup. */ - ValkeyModule_DictIteratorStop(iter); - return VALKEYMODULE_OK; -} - -/* This function must be present on each module. It is used in order to - * register the commands into the server. */ -int ValkeyModule_OnLoad(ValkeyModuleCtx *ctx, ValkeyModuleString **argv, int argc) { - VALKEYMODULE_NOT_USED(argv); - VALKEYMODULE_NOT_USED(argc); - - if (ValkeyModule_Init(ctx, "hellodict", 1, VALKEYMODULE_APIVER_1) == VALKEYMODULE_ERR) return VALKEYMODULE_ERR; - - if (ValkeyModule_CreateCommand(ctx, "hellodict.set", cmd_SET, "write deny-oom", 1, 1, 1) == VALKEYMODULE_ERR) - return VALKEYMODULE_ERR; - - if (ValkeyModule_CreateCommand(ctx, "hellodict.get", cmd_GET, "readonly", 1, 1, 1) == VALKEYMODULE_ERR) - return VALKEYMODULE_ERR; - - if (ValkeyModule_CreateCommand(ctx, "hellodict.keyrange", cmd_KEYRANGE, "readonly", 1, 1, 1) == VALKEYMODULE_ERR) - return VALKEYMODULE_ERR; - - /* Create our global dictionary. Here we'll set our keys and values. */ - Keyspace = ValkeyModule_CreateDict(NULL); - - return VALKEYMODULE_OK; -}