def initialize_tracer(project_id):
exporter = stackdriver_exporter.StackdriverExporter(project_id=project_id)
tracer = opencensus.trace.tracer.Tracer(
exporter=exporter,
sampler=opencensus.trace.tracer.samplers.AlwaysOnSampler())
return tracer
if "DISABLE_PROFILER" in os.environ:
raise KeyError()
else:
logger.info("Profiler enabled.")
initStackdriverProfiling()
except KeyError:
logger.info("Profiler disabled.")
# Tracing
try:
if "DISABLE_TRACING" in os.environ:
raise KeyError()
else:
logger.info("Tracing enabled.")
sampler = samplers.AlwaysOnSampler()
exporter = stackdriver_exporter.StackdriverExporter(
project_id=os.environ.get('GCP_PROJECT_ID'),
transport=AsyncTransport)
tracer_interceptor = server_interceptor.OpenCensusServerInterceptor(
sampler, exporter)
except (KeyError, DefaultCredentialsError):
logger.info("Tracing disabled.")
tracer_interceptor = server_interceptor.OpenCensusServerInterceptor()
except Exception as e:
logger.warn(
f"Exception on Cloud Trace setup: {traceback.format_exc()}, tracing disabled."
)
tracer_interceptor = server_interceptor.OpenCensusServerInterceptor()
start(dummy_mode=True)
app.register_blueprint(export_manager_blueprint, url_prefix='/export_manager')
app.register_blueprint(backup_manager_blueprint, url_prefix='/backup_manager')
app.register_blueprint(dataflow_monitor_blueprint,
url_prefix='/dataflow_monitor')
app.register_blueprint(validation_manager_blueprint,
url_prefix='/validation_manager')
app.register_blueprint(calculation_data_storage_manager_blueprint,
url_prefix='/calculation_data_storage_manager')
if environment.in_gae():
SQLAlchemyEngineManager.init_engines_for_server_postgres_instances()
# Export traces and metrics to stackdriver if running on GAE
if environment.in_gae():
monitoring.register_stackdriver_exporter()
trace_exporter = stackdriver_trace.StackdriverExporter(
project_id=metadata.project_id(), transport=AsyncTransport)
trace_sampler = samplers.ProbabilitySampler(
rate=0.05) # Default is 1 in 10k, trace 1 in 20 instead
else:
trace_exporter = file_exporter.FileExporter(file_name='traces')
trace_sampler = samplers.AlwaysOnSampler()
middleware = FlaskMiddleware(
app,
blacklist_paths=['metadata'], # Don't trace metadata requests
sampler=trace_sampler,
exporter=trace_exporter,
propagator=google_cloud_format.GoogleCloudFormatPropagator())
config_integration.trace_integrations(
['google_cloud_clientlibs', 'requests', 'sqlalchemy'])