Function asyncIteratorWithSpan

packages/shared/src/iterator.ts:175–210

(
  { name, ...options }: AsyncIteratorWithSpanOptions,
  iterator: AsyncIterator<T, TReturn, TNext>,
)

}

export function asyncIteratorWithSpan<T, TReturn, TNext>(
  { name, ...options }: AsyncIteratorWithSpanOptions,
  iterator: AsyncIterator<T, TReturn, TNext>,
): AsyncIteratorClass<T, TReturn, TNext> {
  let span: ReturnType<typeof startSpan> | undefined

  return new AsyncIteratorClass(
    async () => {
      span ??= startSpan(name)

      try {
        const result = await runInSpanContext(span, () => iterator.next())
        span?.addEvent(result.done ? 'completed' : 'yielded')
        return result
      }
      catch (err) {
        setSpanError(span, err, options)
        throw err
      }
    },
    async (reason) => {
      try {
        if (reason !== 'next') {
          await runInSpanContext(span, () => iterator.return?.())
        }
      }
      catch (err) {
        setSpanError(span, err, options)
        throw err
      }
      finally {
        span?.end()
      }
    },
  )
}