io.vertx.ext.web.handler.impl.TimeoutHandlerImpl Maven / Gradle / Ivy
/*
* Copyright 2014 Red Hat, Inc.
*
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* and Apache License v2.0 which accompanies this distribution.
*
* The Eclipse Public License is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* The Apache License v2.0 is available at
* http://www.opensource.org/licenses/apache2.0.php
*
* You may elect to redistribute this code under either of these licenses.
*/
package io.vertx.ext.web.handler.impl;
import io.vertx.core.Timer;
import io.vertx.ext.web.RoutingContext;
import io.vertx.ext.web.handler.TimeoutHandler;
/**
* @author Paulo Lopes
* @author Tim Fox
*
*/
public class TimeoutHandlerImpl implements TimeoutHandler {
private final long timeout;
private final int errorCode;
public TimeoutHandlerImpl(long timeout, int errorCode) {
this.timeout = timeout;
this.errorCode = errorCode;
}
@Override
public void handle(RoutingContext ctx) {
Timer timer = ctx.vertx().timer(timeout);
int handlerId = ctx.addBodyEndHandler(v -> timer.cancel());
timer.onSuccess(v -> {
// If this router has been restarted, the timeout shouldn't fire
if (ctx.removeBodyEndHandler(handlerId)) {
if (!ctx.request().isEnded()) {
ctx.request().resume();
}
ctx.fail(errorCode);
}
});
ctx.next();
}
}