org.noear.solon.socketd.client.websocket.WsSocketClientImp Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of solon.socketd.client.websocket Show documentation

There is a newer version: 2.6.0-M1

Show newest version

package org.noear.solon.socketd.client.websocket;

import org.java_websocket.client.WebSocketClient;
import org.java_websocket.handshake.ServerHandshake;
import org.noear.solon.Solon;
import org.noear.solon.core.event.EventBus;
import org.noear.solon.core.message.Message;
import org.noear.solon.core.message.Session;
import org.noear.solon.socketd.ListenerProxy;
import org.noear.solon.socketd.ProtocolManager;
import org.noear.solon.socketd.SessionFlag;

import java.net.URI;
import java.nio.ByteBuffer;

public class WsSocketClientImp extends WebSocketClient {
    private Session session;

    public WsSocketClientImp(URI serverUri, Session session) {
        super(serverUri);
        this.session = session;
    }

    @Override
    public void onOpen(ServerHandshake serverHandshake) {
        ListenerProxy.getGlobal().onOpen(session);
    }

    @Override
    public void onMessage(String test) {
        try {
            ListenerProxy.getGlobal().onMessage(session, Message.wrap(test).isString(true));
        } catch (Throwable ex) {
            EventBus.push(ex);
        }
    }

    @Override
    public void onMessage(ByteBuffer bytes) {

        try {
            Message message = null;
            if (isWebSocketD()) {
                message = ProtocolManager.decode(bytes);
            } else {
                message = Message.wrap(bytes.array());
            }

            ListenerProxy.getGlobal().onMessage(session, message);
        } catch (Throwable ex) {
            EventBus.push(ex);
        }
    }

    @Override
    public void onClose(int i, String s, boolean b) {
        ListenerProxy.getGlobal().onClose(session);
    }

    @Override
    public void onError(Exception e) {
        ListenerProxy.getGlobal().onError(session, e);
    }

    private boolean isWebSocketD() {
        return Solon.global().enableWebSocketD() || session.flag() == SessionFlag.socketd;
    }
}