Skip to content

Commit

Permalink
⚡ improved annotation processor performance
Browse files Browse the repository at this point in the history
  • Loading branch information
@funkyFangs
funkyFangs committed Jun 7, 2023
1 parent c0ca82e commit 5ed3440
Showing 1 changed file with 56 additions and 42 deletions.
98 changes: 56 additions & 42 deletions ...common/src/main/java/io/funky/fangs/springdoc/customizer/processor/ExamplesProcessor.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -20,9 +20,11 @@
import java.io.IOException;
import java.lang.reflect.Field;
import java.util.*;
import java.util.Map.Entry;

import static com.fasterxml.jackson.annotation.JsonInclude.Include.NON_EMPTY;
import static java.util.Collections.emptyMap;
import static java.util.stream.Collectors.toUnmodifiableMap;
import static javax.tools.StandardLocation.CLASS_OUTPUT;

@SupportedAnnotationTypes("io.funky.fangs.springdoc.customizer.annotation.ExampleDetails")
Expand All @@ -37,53 +39,62 @@ public class ExamplesProcessor extends AbstractProcessor {

private FileObject examplesResourceFile;

@Override
public synchronized void init(ProcessingEnvironment processingEnv) {
super.init(processingEnv);

try {
examplesResourceFile = processingEnv.getFiler()
.createResource(CLASS_OUTPUT, getClass().getPackageName(), EXAMPLES_FILE_NAME,
processingEnv.getElementUtils().getTypeElement(getClass().getCanonicalName()));
}
catch (IOException ioException) {
processingEnv.getMessager().printError(ioException.getMessage());
private FileObject getExamplesResourceFile() {
if (examplesResourceFile == null) {
try {
examplesResourceFile = processingEnv.getFiler()
.createResource(CLASS_OUTPUT, getClass().getPackageName(), EXAMPLES_FILE_NAME,
processingEnv.getElementUtils().getTypeElement(getClass().getCanonicalName()));
}
catch (IOException ioException) {
throw new IllegalStateException("Unable to create examples file", ioException);
}
}

return examplesResourceFile;
}

@Override
public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
// Read map from file
Map<String, Map<String, ExampleDetailsRecord>> elements;
try {
elements = OBJECT_MAPPER.readValue(examplesResourceFile.getCharContent(false).toString(),
EXAMPLES_MAP_TYPE_REFERENCE);
}
catch (IllegalStateException | IOException exception) {
// Map is likely not serialized yet, so create empty one
elements = new HashMap<>();
}
if (!annotations.isEmpty()) {
var examplesResourceFile = getExamplesResourceFile();

// Read map from file
Map<String, Map<String, ExampleDetailsRecord>> elements;
try {
elements = OBJECT_MAPPER.readValue(examplesResourceFile.getCharContent(false).toString(),
EXAMPLES_MAP_TYPE_REFERENCE);
} catch (IllegalStateException | IOException exception) {
// Map is likely not serialized yet, so create empty one
elements = new HashMap<>();
}

var originalState = elements.entrySet()
.stream()
.collect(toUnmodifiableMap(Entry::getKey, entry -> Map.copyOf(entry.getValue())));

// Add fields to map
for (var annotation : annotations) {
for (var element : roundEnv.getElementsAnnotatedWith(annotation)) {
if (element.getModifiers().containsAll(EXAMPLE_MODIFIERS)
&& element instanceof VariableElement fieldElement
&& fieldElement.getEnclosingElement() instanceof TypeElement classElement) {
var detailsRecord = convertToRecord(fieldElement.getAnnotation(ExampleDetails.class));
// Add fields to map
for (var annotation : annotations) {
for (var element : roundEnv.getElementsAnnotatedWith(annotation)) {
if (element.getModifiers().containsAll(EXAMPLE_MODIFIERS)
&& element instanceof VariableElement fieldElement
&& fieldElement.getEnclosingElement() instanceof TypeElement classElement) {
var detailsRecord = convertToRecord(fieldElement.getAnnotation(ExampleDetails.class));

elements.computeIfAbsent(classElement.getQualifiedName().toString(), ignored -> new HashMap<>())
.put(fieldElement.getSimpleName().toString(), detailsRecord);
elements.computeIfAbsent(classElement.getQualifiedName().toString(), ignored -> new HashMap<>())
.put(fieldElement.getSimpleName().toString(), detailsRecord);
}
}
}
}

// Write map to file
try (var outputStream = examplesResourceFile.openOutputStream()) {
outputStream.write(OBJECT_MAPPER.writeValueAsBytes(elements));
}
catch (IOException ignored) {
return false;
if (!elements.equals(originalState)) {
// Write map to file
try (var outputStream = examplesResourceFile.openOutputStream()) {
outputStream.write(OBJECT_MAPPER.writeValueAsBytes(elements));
} catch (IOException ignored) {
return false;
}
}
}

return true;
Expand Down Expand Up @@ -119,12 +130,11 @@ public static Map<Field, ExampleDetailsRecord> getExampleFields() {

return Map.copyOf(result);
}
catch (IOException ignored) {
catch (IOException | IllegalArgumentException ignored) {
return emptyMap();
}
}

@SuppressWarnings("ResultOfMethodCallIgnored") // Method call throws exception to populate class string
private ExampleDetailsRecord convertToRecord(ExampleDetails details) {
var targetRecords = new ArrayList<ExampleTargetRecord>();

Expand All @@ -135,16 +145,20 @@ private ExampleDetailsRecord convertToRecord(ExampleDetails details) {
var typeRecords = new ArrayList<ExampleTypeRecord>();

for (var type : method.types()) {
var mediaTypes = List.of(type.mediaTypes());
var responses = List.of(type.responses());
var exampleType = type.value();

typeRecords.add(new ExampleTypeRecord(type.value(), mediaTypes, responses));
typeRecords.add(new ExampleTypeRecord(exampleType, List.of(type.mediaTypes()),
switch (exampleType) {
case REQUEST -> null;
case RESPONSE -> List.of(type.responses());
}));
}

methodRecords.add(new ExampleMethodRecord(method.name(), typeRecords));
}

try {
//noinspection ResultOfMethodCallIgnored
target.controller();
}
catch (MirroredTypeException exception) {
Expand Down

0 comments on commit 5ed3440

Please sign in to comment.