forked from opentracing/basictracer-go
-
Notifications
You must be signed in to change notification settings - Fork 0
/
dapperish.go
82 lines (70 loc) · 1.94 KB
/
dapperish.go
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
73
74
75
76
77
78
79
80
81
82
package main
import (
"bufio"
"bytes"
"fmt"
"io/ioutil"
"log"
"net/http"
"os"
"runtime"
"strings"
"github.com/opentracing/basictracer-go/examples/dapperish"
opentracing "github.com/opentracing/opentracing-go"
)
func client() {
reader := bufio.NewReader(os.Stdin)
for {
span := opentracing.StartSpan("getInput")
ctx := opentracing.BackgroundContextWithSpan(span)
// Make sure that global baggage propagation works.
span.SetBaggageItem("User", os.Getenv("USER"))
span.LogEventWithPayload("ctx", ctx)
fmt.Print("\n\nEnter text (empty string to exit): ")
text, _ := reader.ReadString('\n')
text = strings.TrimSpace(text)
if len(text) == 0 {
fmt.Println("Exiting.")
os.Exit(0)
}
span.LogEvent(text)
httpClient := &http.Client{}
httpReq, _ := http.NewRequest("POST", "http://localhost:8080/", bytes.NewReader([]byte(text)))
textCarrier := opentracing.HTTPHeaderTextMapCarrier(httpReq.Header)
err := span.Tracer().Inject(span, opentracing.TextMap, textCarrier)
if err != nil {
panic(err)
}
resp, err := httpClient.Do(httpReq)
if err != nil {
span.LogEventWithPayload("error", err)
} else {
span.LogEventWithPayload("got response", resp)
}
span.Finish()
}
}
func server() {
http.HandleFunc("/", func(w http.ResponseWriter, req *http.Request) {
textCarrier := opentracing.HTTPHeaderTextMapCarrier(req.Header)
serverSpan, err := opentracing.GlobalTracer().Join(
"serverSpan", opentracing.TextMap, textCarrier)
if err != nil {
panic(err)
}
serverSpan.SetTag("component", "server")
defer serverSpan.Finish()
fullBody, err := ioutil.ReadAll(req.Body)
if err != nil {
serverSpan.LogEventWithPayload("body read error", err)
}
serverSpan.LogEventWithPayload("got request with body", string(fullBody))
})
log.Fatal(http.ListenAndServe(":8080", nil))
}
func main() {
opentracing.InitGlobalTracer(dapperish.NewTracer("dapperish_tester"))
go server()
go client()
runtime.Goexit()
}