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    *   https://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  package io.netty.example.http.websocketx.server;
17  
18  import io.netty.channel.ChannelHandlerContext;
19  import io.netty.channel.SimpleChannelInboundHandler;
20  import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
21  import io.netty.handler.codec.http.websocketx.WebSocketFrame;
22  import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;
23  
24  import java.util.Locale;
25  
26  /**
27   * Echoes uppercase content of text frames.
28   */
29  public class WebSocketFrameHandler extends SimpleChannelInboundHandler<WebSocketFrame> {
30  
31      @Override
32      protected void channelRead0(ChannelHandlerContext ctx, WebSocketFrame frame) throws Exception {
33          // ping and pong frames already handled
34  
35          if (frame instanceof TextWebSocketFrame) {
36              // Send the uppercase string back.
37              String request = ((TextWebSocketFrame) frame).text();
38              ctx.channel().writeAndFlush(new TextWebSocketFrame(request.toUpperCase(Locale.US)));
39          } else {
40              String message = "unsupported frame type: " + frame.getClass().getName();
41              throw new UnsupportedOperationException(message);
42          }
43      }
44  
45      @Override
46      public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
47          if (evt instanceof WebSocketServerProtocolHandler.HandshakeComplete) {
48              //Channel upgrade to websocket, remove WebSocketIndexPageHandler.
49              ctx.pipeline().remove(WebSocketIndexPageHandler.class);
50          } else {
51              super.userEventTriggered(ctx, evt);
52          }
53      }
54  }