Mercurial Hosting > luan
diff src/org/eclipse/jetty/server/handler/IdleTimeoutHandler.java @ 802:3428c60d7cfc
replace jetty jars with source
author | Franklin Schmidt <fschmidt@gmail.com> |
---|---|
date | Wed, 07 Sep 2016 21:15:48 -0600 |
parents | |
children | 3268ddf919d4 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/org/eclipse/jetty/server/handler/IdleTimeoutHandler.java Wed Sep 07 21:15:48 2016 -0600 @@ -0,0 +1,138 @@ +// +// ======================================================================== +// Copyright (c) 1995-2014 Mort Bay Consulting Pty. Ltd. +// ------------------------------------------------------------------------ +// 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 org.eclipse.jetty.server.handler; + +import java.io.IOException; + +import javax.servlet.AsyncEvent; +import javax.servlet.AsyncListener; +import javax.servlet.ServletException; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; + +import org.eclipse.jetty.io.EndPoint; +import org.eclipse.jetty.server.AbstractHttpConnection; +import org.eclipse.jetty.server.Request; + +/** + * Handler to adjust the idle timeout of requests while dispatched. + * + * <p>Can be applied in jetty.xml with + * <pre> + * <Get id='handler' name='Handler'/> + * <Set name='Handler'> + * <New id='idleTimeoutHandler' class='org.eclipse.jetty.server.handler.IdleTimeoutHandler'> + * <Set name='Handler'><Ref id='handler'/></Set> + * <Set name='IdleTimeoutMs'>5000</Set> + * </New> + * </Set> + * </pre> + */ +public class IdleTimeoutHandler extends HandlerWrapper +{ + private int _idleTimeoutMs = 1000; + private boolean _applyToAsync = false; + + + public boolean isApplyToAsync() + { + return _applyToAsync; + } + + /** + * Should the adjusted idle time be maintained for asynchronous requests + * @param applyToAsync true if alternate idle timeout is applied to asynchronous requests + */ + public void setApplyToAsync(boolean applyToAsync) + { + _applyToAsync = applyToAsync; + } + + public long getIdleTimeoutMs() + { + return _idleTimeoutMs; + } + + /** + * @param idleTimeoutMs The idle timeout in MS to apply while dispatched or async + */ + public void setIdleTimeoutMs(int _idleTimeoutMs) + { + this._idleTimeoutMs = _idleTimeoutMs; + } + + + @Override + public void handle(String target, Request baseRequest, HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException + { + AbstractHttpConnection connection = AbstractHttpConnection.getCurrentConnection(); + final EndPoint endp = connection==null?null:connection.getEndPoint(); + + final int idle_timeout; + if (endp==null) + idle_timeout=-1; + else + { + idle_timeout=endp.getMaxIdleTime(); + endp.setMaxIdleTime(_idleTimeoutMs); + } + + try + { + super.handle(target,baseRequest,request,response); + } + finally + { + if (endp!=null) + { + if (_applyToAsync && request.isAsyncStarted()) + { + request.getAsyncContext().addListener(new AsyncListener() + { + @Override + public void onTimeout(AsyncEvent event) throws IOException + { + } + + @Override + public void onStartAsync(AsyncEvent event) throws IOException + { + } + + @Override + public void onError(AsyncEvent event) throws IOException + { + endp.setMaxIdleTime(idle_timeout); + } + + @Override + public void onComplete(AsyncEvent event) throws IOException + { + endp.setMaxIdleTime(idle_timeout); + } + }); + } + else + { + endp.setMaxIdleTime(idle_timeout); + } + } + } + } +}