Clean up BucketWriters on shutdown (also preventing DelayedCalls leaks in tests).

This commit is contained in:
Itamar Turner-Trauring 2021-11-18 15:42:54 -05:00
parent 1827faf36b
commit 5d915afe1c

View File

@ -136,6 +136,12 @@ class StorageServer(service.MultiService, Referenceable):
# Canaries and disconnect markers for BucketWriters created via Foolscap:
self._bucket_writer_disconnect_markers = {} # type: Dict[BucketWriter,(IRemoteReference, object)]
def stopService(self):
# Cancel any in-progress uploads:
for bw in list(self._bucket_writers.values()):
bw.disconnected()
return service.MultiService.stopService(self)
def __repr__(self):
return "<StorageServer %s>" % (idlib.shortnodeid_b2a(self.my_nodeid),)