Spaces:
Sleeping
Sleeping
File size: 2,425 Bytes
fb49ac2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
import os # noqa: E402
os.environ["MONITOR_SERVICE_NAME"] = "otlp_example" # noqa
os.environ["ANT_OTEL_ENDPOINT"] = "https://antcollector.alipay.com/namespace/aworld/task/aworld/otlp/api/v1/metrics" # noqa
os.environ["OTLP_TRACES_ENDPOINT"] = "https://antcollector.alipay.com/namespace/aworld/task/aworld_trace/otlp/api/v1/traces" # noqa
from aworld.trace.config import ObservabilityConfig
from aworld.logs.util import logger
from aworld.trace.baggage import BaggageContext
from aworld.trace.base import get_tracer_provider
from aworld.trace.instrumentation.requests import instrument_requests
from aworld.trace.instrumentation.flask import instrument_flask
import flask
import threading
import aworld.trace as trace
trace.configure(ObservabilityConfig(
trace_provider="otlp",
trace_backends=["other_otlp"],
trace_base_url="https://antcollector.alipay.com/namespace/aworld/task/aworld_trace/otlp/api/v1/traces",
metrics_provider="otlp",
metrics_backend="antmonitor",
metrics_base_url="https://antcollector.alipay.com/namespace/aworld/task/aworld/otlp/api/v1/metrics"
))
instrument_flask()
instrument_requests()
app = flask.Flask(__name__)
@app.route('/api/test')
def test():
sofa_trace_id = BaggageContext.get_baggage_value("attributes.sofa.traceid")
sofa_rpc_id = BaggageContext.get_baggage_value("attributes.sofa.rpcid")
sofa_pen_attrs = BaggageContext.get_baggage_value(
"attributes.sofa.penattrs")
sofa_sys_pen_attrs = BaggageContext.get_baggage_value(
"attributes.sofa.syspenattrs")
logger.info(
f"test sofa_trace_id={sofa_trace_id}, sofa_rpc_id={sofa_rpc_id}, sofa_pen_attrs={sofa_pen_attrs}, sofa_sys_pen_attrs={sofa_sys_pen_attrs}"
)
return 'Hello, World!'
thread = threading.Thread(target=lambda: app.run(port=7070), daemon=True)
thread.start()
def invoke_api():
import requests
session = requests.session()
session.headers.update({
"SOFA-TraceId": "12345678901234567890123456789012",
"SOFA-RpcId": "0.1.1",
"sofaPenAttrs": "key1=value1&key2=value2",
"sysPenAttrs": "key1=value1&key2=value2"
})
response = session.get('http://localhost:7070/api/test')
logger.info(f"invoke_api response={response.text}")
def main():
logger.info("main running")
invoke_api()
if __name__ == "__main__":
main()
get_tracer_provider().force_flush(1000)
thread.join()
|