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)