/*
* Copyright 2005-2014 Red Hat, Inc.
* Red Hat licenses this file to you 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.hornetq.core.remoting.impl.netty;
import io.netty.buffer.ByteBuf;
import io.netty.channel.Channel;
import io.netty.channel.ChannelDuplexHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.group.ChannelGroup;
import org.hornetq.api.core.HornetQException;
import org.hornetq.core.buffers.impl.ChannelBufferWrapper;
import org.hornetq.core.client.HornetQClientLogger;
import org.hornetq.core.client.HornetQClientMessageBundle;
import org.hornetq.spi.core.remoting.BufferHandler;
import org.hornetq.spi.core.remoting.ConnectionLifeCycleListener;
/**
* Common handler implementation for client and server side handler.
*
* @author <a href="mailto:tlee@redhat.com">Trustin Lee</a>
* @author <a href="mailto:nmaurer@redhat.com">Norman Maurer</a>
* @version $Rev$, $Date$
*/
public class HornetQChannelHandler extends ChannelDuplexHandler
{
private final ChannelGroup group;
private final BufferHandler handler;
private final ConnectionLifeCycleListener listener;
volatile boolean active;
protected HornetQChannelHandler(final ChannelGroup group,
final BufferHandler handler,
final ConnectionLifeCycleListener listener)
{
this.group = group;
this.handler = handler;
this.listener = listener;
}
@Override
public void channelActive(final ChannelHandlerContext ctx) throws Exception
{
group.add(ctx.channel());
ctx.fireChannelActive();
}
@Override
public void channelWritabilityChanged(ChannelHandlerContext ctx) throws Exception
{
// TODO: Think about the id thingy
listener.connectionReadyForWrites(channelId(ctx.channel()), ctx.channel().isWritable());
}
@Override
public void channelRead(final ChannelHandlerContext ctx, final Object msg) throws Exception
{
ByteBuf buffer = (ByteBuf) msg;
handler.bufferReceived(channelId(ctx.channel()), new ChannelBufferWrapper(buffer));
}
@Override
public void channelInactive(final ChannelHandlerContext ctx) throws Exception
{
synchronized (this)
{
if (active)
{
listener.connectionDestroyed(channelId(ctx.channel()));
active = false;
}
}
}
@Override
public void exceptionCaught(final ChannelHandlerContext ctx, final Throwable cause) throws Exception
{
if (!active)
{
return;
}
// We don't want to log this - since it is normal for this to happen during failover/reconnect
// and we don't want to spew out stack traces in that event
// The user has access to this exeception anyway via the HornetQException initial cause
HornetQException me = HornetQClientMessageBundle.BUNDLE.nettyError();
me.initCause(cause);
synchronized (listener)
{
try
{
listener.connectionException(channelId(ctx.channel()), me);
active = false;
}
catch (Exception ex)
{
HornetQClientLogger.LOGGER.errorCallingLifeCycleListener(ex);
}
}
}
protected static int channelId(Channel channel)
{
return channel.hashCode();
}
}