1 /*
2  * JBoss, Home of Professional Open Source.
3  * Copyright 2014 Red Hat, Inc., and individual contributors
4  * as indicated by the @author tags.
5  *
6  * Licensed under the Apache License, Version 2.0 (the "License");
7  * you may not use this file except in compliance with the License.
8  * You may obtain a copy of the License at
9  *
10  *     http://www.apache.org/licenses/LICENSE-2.0
11  *
12  *  Unless required by applicable law or agreed to in writing, software
13  *  distributed under the License is distributed on an "AS IS" BASIS,
14  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15  *  See the License for the specific language governing permissions and
16  *  limitations under the License.
17  */
18
19 package io.undertow.servlet.core;
20
21 import io.undertow.server.HttpServerExchange;
22 import io.undertow.servlet.api.ThreadSetupHandler;
23
24 /**
25  * @author Stuart Douglas
26  */
27 public class ContextClassLoaderSetupAction implements ThreadSetupHandler {
28
29     private final ClassLoader classLoader;
30
31     public ContextClassLoaderSetupAction(final ClassLoader classLoader) {
32         this.classLoader = classLoader;
33     }
34
35     @Override
36     public <T, C> Action<T, C> create(final Action<T, C> action) {
37         return new Action<T, C>() {
38             @Override
39             public T call(HttpServerExchange exchange, C context) throws Exception {
40                 final ClassLoader old = SecurityActions.getContextClassLoader();
41                 SecurityActions.setContextClassLoader(classLoader);
42                 try {
43                     return action.call(exchange, context);
44                 } finally {
45                     SecurityActions.setContextClassLoader(old);
46                 }
47             }
48         };
49     }
50 }
51