View Javadoc
1   /*
2    * Copyright 2012 The Netty Project
3    *
4    * The Netty Project licenses this file to you under the Apache License,
5    * version 2.0 (the "License"); you may not use this file except in compliance
6    * with the License. You may obtain a copy of the License at:
7    *
8    *   http://www.apache.org/licenses/LICENSE-2.0
9    *
10   * Unless required by applicable law or agreed to in writing, software
11   * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
12   * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
13   * License for the specific language governing permissions and limitations
14   * under the License.
15   */
16  //The MIT License
17  //
18  //Copyright (c) 2009 Carl Bystršm
19  //
20  //Permission is hereby granted, free of charge, to any person obtaining a copy
21  //of this software and associated documentation files (the "Software"), to deal
22  //in the Software without restriction, including without limitation the rights
23  //to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
24  //copies of the Software, and to permit persons to whom the Software is
25  //furnished to do so, subject to the following conditions:
26  //
27  //The above copyright notice and this permission notice shall be included in
28  //all copies or substantial portions of the Software.
29  //
30  //THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
31  //IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
32  //FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
33  //AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
34  //LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
35  //OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
36  //THE SOFTWARE.
37  
38  package io.netty.example.http.websocketx.client;
39  
40  import io.netty.channel.Channel;
41  import io.netty.channel.ChannelFuture;
42  import io.netty.channel.ChannelHandlerContext;
43  import io.netty.channel.ChannelPromise;
44  import io.netty.channel.SimpleChannelInboundHandler;
45  import io.netty.handler.codec.http.FullHttpResponse;
46  import io.netty.handler.codec.http.websocketx.CloseWebSocketFrame;
47  import io.netty.handler.codec.http.websocketx.PongWebSocketFrame;
48  import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
49  import io.netty.handler.codec.http.websocketx.WebSocketClientHandshaker;
50  import io.netty.handler.codec.http.websocketx.WebSocketFrame;
51  import io.netty.util.CharsetUtil;
52  
53  public class WebSocketClientHandler extends SimpleChannelInboundHandler<Object> {
54  
55      private final WebSocketClientHandshaker handshaker;
56      private ChannelPromise handshakeFuture;
57  
58      public WebSocketClientHandler(WebSocketClientHandshaker handshaker) {
59          this.handshaker = handshaker;
60      }
61  
62      public ChannelFuture handshakeFuture() {
63          return handshakeFuture;
64      }
65  
66      @Override
67      public void handlerAdded(ChannelHandlerContext ctx) {
68          handshakeFuture = ctx.newPromise();
69      }
70  
71      @Override
72      public void channelActive(ChannelHandlerContext ctx) {
73          handshaker.handshake(ctx.channel());
74      }
75  
76      @Override
77      public void channelInactive(ChannelHandlerContext ctx) {
78          System.out.println("WebSocket Client disconnected!");
79      }
80  
81      @Override
82      public void messageReceived(ChannelHandlerContext ctx, Object msg) {
83          Channel ch = ctx.channel();
84          if (!handshaker.isHandshakeComplete()) {
85              handshaker.finishHandshake(ch, (FullHttpResponse) msg);
86              System.out.println("WebSocket Client connected!");
87              handshakeFuture.setSuccess();
88              return;
89          }
90  
91          if (msg instanceof FullHttpResponse) {
92              FullHttpResponse response = (FullHttpResponse) msg;
93              throw new IllegalStateException(
94                      "Unexpected FullHttpResponse (getStatus=" + response.status() +
95                              ", content=" + response.content().toString(CharsetUtil.UTF_8) + ')');
96          }
97  
98          WebSocketFrame frame = (WebSocketFrame) msg;
99          if (frame instanceof TextWebSocketFrame) {
100             TextWebSocketFrame textFrame = (TextWebSocketFrame) frame;
101             System.out.println("WebSocket Client received message: " + textFrame.text());
102         } else if (frame instanceof PongWebSocketFrame) {
103             System.out.println("WebSocket Client received pong");
104         } else if (frame instanceof CloseWebSocketFrame) {
105             System.out.println("WebSocket Client received closing");
106             ch.close();
107         }
108     }
109 
110     @Override
111     public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
112         cause.printStackTrace();
113         if (!handshakeFuture.isDone()) {
114             handshakeFuture.setFailure(cause);
115         }
116         ctx.close();
117     }
118 }