UNPKG

graphql-yoga

Version:

<div align="center"><img src="/website/public/cover.png" width="720" /></div>

66 lines (65 loc) 2.92 kB
import { isAsyncIterable } from '@envelop/core'; import { handleMaybePromise } from '@whatwg-node/promise-helpers'; import { fakePromise } from '@whatwg-node/server'; import { getResponseInitByRespectingErrors } from '../../error.js'; import { jsonStringifyResultWithoutInternals } from './stringify.js'; export function processMultipartResult(result, fetchAPI) { const headersInit = { Connection: 'keep-alive', 'Content-Type': 'multipart/mixed; boundary="-"', 'Transfer-Encoding': 'chunked', }; const responseInit = getResponseInitByRespectingErrors(result, headersInit); let iterator; const textEncoder = new fetchAPI.TextEncoder(); const readableStream = new fetchAPI.ReadableStream({ start(controller) { if (isAsyncIterable(result)) { iterator = result[Symbol.asyncIterator](); } else { let finished = false; iterator = { next: () => { if (finished) { return fakePromise({ done: true, value: null }); } finished = true; return fakePromise({ done: false, value: result }); }, }; } controller.enqueue(textEncoder.encode('\r\n')); controller.enqueue(textEncoder.encode(`---`)); }, pull(controller) { return handleMaybePromise(() => iterator.next(), ({ done, value }) => { if (value != null) { controller.enqueue(textEncoder.encode('\r\n')); controller.enqueue(textEncoder.encode('Content-Type: application/json; charset=utf-8')); controller.enqueue(textEncoder.encode('\r\n')); const chunk = jsonStringifyResultWithoutInternals(value); const encodedChunk = textEncoder.encode(chunk); controller.enqueue(textEncoder.encode('Content-Length: ' + encodedChunk.byteLength)); controller.enqueue(textEncoder.encode('\r\n')); controller.enqueue(textEncoder.encode('\r\n')); controller.enqueue(encodedChunk); controller.enqueue(textEncoder.encode('\r\n')); controller.enqueue(textEncoder.encode('---')); } if (done) { controller.enqueue(textEncoder.encode('--\r\n')); controller.close(); } }, err => { controller.error(err); }); }, cancel(e) { if (iterator.return) { return handleMaybePromise(() => iterator.return?.(e), () => { }); } }, }); return new fetchAPI.Response(readableStream, responseInit); }