feat: Implement missing replaces() method in all snapshot generators (#764)

Add overridden replaces() method to all Hibernate snapshot generator classes to
explicitly replace the corresponding snapshot generators from core Liquibase.
This ensures that Liquibase uses the Hibernate-specific implementations instead
of the default ones.

Author: filipelautert

src/main/java/liquibase/ext/hibernate/snapshot/CatalogSnapshotGenerator.java

@@ -3,6 +3,7 @@
 import liquibase.exception.DatabaseException;
 import liquibase.snapshot.DatabaseSnapshot;
 import liquibase.snapshot.InvalidExampleException;
+import liquibase.snapshot.SnapshotGenerator;
 import liquibase.structure.DatabaseObject;
 import liquibase.structure.core.Catalog;
 
@@ -26,4 +27,8 @@ protected void addTo(DatabaseObject foundObject, DatabaseSnapshot snapshot) thro
         // Nothing to add to
     }
 
+    @Override
+    public Class<? extends SnapshotGenerator>[] replaces() {
+        return new Class[]{liquibase.snapshot.jvm.CatalogSnapshotGenerator.class};
+    }
 }

src/main/java/liquibase/ext/hibernate/snapshot/ColumnSnapshotGenerator.java

@@ -1,11 +1,11 @@
 package liquibase.ext.hibernate.snapshot;
 
-import java.util.Iterator;
 import java.util.List;
 import java.util.Properties;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
 
+import liquibase.snapshot.SnapshotGenerator;
 import org.hibernate.boot.spi.MetadataImplementor;
 import org.hibernate.dialect.Dialect;
 import org.hibernate.dialect.PostgreSQLDialect;
@@ -226,4 +226,9 @@ protected DataType toDataType(String hibernateType, Integer sqlTypeCode) throws
         return dataType;
     }
 
+    @Override
+    public Class<? extends SnapshotGenerator>[] replaces() {
+        return new Class[]{liquibase.snapshot.jvm.ColumnSnapshotGenerator.class};
+    }
+
 }

src/main/java/liquibase/ext/hibernate/snapshot/ForeignKeySnapshotGenerator.java

@@ -5,6 +5,7 @@
 import liquibase.ext.hibernate.database.HibernateDatabase;
 import liquibase.snapshot.DatabaseSnapshot;
 import liquibase.snapshot.InvalidExampleException;
+import liquibase.snapshot.SnapshotGenerator;
 import liquibase.structure.DatabaseObject;
 import liquibase.structure.core.ForeignKey;
 import liquibase.structure.core.Table;
@@ -85,4 +86,9 @@ protected void addTo(DatabaseObject foundObject, DatabaseSnapshot snapshot) thro
         }
     }
 
+    @Override
+    public Class<? extends SnapshotGenerator>[] replaces() {
+        return new Class[]{ liquibase.snapshot.jvm.ForeignKeySnapshotGenerator.class };
+    }
+
 }

src/main/java/liquibase/ext/hibernate/snapshot/IndexSnapshotGenerator.java

@@ -4,6 +4,7 @@
 import liquibase.exception.DatabaseException;
 import liquibase.snapshot.DatabaseSnapshot;
 import liquibase.snapshot.InvalidExampleException;
+import liquibase.snapshot.SnapshotGenerator;
 import liquibase.structure.DatabaseObject;
 import liquibase.structure.core.*;
 
@@ -91,4 +92,9 @@ private Boolean isUniqueIndex(org.hibernate.mapping.Index hibernateIndex) {
             return false;
         }
     }
+
+    @Override
+    public Class<? extends SnapshotGenerator>[] replaces() {
+        return new Class[]{liquibase.snapshot.jvm.IndexSnapshotGenerator.class};
+    }
 }

src/main/java/liquibase/ext/hibernate/snapshot/PrimaryKeySnapshotGenerator.java

@@ -4,6 +4,7 @@
 import liquibase.exception.DatabaseException;
 import liquibase.snapshot.DatabaseSnapshot;
 import liquibase.snapshot.InvalidExampleException;
+import liquibase.snapshot.SnapshotGenerator;
 import liquibase.structure.DatabaseObject;
 import liquibase.structure.core.Column;
 import liquibase.structure.core.Index;
@@ -67,4 +68,9 @@ protected void addTo(DatabaseObject foundObject, DatabaseSnapshot snapshot) thro
         }
     }
 
+    @Override
+    public Class<? extends SnapshotGenerator>[] replaces() {
+        return new Class[]{liquibase.snapshot.jvm.PrimaryKeySnapshotGenerator.class};
+    }
+
 }

src/main/java/liquibase/ext/hibernate/snapshot/SchemaSnapshotGenerator.java

@@ -3,6 +3,7 @@
 import liquibase.exception.DatabaseException;
 import liquibase.snapshot.DatabaseSnapshot;
 import liquibase.snapshot.InvalidExampleException;
+import liquibase.snapshot.SnapshotGenerator;
 import liquibase.structure.DatabaseObject;
 import liquibase.structure.core.Catalog;
 import liquibase.structure.core.Schema;
@@ -26,4 +27,9 @@ protected void addTo(DatabaseObject foundObject, DatabaseSnapshot snapshot) thro
         // Nothing to do
     }
 
+    @Override
+    public Class<? extends SnapshotGenerator>[] replaces() {
+        return new Class[]{liquibase.snapshot.jvm.SchemaSnapshotGenerator.class};
+    }
+
 }

src/main/java/liquibase/ext/hibernate/snapshot/SequenceSnapshotGenerator.java

@@ -4,6 +4,7 @@
 import liquibase.ext.hibernate.database.HibernateDatabase;
 import liquibase.snapshot.DatabaseSnapshot;
 import liquibase.snapshot.InvalidExampleException;
+import liquibase.snapshot.SnapshotGenerator;
 import liquibase.structure.DatabaseObject;
 import liquibase.structure.core.Schema;
 import liquibase.structure.core.Sequence;
@@ -46,6 +47,8 @@ protected void addTo(DatabaseObject foundObject, DatabaseSnapshot snapshot) thro
         }
     }
 
-
-
+    @Override
+    public Class<? extends SnapshotGenerator>[] replaces() {
+        return new Class[]{ liquibase.snapshot.jvm.SequenceSnapshotGenerator.class };
+    }
 }

src/main/java/liquibase/ext/hibernate/snapshot/TableSnapshotGenerator.java

@@ -7,6 +7,7 @@
 import liquibase.ext.hibernate.snapshot.extension.TableGeneratorSnapshotGenerator;
 import liquibase.snapshot.DatabaseSnapshot;
 import liquibase.snapshot.InvalidExampleException;
+import liquibase.snapshot.SnapshotGenerator;
 import liquibase.structure.DatabaseObject;
 import liquibase.structure.core.Schema;
 import liquibase.structure.core.Table;
@@ -118,4 +119,9 @@ private void addDatabaseObjectToSchema(org.hibernate.mapping.Table join, Schema
         Scope.getCurrentScope().getLog(getClass()).info("Found table " + joinTable.getName());
         schema.addDatabaseObject(snapshotObject(joinTable, snapshot));
     }
+
+    @Override
+    public Class<? extends SnapshotGenerator>[] replaces() {
+        return new Class[]{liquibase.snapshot.jvm.TableSnapshotGenerator.class};
+    }
 }

src/main/java/liquibase/ext/hibernate/snapshot/UniqueConstraintSnapshotGenerator.java

@@ -4,6 +4,7 @@
 import liquibase.exception.DatabaseException;
 import liquibase.snapshot.DatabaseSnapshot;
 import liquibase.snapshot.InvalidExampleException;
+import liquibase.snapshot.SnapshotGenerator;
 import liquibase.structure.DatabaseObject;
 import liquibase.structure.core.Column;
 import liquibase.structure.core.Index;
@@ -112,4 +113,9 @@ protected Index getBackingIndex(UniqueConstraint uniqueConstraint, org.hibernate
         return index;
     }
 
+    @Override
+    public Class<? extends SnapshotGenerator>[] replaces() {
+        return new Class[]{ liquibase.snapshot.jvm.UniqueConstraintSnapshotGenerator.class };
+    }
+
 }

src/main/java/liquibase/ext/hibernate/snapshot/ViewSnapshotGenerator.java

@@ -3,6 +3,7 @@
 import liquibase.exception.DatabaseException;
 import liquibase.snapshot.DatabaseSnapshot;
 import liquibase.snapshot.InvalidExampleException;
+import liquibase.snapshot.SnapshotGenerator;
 import liquibase.structure.DatabaseObject;
 import liquibase.structure.core.Schema;
 import liquibase.structure.core.View;
@@ -24,7 +25,11 @@ protected DatabaseObject snapshotObject(DatabaseObject example, DatabaseSnapshot
     @Override
     protected void addTo(DatabaseObject foundObject, DatabaseSnapshot snapshot) throws DatabaseException, InvalidExampleException {
         // No views in Hibernate mapping
+    }
 
+    @Override
+    public Class<? extends SnapshotGenerator>[] replaces() {
+        return new Class[]{ liquibase.snapshot.jvm.ViewSnapshotGenerator.class };
     }
 
 }