Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.atmosphere.cpr.DefaultAnnotationProcessor Maven / Gradle / Ivy
/*
* Copyright 2015 Async-IO.org
*
* Licensed 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.atmosphere.cpr;
import org.atmosphere.config.AtmosphereAnnotation;
import org.atmosphere.config.service.AsyncSupportListenerService;
import org.atmosphere.config.service.AsyncSupportService;
import org.atmosphere.config.service.AtmosphereFrameworkListenerService;
import org.atmosphere.config.service.AtmosphereHandlerService;
import org.atmosphere.config.service.AtmosphereInterceptorService;
import org.atmosphere.config.service.AtmosphereResourceFactoryService;
import org.atmosphere.config.service.AtmosphereResourceListenerService;
import org.atmosphere.config.service.AtmosphereService;
import org.atmosphere.config.service.BroadcasterCacheInspectorService;
import org.atmosphere.config.service.BroadcasterCacheListenerService;
import org.atmosphere.config.service.BroadcasterCacheService;
import org.atmosphere.config.service.BroadcasterFactoryService;
import org.atmosphere.config.service.BroadcasterFilterService;
import org.atmosphere.config.service.BroadcasterListenerService;
import org.atmosphere.config.service.BroadcasterService;
import org.atmosphere.config.service.EndpointMapperService;
import org.atmosphere.config.service.ManagedService;
import org.atmosphere.config.service.MeteorService;
import org.atmosphere.config.service.UUIDProviderService;
import org.atmosphere.config.service.WebSocketHandlerService;
import org.atmosphere.config.service.WebSocketProcessorService;
import org.atmosphere.config.service.WebSocketProtocolService;
import org.atmosphere.util.IOUtils;
import org.atmosphere.util.annotation.AnnotationDetector;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.servlet.ServletContext;
import javax.servlet.annotation.HandlesTypes;
import java.io.File;
import java.io.FilenameFilter;
import java.io.IOException;
import java.lang.annotation.Annotation;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.atomic.AtomicBoolean;
/**
* An {@link AnnotationProcessor} that selects between a ServletContextInitializer based scanner, and
* a bytecode based scanner based on
*
* @author Jeanfrancois Arcand
*/
public class DefaultAnnotationProcessor implements AnnotationProcessor {
private static final Logger logger = LoggerFactory.getLogger(DefaultAnnotationProcessor.class);
/**
* The attribute name under which the annotations are stored in the servlet context.
*/
public static final String ANNOTATION_ATTRIBUTE = "org.atmosphere.cpr.ANNOTATION_MAP";
// Annotation in java is broken.
private static final Class[] coreAnnotations = {
AtmosphereHandlerService.class,
BroadcasterCacheService.class,
BroadcasterFilterService.class,
BroadcasterFactoryService.class,
BroadcasterService.class,
MeteorService.class,
WebSocketHandlerService.class,
WebSocketProtocolService.class,
AtmosphereInterceptorService.class,
BroadcasterListenerService.class,
AsyncSupportService.class,
AsyncSupportListenerService.class,
WebSocketProcessorService.class,
BroadcasterCacheInspectorService.class,
ManagedService.class,
AtmosphereService.class,
EndpointMapperService.class,
BroadcasterCacheListenerService.class,
AtmosphereAnnotation.class,
AtmosphereResourceFactoryService.class,
AtmosphereFrameworkListenerService.class,
AtmosphereResourceListenerService.class,
UUIDProviderService.class
};
private AnnotationProcessor delegate;
private final AnnotationHandler handler;
private final AtomicBoolean coreAnnotationsFound = new AtomicBoolean();
private final AnnotationDetector.TypeReporter atmosphereReporter = new AnnotationDetector.TypeReporter() {
@SuppressWarnings("unchecked")
@Override
public Class[] annotations() {
return new Class[]{
AtmosphereAnnotation.class
};
}
@Override
public void reportTypeAnnotation(Class annotation, String className) {
try {
coreAnnotationsFound.set(true);
handler.handleProcessor(loadClass(getClass(), className));
} catch (Exception e) {
logger.warn("Error scanning @AtmosphereAnnotation", e);
}
}
};
public DefaultAnnotationProcessor() {
this.handler = new AnnotationHandler();
}
@Override
public void configure(final AtmosphereConfig config) {
ServletContext sc = config.framework().getServletContext();
Map, Set>> annotations= (Map, Set>>) sc.getAttribute(ANNOTATION_ATTRIBUTE);
sc.removeAttribute(ANNOTATION_ATTRIBUTE);
boolean useByteCodeProcessor = config.getInitParameter(ApplicationConfig.BYTECODE_PROCESSOR, false);
boolean scanForAtmosphereAnnotation = false;
if (useByteCodeProcessor || annotations == null || annotations.isEmpty()) {
delegate = new BytecodeBasedAnnotationProcessor(handler);
scanForAtmosphereAnnotation = true;
} else {
Map, Set>> clone = new HashMap, Set>>();
clone.putAll(annotations);
delegate = new ServletContainerInitializerAnnotationProcessor(handler, clone, config.framework());
}
logger.info("AnnotationProcessor {} being used", delegate.getClass());
if (scanForAtmosphereAnnotation) {
scanForAnnotation(config.framework());
}
delegate.configure(config.framework().getAtmosphereConfig());
}
private void scanForAnnotation(AtmosphereFramework f) {
List packages = f.customAnnotationPackages();
AnnotationDetector detector = new AnnotationDetector(atmosphereReporter);
try {
if (packages.size() > 0) {
for (String p : packages) {
logger.trace("Package {} scanned for @AtmosphereAnnotation", p);
detector.detect(p);
}
}
// Now look for application defined annotation
String path = IOUtils.realPath(f.getServletContext(), f.getHandlersPath());
if (path != null) {
detector.detect(new File(path));
}
String pathLibs = IOUtils.realPath(f.getServletContext(), f.getLibPath());
if (pathLibs != null) {
File libFolder = new File(pathLibs);
File jars[] = libFolder.listFiles(new FilenameFilter() {
@Override
public boolean accept(File arg0, String arg1) {
return arg1.endsWith(".jar");
}
});
if (jars != null) {
for (File file : jars) {
detector.detect(file);
}
}
}
// JBoss|vfs with APR issue, or any strange containers may fail. This is a hack for them.
// https://github.com/Atmosphere/atmosphere/issues/1292
if (!coreAnnotationsFound.get()) {
fallbackToManualAnnotatedClasses(getClass(), f, handler);
}
} catch (IOException e) {
logger.warn("Unable to scan annotation", e);
} finally {
detector.destroy();
}
}
private static void fallbackToManualAnnotatedClasses(Class mainClass, AtmosphereFramework f, AnnotationHandler handler) {
logger.warn("Unable to detect annotations. Application may fail to deploy.");
f.annotationScanned(true);
for (Class a : coreAnnotations) {
try {
handler.handleProcessor(loadClass(mainClass, a.getName()));
} catch (Exception e) {
logger.trace("", e);
}
}
}
@Override
public AnnotationProcessor scan(final File rootDir) throws IOException {
delegate.scan(rootDir);
return this;
}
@Override
public AnnotationProcessor scan(final String packageName) throws IOException {
delegate.scan(packageName);
return this;
}
@Override
public AnnotationProcessor scanAll() throws IOException {
delegate.scanAll();
return this;
}
@Override
public void destroy() {
if (delegate != null) {
delegate.destroy();
handler.destroy();
// Help GC with the annotations Map.
delegate = null;
}
}
private static final class ServletContainerInitializerAnnotationProcessor implements AnnotationProcessor {
private final Map, Set>> annotations;
private final AtmosphereFramework framework;
private final AnnotationHandler handler;
/**
* This is not great, but we can't differentiate based on the file, so we just do one scan
* of everything in the war. It would be nice to change to the API to make this a bit cleaner
* but it looks like it is a public API.
*/
private boolean alreadyScanned = false;
private ServletContainerInitializerAnnotationProcessor(AnnotationHandler handler,
final Map, Set>> annotations,
final AtmosphereFramework framework) {
this.annotations = annotations;
this.framework = framework;
this.handler = handler;
}
@Override
public void configure(final AtmosphereConfig config) {
}
@Override
public AnnotationProcessor scan(final File rootDir) throws IOException {
if (alreadyScanned) {
return this;
}
alreadyScanned = true;
Set> atmosphereAnnotatedClasses = annotations.get(AtmosphereAnnotation.class);
boolean handleAtmosphereAnnotation = handleAtmosphereAnnotation(atmosphereAnnotatedClasses);
for (Map.Entry, Set>> entry : annotations.entrySet()) {
for (Class clazz : entry.getValue()) {
handler.handleAnnotation(framework, entry.getKey(), clazz);
}
}
if (handleAtmosphereAnnotation) {
scanForCustomAnnotation();
}
return this;
}
private boolean handleAtmosphereAnnotation(Set> atmosphereAnnotatedClasses) {
boolean scanForCustomizedAnnotation = false;
if (atmosphereAnnotatedClasses != null) {
for (Class clazz : atmosphereAnnotatedClasses) {
handler.handleProcessor(clazz);
}
} else {
fallbackToManualAnnotatedClasses(getClass(),framework, handler);
}
// If larger, a custom annotation has been defined.
if (atmosphereAnnotatedClasses != null && atmosphereAnnotatedClasses.size() >=
AnnotationScanningServletContainerInitializer.class.getAnnotation(HandlesTypes.class).value().length) {
scanForCustomizedAnnotation = true;
}
return scanForCustomizedAnnotation;
}
private void scanForCustomAnnotation() throws IOException {
BytecodeBasedAnnotationProcessor b = new BytecodeBasedAnnotationProcessor(handler);
b.configure(framework.getAtmosphereConfig());
String path = framework.getServletContext().getRealPath(framework.getHandlersPath());
if (path != null) {
b.scan(new File(path)).destroy();
} else {
logger.warn("Unable to scan using File. Scanning classpath");
b.scanAll();
}
}
@Override
public AnnotationProcessor scan(final String packageName) throws IOException {
Set> atmosphereAnnotatedClasses = annotations.get(AtmosphereAnnotation.class);
if (packageName.equals("all") || getClass().getClassLoader().getResource(packageName.replace(".", "/")) != null) {
boolean handleAtmosphereAnnotation = handleAtmosphereAnnotation(atmosphereAnnotatedClasses);
for (Map.Entry, Set>> entry : annotations.entrySet()) {
for (Class clazz : entry.getValue()) {
if (packageName.equals("all") || clazz.getPackage().getName().startsWith(packageName)) {
handler.handleAnnotation(framework, entry.getKey(), clazz);
}
}
}
if (handleAtmosphereAnnotation) {
scanForCustomAnnotation();
}
}
return this;
}
@Override
public AnnotationProcessor scanAll() throws IOException {
return scan("all");
}
@Override
public void destroy() {
annotations.clear();
}
}
private static final class BytecodeBasedAnnotationProcessor implements AnnotationProcessor {
protected AnnotationDetector detector;
protected final AnnotationHandler handler;
public BytecodeBasedAnnotationProcessor(AnnotationHandler handler) {
this.handler = handler;
}
@Override
public void configure(final AtmosphereConfig config) {
final AnnotationDetector.TypeReporter reporter = new AnnotationDetector.TypeReporter() {
@SuppressWarnings("unchecked")
@Override
public Class[] annotations() {
return handler.handledClass();
}
@Override
public void reportTypeAnnotation(Class annotation, String className) {
try {
final Class discoveredClass = loadClass(getClass(), className);
handler.handleAnnotation(config.framework(), annotation, discoveredClass);
} catch (Exception e) {
logger.warn("Could not load discovered class", e);
}
}
};
detector = new AnnotationDetector(reporter);
}
@Override
public AnnotationProcessor scan(File rootDir) throws IOException {
detector.detect(rootDir);
return this;
}
@Override
public AnnotationProcessor scan(String packageName) throws IOException {
logger.trace("Scanning @Service annotations in {}", packageName);
detector.detect(packageName);
return this;
}
@Override
public AnnotationProcessor scanAll() throws IOException {
detector.detect();
return this;
}
@Override
public void destroy() {
if (detector != null) detector.destroy();
}
}
private static Class loadClass(Class thisClass, String className) throws Exception {
try {
return Thread.currentThread().getContextClassLoader().loadClass(className);
} catch (Throwable t) {
return thisClass.getClassLoader().loadClass(className);
}
}
}
