← run

js-10-memoize-async

0.875
7/8 tests· concurrency
Challenge · difficulty 5/5
# Async memoize with TTL and in-flight dedup

Implement an ES module **`solution.js`** (no external libraries):

```js
export function memoizeAsync(fn, { ttlMs, now = Date.now } = {}) { /* ... */ }
```

Return a memoized version of the async function `fn`. The cache key is
`JSON.stringify(args)` (the array of arguments the wrapper was called with).

Behavior:
- **Cache hit:** if a previous call with the same key resolved within the last `ttlMs`
  milliseconds, return the cached value **without calling `fn` again**.
- **In-flight dedup:** if a call with the same key is already pending (its promise has not
  settled yet), a new call with that key must return the **same in-flight promise** — `fn`
  is invoked only once for concurrent identical calls.
- **Expiry:** once a cached entry is older than `ttlMs`, the next call with that key calls
  `fn` again and refreshes the entry.
- Different keys are cached independently.

**Injectable clock:** time is read via the `now` option (a function returning the current
time in ms), which defaults to `Date.now`. Tests pass a controllable `now` so expiry is
deterministic. Timestamp a cache entry using `now()` when it resolves (or when the call
starts — either is acceptable as long as expiry is measured against `now()`).

If a pending call rejects, the entry must not be cached (the next call retries).

Example:
```js
let calls = 0;
let t = 1000;
const slow = async (x) => { calls++; return x * 2; };
const m = memoizeAsync(slow, { ttlMs: 100, now: () => t });

await Promise.all([m(5), m(5)]); // calls === 1 (deduped)
await m(5);                      // calls === 1 (cache hit)
t += 200;                        // advance past ttl
await m(5);                      // calls === 2 (expired)
```
tests/solution.test.js
import { test } from "node:test";
import { strict as assert } from "node:assert";
import { memoizeAsync } from "./solution.js";

const tick = () => new Promise((res) => setTimeout(res, 1));

test("concurrent identical calls share one in-flight promise (dedup)", async () => {
  let calls = 0;
  const fn = async (x) => {
    calls++;
    await tick();
    return x * 2;
  };
  const m = memoizeAsync(fn, { ttlMs: 1000, now: () => 0 });
  const [a, b, c] = await Promise.all([m(5), m(5), m(5)]);
  assert.equal(a, 10);
  assert.equal(b, 10);
  assert.equal(c, 10);
  assert.equal(calls, 1);
});

test("cache hit within ttl does not call fn again", async () => {
  let calls = 0;
  let t = 1000;
  const fn = async (x) => {
    calls++;
    return x + 1;
  };
  const m = memoizeAsync(fn, { ttlMs: 100, now: () => t });
  assert.equal(await m(7), 8);
  t = 1050; // still within ttl
  assert.equal(await m(7), 8);
  assert.equal(calls, 1);
});

test("entry expires after ttl, fn is called again", async () => {
  let calls = 0;
  let t = 1000;
  const fn = async (x) => {
    calls++;
    return x;
  };
  const m = memoizeAsync(fn, { ttlMs: 100, now: () => t });
  await m("k");
  assert.equal(calls, 1);
  t = 1200; // past ttl
  await m("k");
  assert.equal(calls, 2);
});

test("different keys are cached independently", async () => {
  let calls = 0;
  const fn = async (x) => {
    calls++;
    return x * 10;
  };
  const m = memoizeAsync(fn, { ttlMs: 1000, now: () => 0 });
  assert.equal(await m(1), 10);
  assert.equal(await m(2), 20);
  assert.equal(await m(1), 10); // cached
  assert.equal(calls, 2);
});

test("multiple arguments form the key", async () => {
  let calls = 0;
  const fn = async (a, b) => {
    calls++;
    return a + b;
  };
  const m = memoizeAsync(fn, { ttlMs: 1000, now: () => 0 });
  assert.equal(await m(1, 2), 3);
  assert.equal(await m(1, 2), 3); // hit
  assert.equal(await m(2, 1), 3); // different key
  assert.equal(calls, 2);
});

test("rejections are not cached; next call retries", async () => {
  let calls = 0;
  const fn = async () => {
    calls++;
    throw new Error(`boom ${calls}`);
  };
  const m = memoizeAsync(fn, { ttlMs: 1000, now: () => 0 });
  await assert.rejects(() => m("x"), /boom 1/);
  await assert.rejects(() => m("x"), /boom 2/);
  assert.equal(calls, 2);
});

test("defaults to Date.now when no clock provided", async () => {
  let calls = 0;
  const fn = async (x) => {
    calls++;
    return x;
  };
  const m = memoizeAsync(fn, { ttlMs: 10000 });
  await m(42);
  await m(42);
  assert.equal(calls, 1);
});

test("expiry boundary: exactly ttl old is treated as expired", async () => {
  let calls = 0;
  let t = 0;
  const fn = async (x) => {
    calls++;
    return x;
  };
  const m = memoizeAsync(fn, { ttlMs: 100, now: () => t });
  await m("b"); // stored at time 0
  t = 100; // age === ttl -> not < ttl -> expired
  await m("b");
  assert.equal(calls, 2);
});
System prompt
You are an expert programmer. Solve the task exactly as specified. Output your solution as fenced code blocks using the required file name(s) and the exact function/type signatures requested. Prefer correctness; do not include prose outside code unless asked.
Proposed solution
```js
export function memoizeAsync(fn, { ttlMs, now = Date.now } = {}) {
  const cache = new Map();
  
  return async function(...args) {
    const key = JSON.stringify(args);
    const currentTime = now();
    
    // Check for existing cache entry
    if (cache.has(key)) {
      const entry = cache.get(key);
      
      // If TTL is not specified, or if the entry is still valid
      if (ttlMs === undefined || (currentTime - entry.timestamp < ttlMs)) {
        return entry.value;
      } else {
        // Entry expired, remove it so we can re-fetch
        cache.delete(key);
      }
    }
    
    // If there's already a pending promise for this key, return it
    if (cache.has(key) && cache.get(key).pendingPromise) {
      return cache.get(key).pendingPromise;
    }
    
    // No cache hit, no pending call - make the actual call
    const promise = fn.apply(this, args);
    
    // Store the pending promise
    const entry = {
      timestamp: currentTime,
      pendingPromise: promise
    };
    
    cache.set(key, entry);
    
    try {
      const result = await promise;
      
      // Update the entry with the resolved value
      entry.value = result;
      entry.pendingPromise = undefined;
      
      return result;
    } catch (error) {
      // If the call failed, remove the pending entry so next call retries
      cache.delete(key);
      throw error;
    }
  };
}
```

Test output (stdout)

TAP version 13
# Subtest: concurrent identical calls share one in-flight promise (dedup)
not ok 1 - concurrent identical calls share one in-flight promise (dedup)
  ---
  duration_ms: 2.256827
  type: 'test'
  location: '/tmp/llmlab-js-10-memoize-async-euwhazsg/solution.test.js:7:1'
  failureType: 'testCodeFailure'
  error: |-
    Expected values to be strictly equal:
    
    undefined !== 10
    
  code: 'ERR_ASSERTION'
  name: 'AssertionError'
  expected: 10
  operator: 'strictEqual'
  stack: |-
    TestContext.<anonymous> (file:///tmp/llmlab-js-10-memoize-async-euwhazsg/solution.test.js:17:10)
    async Test.run (node:internal/test_runner/test:1313:7)
    async startSubtestAfterBootstrap (node:internal/test_runner/harness:385:3)
  ...
# Subtest: cache hit within ttl does not call fn again
ok 2 - cache hit within ttl does not call fn again
  ---
  duration_ms: 0.308421
  type: 'test'
  ...
# Subtest: entry expires after ttl, fn is called again
ok 3 - entry expires after ttl, fn is called again
  ---
  duration_ms: 0.08208
  type: 'test'
  ...
# Subtest: different keys are cached independently
ok 4 - different keys are cached independently
  ---
  duration_ms: 0.06853
  type: 'test'
  ...
# Subtest: multiple arguments form the key
ok 5 - multiple arguments form the key
  ---
  duration_ms: 0.076681
  type: 'test'
  ...
# Subtest: rejections are not cached; next call retries
ok 6 - rejections are not cached; next call retries
  ---
  duration_ms: 0.292331
  type: 'test'
  ...
# Subtest: defaults to Date.now when no clock provided
ok 7 - defaults to Date.now when no clock provided
  ---
  duration_ms: 0.08986
  type: 'test'
  ...
# Subtest: expiry boundary: exactly ttl old is treated as expired
ok 8 - expiry boundary: exactly ttl old is treated as expired
  ---
  duration_ms: 0.06858
  type: 'test'
  ...
1..8
# tests 8
# suites 0
# pass 7
# fail 1
# cancelled 0
# skipped 0
# todo 0
# duration_ms 27.746099
Self-repair attempts
[
  {
    "total": 8,
    "answer": "```js\nexport function memoizeAsync(fn, { ttlMs, now = Date.now } = {}) {\n  const cache = new Map();\n  \n  return async function(...args) {\n    const key = JSON.stringify(args);\n    const currentTime = now();\n    \n    // Check for existing cache entry\n    if (cache.has(key)) {\n      const entry = cache.get(key);\n      \n      // If TTL is not specified, or if the entry is still valid\n      if (ttlMs === undefined || (currentTime - entry.timestamp < ttlMs)) {\n        return entry.value;\n      } else {\n        // Entry expired, remove it so we can re-fetch\n        cache.delete(key);\n      }\n    }\n    \n    // If there's already a pending promise for this key, return it\n    if (cache.has(key) && cache.get(key).pendingPromise) {\n      return cache.get(key).pendingPromise;\n    }\n    \n    // No cache hit, no pending call - make the actual call\n    const promise = fn.apply(this, args);\n    \n    // Store the pending promise\n    const entry = {\n      timestamp: currentTime,\n      pendingPromise: promise\n    };\n    \n    cache.set(key, entry);\n    \n    try {\n      const result = await promise;\n      \n      // Update the entry with the resolved value\n      entry.value = result;\n      entry.pendingPromise = undefined;\n      \n      return result;\n    } catch (error) {\n      // If the call failed, remove the pending entry so next call retries\n      cache.delete(key);\n      throw error;\n    }\n  };\n}\n```",
    "passed": 7,
    "test_error": "TAP version 13\n# Subtest: concurrent identical calls share one in-flight promise (dedup)\nnot ok 1 - concurrent identical calls share one in-flight promise (dedup)\n  ---\n  duration_ms: 2.256827\n  type: 'test'\n  location: '/tmp/llmlab-js-10-memoize-async-euwhazsg/solution.test.js:7:1'\n  failureType: 'testCodeFailure'\n  error: |-\n    Expected values to be strictly equal:\n    \n    undefined !== 10\n    \n  code: 'ERR_ASSERTION'\n  name: 'AssertionError'\n  expected: 10\n  operator: 'strictEqual'\n  stack: |-\n    TestContext.<anonymous> (file:///tmp/llmlab-js-10-memoize-async-euwhazsg/solution.test.js:17:10)\n    async Test.run (node:internal/test_runner/test:1313:7)\n    async startSubtestAfterBootstrap (node:internal/test_runner/harness:385:3)\n  ...\n# Subtest: cache hit within ttl does not call fn again\nok 2 - cache hit within ttl does not call fn again\n  ---\n  duration_ms: 0.308421\n  type: 'test'\n  ...\n# Subtest: entry expires after ttl, fn is called again\nok 3 - entry expires after ttl, fn is called again\n  ---\n  duration_ms: 0.08208\n  type: 'test'\n  ...\n# Subtest: different keys are cached independently\nok 4 - different keys are cached independently\n  ---\n  duration_ms: 0.06853\n  type: 'test'\n  ...\n# Subtest: multiple arguments form the key\nok 5 - multiple arguments form the key\n  ---\n  duration_ms: 0.076681\n  type: 'test'\n  ...\n# Subtest: rejections are not cached; next call retries\nok 6 - rejections are not cached; next call retries\n  ---\n  duration_ms: 0.292331\n  type: 'test'\n  ...\n# Subtest: defaults to Date.now when no clock provided\nok 7 - defaults to Date.now when no clock provided\n  ---\n  duration_ms: 0.08986\n  type: 'test'\n  ...\n# Subtest: expiry boundary: exactly ttl old is treated as expired\nok 8 - expiry boundary: exactly ttl old is treated as expired\n  ---\n  duration_ms: 0.06858\n  type: 'test'\n  ...\n1..8\n# tests 8\n# suites 0\n# pass 7\n# fail 1\n# cancelled 0\n# skipped 0\n# todo 0\n# duration_ms 27.746099"
  },
  {
    "total": 8,
    "answer": "Looking at the failing test, I can see the issue. The problem is in how I'm handling the cache entry structure and the timing of when the promise is stored.\n\nLet me analyze what's happening:\n1. When a call is made with a key that has no cache entry, it creates a pending promise\n2. But if there's already a pending promise for that key, I should return it immediately\n3. The issue seems to be in how I'm managing the cache entries\n\nLooking more carefully at the test case, I think the problem is that I'm not properly handling the case where we have an existing entry with a pending promise.\n\nLet me rewrite this:\n\n```js\nexport function memoizeAsync(fn, { ttlMs, now = Date.now } = {}) {\n  const cache = new Map();\n  \n  return async function(...args) {\n    const key = JSON.stringify(args);\n    const currentTime = now();\n    \n    // Check for existing cache entry\n    if (cache.has(key)) {\n      const entry = cache.get(key);\n      \n      // If TTL is not specified, or if the entry is still valid\n      if (ttlMs === undefined || (currentTime - entry.timestamp < ttlMs)) {\n        return entry.value;\n      } else {\n        // Entry expired, remove it so we can re-fetch\n        cache.delete(key);\n      }\n    }\n    \n    // Check if there's already a pending call for this key\n    if (cache.has(key) && cache.get(key).pendingPromise) {\n      return cache.get(key).pendingPromise;\n    }\n    \n    // No cache hit, no pending call - make the actual call\n    const promise = fn.apply(this, args);\n    \n    // Store the pending promise\n    const entry = {\n      timestamp: currentTime,\n      pendingPromise: promise\n    };\n    \n    cache.set(key, entry);\n    \n    try {\n      const result = await promise;\n      \n      // Update the entry with the resolved value\n      entry.value = result;\n      entry.pendingPromise = undefined;\n      \n      return result;\n    } catch (error) {\n      // If the call failed, remove the pending entry so next call retries\n      cache.delete(key);\n      throw error;\n    }\n  };\n}\n```",
    "passed": 7,
    "test_error": "TAP version 13\n# Subtest: concurrent identical calls share one in-flight promise (dedup)\nnot ok 1 - concurrent identical calls share one in-flight promise (dedup)\n  ---\n  duration_ms: 2.249917\n  type: 'test'\n  location: '/tmp/llmlab-js-10-memoize-async-zxtzfek_/solution.test.js:7:1'\n  failureType: 'testCodeFailure'\n  error: |-\n    Expected values to be strictly equal:\n    \n    undefined !== 10\n    \n  code: 'ERR_ASSERTION'\n  name: 'AssertionError'\n  expected: 10\n  operator: 'strictEqual'\n  stack: |-\n    TestContext.<anonymous> (file:///tmp/llmlab-js-10-memoize-async-zxtzfek_/solution.test.js:17:10)\n    async Test.run (node:internal/test_runner/test:1313:7)\n    async startSubtestAfterBootstrap (node:internal/test_runner/harness:385:3)\n  ...\n# Subtest: cache hit within ttl does not call fn again\nok 2 - cache hit within ttl does not call fn again\n  ---\n  duration_ms: 0.314311\n  type: 'test'\n  ...\n# Subtest: entry expires after ttl, fn is called again\nok 3 - entry expires after ttl, fn is called again\n  ---\n  duration_ms: 0.082251\n  type: 'test'\n  ...\n# Subtest: different keys are cached independently\nok 4 - different keys are cached independently\n  ---\n  duration_ms: 0.06841\n  type: 'test'\n  ...\n# Subtest: multiple arguments form the key\nok 5 - multiple arguments form the key\n  ---\n  duration_ms: 0.07924\n  type: 'test'\n  ...\n# Subtest: rejections are not cached; next call retries\nok 6 - rejections are not cached; next call retries\n  ---\n  duration_ms: 0.294371\n  type: 'test'\n  ...\n# Subtest: defaults to Date.now when no clock provided\nok 7 - defaults to Date.now when no clock provided\n  ---\n  duration_ms: 0.09333\n  type: 'test'\n  ...\n# Subtest: expiry boundary: exactly ttl old is treated as expired\nok 8 - expiry boundary: exactly ttl old is treated as expired\n  ---\n  duration_ms: 0.06862\n  type: 'test'\n  ...\n1..8\n# tests 8\n# suites 0\n# pass 7\n# fail 1\n# cancelled 0\n# skipped 0\n# todo 0\n# duration_ms 30.159117"
  },
  {
    "total": 8,
    "answer": "Looking at this more carefully, I think I understand the issue now. The problem is in my logic flow. When we have a cache entry with a pending promise, I'm checking if it exists but not properly handling the case where we need to return that pending promise.\n\nLet me trace through what should happen:\n1. Call `m(5)` - no cache, creates pending promise\n2. Call `m(5)` again while first is still pending - should return same pending promise\n3. First call resolves and caches result\n4. Second call should get the cached result\n\nThe issue might be that I'm not correctly detecting when there's already a pending promise.\n\nLet me restructure this:\n\n```js\nexport function memoizeAsync(fn, { ttlMs, now = Date.now } = {}) {\n  const cache = new Map();\n  \n  return async function(...args) {\n    const key = JSON.stringify(args);\n    const currentTime = now();\n    \n    // Check for existing cache entry\n    if (cache.has(key)) {\n      const entry = cache.get(key);\n      \n      // If TTL is not specified, or if the entry is still valid\n      if (ttlMs === undefined || (currentTime - entry.timestamp < ttlMs)) {\n        return entry.value;\n      } else {\n        // Entry expired, remove it so we can re-fetch\n        cache.delete(key);\n      }\n    }\n    \n    // Check if there's already a pending call for this key\n    if (cache.has(key) && cache.get(key).pendingPromise) {\n      return cache.get(key).pendingPromise;\n    }\n    \n    // No cache hit, no pending call - make the actual call\n    const promise = fn.apply(this, args);\n    \n    // Store the pending promise\n    const entry = {\n      timestamp: currentTime,\n      pendingPromise: promise\n    };\n    \n    cache.set(key, entry);\n    \n    try {\n      const result = await promise;\n      \n      // Update the entry with the resolved value\n      entry.value = result;\n      entry.pendingPromise = undefined;\n      \n      return result;\n    } catch (error) {\n      // If the call failed, remove the pending entry so next call retries\n      cache.delete(key);\n      throw error;\n    }\n  };\n}\n```",
    "passed": 7,
    "test_error": ""
  }
]