74 lines
1.8 KiB
Nix
74 lines
1.8 KiB
Nix
{
|
|
pkgs,
|
|
}:
|
|
let
|
|
inherit (pkgs) lib;
|
|
in
|
|
{
|
|
name,
|
|
default_config ? true,
|
|
extra_config ? '''',
|
|
package ? pkgs.valkey,
|
|
port ? 6379,
|
|
bind ? "localhost",
|
|
...
|
|
}:
|
|
|
|
let
|
|
|
|
generateValkeyConfig = ''
|
|
cat > "$RUNTIME_PATH"/${name}.conf <<EOF
|
|
${lib.optionalString default_config ''
|
|
port ${builtins.toString port}
|
|
${lib.optionalString (bind != null) "bind ${bind}"}
|
|
${lib.optionalString (port == 0) "unixsocket \"$RUNTIME_PATH/${name}.sock\""}
|
|
${lib.optionalString (port == 0) "unixsocketperm 700"}
|
|
''}
|
|
${extra_config}
|
|
EOF
|
|
'';
|
|
|
|
script = pkgs.writeShellApplication {
|
|
name = "run-valkey";
|
|
text = ''
|
|
set -euo pipefail
|
|
VALKEYDATA="$RUNTIME_PATH"/${name}/
|
|
${
|
|
if (port == 0) then
|
|
''
|
|
VALKEY_UNIX_SOCKET = "$RUNTIME_PATH"/${name}.sock
|
|
exportVALKEY_UNIX_SOCKET
|
|
echo "Starting valkey on unix socket ''${VALKEY_UNIX_SOCKET}"
|
|
''
|
|
else
|
|
''
|
|
echo "Starting valkey on port ${builtins.toString port}"
|
|
''
|
|
}
|
|
export VALKEYDATA
|
|
if [ ! -d "$VALKEYDATA" ]; then
|
|
mkdir -p "$VALKEYDATA"
|
|
fi
|
|
${generateValkeyConfig}
|
|
${package}/bin/valkey-server "$RUNTIME_PATH"/${name}.conf --daemonize no --dir "$VALKEYDATA"
|
|
'';
|
|
};
|
|
tcpPing = "${package}/bin/valkey-cli -p ${builtins.toString port} ping";
|
|
unixPing = "${package}/bin/valkey-cli -s \"$RUNTIME_PATH\"/${name}.sock ping";
|
|
in
|
|
{
|
|
processes."${name}" = {
|
|
command = "${script}/bin/run-valkey";
|
|
readiness_probe = {
|
|
exec.command = if port == 0 then unixPing else tcpPing;
|
|
initial_delay_seconds = 2;
|
|
period_seconds = 10;
|
|
timeout_seconds = 4;
|
|
success_threshold = 1;
|
|
failure_threshold = 3;
|
|
};
|
|
availability.restart = "on_failure";
|
|
};
|
|
|
|
}
|