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]