Java WebSocketFrame类代码示例

import org.jboss.netty.handler.codec.http.websocketx.WebSocketFrame; //导入依赖的package包/类
private void handleWebSocketFrame(final ChannelHandlerContext ctx, WebSocketFrame frame)
        throws Exception {
    if (frame instanceof CloseWebSocketFrame) {
        this.logger.info(Text.WS_CONNECTION_CLOSED_BY,
                ((CloseWebSocketFrame) frame).getStatusCode(),
                this.closedReason = ((CloseWebSocketFrame) frame).getReasonText());
        ctx.getChannel().close();
        return;
    } else if (frame instanceof BinaryWebSocketFrame) {
        if (!((BinaryWebSocketFrame) frame).isFinalFragment()) {
            this.logger.warn(Text.WS_NOT_FINAL);
            return;
        }
        if (this.channelHandler != null) {
            // if not final frame,
            // should wait until final frame received
            // https://github.com/wsky/top-link/issues/5
            ChannelBuffer buffer = ((BinaryWebSocketFrame) frame).getBinaryData();
            this.channelHandler.onMessage(this.createContext(buffer.toByteBuffer()));
        }
    }
    if (this.logger.isDebugEnabled()) {
        this.logger.debug("unhandled frame: %s", frame);
    }
}
 

import org.jboss.netty.handler.codec.http.websocketx.WebSocketFrame; //导入依赖的package包/类
@Override
protected Object decode(ChannelHandlerContext ctx, Channel channel, ChannelBuffer buffer,
        State state) throws Exception {
    Object o = super.decode(ctx, channel, buffer, state);

    if (o == null) {
        return null;
    }

    if (!((o instanceof BinaryWebSocketFrame) || (o instanceof TextWebSocketFrame))) {
        return o;
    }

    // collect binary/text frame only
    this.cumulation.add((WebSocketFrame) o);
    // always return null making calldecode break and do not raise
    // unfoldAndFireMessageReceived
    return null;
}
 

import org.jboss.netty.handler.codec.http.websocketx.WebSocketFrame; //导入依赖的package包/类
private void handleWebSocketFrame(ChannelHandlerContext ctx, WebSocketFrame frame)
        throws Exception {
    if (frame instanceof CloseWebSocketFrame) {
        this.logger.info(Text.WS_CONNECTION_CLOSED_BY,
                ((CloseWebSocketFrame) frame).getStatusCode(),
                this.closedReason = ((CloseWebSocketFrame) frame).getReasonText());
        CloseWebSocketFrame closeFrame = (CloseWebSocketFrame) frame;
        this.clear(ctx);
        this.logger.warn(Text.WS_CONNECTION_CLOSED_BY, closeFrame.getStatusCode(),
                closeFrame.getReasonText());
    } else if (frame instanceof BinaryWebSocketFrame) {
        if (!((BinaryWebSocketFrame) frame).isFinalFragment()) {
            this.logger.warn(Text.WS_NOT_FINAL);
            return;
        }
        if (this.haveHandler()) {
            ChannelBuffer buffer = ((BinaryWebSocketFrame) frame).getBinaryData();
            this.getHandler().onMessage(this.createContext(buffer.toByteBuffer()));
        }
    }
}
 

import org.jboss.netty.handler.codec.http.websocketx.WebSocketFrame; //导入依赖的package包/类
@Override
public void frameReceived(WebSocketFrame msg) {

	if (socketChannel != null) {
		if (socketChannel.isConnected()) {
			if (log.isDebugEnabled()) {
				log.debug("Forwarding frame to socket "
						+ socketChannel.getRemoteAddress());
			}

			reportInputBytes(msg.getBinaryData()
					.readableBytes());
			socketChannel.write(msg.getBinaryData());
		} else {
			if (log.isDebugEnabled()) {
				log.debug("Forwarding socket is no longer connected for "
						+ socketChannel.getRemoteAddress());
			}
			close();
		}
	}

	
}
 

import org.jboss.netty.handler.codec.http.websocketx.WebSocketFrame; //导入依赖的package包/类
private void processWebsocketFrame(WebSocketFrame msg, Channel channel) {

		if (log.isDebugEnabled()) {
			log.debug("Received websocket frame from "
					+ channel.getRemoteAddress());
		}
		// Check for closing frame
		if (msg instanceof CloseWebSocketFrame) {
			// handshaker.close(channel, (CloseWebSocketFrame) msg);
			return;
		} else if (!(msg instanceof BinaryWebSocketFrame)) {
			// Close
		}

		NettyWebsocketClient nettyClient = (NettyWebsocketClient) channel.getAttachment();
		nettyClient.frameReceived(msg);
		
	}
 

import org.jboss.netty.handler.codec.http.websocketx.WebSocketFrame; //导入依赖的package包/类
/**
 * {@inheritDoc}
 * @see org.jboss.netty.channel.ChannelDownstreamHandler#handleDownstream(org.jboss.netty.channel.ChannelHandlerContext, org.jboss.netty.channel.ChannelEvent)
 */
@Override
public void handleDownstream(ChannelHandlerContext ctx, ChannelEvent e) throws Exception {
	Channel channel = e.getChannel();
	if(!channel.isOpen()) return;
	if(!(e instanceof MessageEvent)) {
           ctx.sendDownstream(e);
           return;
       }
	Object message = ((MessageEvent)e).getMessage();
	if((message instanceof HttpResponse) || (message instanceof WebSocketFrame)) {
		ctx.sendDownstream(e);
		return;
	}
	if((message instanceof ChannelBuffer)) {
		ctx.sendDownstream(new DownstreamMessageEvent(channel, Channels.future(channel), new TextWebSocketFrame((ChannelBuffer)message), channel.getRemoteAddress()));
	} else if((message instanceof JsonNode)) {  			
		String json = marshaller.writeValueAsString(message);
		ctx.sendDownstream(new DownstreamMessageEvent(channel, Channels.future(channel), new TextWebSocketFrame(json), channel.getRemoteAddress()));			
	} else if((message instanceof ChannelBufferizable)) {
		ctx.sendDownstream(new DownstreamMessageEvent(channel, Channels.future(channel), new TextWebSocketFrame(((Netty3ChannelBufferizable)message).toChannelBuffer()), channel.getRemoteAddress()));
	} else if((message instanceof CharSequence)) {
		ctx.sendDownstream(new DownstreamMessageEvent(channel, Channels.future(channel), new TextWebSocketFrame(marshaller.writeValueAsString(message)), channel.getRemoteAddress()));
	} else if((message instanceof JSONResponse)) {				
		ObjectMapper mapper = (ObjectMapper)((JSONResponse)message).getChannelOption("mapper", TSDBTypeSerializer.DEFAULT.getMapper());			
		ctx.sendDownstream(new DownstreamMessageEvent(channel, Channels.future(channel), new TextWebSocketFrame(mapper.writeValueAsString(message)), channel.getRemoteAddress()));					
	} else {
           ctx.sendUpstream(e);
	}		
}
 

import org.jboss.netty.handler.codec.http.websocketx.WebSocketFrame; //导入依赖的package包/类
/**
 * {@inheritDoc}
 * @see org.jboss.netty.channel.ChannelUpstreamHandler#handleUpstream(org.jboss.netty.channel.ChannelHandlerContext, org.jboss.netty.channel.ChannelEvent)
 */
@Override
public void handleUpstream(ChannelHandlerContext ctx, ChannelEvent e) throws Exception {
	log.warn("ChannelEvent: {}", e);
	if(e instanceof MessageEvent) {
		Object message = ((MessageEvent)e).getMessage();
		if (message instanceof HttpRequest) {
			handleRequest(ctx, (HttpRequest) message, (MessageEvent)e);
		} else if (message instanceof WebSocketFrame) {
			handleRequest(ctx, (WebSocketFrame) message);
		}
	} else {
		ctx.sendUpstream(e);
	}			
}
 

import org.jboss.netty.handler.codec.http.websocketx.WebSocketFrame; //导入依赖的package包/类
private boolean send(WebSocketFrame frame, final SendHandler sendHandler, int timeout)
        throws ChannelException {
    // do not support fragmentation
    frame.setFinalFragment(true);

    // weather sendSync enabled
    final CountDownLatch latch = timeout > 0 ? new CountDownLatch(1) : null;

    this.channel.write(frame).addListener(new ChannelFutureListener() {

        @Override
        public void operationComplete(ChannelFuture future) throws Exception {
            if (latch != null) {
                latch.countDown();
            } else if (sendHandler != null) {
                sendHandler.onSendComplete(future.isSuccess());
            }
        }
    });

    if (latch == null) {
        return true;
    }

    // boolean success = false;
    try {
        return latch.await(timeout, TimeUnit.MILLISECONDS);
    } catch (InterruptedException e) {
        throw new ChannelException(Text.WS_SEND_SYNC_ERROR, e);
    } finally {
        // not safe, timeout message already in netty sendbuffer and will be
        // sent later,
        // if buffer clear in onSendComplete, client will got wrong message
        // if (sendHandler != null)
        // sendHandler.onSendComplete(success);
    }
}
 

import org.jboss.netty.handler.codec.http.websocketx.WebSocketFrame; //导入依赖的package包/类
@SuppressWarnings("unchecked")
@Override
public void messageReceived(ChannelHandlerContext ctx, MessageEvent e) throws Exception {
    Object msg = e.getMessage();
    if (msg instanceof HttpRequest) {
        this.handleHttpRequest(ctx, (HttpRequest) msg);
    } else if (msg instanceof WebSocketFrame) {
        this.handleWebSocketFrame(ctx, (WebSocketFrame) msg);
    } else if (msg instanceof List<?>) {
        this.handleWebSocketFrame(ctx, (List<WebSocketFrame>) msg);
    }
}
 

import org.jboss.netty.handler.codec.http.websocketx.WebSocketFrame; //导入依赖的package包/类
@Override
public void messageReceived(ChannelHandlerContext ctx, MessageEvent e) throws Exception {
    this.cumulation = new ArrayList<WebSocketFrame>();
    super.messageReceived(ctx, e);
    // batch FireMessageReceived
    if (this.cumulation.size() > 0) {
        Channels.fireMessageReceived(ctx, cumulation, e.getRemoteAddress());
    }
}
 

import org.jboss.netty.handler.codec.http.websocketx.WebSocketFrame; //导入依赖的package包/类
@Override
public void messageReceived(ChannelHandlerContext ctx, MessageEvent e) throws Exception {
    if (!this.handshaker.isHandshakeComplete()) {
        this.handleHandshake(ctx, (HttpResponse) e.getMessage());
    }
    if (e.getMessage() instanceof WebSocketFrame) {
        this.handleWebSocketFrame(ctx, (WebSocketFrame) e.getMessage());
    }
}
 

import org.jboss.netty.handler.codec.http.websocketx.WebSocketFrame; //导入依赖的package包/类
@Override
public void messageReceived(ChannelHandlerContext ctx, MessageEvent e) throws Exception
{
	Object msg = e.getMessage();
	if (msg instanceof HttpRequest)
	{
		this.handleHttpRequest(ctx, (HttpRequest) msg);
	}
	else if (msg instanceof WebSocketFrame)
	{
		this.handleWebSocketFrame(ctx, (WebSocketFrame) msg);
	}
}
 

import org.jboss.netty.handler.codec.http.websocketx.WebSocketFrame; //导入依赖的package包/类
@Override
public void messageReceived(ChannelHandlerContext ctx, MessageEvent e) throws Exception
{
	Channel ch = ctx.getChannel();
	if (!this.handshaker.isHandshakeComplete())
	{
		this.ch = ch;
		this.handshaker.finishHandshake(ch, (HttpResponse) e.getMessage());
		this.t.notifyOpen();
		return;
	}
	
	if (e.getMessage() instanceof HttpResponse)
	{
		HttpResponse response = (HttpResponse) e.getMessage();
		throw new Exception("Unexpected HttpResponse (status=" + response.getStatus() + ", content=" + response.getContent().toString(CharsetUtil.UTF_8) + ")");
	}
	
	WebSocketFrame frame = (WebSocketFrame) e.getMessage();
	if (frame instanceof BinaryWebSocketFrame)
	{
		BinaryWebSocketFrame f = (BinaryWebSocketFrame) frame;
		ChannelBuffer binaryData = f.getBinaryData();
		byte[] arr = new byte[binaryData.readableBytes()];
		binaryData.readBytes(arr);
		this.t.notifyMessage(arr);

	}
	else if (frame instanceof PongWebSocketFrame)
	{
		System.out.println("WebSocket Client received pong");
	}
	else if (frame instanceof CloseWebSocketFrame)
	{
		System.out.println("WebSocket Client received closing");
		ch.close();
	}
}
 

import org.jboss.netty.handler.codec.http.websocketx.WebSocketFrame; //导入依赖的package包/类
@Override
public void messageReceived(ChannelHandlerContext ctx, MessageEvent e) throws Exception {
    Object msg = e.getMessage();
    if (msg instanceof HttpRequest) {
        handleHttpRequest(ctx, (HttpRequest) msg);
    } else if (msg instanceof WebSocketFrame) {
        handleWebSocketFrame(ctx, (WebSocketFrame) msg);
    }
}
 

import org.jboss.netty.handler.codec.http.websocketx.WebSocketFrame; //导入依赖的package包/类
private void handleWebSocketFrame(ChannelHandlerContext ctx, WebSocketFrame frame) {
    Connect connect = connects.getConnect(ctx.getChannel().getId() + "");
    try {
        if (frame instanceof CloseWebSocketFrame) {
            handshaker.close(ctx.getChannel(), (CloseWebSocketFrame) frame);
        } else if (frame instanceof PingWebSocketFrame) {
            ctx.getChannel().write(new PongWebSocketFrame(frame.getBinaryData()));
        } else if (frame instanceof TextWebSocketFrame) {
            String message = ((TextWebSocketFrame) frame).getText();
            Jsonx cc = Jsonx.create(message);
            SocketRequest request = cc.toObject(new SocketRequest());
            ObjectSnooper.snoop(request).set("session", connect.getSession());
            BaseSocketService service = SocketServiceMapping.getSocketServiceMapping().getService(cc.get("type").toString());
            service.setConnect(connect);
            service.setRequest(request);
            service.setConnections(connects);
            if (service != null) {
                service.doRoutResult();
            } else {
                throw new Exception("service can not support");
            }
        } else {
            throw new Exception("data error,");
        }
    } catch (Exception e) {
        e.printStackTrace();
        connect.send("{\"code\":0,\"mes\":" + e.getMessage() + "}");
    }
}