org.rzo.netty.ahessian.heartbeat.HeartbeatHandlerInbound.java Source code

Introduction

Here is the source code for org.rzo.netty.ahessian.heartbeat.HeartbeatHandlerInbound.java

Source

/*******************************************************************************
 * Copyright  2015 rzorzorzo@users.sf.net
 *
 * 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.rzo.netty.ahessian.heartbeat;

import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;
import io.netty.util.Timeout;
import io.netty.util.Timer;
import io.netty.util.TimerTask;

import java.util.concurrent.atomic.AtomicLong;

import org.rzo.netty.ahessian.Constants;

public class HeartbeatHandlerInbound extends ChannelInboundHandlerAdapter {
    volatile AtomicLong _lastCalled = new AtomicLong();
    volatile ChannelHandlerContext _ctx;
    final IntervalTimer _intervalTimer;
    final String _name;
    TimedOutAction _action;

    public HeartbeatHandlerInbound(final String name, final Timer timer, final long timeout) {
        _name = name;
        final TimerTask task = new TimerTask() {
            public void run(Timeout nTimeout) throws Exception {
                if (((getLastCalled() + timeout) <= System.currentTimeMillis()) && isConnected())
                    try {
                        _action.timedOut(_ctx);
                    } catch (Exception e) {
                        Constants.ahessianLogger.warn("", e);
                    }
            }

        };
        _intervalTimer = new IntervalTimer(timer, task, timeout);
    }

    long getLastCalled() {
        return _lastCalled.get();
    }

    boolean isConnected() {
        return _ctx != null && _ctx.channel().isActive();
    }

    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        _ctx = null;
        _intervalTimer.stop();
        ctx.fireChannelInactive();
    }

    protected void ping() {
        _lastCalled.set(System.currentTimeMillis());
    }

    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        ping();
        _ctx = ctx;
        _intervalTimer.setName(_name + ":" + _ctx.channel().hashCode());

        Constants.ahessianLogger
                .info("AbstractHeartBeatHandler scheduler started: " + _intervalTimer.getInterval());
        _intervalTimer.start();
        ctx.fireChannelActive();
    }

    public void setAction(TimedOutAction action) {
        _action = action;
    }

}