com.calclab.emite.reconnect.client.SessionReconnect.java Source code

Introduction

Here is the source code for com.calclab.emite.reconnect.client.SessionReconnect.java

Source

/*
 * ((e)) emite: A pure Google Web Toolkit XMPP library
 * Copyright (c) 2008-2011 The Emite development team
 * 
 * This file is part of Emite.
 *
 * Emite is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation, either version 3
 * of the License, or (at your option) any later version.
 *
 * Emite is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with Emite.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.calclab.emite.reconnect.client;

import java.util.logging.Logger;

import com.calclab.emite.core.client.conn.ConnectionStateChangedEvent;
import com.calclab.emite.core.client.conn.ConnectionStateChangedEvent.ConnectionState;
import com.calclab.emite.core.client.conn.ConnectionStateChangedHandler;
import com.calclab.emite.core.client.conn.XmppConnection;
import com.calclab.emite.core.client.events.StateChangedEvent;
import com.calclab.emite.core.client.events.StateChangedHandler;
import com.calclab.emite.core.client.xmpp.sasl.AuthorizationResultEvent;
import com.calclab.emite.core.client.xmpp.sasl.AuthorizationResultHandler;
import com.calclab.emite.core.client.xmpp.sasl.SASLManager;
import com.calclab.emite.core.client.xmpp.session.Credentials;
import com.calclab.emite.core.client.xmpp.session.SessionStates;
import com.calclab.emite.core.client.xmpp.session.XmppSession;
import com.google.gwt.user.client.Timer;
import com.google.inject.Inject;

public class SessionReconnect {

    private static final Logger logger = Logger.getLogger(SessionReconnect.class.getName());

    private boolean shouldReconnect;
    private Credentials lastSuccessfulCredentials;
    protected int reconnectionAttempts;

    @Inject
    public SessionReconnect(final XmppConnection connection, final XmppSession session,
            final SASLManager saslManager) {
        shouldReconnect = false;
        reconnectionAttempts = 0;
        logger.info("RECONNECT BEHAVIOUR");

        saslManager.addAuthorizationResultHandler(new AuthorizationResultHandler() {
            @Override
            public void onAuthorization(final AuthorizationResultEvent event) {
                lastSuccessfulCredentials = event.getCredentials();
            }
        });

        session.addSessionStateChangedHandler(true, new StateChangedHandler() {
            @Override
            public void onStateChanged(final StateChangedEvent event) {
                if (event.is(SessionStates.connecting)) {
                    shouldReconnect = false;
                } else if (event.is(SessionStates.disconnected) && shouldReconnect) {
                    if (lastSuccessfulCredentials != null) {
                        final double seconds = Math.pow(2, reconnectionAttempts - 1);
                        new Timer() {
                            @Override
                            public void run() {
                                logger.info("Reconnecting...");
                                if (shouldReconnect) {
                                    shouldReconnect = false;
                                    session.login(lastSuccessfulCredentials);
                                }
                            }
                        }.schedule((int) (1000 * seconds));
                        logger.info("Reconnecting in " + seconds + " seconds.");
                    }
                } else if (event.is(SessionStates.ready)) {
                    logger.finer("CLEAR RECONNECTION ATTEMPS");
                    reconnectionAttempts = 0;
                }
            }
        });

        connection.addConnectionStateChangedHandler(new ConnectionStateChangedHandler() {
            @Override
            public void onStateChanged(final ConnectionStateChangedEvent event) {
                if (event.is(ConnectionState.error) || event.is(ConnectionState.waitingForRetry)) {
                    shouldReconnect();
                }
            }
        });

    }

    private void shouldReconnect() {
        shouldReconnect = true;
        reconnectionAttempts++;
    }
}