mirror of https://github.com/pf4j/pf4j.git
Decebal Suiu
11 years ago
16 changed files with 632 additions and 455 deletions
@ -0,0 +1,135 @@
|
||||
/* |
||||
* Copyright 2013 Decebal Suiu |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this work except in compliance with |
||||
* the License. You may obtain a copy of the License in the LICENSE file, or 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 ro.fortsoft.pf4j; |
||||
|
||||
import java.io.BufferedReader; |
||||
import java.io.FileNotFoundException; |
||||
import java.io.IOException; |
||||
import java.io.Reader; |
||||
import java.io.Writer; |
||||
import java.util.ArrayList; |
||||
import java.util.HashSet; |
||||
import java.util.List; |
||||
import java.util.Set; |
||||
|
||||
import javax.annotation.processing.AbstractProcessor; |
||||
import javax.annotation.processing.RoundEnvironment; |
||||
import javax.lang.model.SourceVersion; |
||||
import javax.lang.model.element.Element; |
||||
import javax.lang.model.element.TypeElement; |
||||
import javax.tools.Diagnostic.Kind; |
||||
import javax.tools.FileObject; |
||||
import javax.tools.StandardLocation; |
||||
|
||||
/** |
||||
* @author Decebal Suiu |
||||
*/ |
||||
public class ExtensionsIndexer extends AbstractProcessor { |
||||
|
||||
public static final String EXTENSIONS_RESOURCE = "META-INF/extensions.idx"; |
||||
|
||||
private List<TypeElement> extensions = new ArrayList<TypeElement>(); |
||||
|
||||
@Override |
||||
public SourceVersion getSupportedSourceVersion() { |
||||
return SourceVersion.latest(); |
||||
} |
||||
|
||||
@Override |
||||
public Set<String> getSupportedAnnotationTypes() { |
||||
Set<String> annotationTypes = new HashSet<String>(); |
||||
annotationTypes.add(Extension.class.getName()); |
||||
|
||||
return annotationTypes; |
||||
} |
||||
|
||||
@Override |
||||
public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) { |
||||
if (roundEnv.processingOver()) { |
||||
return false; |
||||
} |
||||
|
||||
for (Element element : roundEnv.getElementsAnnotatedWith(Extension.class)) { |
||||
if (!(element instanceof TypeElement)) { |
||||
continue; |
||||
} |
||||
|
||||
TypeElement typeElement = (TypeElement) element; |
||||
String message = "Extension found in " + processingEnv.getElementUtils().getBinaryName(typeElement).toString(); |
||||
processingEnv.getMessager().printMessage(Kind.NOTE, message); |
||||
extensions.add(typeElement); |
||||
} |
||||
|
||||
/* |
||||
if (!roundEnv.processingOver()) { |
||||
return false; |
||||
} |
||||
*/ |
||||
|
||||
write(); |
||||
|
||||
return false; |
||||
// return true; // no further processing of this annotation type
|
||||
} |
||||
|
||||
private void write() { |
||||
Set<String> entries = new HashSet<String>(); |
||||
for (TypeElement typeElement : extensions) { |
||||
entries.add(processingEnv.getElementUtils().getBinaryName(typeElement).toString()); |
||||
} |
||||
|
||||
read(entries); // read old entries
|
||||
write(entries); // write entries
|
||||
} |
||||
|
||||
private void write(Set<String> entries) { |
||||
try { |
||||
FileObject file = processingEnv.getFiler().createResource(StandardLocation.CLASS_OUTPUT, "", EXTENSIONS_RESOURCE); |
||||
Writer writer = file.openWriter(); |
||||
for (String entry : entries) { |
||||
writer.write(entry); |
||||
writer.write("\n"); |
||||
} |
||||
writer.close(); |
||||
} catch (FileNotFoundException e) { |
||||
// it's the first time, create the file
|
||||
} catch (IOException e) { |
||||
processingEnv.getMessager().printMessage(Kind.ERROR, e.toString()); |
||||
} |
||||
} |
||||
|
||||
private void read(Set<String> entries) { |
||||
try { |
||||
FileObject file = processingEnv.getFiler().getResource(StandardLocation.CLASS_OUTPUT, "", EXTENSIONS_RESOURCE); |
||||
readIndex(file.openReader(true), entries); |
||||
} catch (FileNotFoundException e) { |
||||
} catch (IOException e) { |
||||
// thrown by Eclipse JDT when not found
|
||||
} catch (UnsupportedOperationException e) { |
||||
// java6 does not support reading old index files
|
||||
} |
||||
} |
||||
|
||||
public static void readIndex(Reader reader, Set<String> entries) throws IOException { |
||||
BufferedReader bufferedReader = new BufferedReader(reader); |
||||
|
||||
String line; |
||||
while ((line = bufferedReader.readLine()) != null) { |
||||
entries.add(line); |
||||
} |
||||
|
||||
reader.close(); |
||||
} |
||||
|
||||
} |
||||
|
||||
@ -1,80 +0,0 @@
|
||||
/* |
||||
* Copyright 2012 Decebal Suiu |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this work except in compliance with |
||||
* the License. You may obtain a copy of the License in the LICENSE file, or 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 ro.fortsoft.pf4j; |
||||
|
||||
import java.lang.reflect.AnnotatedElement; |
||||
import java.util.ArrayList; |
||||
import java.util.Iterator; |
||||
import java.util.List; |
||||
|
||||
import org.slf4j.Logger; |
||||
import org.slf4j.LoggerFactory; |
||||
|
||||
import net.java.sezpoz.Index; |
||||
import net.java.sezpoz.IndexItem; |
||||
|
||||
/** |
||||
* Using Sezpoz(http://sezpoz.java.net/) for extensions discovery.
|
||||
* |
||||
* @author Decebal Suiu |
||||
*/ |
||||
public class SezpozExtensionFinder implements ExtensionFinder { |
||||
|
||||
private static final Logger log = LoggerFactory.getLogger(SezpozExtensionFinder.class); |
||||
|
||||
private volatile List<IndexItem<Extension, Object>> indices; |
||||
private ClassLoader classLoader; |
||||
|
||||
public SezpozExtensionFinder(ClassLoader classLoader) { |
||||
this.classLoader = classLoader; |
||||
} |
||||
|
||||
@Override |
||||
public <T> List<ExtensionWrapper<T>> find(Class<T> type) { |
||||
log.debug("Find extensions for {}", type); |
||||
List<ExtensionWrapper<T>> result = new ArrayList<ExtensionWrapper<T>>(); |
||||
getIndices(); |
||||
// System.out.println("indices = "+ indices);
|
||||
for (IndexItem<Extension, Object> item : indices) { |
||||
try { |
||||
AnnotatedElement element = item.element(); |
||||
Class<?> extensionType = (Class<?>) element; |
||||
log.debug("Checking extension type {}", extensionType); |
||||
if (type.isAssignableFrom(extensionType)) { |
||||
Object instance = item.instance(); |
||||
if (instance != null) { |
||||
log.debug("Added extension {}", extensionType); |
||||
result.add(new ExtensionWrapper<T>(type.cast(instance), item.annotation().ordinal())); |
||||
} |
||||
} |
||||
} catch (InstantiationException e) { |
||||
log.error(e.getMessage(), e); |
||||
} |
||||
} |
||||
|
||||
return result; |
||||
} |
||||
|
||||
private List<IndexItem<Extension, Object>> getIndices() { |
||||
if (indices == null) { |
||||
indices = new ArrayList<IndexItem<Extension, Object>>(); |
||||
Iterator<IndexItem<Extension, Object>> it = Index.load(Extension.class, Object.class, classLoader).iterator(); |
||||
while (it.hasNext()) { |
||||
indices.add(it.next()); |
||||
} |
||||
} |
||||
|
||||
return indices; |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1 @@
|
||||
ro.fortsoft.pf4j.ExtensionsIndexer |
||||
Loading…
Reference in new issue