foster/denoland-deno
1
0
Fork 0
mirror of https://github.com/denoland/deno.git synced 2024-12-22 07:14:47 -05:00
Code Issues Packages 1 Wiki Activity

fix(ext/kv): same expireIn should generate same expireAt (#20396)

Browse source 
Co-authored-by: Luca Casonato <hello@lcas.dev>
This commit is contained in:
Heyang Zhou 2023-09-08 23:20:28 +08:00 committed by GitHub
parent 17276a1df9
commit 375d8a5bd5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 18 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
ext/kv/01_db.ts
View file

@ -134,11 +134,8 @@ class Kv {
value = serializeValue(value);
const checks: Deno.AtomicCheck[] = [];
const expireAt = typeof options?.expireIn === "number"
? Date.now() + options.expireIn
: undefined;
const mutations = [
[key, "set", value, expireAt],
[key, "set", value, options?.expireIn],
];
const versionstamp = await core.opAsync(
@ -340,7 +337,7 @@ class AtomicOperation {
const key = mutation.key;
let type: string;
let value: RawValue | null;
let expireAt: number | undefined = undefined;
let expireIn: number | undefined = undefined;
switch (mutation.type) {
case "delete":
type = "delete";
@ -350,7 +347,7 @@ class AtomicOperation {
break;
case "set":
if (typeof mutation.expireIn === "number") {
expireAt = Date.now() + mutation.expireIn;
expireIn = mutation.expireIn;
}
/* falls through */
case "sum":
@ -365,7 +362,7 @@ class AtomicOperation {
default:
throw new TypeError("Invalid mutation type");
}
this.#mutations.push([key, type, value, expireAt]);
this.#mutations.push([key, type, value, expireIn]);
}
return this;
}
@ -390,10 +387,12 @@ class AtomicOperation {
value: unknown,
options?: { expireIn?: number },
): this {
const expireAt = typeof options?.expireIn === "number"
? Date.now() + options.expireIn
: undefined;
this.#mutations.push([key, "set", serializeValue(value), expireAt]);
this.#mutations.push([
key,
"set",
serializeValue(value),
options?.expireIn,
]);
return this;
}

12
ext/kv/lib.rs
View file

@ -12,6 +12,7 @@ use std::cell::RefCell;
use std::num::NonZeroU32;
use std::rc::Rc;
use chrono::Utc;
use codec::decode_key;
use codec::encode_key;
use deno_core::anyhow::Context;
@ -383,9 +384,11 @@ impl TryFrom<V8KvCheck> for KvCheck {
type V8KvMutation = (KvKey, String, Option<FromV8Value>, Option<u64>);
impl TryFrom<V8KvMutation> for KvMutation {
impl TryFrom<(V8KvMutation, u64)> for KvMutation {
type Error = AnyError;
fn try_from(value: V8KvMutation) -> Result<Self, AnyError> {
fn try_from(
(value, current_timstamp): (V8KvMutation, u64),
) -> Result<Self, AnyError> {
let key = encode_v8_key(value.0)?;
let kind = match (value.1.as_str(), value.2) {
("set", Some(value)) => MutationKind::Set(value.try_into()?),
@ -405,7 +408,7 @@ impl TryFrom<V8KvMutation> for KvMutation {
Ok(KvMutation {
key,
kind,
expire_at: value.3,
expire_at: value.3.map(|expire_in| current_timstamp + expire_in),
})
}
}
@ -606,6 +609,7 @@ async fn op_kv_atomic_write<DBH>(
where
DBH: DatabaseHandler + 'static,
{
let current_timestamp = Utc::now().timestamp_millis() as u64;
let db = {
let state = state.borrow();
let resource =
@ -631,7 +635,7 @@ where
.with_context(|| "invalid check")?;
let mutations = mutations
.into_iter()
.map(TryInto::try_into)
.map(|mutation| TryFrom::try_from((mutation, current_timestamp)))
.collect::<Result<Vec<KvMutation>, AnyError>>()
.with_context(|| "invalid mutation")?;
let enqueues = enqueues