LBRYFoundation/lbry-sdk
1
0
Fork 0
mirror of https://github.com/LBRYFoundation/lbry-sdk.git synced 2025-09-04 04:45:09 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

return true/false if clean was performed

Browse source
This commit is contained in:
Lex Berezhny 2021-08-16 17:02:13 -04:00
parent 2535b8adef
commit 2cd5d75a2e
2 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
lbry/blob/disk_space_manager.py
View file

@ -28,7 +28,7 @@ class DiskSpaceManager:
def clean(self): def clean(self):
if not self.config.blob_storage_limit: if not self.config.blob_storage_limit:
return return False
used = 0 used = 0
files = [] files = []
data_dir = os.path.join(self.config.data_dir, 'blobfiles') data_dir = os.path.join(self.config.data_dir, 'blobfiles')
@ -39,11 +39,14 @@ class DiskSpaceManager:
files.append((file_stats.st_mtime, file_stats.st_size, file.path)) files.append((file_stats.st_mtime, file_stats.st_size, file.path))
files.sort() files.sort()
available = (self.config.blob_storage_limit*1024*1024) - used available = (self.config.blob_storage_limit*1024*1024) - used
cleaned = False
for _, file_size, file in files: for _, file_size, file in files:
available += file_size available += file_size
if available > 0: if available > 0:
break break
os.remove(file) os.remove(file)
cleaned = True
return cleaned
async def cleaning_loop(self): async def cleaning_loop(self):
while self.running: while self.running:

3
tests/unit/blob/test_disk_space_manager.py
View file

@ -24,6 +24,7 @@ class ConfigurationTests(unittest.TestCase):
with open(os.path.join(config.data_dir, 'blobfiles', f'3mb-{file_no}'), 'w') as blob: with open(os.path.join(config.data_dir, 'blobfiles', f'3mb-{file_no}'), 'w') as blob:
blob.write('0' * 1 * 1024 * 1024) blob.write('0' * 1 * 1024 * 1024)
self.assertEqual(10, dsm.space_used_mb) self.assertEqual(10, dsm.space_used_mb)
dsm.clean() self.assertTrue(dsm.clean())
self.assertEqual(5, dsm.space_used_mb) self.assertEqual(5, dsm.space_used_mb)
self.assertFalse(dsm.clean())