fix: compatibility with set headers option (#763)

webpack · Nov 10, 2020 · 7c4cac5 · 7c4cac5
1 parent 441b449
commit 7c4cac5
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/src/middleware.js b/src/middleware.js
@@ -79,8 +79,19 @@ export default function wrapper(context) {
       }
 
       if (headers) {
-        for (const name of Object.keys(headers)) {
-          res.setHeader(name, headers[name]);
+        const names = Object.keys(headers);
+
+        // Express API
+        if(res.set) {
+          for (const name of names) {
+            res.set(name, headers[name]);
+          }
+        }
+        // Node.js API
+        else {
+          for (const name of names) {
+            res.setHeader(name, headers[name]);
+          }
         }
       }