| /* |
| * |
| * 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.qpid.server.model; |
| |
| import java.io.IOException; |
| import java.io.OutputStreamWriter; |
| import java.io.PrintWriter; |
| import java.util.Collections; |
| import java.util.Set; |
| |
| import javax.annotation.processing.AbstractProcessor; |
| import javax.annotation.processing.Filer; |
| import javax.annotation.processing.RoundEnvironment; |
| import javax.lang.model.SourceVersion; |
| import javax.lang.model.element.Element; |
| import javax.lang.model.element.ElementKind; |
| import javax.lang.model.element.ExecutableElement; |
| import javax.lang.model.element.PackageElement; |
| import javax.lang.model.element.TypeElement; |
| import javax.lang.model.element.VariableElement; |
| import javax.lang.model.type.TypeMirror; |
| import javax.tools.Diagnostic; |
| import javax.tools.JavaFileObject; |
| |
| import org.apache.qpid.server.License; |
| |
| public class ConfiguredObjectFactoryGenerator extends AbstractProcessor |
| { |
| @Override |
| public SourceVersion getSupportedSourceVersion() |
| { |
| return SourceVersion.latest(); |
| } |
| |
| @Override |
| public Set<String> getSupportedAnnotationTypes() |
| { |
| return Collections.singleton(ManagedObjectFactoryConstructor.class.getName()); |
| } |
| |
| @Override |
| public boolean process(final Set<? extends TypeElement> annotations, final RoundEnvironment roundEnv) |
| { |
| |
| if(roundEnv.processingOver()) |
| { |
| |
| return true; |
| } |
| |
| Filer filer = processingEnv.getFiler(); |
| |
| try |
| { |
| |
| for (Element e : roundEnv.getElementsAnnotatedWith(ManagedObjectFactoryConstructor.class)) |
| { |
| if(e.getKind() == ElementKind.CONSTRUCTOR) |
| { |
| ExecutableElement constructorElement = (ExecutableElement) e; |
| String factoryName = generateObjectFactory(filer, constructorElement); |
| } |
| } |
| |
| } |
| catch (Exception e) |
| { |
| processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "Error: " + e.getLocalizedMessage()); |
| } |
| |
| return true; |
| } |
| |
| private String generateObjectFactory(final Filer filer, final ExecutableElement constructorElement) |
| { |
| TypeElement classElement = (TypeElement) constructorElement.getEnclosingElement(); |
| String factoryName = classElement.getQualifiedName().toString() + "Factory"; |
| String factorySimpleName = classElement.getSimpleName().toString() + "Factory"; |
| String objectSimpleName = classElement.getSimpleName().toString(); |
| processingEnv.getMessager().printMessage(Diagnostic.Kind.NOTE, "Generating factory file for " + classElement.getQualifiedName().toString()); |
| final ManagedObjectFactoryConstructor annotation = |
| constructorElement.getAnnotation(ManagedObjectFactoryConstructor.class); |
| PackageElement packageElement = (PackageElement) classElement.getEnclosingElement(); |
| |
| try |
| { |
| JavaFileObject factoryFile = filer.createSourceFile(factoryName); |
| PrintWriter pw = new PrintWriter(new OutputStreamWriter(factoryFile.openOutputStream(), "UTF-8")); |
| pw.println("/*"); |
| for(String headerLine : License.LICENSE) |
| { |
| pw.println(" *" + headerLine); |
| } |
| pw.println(" */"); |
| pw.println(); |
| pw.print("package "); |
| pw.print(packageElement.getQualifiedName()); |
| pw.println(";"); |
| pw.println(); |
| |
| pw.println("import java.util.Map;"); |
| pw.println(); |
| pw.println("import org.apache.qpid.server.model.AbstractConfiguredObjectTypeFactory;"); |
| pw.println("import org.apache.qpid.server.model.ConfiguredObject;"); |
| pw.println("import org.apache.qpid.server.plugin.PluggableService;"); |
| if(annotation.conditionallyAvailable()) |
| { |
| pw.println("import org.apache.qpid.server.plugin.ConditionallyAvailable;"); |
| } |
| pw.println(); |
| pw.println("@PluggableService"); |
| pw.println("public final class " + factorySimpleName + " extends AbstractConfiguredObjectTypeFactory<"+ objectSimpleName +">"); |
| if(annotation.conditionallyAvailable()) |
| { |
| pw.println(" implements ConditionallyAvailable"); |
| } |
| pw.println("{"); |
| pw.println(" public " + factorySimpleName + "()"); |
| pw.println(" {"); |
| pw.println(" super(" + objectSimpleName + ".class);"); |
| pw.println(" }"); |
| pw.println(); |
| pw.println(" @Override"); |
| pw.println(" protected "+objectSimpleName+" createInstance(final Map<String, Object> attributes, final ConfiguredObject<?>... parents)"); |
| pw.println(" {"); |
| pw.print(" return new "+objectSimpleName+"(attributes"); |
| boolean first = true; |
| for(VariableElement param : constructorElement.getParameters()) |
| { |
| if(first) |
| { |
| first = false; |
| } |
| else |
| { |
| TypeMirror erasureType = processingEnv.getTypeUtils().erasure(param.asType()); |
| pw.print(", getParent("+erasureType.toString()+".class,parents)"); |
| } |
| } |
| pw.println(");"); |
| pw.println(" }"); |
| if(annotation.conditionallyAvailable()) |
| { |
| pw.println(); |
| pw.println(" @Override"); |
| pw.println(" public boolean isAvailable()"); |
| pw.println(" {"); |
| pw.println(" return " + objectSimpleName + ".isAvailable();"); |
| pw.println(" }"); |
| |
| } |
| |
| pw.println("}"); |
| |
| pw.close(); |
| } |
| catch (IOException e) |
| { |
| processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, |
| "Failed to write factory file: " |
| + factoryName |
| + " - " |
| + e.getLocalizedMessage()); |
| } |
| |
| return factoryName; |
| } |
| |
| } |