001    /**
002     *  Licensed to the Apache Software Foundation (ASF) under one or more
003     *  contributor license agreements.  See the NOTICE file distributed with
004     *  this work for additional information regarding copyright ownership.
005     *  The ASF licenses this file to You under the Apache License, Version 2.0
006     *  (the "License"); you may not use this file except in compliance with
007     *  the License.  You may obtain a copy of the License at
008     *
009     *     http://www.apache.org/licenses/LICENSE-2.0
010     *
011     *  Unless required by applicable law or agreed to in writing, software
012     *  distributed under the License is distributed on an "AS IS" BASIS,
013     *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014     *  See the License for the specific language governing permissions and
015     *  limitations under the License.
016     */
017    package org.apache.geronimo.tomcat.interceptor;
018    
019    import javax.naming.Context;
020    import javax.servlet.ServletRequest;
021    import javax.servlet.ServletResponse;
022    
023    import org.apache.geronimo.naming.java.RootContext;
024    
025    public class ComponentContextBeforeAfter implements BeforeAfter{
026        private final BeforeAfter next;
027        private final int index;
028        private final Context componentContext;
029    
030        public ComponentContextBeforeAfter(BeforeAfter next, int index, Context componentContext) {
031            this.next = next;
032            this.index = index;
033            this.componentContext = componentContext;
034        }
035    
036        public void before(Object[] context, ServletRequest httpRequest, ServletResponse httpResponse, int dispatch) {
037            context[index] = RootContext.getComponentContext();
038            RootContext.setComponentContext(componentContext);
039            if (next != null) {
040                next.before(context, httpRequest, httpResponse, 0);
041            }
042        }
043    
044        public void after(Object[] context, ServletRequest httpRequest, ServletResponse httpResponse, int dispatch) {
045            if (next != null) {
046                next.after(context, httpRequest, httpResponse, 0);
047            }
048            RootContext.setComponentContext((Context) context[index]);
049        }
050    
051    }