JavaxWebsocketChannel.java

/*
 * Copyright 2014-2016 Red Hat, Inc, and individual contributors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.projectodd.wunderboss.web.async.websocket;

import org.projectodd.wunderboss.web.async.WebsocketUtil;

import javax.websocket.CloseReason;
import javax.websocket.Endpoint;
import javax.websocket.EndpointConfig;
import javax.websocket.MessageHandler;
import javax.websocket.SendHandler;
import javax.websocket.SendResult;
import javax.websocket.Session;
import java.io.IOException;
import java.nio.ByteBuffer;

public class JavaxWebsocketChannel extends WebsocketChannelSkeleton {

    public JavaxWebsocketChannel(final OnOpen onOpen,
                                 final OnError onError,
                                 final OnClose onClose,
                                 final OnMessage onMessage) {
        super(onOpen, onError, onClose, onMessage);
    }

    @Override
    public Endpoint endpoint() {
        final JavaxWebsocketChannel channel = this;
        return new Endpoint() {
            @Override
            public void onOpen(final Session session, final EndpointConfig endpointConfig) {
                session.addMessageHandler(new MessageHandler.Whole<String>() {
                    @Override
                    public void onMessage(String s) {
                        notifyMessage(s);
                    }
                });
                session.addMessageHandler(new MessageHandler.Whole<byte[]>() {
                    @Override
                    public void onMessage(byte[] s) {
                        notifyMessage(s);
                    }
                });
                channel.setUnderlyingChannel(session);
                channel.notifyOpen(endpointConfig);
            }

            @Override
            public void onClose(final Session _, final CloseReason closeReason) {
                notifyClose(closeReason.getCloseCode().getCode(),
                            closeReason.getReasonPhrase());
            }

            @Override
            public void onError(final Session _, final Throwable error) {
                notifyError(error);
            }
        };
    }

    @Override
    public void setUnderlyingChannel(Object channel) {
        this.session = (Session)channel;
        setTimeoutOnSession();
    }

    @Override
    public boolean isOpen() {
        return this.session != null &&
                this.session.isOpen();
    }

    @Override
    public boolean send(Object message,
                        final boolean shouldClose,
                        final OnComplete onComplete) throws Exception {
        if (!isOpen()) {
            return false;
        }

        SendHandler handler = new SendHandler() {
            @Override
            public void onResult(SendResult sendResult) {
                Throwable ex = sendResult.getException();
                if (sendResult.isOK()) {
                    if (shouldClose) {
                        try {
                            close();
                        } catch (IOException e) {
                            ex = e;
                        }
                    }
                }
                notifyComplete(onComplete, ex);
            }
        };

        if (message == null) {
            handler.onResult(new SendResult());
        } else if (message instanceof String) {
            this.session.getAsyncRemote().sendText((String)message, handler);
        } else if (message instanceof byte[]) {
            this.session.getAsyncRemote().sendBinary(ByteBuffer.wrap((byte[])message), handler);
        } else {
            throw WebsocketUtil.wrongMessageType(message.getClass());
        }


        return true;
    }

    @Override
    public void close() throws IOException {
        if (isOpen()) {
            this.session.close();
        }
    }

    @Override
    public void setTimeout(long timeout) {
        this.timeout = timeout;
        setTimeoutOnSession();
    }

    private void setTimeoutOnSession() {
        if (this.session != null &&
                this.timeout >= 0) {
            this.session.setMaxIdleTimeout(this.timeout);
        }
    }

    private Session session;
    private long timeout = -1;
}