Documentation: Valkey modules API

The modules documentation is composed of the following pages:

  • Introduction to Valkey modules (this file). An overview about Valkey Modules system and API. It's a good idea to start your reading here.
  • Implementing native data types covers the implementation of native data types into modules.
  • Blocking operations shows how to write blocking commands that will not reply immediately, but will block the client, without blocking the Valkey server, and will provide a reply whenever will be possible.
  • Valkey modules API reference is generated from module.c top comments of ValkeyModule functions. It is a good reference in order to understand how each function works.

Valkey modules make it possible to extend Valkey functionality using external modules, rapidly implementing new Valkey commands with features similar to what can be done inside the core itself.

Valkey modules are dynamic libraries that can be loaded into Valkey at startup, or using the MODULE LOAD command. Valkey exports a C API, in the form of a single C header file called valkeymodule.h. Modules are meant to be written in C, however it will be possible to use C++ or other languages that have C binding functionalities.

Modules are designed in order to be loaded into different versions of Valkey, so a given module does not need to be designed, or recompiled, in order to run with a specific version of Valkey. For this reason, the module will register to the Valkey core using a specific API version. The current API version is "1".

Loading modules

In order to test the module you are developing, you can load the module using the following valkey.conf configuration directive:

loadmodule /path/to/mymodule.so

It is also possible to load a module at runtime using the following command:

MODULE LOAD /path/to/mymodule.so

In order to list all loaded modules, use:

MODULE LIST

Finally, you can unload (and later reload if you wish) a module using the following command:

MODULE UNLOAD mymodule

Note that mymodule above is not the filename without the .so suffix, but instead, the name the module used to register itself into the Valkey core. The name can be obtained using MODULE LIST. However it is good practice that the filename of the dynamic library is the same as the name the module uses to register itself into the Valkey core.

The simplest module you can write

In order to show the different parts of a module, here we'll show a very simple module that implements a command that outputs a random number.

#include "valkeymodule.h"
#include <stdlib.h>

int HelloworldRand_ValkeyCommand(ValkeyModuleCtx *ctx, ValkeyModuleString **argv, int argc) {
    ValkeyModule_ReplyWithLongLong(ctx,rand());
    return VALKEYMODULE_OK;
}

int ValkeyModule_OnLoad(ValkeyModuleCtx *ctx, ValkeyModuleString **argv, int argc) {
    if (ValkeyModule_Init(ctx,"helloworld",1,VALKEYMODULE_APIVER_1)
        == VALKEYMODULE_ERR) return VALKEYMODULE_ERR;

    if (ValkeyModule_CreateCommand(ctx,"helloworld.rand",
        HelloworldRand_ValkeyCommand, "fast random",
        0, 0, 0) == VALKEYMODULE_ERR)
        return VALKEYMODULE_ERR;

    return VALKEYMODULE_OK;
}

The example module has two functions. One implements a command called HELLOWORLD.RAND. This function is specific of that module. However the other function called ValkeyModule_OnLoad() must be present in each Valkey module. It is the entry point for the module to be initialized, register its commands, and potentially other private data structures it uses.

Note that it is a good idea for modules to call commands with the name of the module followed by a dot, and finally the command name, like in the case of HELLOWORLD.RAND. This way it is less likely to have collisions.

Note that if different modules have colliding commands, they'll not be able to work in Valkey at the same time, since the function ValkeyModule_CreateCommand will fail in one of the modules, so the module loading will abort returning an error condition.

Module initialization

The above example shows the usage of the function ValkeyModule_Init(). It should be the first function called by the module OnLoad function. The following is the function prototype:

int ValkeyModule_Init(ValkeyModuleCtx *ctx, const char *modulename,
                     int module_version, int api_version);

The Init function announces the Valkey core that the module has a given name, its version (that is reported by MODULE LIST), and that is willing to use a specific version of the API.

If the API version is wrong, the name is already taken, or there are other similar errors, the function will return VALKEYMODULE_ERR, and the module OnLoad function should return ASAP with an error.

Before the Init function is called, no other API function can be called, otherwise the module will segfault and the Valkey instance will crash.

The second function called, ValkeyModule_CreateCommand, is used in order to register commands into the Valkey core. The following is the prototype:

int ValkeyModule_CreateCommand(ValkeyModuleCtx *ctx, const char *name,
                              ValkeyModuleCmdFunc cmdfunc, const char *strflags,
                              int firstkey, int lastkey, int keystep);

As you can see, most Valkey modules API calls all take as first argument the context of the module, so that they have a reference to the module calling it, to the command and client executing a given command, and so forth.

To create a new command, the above function needs the context, the command's name, a pointer to the function implementing the command, the command's flags and the positions of key names in the command's arguments.

The function that implements the command must have the following prototype:

int mycommand(ValkeyModuleCtx *ctx, ValkeyModuleString **argv, int argc);

The command function arguments are just the context, that will be passed to all the other API calls, the command argument vector, and total number of arguments, as passed by the user.

As you can see, the arguments are provided as pointers to a specific data type, the ValkeyModuleString. This is an opaque data type you have API functions to access and use, direct access to its fields is never needed.

Zooming into the example command implementation, we can find another call:

int ValkeyModule_ReplyWithLongLong(ValkeyModuleCtx *ctx, long long integer);

This function returns an integer to the client that invoked the command, exactly like other Valkey commands do, like for example INCR or SCARD.

Module cleanup

In most cases, there is no need for special cleanup. When a module is unloaded, Valkey will automatically unregister commands and unsubscribe from notifications. However in the case where a module contains some persistent memory or configuration, a module may include an optional ValkeyModule_OnUnload function. If a module provides this function, it will be invoked during the module unload process. The following is the function prototype:

int ValkeyModule_OnUnload(ValkeyModuleCtx *ctx);

The OnUnload function may prevent module unloading by returning VALKEYMODULE_ERR. Otherwise, VALKEYMODULE_OK should be returned.

Setup and dependencies of a Valkey module

Valkey modules don't depend on Valkey or some other library, nor they need to be compiled with a specific valkeymodule.h file. In order to create a new module, just copy a recent version of valkeymodule.h in your source tree, link all the libraries you want, and create a dynamic library having the ValkeyModule_OnLoad() function symbol exported.

The module will be able to load into different versions of Valkey.

A module can be designed to support both newer and older Redis OSS versions where certain API functions are not available in all versions. If an API function is not implemented in the currently running Redis OSS version, the function pointer is set to NULL. This allows the module to check if a function exists before using it:

if (ValkeyModule_SetCommandInfo != NULL) {
    ValkeyModule_SetCommandInfo(cmd, &info);
}

In recent versions of valkeymodule.h, a convenience macro RMAPI_FUNC_SUPPORTED(funcname) is defined. Using the macro or just comparing with NULL is a matter of personal preference.

Passing configuration parameters to Valkey modules

When the module is loaded with the MODULE LOAD command, or using the loadmodule directive in the valkey.conf file, the user is able to pass configuration parameters to the module by adding arguments after the module file name:

loadmodule mymodule.so foo bar 1234

In the above example the strings foo, bar and 1234 will be passed to the module OnLoad() function in the argv argument as an array of ValkeyModuleString pointers. The number of arguments passed is into argc.

The way you can access those strings will be explained in the rest of this document. Normally the module will store the module configuration parameters in some static global variable that can be accessed module wide, so that the configuration can change the behavior of different commands.

Working with ValkeyModuleString objects

The command argument vector argv passed to module commands, and the return value of other module APIs functions, are of type ValkeyModuleString.

Usually you directly pass module strings to other API calls, however sometimes you may need to directly access the string object.

There are a few functions in order to work with string objects:

const char *ValkeyModule_StringPtrLen(ValkeyModuleString *string, size_t *len);

The above function accesses a string by returning its pointer and setting its length in len. You should never write to a string object pointer, as you can see from the const pointer qualifier.

However, if you want, you can create new string objects using the following API:

ValkeyModuleString *ValkeyModule_CreateString(ValkeyModuleCtx *ctx, const char *ptr, size_t len);

The string returned by the above command must be freed using a corresponding call to ValkeyModule_FreeString():

void ValkeyModule_FreeString(ValkeyModuleString *str);

However if you want to avoid having to free strings, the automatic memory management, covered later in this document, can be a good alternative, by doing it for you.

Note that the strings provided via the argument vector argv never need to be freed. You only need to free new strings you create, or new strings returned by other APIs, where it is specified that the returned string must be freed.

Creating strings from numbers or parsing strings as numbers

Creating a new string from an integer is a very common operation, so there is a function to do this:

ValkeyModuleString *mystr = ValkeyModule_CreateStringFromLongLong(ctx,10);

Similarly in order to parse a string as a number:

long long myval;
if (ValkeyModule_StringToLongLong(ctx,argv[1],&myval) == VALKEYMODULE_OK) {
    /* Do something with 'myval' */
}

Accessing Valkey keys from modules

Most Valkey modules, in order to be useful, have to interact with the Valkey data space (this is not always true, for example an ID generator may never touch Valkey keys). Valkey modules have two different APIs in order to access the Valkey data space, one is a low level API that provides very fast access and a set of functions to manipulate Valkey data structures. The other API is more high level, and allows to call Valkey commands and fetch the result, similarly to how Lua scripts access Valkey.

The high level API is also useful in order to access Valkey functionalities that are not available as APIs.

In general modules developers should prefer the low level API, because commands implemented using the low level API run at a speed comparable to the speed of native Valkey commands. However there are definitely use cases for the higher level API. For example often the bottleneck could be processing the data and not accessing it.

Also note that sometimes using the low level API is not harder compared to the higher level one.

Calling Valkey commands

The high level API to access Valkey is the sum of the ValkeyModule_Call() function, together with the functions needed in order to access the reply object returned by Call().

ValkeyModule_Call uses a special calling convention, with a format specifier that is used to specify what kind of objects you are passing as arguments to the function.

Valkey commands are invoked just using a command name and a list of arguments. However when calling commands, the arguments may originate from different kind of strings: null-terminated C strings, ValkeyModuleString objects as received from the argv parameter in the command implementation, binary safe C buffers with a pointer and a length, and so forth.

For example if I want to call INCRBY using a first argument (the key) a string received in the argument vector argv, which is an array of ValkeyModuleString object pointers, and a C string representing the number "10" as second argument (the increment), I'll use the following function call:

ValkeyModuleCallReply *reply;
reply = ValkeyModule_Call(ctx,"INCRBY","sc",argv[1],"10");

The first argument is the context, and the second is always a null terminated C string with the command name. The third argument is the format specifier where each character corresponds to the type of the arguments that will follow. In the above case "sc" means a ValkeyModuleString object, and a null terminated C string. The other arguments are just the two arguments as specified. In fact argv[1] is a ValkeyModuleString and "10" is a null terminated C string.

This is the full list of format specifiers:

  • c -- Null terminated C string pointer.
  • b -- C buffer, two arguments needed: C string pointer and size_t length.
  • s -- ValkeyModuleString as received in argv or by other Valkey module APIs returning a ValkeyModuleString object.
  • l -- Long long integer.
  • v -- Array of ValkeyModuleString objects.
  • ! -- This modifier just tells the function to replicate the command to replicas and AOF. It is ignored from the point of view of arguments parsing.
  • A -- This modifier, when ! is given, tells to suppress AOF propagation: the command will be propagated only to replicas.
  • R -- This modifier, when ! is given, tells to suppress replicas propagation: the command will be propagated only to the AOF if enabled.

The function returns a ValkeyModuleCallReply object on success, on error NULL is returned.

NULL is returned when the command name is invalid, the format specifier uses characters that are not recognized, or when the command is called with the wrong number of arguments. In the above cases the errno var is set to EINVAL. NULL is also returned when, in an instance with Cluster enabled, the target keys are about non local hash slots. In this case errno is set to EPERM.

Working with ValkeyModuleCallReply objects.

ValkeyModuleCall returns reply objects that can be accessed using the ValkeyModule_CallReply* family of functions.

In order to obtain the type or reply (corresponding to one of the data types supported by the Valkey protocol), the function ValkeyModule_CallReplyType() is used:

reply = ValkeyModule_Call(ctx,"INCRBY","sc",argv[1],"10");
if (ValkeyModule_CallReplyType(reply) == VALKEYMODULE_REPLY_INTEGER) {
    long long myval = ValkeyModule_CallReplyInteger(reply);
    /* Do something with myval. */
}

Valid reply types are:

  • VALKEYMODULE_REPLY_STRING Bulk string or status replies.
  • VALKEYMODULE_REPLY_ERROR Errors.
  • VALKEYMODULE_REPLY_INTEGER Signed 64 bit integers.
  • VALKEYMODULE_REPLY_ARRAY Array of replies.
  • VALKEYMODULE_REPLY_NULL NULL reply.

Strings, errors and arrays have an associated length. For strings and errors the length corresponds to the length of the string. For arrays the length is the number of elements. To obtain the reply length the following function is used:

size_t reply_len = ValkeyModule_CallReplyLength(reply);

In order to obtain the value of an integer reply, the following function is used, as already shown in the example above:

long long reply_integer_val = ValkeyModule_CallReplyInteger(reply);

Called with a reply object of the wrong type, the above function always returns LLONG_MIN.

Sub elements of array replies are accessed this way:

ValkeyModuleCallReply *subreply;
subreply = ValkeyModule_CallReplyArrayElement(reply,idx);

The above function returns NULL if you try to access out of range elements.

Strings and errors (which are like strings but with a different type) can be accessed using in the following way, making sure to never write to the resulting pointer (that is returned as a const pointer so that misusing must be pretty explicit):

size_t len;
char *ptr = ValkeyModule_CallReplyStringPtr(reply,&len);

If the reply type is not a string or an error, NULL is returned.

ValkeyCallReply objects are not the same as module string objects (ValkeyModuleString types). However sometimes you may need to pass replies of type string or integer, to API functions expecting a module string.

When this is the case, you may want to evaluate if using the low level API could be a simpler way to implement your command, or you can use the following function in order to create a new string object from a call reply of type string, error or integer:

ValkeyModuleString *mystr = ValkeyModule_CreateStringFromCallReply(myreply);

If the reply is not of the right type, NULL is returned. The returned string object should be released with ValkeyModule_FreeString() as usually, or by enabling automatic memory management (see corresponding section).

Releasing call reply objects

Reply objects must be freed using ValkeyModule_FreeCallReply. For arrays, you need to free only the top level reply, not the nested replies. Currently the module implementation provides a protection in order to avoid crashing if you free a nested reply object for error, however this feature is not guaranteed to be here forever, so should not be considered part of the API.

If you use automatic memory management (explained later in this document) you don't need to free replies (but you still could if you wish to release memory ASAP).

Returning values from Valkey commands

Like normal Valkey commands, new commands implemented via modules must be able to return values to the caller. The API exports a set of functions for this goal, in order to return the usual types of the Valkey protocol, and arrays of such types as elements. Also errors can be returned with any error string and code (the error code is the initial uppercase letters in the error message, like the "BUSY" string in the "BUSY the sever is busy" error message).

All the functions to send a reply to the client are called ValkeyModule_ReplyWith<something>.

To return an error, use:

ValkeyModule_ReplyWithError(ValkeyModuleCtx *ctx, const char *err);

There is a predefined error string for key of wrong type errors:

VALKEYMODULE_ERRORMSG_WRONGTYPE

Example usage:

ValkeyModule_ReplyWithError(ctx,"ERR invalid arguments");

We already saw how to reply with a long long in the examples above:

ValkeyModule_ReplyWithLongLong(ctx,12345);

To reply with a simple string, that can't contain binary values or newlines, (so it's suitable to send small words, like "OK") we use:

ValkeyModule_ReplyWithSimpleString(ctx,"OK");

It's possible to reply with "bulk strings" that are binary safe, using two different functions:

int ValkeyModule_ReplyWithStringBuffer(ValkeyModuleCtx *ctx, const char *buf, size_t len);

int ValkeyModule_ReplyWithString(ValkeyModuleCtx *ctx, ValkeyModuleString *str);

The first function gets a C pointer and length. The second a ValkeyModuleString object. Use one or the other depending on the source type you have at hand.

In order to reply with an array, you just need to use a function to emit the array length, followed by as many calls to the above functions as the number of elements of the array are:

ValkeyModule_ReplyWithArray(ctx,2);
ValkeyModule_ReplyWithStringBuffer(ctx,"age",3);
ValkeyModule_ReplyWithLongLong(ctx,22);

To return nested arrays is easy, your nested array element just uses another call to ValkeyModule_ReplyWithArray() followed by the calls to emit the sub array elements.

Returning arrays with dynamic length

Sometimes it is not possible to know beforehand the number of items of an array. As an example, think of a Valkey module implementing a FACTOR command that given a number outputs the prime factors. Instead of factorializing the number, storing the prime factors into an array, and later produce the command reply, a better solution is to start an array reply where the length is not known, and set it later. This is accomplished with a special argument to ValkeyModule_ReplyWithArray():

ValkeyModule_ReplyWithArray(ctx, VALKEYMODULE_POSTPONED_LEN);

The above call starts an array reply so we can use other ReplyWith calls in order to produce the array items. Finally in order to set the length, use the following call:

ValkeyModule_ReplySetArrayLength(ctx, number_of_items);

In the case of the FACTOR command, this translates to some code similar to this:

ValkeyModule_ReplyWithArray(ctx, VALKEYMODULE_POSTPONED_LEN);
number_of_factors = 0;
while(still_factors) {
    ValkeyModule_ReplyWithLongLong(ctx, some_factor);
    number_of_factors++;
}
ValkeyModule_ReplySetArrayLength(ctx, number_of_factors);

Another common use case for this feature is iterating over the arrays of some collection and only returning the ones passing some kind of filtering.

It is possible to have multiple nested arrays with postponed reply. Each call to SetArray() will set the length of the latest corresponding call to ReplyWithArray():

ValkeyModule_ReplyWithArray(ctx, VALKEYMODULE_POSTPONED_LEN);
// ... generate 100 elements ...
ValkeyModule_ReplyWithArray(ctx, VALKEYMODULE_POSTPONED_LEN);
// ... generate 10 elements ...
ValkeyModule_ReplySetArrayLength(ctx, 10);
ValkeyModule_ReplySetArrayLength(ctx, 100);

This creates a 100 items array having as last element a 10 items array.

Arity and type checks

Often commands need to check that the number of arguments and type of the key is correct. In order to report a wrong arity, there is a specific function called ValkeyModule_WrongArity(). The usage is trivial:

if (argc != 2) return ValkeyModule_WrongArity(ctx);

Checking for the wrong type involves opening the key and checking the type:

ValkeyModuleKey *key = ValkeyModule_OpenKey(ctx,argv[1],
    VALKEYMODULE_READ|VALKEYMODULE_WRITE);

int keytype = ValkeyModule_KeyType(key);
if (keytype != VALKEYMODULE_KEYTYPE_STRING &&
    keytype != VALKEYMODULE_KEYTYPE_EMPTY)
{
    ValkeyModule_CloseKey(key);
    return ValkeyModule_ReplyWithError(ctx,VALKEYMODULE_ERRORMSG_WRONGTYPE);
}

Note that you often want to proceed with a command both if the key is of the expected type, or if it's empty.

Low level access to keys

Low level access to keys allow to perform operations on value objects associated to keys directly, with a speed similar to what Valkey uses internally to implement the built-in commands.

Once a key is opened, a key pointer is returned that will be used with all the other low level API calls in order to perform operations on the key or its associated value.

Because the API is meant to be very fast, it cannot do too many run-time checks, so the user must be aware of certain rules to follow:

  • Opening the same key multiple times where at least one instance is opened for writing, is undefined and may lead to crashes.
  • While a key is open, it should only be accessed via the low level key API. For example opening a key, then calling DEL on the same key using the ValkeyModule_Call() API will result into a crash. However it is safe to open a key, perform some operation with the low level API, closing it, then using other APIs to manage the same key, and later opening it again to do some more work.

In order to open a key the ValkeyModule_OpenKey function is used. It returns a key pointer, that we'll use with all the next calls to access and modify the value:

ValkeyModuleKey *key;
key = ValkeyModule_OpenKey(ctx,argv[1],VALKEYMODULE_READ);

The second argument is the key name, that must be a ValkeyModuleString object. The third argument is the mode: VALKEYMODULE_READ or VALKEYMODULE_WRITE. It is possible to use | to bitwise OR the two modes to open the key in both modes. Currently a key opened for writing can also be accessed for reading but this is to be considered an implementation detail. The right mode should be used in sane modules.

You can open non existing keys for writing, since the keys will be created when an attempt to write to the key is performed. However when opening keys just for reading, ValkeyModule_OpenKey will return NULL if the key does not exist.

Once you are done using a key, you can close it with:

ValkeyModule_CloseKey(key);

Note that if automatic memory management is enabled, you are not forced to close keys. When the module function returns, Valkey will take care to close all the keys which are still open.

Getting the key type

In order to obtain the value of a key, use the ValkeyModule_KeyType() function:

int keytype = ValkeyModule_KeyType(key);

It returns one of the following values:

VALKEYMODULE_KEYTYPE_EMPTY
VALKEYMODULE_KEYTYPE_STRING
VALKEYMODULE_KEYTYPE_LIST
VALKEYMODULE_KEYTYPE_HASH
VALKEYMODULE_KEYTYPE_SET
VALKEYMODULE_KEYTYPE_ZSET

The above are just the usual Valkey key types, with the addition of an empty type, that signals the key pointer is associated with an empty key that does not yet exists.

Creating new keys

To create a new key, open it for writing and then write to it using one of the key writing functions. Example:

ValkeyModuleKey *key;
key = ValkeyModule_OpenKey(ctx,argv[1],VALKEYMODULE_WRITE);
if (ValkeyModule_KeyType(key) == VALKEYMODULE_KEYTYPE_EMPTY) {
    ValkeyModule_StringSet(key,argv[2]);
}

Deleting keys

Just use:

ValkeyModule_DeleteKey(key);

The function returns VALKEYMODULE_ERR if the key is not open for writing. Note that after a key gets deleted, it is setup in order to be targeted by new key commands. For example ValkeyModule_KeyType() will return it is an empty key, and writing to it will create a new key, possibly of another type (depending on the API used).

Managing key expires (TTLs)

To control key expires two functions are provided, that are able to set, modify, get, and unset the time to live associated with a key.

One function is used in order to query the current expire of an open key:

mstime_t ValkeyModule_GetExpire(ValkeyModuleKey *key);

The function returns the time to live of the key in milliseconds, or VALKEYMODULE_NO_EXPIRE as a special value to signal the key has no associated expire or does not exist at all (you can differentiate the two cases checking if the key type is VALKEYMODULE_KEYTYPE_EMPTY).

In order to change the expire of a key the following function is used instead:

int ValkeyModule_SetExpire(ValkeyModuleKey *key, mstime_t expire);

When called on a non existing key, VALKEYMODULE_ERR is returned, because the function can only associate expires to existing open keys (non existing open keys are only useful in order to create new values with data type specific write operations).

Again the expire time is specified in milliseconds. If the key has currently no expire, a new expire is set. If the key already have an expire, it is replaced with the new value.

If the key has an expire, and the special value VALKEYMODULE_NO_EXPIRE is used as a new expire, the expire is removed, similarly to the Valkey PERSIST command. In case the key was already persistent, no operation is performed.

Obtaining the length of values

There is a single function in order to retrieve the length of the value associated to an open key. The returned length is value-specific, and is the string length for strings, and the number of elements for the aggregated data types (how many elements there is in a list, set, sorted set, hash).

size_t len = ValkeyModule_ValueLength(key);

If the key does not exist, 0 is returned by the function:

String type API

Setting a new string value, like the Valkey SET command does, is performed using:

int ValkeyModule_StringSet(ValkeyModuleKey *key, ValkeyModuleString *str);

The function works exactly like the Valkey SET command itself, that is, if there is a prior value (of any type) it will be deleted.

Accessing existing string values is performed using DMA (direct memory access) for speed. The API will return a pointer and a length, so that's possible to access and, if needed, modify the string directly.

size_t len, j;
char *myptr = ValkeyModule_StringDMA(key,&len,VALKEYMODULE_WRITE);
for (j = 0; j < len; j++) myptr[j] = 'A';

In the above example we write directly on the string. Note that if you want to write, you must be sure to ask for WRITE mode.

DMA pointers are only valid if no other operations are performed with the key before using the pointer, after the DMA call.

Sometimes when we want to manipulate strings directly, we need to change their size as well. For this scope, the ValkeyModule_StringTruncate function is used. Example:

ValkeyModule_StringTruncate(mykey,1024);

The function truncates, or enlarges the string as needed, padding it with zero bytes if the previous length is smaller than the new length we request. If the string does not exist since key is associated to an open empty key, a string value is created and associated to the key.

Note that every time StringTruncate() is called, we need to re-obtain the DMA pointer again, since the old may be invalid.

For a complete list of string type functions, see Key API for String type in the Modules API reference.

List type API

It's possible to push and pop values from list values:

int ValkeyModule_ListPush(ValkeyModuleKey *key, int where, ValkeyModuleString *ele);
ValkeyModuleString *ValkeyModule_ListPop(ValkeyModuleKey *key, int where);

In both the APIs the where argument specifies if to push or pop from tail or head, using the following macros:

VALKEYMODULE_LIST_HEAD
VALKEYMODULE_LIST_TAIL

Elements returned by ValkeyModule_ListPop() are like strings created with ValkeyModule_CreateString(), they must be released with ValkeyModule_FreeString() or by enabling automatic memory management.

For a complete list of set type functions, see Key API for List type in the Modules API reference.

Set type API

A direct API to set type keys is not yet implemented. Use the ValkeyModule_Call API with set commands like SADD to access keys of type set.

Sorted set type API

See the Key API for Sorted Set type section in the Modules API reference.

Hash type API

See Key API for Hash type in the Modules API reference.

Replicating commands

If you want to use module commands exactly like normal Valkey commands, in the context of replicated Valkey instances, or using the AOF file for persistence, it is important for module commands to handle their replication in a consistent way.

When using the higher level APIs to invoke commands, replication happens automatically if you use the "!" modifier in the format string of ValkeyModule_Call() as in the following example:

reply = ValkeyModule_Call(ctx,"INCRBY","!sc",argv[1],"10");

As you can see the format specifier is "!sc". The bang is not parsed as a format specifier, but it internally flags the command as "must replicate".

If you use the above programming style, there are no problems. However sometimes things are more complex than that, and you use the low level API. In this case, if there are no side effects in the command execution, and it consistently always performs the same work, what is possible to do is to replicate the command verbatim as the user executed it. To do that, you just need to call the following function:

ValkeyModule_ReplicateVerbatim(ctx);

When you use the above API, you should not use any other replication function since they are not guaranteed to mix well.

However this is not the only option. It's also possible to exactly tell Valkey what commands to replicate as the effect of the command execution, using an API similar to ValkeyModule_Call() but that instead of calling the command sends it to the AOF / replicas stream. Example:

ValkeyModule_Replicate(ctx,"INCRBY","cl","foo",my_increment);

It's possible to call ValkeyModule_Replicate multiple times, and each will emit a command. All the sequence emitted is wrapped between a MULTI/EXEC transaction, so that the AOF and replication effects are the same as executing a single command.

Note that Call() replication and Replicate() replication have a rule, in case you want to mix both forms of replication (not necessarily a good idea if there are simpler approaches). Commands replicated with Call() are always the first emitted in the final MULTI/EXEC block, while all the commands emitted with Replicate() will follow.

Automatic memory management

Normally when writing programs in the C language, programmers need to manage memory manually. This is why the Valkey modules API has functions to release strings, close open keys, free replies, and so forth.

However given that commands are executed in a contained environment and with a set of strict APIs, Valkey is able to provide automatic memory management to modules, at the cost of some performance (most of the time, a very low cost).

When automatic memory management is enabled:

  1. You don't need to close open keys.
  2. You don't need to free replies.
  3. You don't need to free ValkeyModuleString objects.

However you can still do it, if you want. For example, automatic memory management may be active, but inside a loop allocating a lot of strings, you may still want to free strings no longer used.

In order to enable automatic memory management, just call the following function at the start of the command implementation:

ValkeyModule_AutoMemory(ctx);

Automatic memory management is usually the way to go, however experienced C programmers may not use it in order to gain some speed and memory usage benefit.

Allocating memory into modules

Normal C programs use malloc() and free() in order to allocate and release memory dynamically. While in Valkey modules the use of malloc is not technically forbidden, it is a lot better to use the Valkey Modules specific functions, that are exact replacements for malloc, free, realloc and strdup. These functions are:

void *ValkeyModule_Alloc(size_t bytes);
void* ValkeyModule_Realloc(void *ptr, size_t bytes);
void ValkeyModule_Free(void *ptr);
void ValkeyModule_Calloc(size_t nmemb, size_t size);
char *ValkeyModule_Strdup(const char *str);

They work exactly like their libc equivalent calls, however they use the same allocator Valkey uses, and the memory allocated using these functions is reported by the INFO command in the memory section, is accounted when enforcing the maxmemory policy, and in general is a first citizen of the Valkey executable. On the contrary, the method allocated inside modules with libc malloc() is transparent to Valkey.

Another reason to use the modules functions in order to allocate memory is that, when creating native data types inside modules, the RDB loading functions can return deserialized strings (from the RDB file) directly as ValkeyModule_Alloc() allocations, so they can be used directly to populate data structures after loading, instead of having to copy them to the data structure.

Pool allocator

Sometimes in commands implementations, it is required to perform many small allocations that will be not retained at the end of the command execution, but are just functional to execute the command itself.

This work can be more easily accomplished using the Valkey pool allocator:

void *ValkeyModule_PoolAlloc(ValkeyModuleCtx *ctx, size_t bytes);

It works similarly to malloc(), and returns memory aligned to the next power of two of greater or equal to bytes (for a maximum alignment of 8 bytes). However it allocates memory in blocks, so it the overhead of the allocations is small, and more important, the memory allocated is automatically released when the command returns.

So in general short living allocations are a good candidates for the pool allocator.

Writing commands compatible with Valkey Cluster

See the Modules API reference for the following commands:

© Valkey contributors, 2024.

Valkey and the Valkey logo are trademarks of LF Projects, LLC.

Trademark Policy. Privacy Policy. Terms of Use. For other policies, please see lfprojects.org.

Valkey includes certain 3-Clause BSD-licensed Redis code from Redis Ltd. and other source code. Redis Ltd. is not the source of that other source code. Redis is a registered trademark of Redis Ltd.

Logo design by Ziyang.