com.hotels.styx.server.netty.handlers.RequestTimeoutHandler Maven / Gradle / Ivy
/**
* Copyright (C) 2013-2017 Expedia Inc.
*
* 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 com.hotels.styx.server.netty.handlers;
import com.hotels.styx.server.RequestTimeoutException;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;
import io.netty.handler.codec.http.HttpRequest;
import io.netty.handler.codec.http.LastHttpContent;
import io.netty.handler.timeout.IdleState;
import io.netty.handler.timeout.IdleStateEvent;
/**
* Sends a 408 Request Timeout response if a request times out.
*
*/
public class RequestTimeoutHandler extends ChannelInboundHandlerAdapter {
private boolean requestOngoing;
private Object msg;
@Override
public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
if (msg instanceof HttpRequest && ((HttpRequest) msg).getDecoderResult().isSuccess()) {
requestOngoing = true;
this.msg = msg;
} else if (msg instanceof LastHttpContent) {
requestOngoing = false;
}
super.channelRead(ctx, msg);
}
@Override
public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
if (evt instanceof IdleStateEvent) {
IdleStateEvent e = (IdleStateEvent) evt;
if (e.state() == IdleState.READER_IDLE && requestOngoing) {
throw new RequestTimeoutException("message=" + String.valueOf(msg));
}
}
super.userEventTriggered(ctx, evt);
}
}