001    // Copyright 2009 The Apache Software Foundation
002    //
003    // Licensed under the Apache License, Version 2.0 (the "License");
004    // you may not use this file except in compliance with the License.
005    // You may obtain a copy of the License at
006    //
007    //     http://www.apache.org/licenses/LICENSE-2.0
008    //
009    // Unless required by applicable law or agreed to in writing, software
010    // distributed under the License is distributed on an "AS IS" BASIS,
011    // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
012    // See the License for the specific language governing permissions and
013    // limitations under the License.
014    
015    package org.apache.tapestry5.internal.test;
016    
017    import org.apache.tapestry5.services.RequestHandler;
018    import org.apache.tapestry5.services.Request;
019    import org.apache.tapestry5.services.Response;
020    import org.apache.tapestry5.services.RequestFilter;
021    import org.apache.tapestry5.ioc.services.PerthreadManager;
022    
023    import java.io.IOException;
024    
025    /**
026     * Makes sure that {@link org.apache.tapestry5.ioc.services.PerthreadManager#cleanup()} is invoked at the end of each
027     * request (normally handled by {@link org.apache.tapestry5.TapestryFilter}).
028     */
029    public class EndOfRequestCleanupFilter implements RequestFilter
030    {
031        private final PerthreadManager perThreadManager;
032    
033        public EndOfRequestCleanupFilter(PerthreadManager perThreadManager)
034        {
035            this.perThreadManager = perThreadManager;
036        }
037    
038        public boolean service(Request request, Response response, RequestHandler handler) throws IOException
039        {
040            try
041            {
042                return handler.service(request, response);
043            }
044            finally
045            {
046                perThreadManager.cleanup();
047            }
048        }
049    }