Adding a configuration object

Arun Gupta · Arun Gupta · commit 0f52ac4d368e · 2013-04-03T18:16:03.000-07:00
Former-commit-id: 7de2bf5676e766a4ecb4d7a39d178ab56bad05ff
diff --git a/samples/websocket/websocket-client-config/src/main/java/org/sample/client/MyClient.java b/samples/websocket/websocket-client-config/src/main/java/org/sample/client/MyClient.java
@@ -51,14 +51,14 @@
 /**
  * @author Arun Gupta
  */
-@ClientEndpoint
+@ClientEndpoint(configurator = MyConfigurator.class)
 public class MyClient {
     @OnOpen
     public void onOpen(Session session) {
         System.out.println("Connected to endpoint: " + session.getBasicRemote());
         try {
             String name = "Duke";
-            System.out.println("Sending message to endpoint: " + name);
+            System.out.println("Sending message from client -> endpoint: " + name);
             session.getBasicRemote().sendText(name);
         } catch (IOException ex) {
             Logger.getLogger(MyClient.class.getName()).log(Level.SEVERE, null, ex);
diff --git a/samples/websocket/websocket-client-config/src/main/java/org/sample/client/MyConfigurator.java b/samples/websocket/websocket-client-config/src/main/java/org/sample/client/MyConfigurator.java
@@ -0,0 +1,72 @@
+/*
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
+ *
+ * Copyright (c) 2013 Oracle and/or its affiliates. All rights reserved.
+ *
+ * The contents of this file are subject to the terms of either the GNU
+ * General Public License Version 2 only ("GPL") or the Common Development
+ * and Distribution License("CDDL") (collectively, the "License").  You
+ * may not use this file except in compliance with the License.  You can
+ * obtain a copy of the License at
+ * https://glassfish.dev.java.net/public/CDDL+GPL_1_1.html
+ * or packager/legal/LICENSE.txt.  See the License for the specific
+ * language governing permissions and limitations under the License.
+ *
+ * When distributing the software, include this License Header Notice in each
+ * file and include the License file at packager/legal/LICENSE.txt.
+ *
+ * GPL Classpath Exception:
+ * Oracle designates this particular file as subject to the "Classpath"
+ * exception as provided by Oracle in the GPL Version 2 section of the License
+ * file that accompanied this code.
+ *
+ * Modifications:
+ * If applicable, add the following below the License Header, with the fields
+ * enclosed by brackets [] replaced by your own identifying information:
+ * "Portions Copyright [year] [name of copyright owner]"
+ *
+ * Contributor(s):
+ * If you wish your version of this file to be governed by only the CDDL or
+ * only the GPL Version 2, indicate your decision by adding "[Contributor]
+ * elects to include this software in this distribution under the [CDDL or GPL
+ * Version 2] license."  If you don't indicate a single choice of license, a
+ * recipient has the option to distribute your version of this file under
+ * either the CDDL, the GPL Version 2 or to extend the choice of license to
+ * its licensees as provided above.  However, if you add GPL Version 2 code
+ * and therefore, elected the GPL Version 2 license, then the option applies
+ * only if the new code is made subject to such option by the copyright
+ * holder.
+ */
+package org.sample.client;
+
+import java.util.List;
+import java.util.Map;
+import javax.websocket.ClientEndpointConfig;
+import javax.websocket.HandshakeResponse;
+
+/**
+ * @author Arun Gupta
+ */
+ public class MyConfigurator extends ClientEndpointConfig.Configurator {
+
+    @Override
+    public void beforeRequest(Map<String, List<String>> headers) {
+        System.out.println("beforeRequest:");
+        for (String h : headers.keySet()) {
+            for (String k : headers.get(h)) {
+                System.out.println("Header: " + h + ", " + k);
+            }
+        }
+
+    }
+
+    @Override
+    public void afterResponse(HandshakeResponse response) {
+        System.out.println("afterResponse:");
+        for (String h : response.getHeaders().keySet()) {
+            for (String k : response.getHeaders().get(h)) {
+                System.out.println("Header: " + h + ", " + k);
+            }
+        }
+    }
+}
diff --git a/samples/websocket/websocket-client-config/src/main/java/org/sample/client/TestClient.java b/samples/websocket/websocket-client-config/src/main/java/org/sample/client/TestClient.java
@@ -83,8 +83,7 @@ protected void processRequest(HttpServletRequest request, HttpServletResponse re
             WebSocketContainer container = ContainerProvider.getWebSocketContainer();
             String uri = "ws://localhost:8080" + request.getContextPath() + "/websocket";
             out.println("Connecting to " + uri);
-//            container.connectToServer(MyClient.class, URI.create(uri));
-            container.connectToServer(MyEndpoint.class, URI.create(uri));
+            container.connectToServer(MyClient.class, URI.create(uri));
             out.println("<br><br>Look in server.log for log messages from message exchange between client/server.");
             
             out.println("</body>");
