JSR-356 adding support for ClientEndpointConfig.Configurator

1 files changed, 17 insertions, 0 deletions

diff --git a/jetty-websocket/websocket-api/src/main/java/org/eclipse/jetty/websocket/api/UpgradeRequest.java b/jetty-websocket/websocket-api/src/main/java/org/eclipse/jetty/websocket/api/UpgradeRequest.java
index cb35c3aecd..9183496c1d 100644
--- a/jetty-websocket/websocket-api/src/main/java/org/eclipse/jetty/websocket/api/UpgradeRequest.java
+++ b/jetty-websocket/websocket-api/src/main/java/org/eclipse/jetty/websocket/api/UpgradeRequest.java
@@ -75,6 +75,11 @@ public class UpgradeRequest
         }
     }
 
+    public void clearHeaders()
+    {
+        headers.clear();
+    }
+
     public List<HttpCookie> getCookies()
     {
         return cookies;
@@ -242,6 +247,18 @@ public class UpgradeRequest
         setHeader(name.toLowerCase(Locale.ENGLISH),values);
     }
 
+    public void setHeaders(Map<String, List<String>> headers)
+    {
+        clearHeaders();
+
+        for (Map.Entry<String, List<String>> entry : headers.entrySet())
+        {
+            String name = entry.getKey();
+            List<String> values = entry.getValue();
+            setHeader(name,values);
+        }
+    }
+
     public void setHttpVersion(String httpVersion)
     {
         this.httpVersion = httpVersion;