1 /*
2  * JBoss, Home of Professional Open Source.
3  * Copyright 2017 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 org.jboss.threads;
20
21 class ContextClassLoaderSavingRunnable implements Runnable {
22
23     private final ClassLoader loader;
24     private final Runnable delegate;
25
26     ContextClassLoaderSavingRunnable(final ClassLoader loader, final Runnable delegate) {
27         this.loader = loader;
28         this.delegate = delegate;
29     }
30
31     public void run() {
32         final Thread currentThread = Thread.currentThread();
33         final ClassLoader old = JBossExecutors.getAndSetContextClassLoader(currentThread, loader);
34         try {
35             delegate.run();
36         } finally {
37             JBossExecutors.setContextClassLoader(currentThread, old);
38         }
39     }
40
41     public String toString() {
42         return "Context class loader saving " + delegate.toString();
43     }
44 }
45