def setUp(self):
initialize_logging(_log_path)
self.tearDown()
# clear out any old stats
space_accounting_database = SpaceAccountingDatabase()
space_accounting_database.clear_collection_stats(_collection_id)
space_accounting_database.commit()
self._space_accounting_server_process = start_space_accounting_server(
_local_node_name, _space_accounting_server_address, _space_accounting_pipeline_address
)
poll_result = poll_process(self._space_accounting_server_process)
self.assertEqual(poll_result, None)
def setUp(self):
initialize_logging(_log_path)
self.tearDown()
# clear out any old stats
space_accounting_database = SpaceAccountingDatabase()
space_accounting_database.clear_collection_stats(_collection_id)
space_accounting_database.commit()
self._space_accounting_server_process = \
start_space_accounting_server(
_local_node_name,
_space_accounting_server_address,
_space_accounting_pipeline_address
)
poll_result = poll_process(self._space_accounting_server_process)
self.assertEqual(poll_result, None)
def _flush_to_database(self, hour):
if hour not in self._state["data"]:
self._log.warn("no data for %s" % (hour, ))
return
self._log.info("storing data for %s" % (hour, ))
space_accounting_database = SpaceAccountingDatabase()
for collection_id, events in self._state["data"][hour].items():
space_accounting_database.store_collection_stats(
collection_id,
hour,
events.get("bytes_added", 0),
events.get("bytes_removed", 0),
events.get("bytes_retrieved", 0)
)
space_accounting_database.commit()
del self._state["data"][hour]
def _flush_to_database(self, hour):
if hour not in self._state["data"]:
self._log.warn("no data for %s" % (hour, ))
return
self._log.info("storing data for %s" % (hour, ))
space_accounting_database = SpaceAccountingDatabase()
for collection_id, events in self._state["data"][hour].items():
space_accounting_database.store_collection_stats(
collection_id,
hour,
events.get("bytes_added", 0),
events.get("bytes_removed", 0),
events.get("bytes_retrieved", 0)
)
space_accounting_database.commit()
del self._state["data"][hour]