use std::time::Duration;
use valkey_module::alloc::ValkeyAlloc;
use valkey_module::{valkey_module, Context, NextArg, ValkeyError, ValkeyResult, ValkeyString};
fn expire_cmd(ctx: &Context, args: Vec<ValkeyString>) -> ValkeyResult {
if args.len() < 3 {
return Err(ValkeyError::WrongArity);
}
let mut args = args.into_iter().skip(1);
let key_name = args.next_arg()?;
let ttl_sec = args.next_i64()?;
let key = ctx.open_key_writable(&key_name);
if ttl_sec >= 0 {
key.set_expire(Duration::new(ttl_sec as u64, 0))
} else {
key.remove_expire()
}
}
valkey_module! {
name: "expire",
version: 1,
allocator: (ValkeyAlloc, ValkeyAlloc),
data_types: [],
commands: [
["expire.cmd", expire_cmd, "write fast deny-oom", 1, 1, 1],
],
}