External-Mirrors/keycloak
2
0
Fork 0
mirror of https://github.com/keycloak/keycloak.git synced 2026-01-10 15:32:05 -03:30
Code Issues Packages Projects Releases Wiki Activity

Check if PK for DATABASECHANGELOG already exists

Browse Source 
Closes #41082

Signed-off-by: rmartinc <rmartinc@redhat.com>
This commit is contained in:
Ricardo Martin 2025-07-11 16:06:08 +02:00 committed by GitHub
parent 8a694a585b
commit 164274ac51
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 36 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
model/jpa/src/main/java/org/keycloak/connections/jpa/updater/liquibase/conn/CustomChangeLogHistoryService.java
View File

@ -40,7 +40,12 @@ import liquibase.database.core.MySQLDatabase;
import liquibase.exception.DatabaseException;
import liquibase.executor.jvm.ChangelogJdbcMdcListener;
import liquibase.logging.LogFactory;
import liquibase.snapshot.InvalidExampleException;
import liquibase.snapshot.SnapshotGeneratorFactory;
import liquibase.statement.core.AddPrimaryKeyStatement;
import liquibase.structure.core.PrimaryKey;
import liquibase.structure.core.Schema;
import liquibase.structure.core.Table;
/**
*
@ -62,12 +67,11 @@ public class CustomChangeLogHistoryService extends StandardChangeLogHistoryServi
if (serviceInitialized) return;
AddPrimaryKeyStatement pkStatement = new AddPrimaryKeyStatement(getLiquibaseCatalogName(), getLiquibaseSchemaName(), getDatabaseChangeLogTableName(),
ColumnConfig.arrayFromNames("ID, AUTHOR, FILENAME"), "PK_DATABASECHANGELOG");
ChangelogJdbcMdcListener.execute(getDatabase(), ex -> ex.execute(pkStatement));
getDatabase().commit();
serviceInitialized = true;
if (!existsDatabaseChangelogPK()) {
createDatabaseChangelogPK();
}
}
@Override
@ -136,4 +140,31 @@ public class CustomChangeLogHistoryService extends StandardChangeLogHistoryServi
public int getPriority() {
return super.getPriority() + 1; // Ensure bigger priority than StandardChangeLogHistoryService
}
private boolean existsDatabaseChangelogPK() throws DatabaseException {
try {
PrimaryKey example = new PrimaryKey();
Table table = new Table();
table.setSchema(new Schema(getLiquibaseCatalogName(), getLiquibaseSchemaName()));
table.setName(getDatabaseChangeLogTableName());
example.setTable(table);
return SnapshotGeneratorFactory.getInstance().has(example, getDatabase());
} catch (InvalidExampleException e) {
throw new DatabaseException(e);
}
}
private void createDatabaseChangelogPK() throws DatabaseException {
AddPrimaryKeyStatement pkStatement = new AddPrimaryKeyStatement(getLiquibaseCatalogName(), getLiquibaseSchemaName(), getDatabaseChangeLogTableName(),
ColumnConfig.arrayFromNames("ID, AUTHOR, FILENAME"), "PK_DATABASECHANGELOG");
try {
ChangelogJdbcMdcListener.execute(getDatabase(), ex -> ex.execute(pkStatement));
getDatabase().commit();
} catch (DatabaseException e) {
// if PK already exists just ignore the exception
if (!existsDatabaseChangelogPK()) {
throw e;
}
}
}
}