liquibase.ext.hibernate.diff.ChangedUniqueConstraintChangeGenerator Maven / Gradle / Ivy

Go to download
package liquibase.ext.hibernate.diff;

import liquibase.change.Change;
import liquibase.database.Database;
import liquibase.diff.ObjectDifferences;
import liquibase.diff.output.DiffOutputControl;
import liquibase.diff.output.changelog.ChangeGeneratorChain;
import liquibase.diff.output.changelog.ChangedObjectChangeGenerator;
import liquibase.ext.hibernate.database.HibernateDatabase;
import liquibase.structure.DatabaseObject;
import liquibase.structure.core.UniqueConstraint;

/**
 * Hibernate doesn't know about all the variations that occur with unique constraint but just whether they exists or not.
 * To prevent changing customized constraints, we suppress all changes with hibernate.
 */
public class ChangedUniqueConstraintChangeGenerator implements ChangedObjectChangeGenerator {
    @Override
    public int getPriority(Class objectType, Database database) {

        if (UniqueConstraint.class.isAssignableFrom(objectType)) {
            return PRIORITY_ADDITIONAL;
        }
        return PRIORITY_NONE;
    }

    @Override
    public Class[] runBeforeTypes() {
        return null;
    }

    @Override
    public Class[] runAfterTypes() {
        return null;
    }

    @Override
    public Change[] fixChanged(DatabaseObject changedObject, ObjectDifferences differences, DiffOutputControl control, Database referenceDatabase, Database comparisonDatabase, ChangeGeneratorChain chain) {
        if (referenceDatabase instanceof HibernateDatabase || comparisonDatabase instanceof HibernateDatabase) {
            return null;
        } else {
            return chain.fixChanged(changedObject, differences, control, referenceDatabase, comparisonDatabase);
        }
    }
}