Delete resources which failed checksum validation

mlcommons · Feb 10, 2025 · d962da3 · d962da3
1 parent 399ee37
commit d962da3
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/flutter/lib/resources/cache_manager.dart b/flutter/lib/resources/cache_manager.dart
@@ -73,6 +73,16 @@ class CacheManager {
     }
   }
 
+  Future<void> deleteFiles(List<String> resources) async {
+    for (final resource in resources) {
+      final filePath = get(resource);
+      if (filePath == null) continue;
+      final file = File(filePath);
+      if (await file.exists()) await file.delete();
+      print('Deleted resource $resource stored at ${file.path}');
+    }
+  }
+
   Future<void> purgeOutdatedCache(int atLeastDaysOld) async {
     var currentResources = <String>[];
     for (var r in _resourcesMap.values) {

diff --git a/flutter/lib/resources/resource_manager.dart b/flutter/lib/resources/resource_manager.dart
@@ -129,8 +129,11 @@ class ResourceManager {
 
       final checksumFailed = await validateResourcesChecksum(resources);
       if (checksumFailed.isNotEmpty) {
-        final mismatchedPaths = checksumFailed.map((e) => '\n${e.path}').join();
-        throw 'Checksum validation failed for: $mismatchedPaths';
+        final checksumFailedPathString =
+            checksumFailed.map((e) => '\n${e.path}').join();
+        final checksumFailedPaths = checksumFailed.map((e) => e.path).toList();
+        await cacheManager.deleteFiles(checksumFailedPaths);
+        throw 'Checksum validation failed for: $checksumFailedPathString. \nPlease download the missing files again.';
       }
 
       // delete downloaded archives to free up disk space