Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(db): fix migration chain validation

Browse Source
This commit is contained in:
psychedelicious 2023-12-11 11:15:09 +11:00
parent b3f92e0547
commit 8726b203d4
1 changed files with 10 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
invokeai/app/services/shared/sqlite/sqlite_migrator.py
View File

@ -81,27 +81,23 @@ class MigrationSet:
# register() ensures that there is only one migration with a given from_version, so this is safe.
return next((m for m in self._migrations if m.from_version == from_version), None)
def validate_migration_path(self) -> None:
def validate_migration_chain(self) -> None:
"""
Validates that the migrations form a single path of migrations from version 0 to the latest version.
Validates that the migrations form a single chain of migrations from version 0 to the latest version.
Raises a MigrationError if there is a problem.
"""
if self.count == 0:
return
if self.latest_version == 0:
return
current_version = 0
touched_count = 0
while current_version < self.latest_version:
migration = self.get(current_version)
if migration is None:
raise MigrationError(f"Missing migration from {current_version}")
current_version = migration.to_version
touched_count += 1
if current_version != self.latest_version:
raise MigrationError(f"Missing migration to {self.latest_version}")
next_migration = self.get(from_version=0)
touched_count = 1
while next_migration is not None:
next_migration = self.get(next_migration.to_version)
if next_migration is not None:
touched_count += 1
if touched_count != self.count:
raise MigrationError("Migration path is not contiguous")
raise MigrationError("Migration chain is fragmented")
@property
def count(self) -> int:
@ -178,7 +174,7 @@ class SQLiteMigrator:
"""Migrates the database to the latest version."""
with self._lock:
# This throws if there is a problem.
self._migrations.validate_migration_path()
self._migration_set.validate_migration_chain()
self._create_migrations_table(cursor=self._cursor)
if self._migrations.count == 0: