codex-storage/deluge
1
0
Fork 0
mirror of https://github.com/codex-storage/deluge.git synced 2025-02-17 13:56:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

save/load dht state - close #359

Browse Source
This commit is contained in:
Marcos Pinto 2008-07-26 03:36:15 +00:00
parent fb09106928
commit 92f2b369ca
1 changed files with 23 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
deluge/core/core.py
View File

@ -342,6 +342,8 @@ class Core(
"""This is called by a thread from shutdown()""" """This is called by a thread from shutdown()"""
log.info("Shutting down core..") log.info("Shutting down core..")
self._should_shutdown = True self._should_shutdown = True
# Shutdown dht
self._on_set_dht(False, False)
# Shutdown the socket # Shutdown the socket
try: try:
self.socket.shutdown(socket.SHUT_RDWR) self.socket.shutdown(socket.SHUT_RDWR)
@ -765,12 +767,32 @@ class Core(
def _on_set_dht(self, key, value): def _on_set_dht(self, key, value):
log.debug("dht value set to %s", value) log.debug("dht value set to %s", value)
state_file = deluge.common.get_default_config_dir('dht.state')
if value: if value:
if os.path.exists(state_file):
try:
dht_data = open(state_file, 'rb')
state_contents = dht_data.readlines()
dht_data.close()
except IOError:
log.warning("failed to read dht state file")
else:
state_contents = None
try:
self.session.start_dht(state_contents)
except:
log.warning("restoring old dht state failed")
self.session.start_dht(None) self.session.start_dht(None)
self.session.add_dht_router("router.bittorrent.com", 6881) self.session.add_dht_router("router.bittorrent.com", 6881)
self.session.add_dht_router("router.utorrent.com", 6881) self.session.add_dht_router("router.utorrent.com", 6881)
self.session.add_dht_router("router.bitcomet.com", 6881) self.session.add_dht_router("router.bitcomet.com", 6881)
else: else:
try:
dht_data = open(state_file, 'wb')
dht_data.writelines('%s' % (self.session.dht_state()))
dht_data.close()
except IOError:
log.warning("failed to save dht state to file")
self.session.stop_dht() self.session.stop_dht()
def _on_set_upnp(self, key, value): def _on_set_upnp(self, key, value):