com.iteaj.iot.client.websocket.impl.DefaultWebSocketClientProtocolHandle Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of iot-client Show documentation

基于netty的tcp, http, udp等协议的客户端

The newest version!

package com.iteaj.iot.client.websocket.impl;

import com.iteaj.iot.Message;
import com.iteaj.iot.client.ClientProtocolHandle;
import com.iteaj.iot.client.websocket.WebSocketClientListener;
import com.iteaj.iot.websocket.WebSocketConnectHead;
import com.iteaj.iot.websocket.WebSocketException;
import com.iteaj.iot.websocket.WebSocketFrameType;

public class DefaultWebSocketClientProtocolHandle implements ClientProtocolHandle {

    private DefaultWebSocketClientComponent component;

    public DefaultWebSocketClientProtocolHandle(DefaultWebSocketClientComponent component) {
        this.component = component;
    }

    @Override
    public Object handle(DefaultWebSocketClientProtocol protocol) {

        DefaultWebSocketClientMessage requestMessage = protocol.requestMessage();
        WebSocketClientListener listener = requestMessage.getProperties().getListener();
        if(listener != null) {
            Message.MessageHead head = requestMessage.getHead();
            if(head instanceof WebSocketConnectHead) {
                listener.onConnect(protocol);
            } else {
                WebSocketFrameType type = requestMessage.frameType();
                switch (type) {
                    case Text: listener.onText(protocol); break;
                    case Close: listener.onClose(protocol); break;
                    case Binary: listener.onBinary(protocol); break;
                    default: throw new WebSocketException("不支持的事件["+type+"]");
                }
            }
        }

        return null;
    }
}