ekimekim's suggestions

pull/58/head
Christopher Usher 6 years ago
parent 9b28765ff2
commit dd246e1343

@ -189,7 +189,7 @@ class BackfillerManager(object):
stop will exit the loop.""" stop will exit the loop."""
self.logger.info('Starting') self.logger.info('Starting')
if self.node_database is not None: if self.node_database is not None:
self.db_manager = database.DBManager(dsn=self.node_database) self.connection = database.DBManager(dsn=self.node_database).get_conn()
failures = 0 failures = 0
@ -254,12 +254,11 @@ class BackfillerManager(object):
if self.node_database is not None: if self.node_database is not None:
self.logger.info('Fetching list of nodes from {}'.format( self.logger.info('Fetching list of nodes from {}'.format(
urlparse.urlparse(self.node_database).hostname)) urlparse.urlparse(self.node_database).hostname))
conn = self.db_manager.get_conn() results = database.query(self.connection, """
results = database.query(conn, """ SELECT name, url
SELECT name, url, backfill_from FROM nodes
FROM nodes""") WHERE backfill_from""")
for row in results: for row in results:
if row.backfill_from:
nodes[row.name] = row.url nodes[row.name] = row.url
nodes.pop(self.localhost, None) nodes.pop(self.localhost, None)

Loading…
Cancel
Save