fix(koa): restore writeHead once we're done

graphile · Jun 22, 2018 · 50be982 · 50be982
1 parent e3bf765
commit 50be982
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/postgraphile/http/createPostGraphileHttpRequestHandler.js b/src/postgraphile/http/createPostGraphileHttpRequestHandler.js
@@ -767,12 +767,17 @@ export default function createPostGraphileHttpRequestHandler(options) {
 
       // Hack the req object so we can get back to ctx
       ctx.req._koaCtx = ctx
+
+      const oldWriteHead = ctx.res.writeHead
       ctx.res.writeHead = () => {
         throw new Error('res.writeHead not supported in Koa environment')
       }
+
+      const oldSend = ctx.res.send
       ctx.res.send = () => {
         throw new Error('res.send not supported in Koa environment')
       }
+
       const oldEnd = ctx.res.end
       ctx.res.end = (body) => {
         ctx.response.body = body
@@ -786,6 +791,8 @@ export default function createPostGraphileHttpRequestHandler(options) {
         try {
           result = await requestHandler(ctx.req, ctx.res, next)
         } finally {
+          ctx.res.writeHead = oldWriteHead
+          ctx.res.send = oldSend
           ctx.res.end = oldEnd
           if (ctx.res.statusCode && ctx.res.statusCode !== 200) {
             ctx.response.status = ctx.res.statusCode