add customized headers support

DanielYWoo · DanielYWoo · commit 954ab8b7e975 · 2012-04-26T10:53:09.000+08:00
diff --git a/src/org/java_websocket/WebSocketClient.java b/src/org/java_websocket/WebSocketClient.java
@@ -68,6 +68,10 @@ public WebSocketClient( URI serverURI ) {
 	 * must call <var>connect</var> first to initiate the socket connection.
 	 */
 	public WebSocketClient( URI serverUri , Draft draft ) {
+		this(serverUri, draft, null);
+	}
+
+	public WebSocketClient( URI serverUri , Draft draft, Map<String, String> headers) {
 		if( serverUri == null ) {
 			throw new IllegalArgumentException();
 		}
@@ -76,6 +80,7 @@ public WebSocketClient( URI serverUri , Draft draft ) {
 		}
 		this.uri = serverUri;
 		this.draft = draft;
+		this.headers = headers;
 	}
 
 	/**
@@ -279,6 +284,11 @@ private void sendHandshake() throws IOException , InvalidHandshakeException , In
 		HandshakeImpl1Client handshake = new HandshakeImpl1Client();
 		handshake.setResourceDescriptor( path );
 		handshake.put( "Host", host );
+		if (headers != null) {
+			for (Map.Entry<String, String> kv : headers.entrySet()) {
+				handshake.put(kv.getKey(), kv.getValue());
+			}
+		}
 		conn.startHandshake( handshake );
 	}
 
