feat: support both enable/disable attribute for multi value headers o…

…n ALB (CodeGenieApp#392)
suzukalight · Oct 13, 2021 · a5cb5b5 · a5cb5b5
1 parent 37539fb
commit a5cb5b5
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/src/event-sources/aws/alb.js b/src/event-sources/aws/alb.js
@@ -38,16 +38,24 @@ const getRequestValuesFromAlbEvent = ({ event }) => {
 }
 
 const getResponseToAlb = ({
+ event,
  statusCode,
  body,
- headers,
+ headers: responseHeaders,
  isBase64Encoded
 }) => {
- const multiValueHeaders = getMultiValueHeaders({ headers })
+ const multiValueHeaders = !event.headers ? getMultiValueHeaders({ headers: responseHeaders }) : undefined
+ const headers = event.headers
+ ? Object.entries(responseHeaders).reduce((acc, [k, v]) => {
+ acc[k] = Array.isArray(v) ? v[0] : v
+ return acc
+ }, {})
+ : undefined
 
  return {
  statusCode,
  body,
+ headers,
  multiValueHeaders,
  isBase64Encoded
  }

diff --git a/src/transport.js b/src/transport.js
@@ -10,6 +10,7 @@ function forwardResponse ({
  response,
  resolver,
  eventSource,
+ event,
  log
 }) {
  const statusCode = response.statusCode
@@ -30,6 +31,7 @@ function forwardResponse ({
  })
 
  const successResponse = eventSource.getResponse({
+ event,
  statusCode,
  body,
  headers,
@@ -158,6 +160,7 @@ async function forwardRequestToNodeServer ({
  response,
  resolver,
  eventSource,
+ event,
  log
  })
  return response