AsyncLocalStorage for iterators and generators

[himself65]

Version

18, 20, 22

Platform

macos

Subsystem

async_hooks

What steps will reproduce the bug?

import { AsyncLocalStorage } from 'node:async_hooks'

const als = new AsyncLocalStorage()

const iter = als.run(10, function * () {
  yield 1 + (als.getStore() ?? 0)
  yield 2 + (als.getStore() ?? 0)
})

for await (const value of iter) {
  console.log(value)
}

const changed = als.run(10, function () {
  const iter = AsyncLocalStorage.bind(function * () {
    yield 1 + (als.getStore() ?? 0)
    yield 2 + (als.getStore() ?? 0)
  })()
  iter[Symbol.iterator] = AsyncLocalStorage.bind(iter[Symbol.iterator])
  return iter;
})

for (const value of changed) {
  console.log(value)
}

How often does it reproduce? Is there a required condition?

No response

What is the expected behavior? Why is that the expected behavior?

11
12
11
12

What do you see instead?

1
2
1
2

Additional information

No response

[himself65]

workaround:

const changed = als.run(10, function () {
  const iter = function * () {
    yield 1 + (als.getStore() ?? 0)
    yield 2 + (als.getStore() ?? 0)
  }
  const sp = AsyncLocalStorage.snapshot();
  return function * () {
    const f = iter()
    while (true) {
      const { value, done } = sp(() => f.next())
      if (done) {
        break
      }
      yield value
    }
  }()
})

[dead-claudia]

This feels like a dupe of #42237.