feat: Support Baseline view

Refs: #305
SchweizerischeBundesbahnen · Dec 9, 2024 · 233781b · 233781b
1 parent 4d66554
commit 233781b
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/main/java/ch/sbb/polarion/extension/pdf_exporter/service/PdfExporterPolarionService.java b/src/main/java/ch/sbb/polarion/extension/pdf_exporter/service/PdfExporterPolarionService.java
@@ -207,7 +207,7 @@ private boolean sameDocument(@Nullable String projectId, @NotNull String spaceId
     }
 
     @SneakyThrows
-    public <T> T executeInBaseline(@Nullable String baselineRevision, @NotNull ReadOnlyTransaction transaction, Callable<T> callable) {
+    public <T> T executeInBaseline(@Nullable String baselineRevision, @NotNull ReadOnlyTransaction transaction, @NotNull Callable<T> callable) {
         if (baselineRevision == null) {
             return callable.call();
         } else {
@@ -221,4 +221,12 @@ public <T> T executeInBaseline(@Nullable String baselineRevision, @NotNull ReadO
         }
     }
 
+    @SneakyThrows
+    public <T> T executeInBaseline(@Nullable String baselineRevision, @NotNull ReadOnlyTransaction transaction, @NotNull Runnable runnable) {
+        return executeInBaseline(baselineRevision, transaction, () -> {
+            runnable.run();
+            return null;
+        });
+    }
+
 }