/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.openejb.cdi;
import org.apache.openejb.util.LogCategory;
import org.apache.openejb.util.Logger;
import org.apache.openejb.util.classloader.ClassLoaderAwareHandler;
import org.apache.webbeans.service.DefaultLoaderService;
import org.apache.webbeans.spi.LoaderService;
import org.apache.webbeans.spi.plugins.OpenWebBeansPlugin;
import javax.enterprise.inject.spi.Extension;
import java.lang.reflect.Proxy;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;
/**
* @version $Rev$ $Date$
*/
public class OptimizedLoaderService implements LoaderService {
private static final Logger log = Logger.getInstance(LogCategory.OPENEJB.createChild("cdi"), OptimizedLoaderService.class);
public static final ThreadLocal<Collection<String>> ADDITIONAL_EXTENSIONS = new ThreadLocal<Collection<String>>();
private final LoaderService loaderService;
public OptimizedLoaderService() {
this(new DefaultLoaderService());
}
public OptimizedLoaderService(final LoaderService loaderService) {
this.loaderService = loaderService;
}
@Override
public <T> List<T> load(final Class<T> serviceType) {
return load(serviceType, Thread.currentThread().getContextClassLoader());
}
@Override
public <T> List<T> load(final Class<T> serviceType, final ClassLoader classLoader) {
// ServiceLoader is expensive (can take up to a half second). This is an optimization
if (OpenWebBeansPlugin.class.equals(serviceType)) {
return loadWebBeansPlugins(classLoader);
}
// As far as we know, this only is reached for CDI Extension discovery
if (Extension.class.equals(serviceType)) {
return (List<T>) loadExtensions(classLoader);
}
return loaderService.load(serviceType, classLoader);
}
protected List<? extends Extension> loadExtensions(final ClassLoader classLoader) {
final List<Extension> list = loaderService.load(Extension.class, classLoader);
final Collection<String> additional = ADDITIONAL_EXTENSIONS.get();
if (additional != null) {
for (final String name : additional) {
try {
list.add(Extension.class.cast(classLoader.loadClass(name).newInstance()));
} catch (final Exception ignored) {
// no-op
}
}
}
final Collection<Extension> extensionCopy = new ArrayList<>(list);
final Iterator<Extension> it = list.iterator();
while (it.hasNext()) {
if (it.hasNext()) {
if (isFiltered(extensionCopy, it.next())) {
it.remove();
}
}
}
return list;
}
// mainly intended to avoid conflicts between internal and overrided spec extensions
private boolean isFiltered(final Collection<Extension> extensions, final Extension next) {
final String name = next.getClass().getName();
switch (name) {
case "org.apache.bval.cdi.BValExtension":
for (final Extension e : extensions) {
final String en = e.getClass().getName();
// org.hibernate.validator.internal.cdi.ValidationExtension but allowing few evolutions of packages
if (en.startsWith("org.hibernate.validator.") && en.endsWith("ValidationExtension")) {
return true;
}
}
break;
default:
}
return false;
}
private <T> List<T> loadWebBeansPlugins(final ClassLoader loader) {
final String[] knownPlugins = {
"org.apache.openejb.cdi.CdiPlugin",
"org.apache.geronimo.openejb.cdi.GeronimoWebBeansPlugin"
};
final String[] loaderAwareKnownPlugins = {
"org.apache.webbeans.jsf.plugin.OpenWebBeansJsfPlugin"
};
final List<T> list = new ArrayList<T>();
for (final String name : knownPlugins) {
final Class<T> clazz;
try {
clazz = (Class<T>) loader.loadClass(name);
} catch (final ClassNotFoundException e) {
// ignore
continue;
}
try {
list.add(clazz.newInstance());
} catch (final Exception e) {
log.error("Unable to load OpenWebBeansPlugin: " + name);
}
}
for (final String name : loaderAwareKnownPlugins) {
final Class<T> clazz;
try {
clazz = (Class<T>) loader.loadClass(name);
} catch (final ClassNotFoundException e) {
// ignore
continue;
}
try {
list.add((T) Proxy.newProxyInstance(loader, new Class<?>[]{OpenWebBeansPlugin.class}, new ClassLoaderAwareHandler(clazz.getSimpleName(), clazz.newInstance(), loader)));
} catch (final Exception e) {
log.error("Unable to load OpenWebBeansPlugin: " + name);
}
}
return list;
}
}