MinecraftMods/LibZontreck
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add a sanity check for migrations, only execute when migration is newer than current table

Browse source
This commit is contained in:
zontreck 2024-04-24 00:19:41 -07:00
parent b534287c51
commit e7f289d709
2 changed files with 46 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

46
src/main/java/dev/zontreck/libzontreck/memory/world/DatabaseMigrations.java
View file

@ -1,6 +1,7 @@
package dev.zontreck.libzontreck.memory.world;
import com.google.common.collect.Lists;
import dev.zontreck.libzontreck.LibZontreck;
import dev.zontreck.libzontreck.events.RegisterMigrationsEvent;
import net.minecraftforge.common.MinecraftForge;
@ -133,9 +134,52 @@ public class DatabaseMigrations
List<Migration> migration = Lists.reverse(migrations);
Migration lastTableChecked = null;
for(Migration m : migration)
{
m.execute();
if(lastTableChecked == null) lastTableChecked = getCurrentTable(m.tableID);
else {
if(lastTableChecked.tableID != m.tableID) lastTableChecked = getCurrentTable(m.tableID);
}
if(m.version > lastTableChecked.version) {
LibZontreck.LOGGER.info("Executing migration " + m.tableID + ":" + m.version);
m.execute();
} else {
LibZontreck.LOGGER.info("Skipping migration on table " + m.tableID + "; Current table version is " + lastTableChecked.version);
}
}
}
/**
* Gets the current table's version using the Migration structure for data fields. Will be null if there is an error on any table except for migrations.
* @return
*/
private static Migration getCurrentTable(String tableID)
{
try{
PreparedStatement pst = DatabaseWrapper.get().prepareStatement("SELECT * FROM `migrations` WHERE tableID=?;");
pst.setString(0, tableID);
var result = pst.executeQuery();
if(!result.next())
{
return builder().withTableID(tableID).withVersion(0);
}else {
return builder().withTableID(tableID).withVersion(result.getInt("version"));
}
}catch (SQLException ex)
{
if(tableID == "migrations")
{
return builder().withTableID(tableID)
.withVersion(0);
}
ex.printStackTrace();
return null;
}
}