def _accumulate_distributions(future: Future): """ Saves created distribution parameters to global variable. """ global _name_dists task_data = future.result() # type: _TaskData orbital_elems = task_data.orbiral_elems names = task_data.names variations_matrix = task_data.variations_matrix asteroid_variations = variations_matrix[0] dists = _make_dists(orbital_elems, names, asteroid_variations) _name_dists.append((task_data.name, dists))
def published_callback(topic: str, start_time: float, fut: Future) -> None: # Record the metrics finish_time = time.time() exception = fut.exception() if exception: error = str(exception.__class__.__name__) PUBLISHED_MESSAGES.labels(stream_id=topic, partition=-1, error=error).inc() else: metadata = fut.result() PUBLISHED_MESSAGES.labels(stream_id=topic, partition=metadata.partition, error=NOERROR).inc() PRODUCER_TOPIC_OFFSET.labels(stream_id=topic, partition=metadata.partition).set( metadata.offset) PUBLISHED_MESSAGES_TIME.labels(stream_id=topic).observe(finish_time - start_time)