/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF 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 com.alibaba.jstorm.message.netty;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import org.jboss.netty.channel.Channel;
import org.jboss.netty.channel.ChannelHandlerContext;
import org.jboss.netty.channel.ChannelStateEvent;
import org.jboss.netty.channel.ChildChannelStateEvent;
import org.jboss.netty.channel.ExceptionEvent;
import org.jboss.netty.channel.MessageEvent;
import org.jboss.netty.channel.SimpleChannelUpstreamHandler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import backtype.storm.messaging.ControlMessage;
import backtype.storm.messaging.TaskMessage;
class StormServerHandler extends SimpleChannelUpstreamHandler {
private static final Logger LOG = LoggerFactory.getLogger(StormServerHandler.class);
private NettyServer server;
private Map<Channel, Integer> failureCounters;
StormServerHandler(NettyServer server) {
this.server = server;
failureCounters = new ConcurrentHashMap<>();
}
protected int getFailureCounter(Channel channel) {
Integer num = failureCounters.get(channel);
if (num == null) {
return 0;
}
return num;
}
protected void incFailureCounter(Channel channel) {
Integer num = failureCounters.get(channel);
if (num == null) {
num = 0;
}
num = num + 1;
failureCounters.put(channel, num);
}
protected void removeFailureCounter(Channel channel) {
failureCounters.remove(channel);
}
@Override
public void channelConnected(ChannelHandlerContext ctx, ChannelStateEvent e) {
LOG.info("Connection established {}", e.getChannel().getRemoteAddress());
server.addChannel(e.getChannel());
}
@Override
public void childChannelClosed(ChannelHandlerContext ctx, ChildChannelStateEvent e) throws Exception {
super.childChannelClosed(ctx, e);
LOG.info("Connection closed {}", e.getChildChannel().getRemoteAddress());
MessageDecoder.removeTransmitHistogram(e.getChildChannel());
}
@Override
public void channelDisconnected(ChannelHandlerContext ctx, ChannelStateEvent e) throws Exception {
super.channelDisconnected(ctx, e);
LOG.info("Connection channelDisconnected {}", e.getChannel().getRemoteAddress());
MessageDecoder.removeTransmitHistogram(e.getChannel());
server.getChannelGroup().remove(e.getChannel());
}
@Override
public void channelClosed(ChannelHandlerContext ctx, ChannelStateEvent e) throws Exception {
super.channelClosed(ctx, e);
LOG.info("Connection channelClosed {}", e.getChannel().getRemoteAddress());
MessageDecoder.removeTransmitHistogram(e.getChannel());
}
@Override
public void messageReceived(ChannelHandlerContext ctx, MessageEvent e) {
Object msg = e.getMessage();
if (msg == null)
return;
// end of batch?
if (msg == ControlMessage.EOB_MESSAGE) {
return;
} else if (msg instanceof ControlMessage) {
// LOG.debug("Receive ...{}", msg);
return;
}
// enqueue the received message for processing
try {
server.enqueue((TaskMessage) msg, ctx.getChannel());
} catch (Exception e1) {
LOG.warn("Failed to enqueue a request message " + e.toString(), e1);
// Channel channel = ctx.getChannel();
// incFailureCounter(channel);
}
}
@Override
public void exceptionCaught(ChannelHandlerContext ctx, ExceptionEvent e) {
// removeFailureCounter(e.getChannel());
if (e.getChannel() != null) {
LOG.info("Channel occur exception {}, cause={}", e.getChannel().getRemoteAddress(), e.getCause());
}
server.closeChannel(e.getChannel());
}
}