Clear Avro cache on reload

Fixes quarkusio#18776
gsmet · Jul 28, 2021 · e5ac1e2 · e5ac1e2
1 parent 8394756
commit e5ac1e2
Show file tree

Hide file tree

Showing 3 changed files with 42 additions and 1 deletion.
diff --git a/extensions/avro/deployment/src/main/java/io/quarkus/avro/deployment/AvroProcessor.java b/extensions/avro/deployment/src/main/java/io/quarkus/avro/deployment/AvroProcessor.java
@@ -7,15 +7,27 @@
 import org.jboss.jandex.AnnotationTarget;
 import org.jboss.jandex.DotName;
 
+import io.quarkus.avro.runtime.AvroRecorder;
 import io.quarkus.deployment.annotations.BuildProducer;
 import io.quarkus.deployment.annotations.BuildStep;
+import io.quarkus.deployment.annotations.ExecutionTime;
+import io.quarkus.deployment.annotations.Record;
 import io.quarkus.deployment.builditem.CombinedIndexBuildItem;
+import io.quarkus.deployment.builditem.LaunchModeBuildItem;
 import io.quarkus.deployment.builditem.nativeimage.NativeImageConfigBuildItem;
 import io.quarkus.deployment.builditem.nativeimage.NativeImageSystemPropertyBuildItem;
 import io.quarkus.deployment.builditem.nativeimage.ReflectiveClassBuildItem;
 
 public class AvroProcessor {
 
+    @BuildStep
+    @Record(ExecutionTime.STATIC_INIT)
+    void clearCaches(AvroRecorder recorder, LaunchModeBuildItem launchModeBuildItem) {
+        if (launchModeBuildItem.getLaunchMode().isDevOrTest()) {
+            recorder.clearStaticCaches();
+        }
+    }
+
     @BuildStep
     public void build(CombinedIndexBuildItem indexBuildItem,
             BuildProducer<ReflectiveClassBuildItem> reflectiveClass,

diff --git a/extensions/avro/runtime/src/main/java/io/quarkus/avro/runtime/AvroRecorder.java b/extensions/avro/runtime/src/main/java/io/quarkus/avro/runtime/AvroRecorder.java
@@ -0,0 +1,29 @@
+package io.quarkus.avro.runtime;
+
+import java.lang.reflect.Field;
+import java.util.Map;
+
+import org.apache.avro.specific.SpecificData;
+import org.jboss.logging.Logger;
+
+import io.quarkus.runtime.annotations.Recorder;
+
+@Recorder
+public class AvroRecorder {
+
+    private static final Logger log = Logger.getLogger(AvroRecorder.class);
+
+    public void clearStaticCaches() {
+        try {
+            Field instanceField = SpecificData.class.getDeclaredField("INSTANCE");
+            instanceField.setAccessible(true);
+            SpecificData data = (SpecificData) instanceField.get(null);
+            Field classCache = SpecificData.class.getDeclaredField("classCache");
+            classCache.setAccessible(true);
+            Map<String, Class> classCacheMap = (Map<String, Class>) classCache.get(data);
+            classCacheMap.clear();
+        } catch (Throwable t) {
+            log.error("Failed to clear Avro cache", t);
+        }
+    }
+}
diff --git a/...quarkus/avro/graal/AvroSubstitutions.java → ...avro/runtime/graal/AvroSubstitutions.java b/...quarkus/avro/graal/AvroSubstitutions.java → ...avro/runtime/graal/AvroSubstitutions.java
@@ -1,4 +1,4 @@
-package io.quarkus.avro.graal;
+package io.quarkus.avro.runtime.graal;
 
 import java.lang.reflect.Constructor;
 import java.util.HashMap;