|
| 1 | +import dataclasses |
| 2 | +import enum |
| 3 | +from typing import List |
| 4 | + |
| 5 | +from typing_extensions import override |
| 6 | + |
| 7 | +from prime_backup.action import Action |
| 8 | +from prime_backup.db.access import DbAccess |
| 9 | +from prime_backup.db.session import DbSession |
| 10 | +from prime_backup.types.chunk_group_info import BlobChunkGroupBindingInfo |
| 11 | + |
| 12 | + |
| 13 | +class BadBlobChunkGroupBindingItemType(enum.Enum): |
| 14 | + orphan = enum.auto() |
| 15 | + bad_storage_method = enum.auto() |
| 16 | + |
| 17 | + |
| 18 | +@dataclasses.dataclass(frozen=True) |
| 19 | +class BadBlobChunkGroupBindingItem: |
| 20 | + binding: BlobChunkGroupBindingInfo |
| 21 | + typ: BadBlobChunkGroupBindingItemType |
| 22 | + desc: str |
| 23 | + |
| 24 | + |
| 25 | +@dataclasses.dataclass |
| 26 | +class ValidateBlobChunkGroupBindingsResult: |
| 27 | + total: int = 0 |
| 28 | + bad_bindings: List[BadBlobChunkGroupBindingItem] = dataclasses.field(default_factory=list) |
| 29 | + |
| 30 | + @property |
| 31 | + def ok(self) -> int: |
| 32 | + return self.total - self.bad |
| 33 | + |
| 34 | + @property |
| 35 | + def bad(self) -> int: |
| 36 | + return len(self.bad_bindings) |
| 37 | + |
| 38 | + def add_bad(self, binding: BlobChunkGroupBindingInfo, typ: BadBlobChunkGroupBindingItemType, msg: str): |
| 39 | + self.bad_bindings.append(BadBlobChunkGroupBindingItem(binding, typ, msg)) |
| 40 | + |
| 41 | + |
| 42 | +class ValidateBlobChunkGroupBindingsAction(Action[ValidateBlobChunkGroupBindingsResult]): |
| 43 | + """ |
| 44 | + NOTE: BlobChunkGroupBinding's .chunk_group_offset and .chunk_group_id checks are done in ValidateBlobsAction |
| 45 | + """ |
| 46 | + |
| 47 | + @override |
| 48 | + def run(self) -> ValidateBlobChunkGroupBindingsResult: |
| 49 | + self.logger.info('Scanning all blob chunk group bindings for orphan check') |
| 50 | + result = ValidateBlobChunkGroupBindingsResult() |
| 51 | + |
| 52 | + session: DbSession |
| 53 | + with DbAccess.open_session() as session: |
| 54 | + self.logger.info('Scanning all blob chunk group bindings for orphan check') |
| 55 | + result.total = session.get_blob_chunk_group_binding_count() |
| 56 | + |
| 57 | + for lor in session.list_orphan_blob_chunk_group_bindings(limit=1000): |
| 58 | + if lor.blob is None: |
| 59 | + result.add_bad(BlobChunkGroupBindingInfo.of(lor.binding), BadBlobChunkGroupBindingItemType.orphan, f'orphan binding refers to a non-existent blob {lor.binding.blob_id}') |
| 60 | + else: |
| 61 | + result.add_bad(BlobChunkGroupBindingInfo.of(lor.binding), BadBlobChunkGroupBindingItemType.bad_storage_method, f'Chunk group binding has invalid storage method {lor.blob.storage_method}') |
| 62 | + |
| 63 | + self.logger.info('BlobChunkGroupBindingInfo validation done: total {}, ok {}, bad {}'.format( |
| 64 | + result.total, result.ok, result.bad, |
| 65 | + )) |
| 66 | + return result |
0 commit comments