logos-storage/bittorrent-benchmarks
1
0
Fork 0
mirror of https://github.com/logos-storage/bittorrent-benchmarks.git synced 2026-01-06 23:13:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: check that cid is not none before removing

Browse Source
This commit is contained in:
gmega 2025-01-27 20:02:32 -03:00
parent ee67a92726
commit 34a179b110
No known key found for this signature in database
GPG Key ID: 6290D34EAD824B18
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
benchmarks/core/experiments/static_experiment.py
View File

@ -110,14 +110,17 @@ class StaticDisseminationExperiment(
def teardown(self, exception: Optional[Exception] = None): def teardown(self, exception: Optional[Exception] = None):
def _remove(element: Tuple[int, Node[TNetworkHandle, TInitialMetadata]]): def _remove(element: Tuple[int, Node[TNetworkHandle, TInitialMetadata]]):
index, node = element index, node = element
assert self._cid is not None # to please mypy # This means this node didn't even get to seed anything.
if self._cid is None:
return element
# Since teardown might be called as the result of an exception, it's expected
# that not all removes will succeed, so we don't check their result.
node.remove(self._cid) node.remove(self._cid)
logger.info("Node %d (%s) removed file", index + 1, node.name) logger.info("Node %d (%s) removed file", index + 1, node.name)
return element return element
try: try:
# Since teardown might be called as the result of an exception, it's expected
# that not all removes will succeed, so we don't check their result.
ensure_successful( ensure_successful(
[ [
self._executor.submit(_remove, (i, node)) self._executor.submit(_remove, (i, node))