Fix issue with Liquibase and H2 database

Fixes: quarkusio#40575 (cherry picked from commit 509ec82)
gsmet · May 22, 2024 · 4a2d658 · 4a2d658
1 parent 3be9ed1
commit 4a2d658
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/extensions/jdbc/jdbc-h2/runtime/pom.xml b/extensions/jdbc/jdbc-h2/runtime/pom.xml
@@ -52,6 +52,9 @@
                     <parentFirstArtifacts>
                         <parentFirstArtifact>com.h2database:h2</parentFirstArtifact>
                     </parentFirstArtifacts>
+                    <capabilities>
+                        <provides>io.quarkus.jdbc.h2</provides>
+                    </capabilities>
                 </configuration>
             </plugin>
             <plugin>

diff --git a/...iquibase/deployment/src/main/java/io/quarkus/liquibase/deployment/LiquibaseProcessor.java b/...iquibase/deployment/src/main/java/io/quarkus/liquibase/deployment/LiquibaseProcessor.java
@@ -16,6 +16,7 @@
 import java.util.Set;
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.function.BiConsumer;
+import java.util.function.Predicate;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
 
@@ -36,6 +37,7 @@
 import io.quarkus.arc.deployment.SyntheticBeanBuildItem;
 import io.quarkus.arc.processor.DotNames;
 import io.quarkus.datasource.common.runtime.DataSourceUtil;
+import io.quarkus.deployment.Capabilities;
 import io.quarkus.deployment.Feature;
 import io.quarkus.deployment.annotations.BuildProducer;
 import io.quarkus.deployment.annotations.BuildStep;
@@ -101,6 +103,7 @@ void nativeImageConfiguration(
             LiquibaseBuildTimeConfig liquibaseBuildConfig,
             List<JdbcDataSourceBuildItem> jdbcDataSourceBuildItems,
             CombinedIndexBuildItem combinedIndex,
+            Capabilities capabilities,
             BuildProducer<ReflectiveClassBuildItem> reflective,
             BuildProducer<NativeImageResourceBuildItem> resource,
             BuildProducer<ServiceProviderBuildItem> services,
@@ -209,7 +212,7 @@ void nativeImageConfiguration(
         // CommandStep implementations are needed
         consumeService(liquibase.command.CommandStep.class, (serviceClass, implementations) -> {
             var filteredImpls = implementations.stream()
-                    .filter(not("liquibase.command.core.StartH2CommandStep"::equals))
+                    .filter(commandStepPredicate(capabilities))
                     .toArray(String[]::new);
             services.produce(new ServiceProviderBuildItem(serviceClass.getName(), filteredImpls));
             reflective.produce(ReflectiveClassBuildItem.builder(filteredImpls).constructors().build());
@@ -247,6 +250,14 @@ void nativeImageConfiguration(
         resourceBundle.produce(new NativeImageResourceBundleBuildItem("liquibase/i18n/liquibase-core"));
     }
 
+    private static Predicate<String> commandStepPredicate(Capabilities capabilities) {
+        if (capabilities.isPresent("io.quarkus.jdbc.h2")) {
+            return (s) -> true;
+        } else {
+            return not("liquibase.command.core.StartH2CommandStep"::equals);
+        }
+    }
+
     private void consumeService(Class<?> serviceClass, BiConsumer<Class<?>, Collection<String>> consumer) {
         try {
             String service = "META-INF/services/" + serviceClass.getName();