UNPKG

graphql-yoga

Version:

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

58 lines (57 loc) 2.46 kB
import { isAsyncIterable } from '@envelop/core'; import { getResponseInitByRespectingErrors } from '../../error.js'; import { jsonStringifyResult } 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 Promise.resolve({ done: true, value: null }); } finished = true; return Promise.resolve({ done: false, value: result }); }, }; } controller.enqueue(textEncoder.encode(`---`)); }, async pull(controller) { const { done, value } = await iterator.next(); 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 = jsonStringifyResult(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(); } }, async cancel(e) { await iterator.return?.(e); }, }); return new fetchAPI.Response(readableStream, responseInit); }