-
Notifications
You must be signed in to change notification settings - Fork 1
/
Generating sample jaeger traces
57 lines (45 loc) · 1.73 KB
/
Generating sample jaeger traces
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
import requests
import sys
import time
from lib.tracing import init_tracer
from opentracing_instrumentation.request_context import get_current_span, span_in_context
from opentracing.ext import tags
from opentracing.propagation import Format
def say_hello(hello_to, greeting):
with tracer.start_span('say-hello') as span:
span.set_tag('hello-to', hello_to)
span.set_baggage_item('greeting', greeting)
with span_in_context(span):
hello_str = format_string(hello_to)
print_hello(hello_str)
def format_string(hello_to):
with tracer.start_span('format', child_of=get_current_span()) as span:
with span_in_context(span):
hello_str = http_get(8081, 'format', 'helloTo', hello_to)
span.log_kv({'event': 'string-format', 'value': hello_str})
return hello_str
def print_hello(hello_str):
with tracer.start_span('println', child_of=get_current_span()) as span:
with span_in_context(span):
http_get(8082, 'publish', 'helloStr', hello_str)
span.log_kv({'event': 'println'})
def http_get(port, path, param, value):
url = 'http://localhost:%s/%s' % (port, path)
span = get_current_span()
span.set_tag(tags.HTTP_METHOD, 'GET')
span.set_tag(tags.HTTP_URL, url)
span.set_tag(tags.SPAN_KIND, tags.SPAN_KIND_RPC_CLIENT)
headers = {}
tracer.inject(span, Format.HTTP_HEADERS, headers)
r = requests.get(url, params={param: value}, headers=headers)
assert r.status_code == 200
return r.text
# main
assert len(sys.argv) == 3
tracer = init_tracer('hello-world')
hello_to = sys.argv[1]
greeting = sys.argv[2]
say_hello(hello_to, greeting)
# yield to IOLoop to flush the spans
time.sleep(2)
tracer.close()