Fixed a bug where the builder did not reset the status after throwing…

… an exception
AmayaFramework · Feb 21, 2024 · 3167697 · 3167697
1 parent bbe6b28
commit 3167697
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/src/main/java/io/github/amayaframework/di/CheckedProviderBuilder.java b/src/main/java/io/github/amayaframework/di/CheckedProviderBuilder.java
@@ -93,8 +93,7 @@ protected void buildArtifacts(Map<Class<?>, ClassScheme> schemes, LazyProvider p
         }
     }
 
-    @Override
-    public ServiceProvider build() {
+    private ServiceProvider uncheckedBuild() {
         // Build class schemes
         var schemes = makeSchemes();
         // Build dependency graph
@@ -131,4 +130,17 @@ public ServiceProvider build() {
         reset();
         return new ServiceProviderImpl(repository);
     }
+
+    @Override
+    public ServiceProvider build() {
+        try {
+            return uncheckedBuild();
+        } catch (Error | RuntimeException e) {
+            reset();
+            throw e;
+        } catch (Throwable e) {
+            reset();
+            throw new RuntimeException(e);
+        }
+    }
 }