package com.mycompany.jdk16annotation;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;
import javax.annotation.processing.AbstractProcessor;
import javax.annotation.processing.RoundEnvironment;
import javax.annotation.processing.SupportedSourceVersion;
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;
@SupportedSourceVersion(SourceVersion.RELEASE_6)
public class ServiceProviderProcessor extends AbstractProcessor {
public @Override Set<String> getSupportedAnnotationTypes() {
return new HashSet<String>(Arrays.asList(
ServiceProvider.class.getCanonicalName()
));
}
/** public for ServiceLoader */
public ServiceProviderProcessor() {}
@Override
public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
if (roundEnv.errorRaised()) {
return false;
}
if (roundEnv.processingOver()) {
writeServices();
return true;
} else {
return true;
}
}
private void writeServices() {
try {
FileObject out = processingEnv.getFiler().createResource(StandardLocation.CLASS_OUTPUT, "", "META-INF/one",new Element[0]);
OutputStream os = out.openOutputStream();
OutputStream os2 = processingEnv.getFiler().createSourceFile("org.Milos", new Element[0]).openOutputStream();
OutputStreamWriter osr = new OutputStreamWriter(os2);
try {
PrintWriter w = new PrintWriter(new OutputStreamWriter(os, "UTF-8"));
w.write("test");
w.flush();
String clazz = "package org;\n class Milos {}";
osr.write(clazz.toCharArray());
osr.flush();
} finally {
osr.close();
os.close();
}
} catch (IOException x) {
processingEnv.getMessager().printMessage(Kind.ERROR, "Failed to write to one: " + x.toString());
}
}
}