diff options
| -rw-r--r-- | bcel-builder/src/main/java/org/aspectj/apache/bcel/util/ThreadLocalAwareRepository.java | 111 |
1 files changed, 111 insertions, 0 deletions
diff --git a/bcel-builder/src/main/java/org/aspectj/apache/bcel/util/ThreadLocalAwareRepository.java b/bcel-builder/src/main/java/org/aspectj/apache/bcel/util/ThreadLocalAwareRepository.java new file mode 100644 index 000000000..414a4b9af --- /dev/null +++ b/bcel-builder/src/main/java/org/aspectj/apache/bcel/util/ThreadLocalAwareRepository.java @@ -0,0 +1,111 @@ + +/* ==================================================================== + * The Apache Software License, Version 1.1 + * + * Copyright (c) 2001 The Apache Software Foundation. All rights + * reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in + * the documentation and/or other materials provided with the + * distribution. + * + * 3. The end-user documentation included with the redistribution, + * if any, must include the following acknowledgment: + * "This product includes software developed by the + * Apache Software Foundation (http://www.apache.org/)." + * Alternately, this acknowledgment may appear in the software itself, + * if and wherever such third-party acknowledgments normally appear. + * + * 4. The names "Apache" and "Apache Software Foundation" and + * "Apache BCEL" must not be used to endorse or promote products + * derived from this software without prior written permission. For + * written permission, please contact apache@apache.org. + * + * 5. Products derived from this software may not be called "Apache", + * "Apache BCEL", nor may "Apache" appear in their name, without + * prior written permission of the Apache Software Foundation. + * + * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED + * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES + * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR + * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, + * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT + * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF + * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND + * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, + * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT + * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF + * SUCH DAMAGE. + * ==================================================================== + * + * This software consists of voluntary contributions made by many + * individuals on behalf of the Apache Software Foundation. For more + * information on the Apache Software Foundation, please see + * <http://www.apache.org/>. + */ +package org.aspectj.apache.bcel.util; + +import java.lang.ref.WeakReference; + +import org.aspectj.apache.bcel.classfile.JavaClass; + +public class ThreadLocalAwareRepository implements Repository { + + private static ThreadLocal<WeakReference<Repository>> threadLocal = new ThreadLocal<>(); + + public static void setCurrentRepository(Repository repository) { + threadLocal.set(new WeakReference<>(repository)); + } + + private static Repository currentRepository() { + WeakReference<Repository> ref = threadLocal.get(); + Repository repo = null; + if (ref != null) { + repo = ref.get(); + } + if (repo == null) { + repo = SyntheticRepository.getInstance(); + } + return repo; + } + + @Override + public void storeClass(JavaClass clazz) { + currentRepository().storeClass(clazz); + } + + @Override + public void removeClass(JavaClass clazz) { + currentRepository().removeClass(clazz); + } + + @Override + public JavaClass findClass(String className) { + return currentRepository().findClass(className); + } + + @Override + public JavaClass loadClass(String className) throws ClassNotFoundException { + return currentRepository().loadClass(className); + } + + @Override + public JavaClass loadClass(Class clazz) throws ClassNotFoundException { + return currentRepository().loadClass(clazz); + } + + @Override + public void clear() { + currentRepository().clear(); + } + +} |