org.lastbamboo.common.turn.client.TurnClientIoHandler Maven / Gradle / Ivy

Go to download
package org.lastbamboo.common.turn.client;

import org.littleshoot.mina.common.IdleStatus;
import org.littleshoot.mina.common.IoHandler;
import org.littleshoot.mina.common.IoHandlerAdapter;
import org.littleshoot.mina.common.IoSession;
import org.littleshoot.mina.util.SessionUtil;
import org.littleshoot.stun.stack.message.StunMessage;
import org.littleshoot.stun.stack.message.StunMessageVisitor;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * {@link IoHandler} for the TURN client connection to the TURN server.
 */
public class TurnClientIoHandler extends IoHandlerAdapter {

    private final Logger LOG = LoggerFactory.getLogger(getClass());
    
    private final StunMessageVisitor visitor;

    /**
     * Creates a new {@link IoHandler} for the TURN proxy/client.
     * 
     * @param visitor The class for visiting read messages.
     */
    public TurnClientIoHandler(final StunMessageVisitor visitor) {
        this.visitor = visitor;
    }

    @Override
    public void messageReceived(final IoSession session, final Object message) {
        final StunMessage vsm = (StunMessage) message;
        vsm.accept(this.visitor);
    }

    @Override
    public void sessionClosed(final IoSession session) {
        // This is taken care of through an IoServiceListener.
    }

    @Override
    public void sessionCreated(final IoSession session) {
        SessionUtil.initialize(session);

        // We consider the session idle fairly quickly to free up expensive
        // resources.
        // session.setIdleTime(IdleStatus.WRITER_IDLE, 80);
    }

    @Override
    public void sessionIdle(final IoSession session, final IdleStatus status) {
        LOG.debug("Session idle...closing TURN connection to server...");

        // final AllocateRequest request = new AllocateRequest();
        // session.write(request);

        // We used to issue a new allocate request in this case to make sure
        // the server keeps the TURN connection up. This only makes sense
        // for maintaining long-lived TURN connections, though. In our case,
        // if a connection is idle, it's not being used for a file transfer,
        // so we should close it.

        // Note we ideally should not rely on this behavior, instead closing
        // TURN connections whenever we know they're not used.
        session.close();
    }

    @Override
    public void exceptionCaught(final IoSession session, final Throwable cause){
        LOG.warn("Caught exception", cause);
    }
}