This repository has been archived by the owner on Mar 6, 2023. It is now read-only.
/
dapperish.go
89 lines (76 loc) 路 2.17 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
83
84
85
86
87
88
89
package main
import (
"bufio"
"bytes"
"fmt"
"io/ioutil"
golog "log"
"net/http"
"os"
"runtime"
"strings"
"golang.org/x/net/context"
"github.com/opentracing/basictracer-go/examples/dapperish"
opentracing "github.com/opentracing/opentracing-go"
"github.com/opentracing/opentracing-go/ext"
"github.com/opentracing/opentracing-go/log"
)
func client() {
reader := bufio.NewReader(os.Stdin)
for {
span := opentracing.StartSpan("getInput")
ctx := opentracing.ContextWithSpan(context.Background(), span)
// Make sure that global baggage propagation works.
span.SetBaggageItem("User", os.Getenv("USER"))
span.LogFields(log.Object("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.LogFields(log.String("user text", text))
httpClient := &http.Client{}
httpReq, _ := http.NewRequest("POST", "http://localhost:8080/", bytes.NewReader([]byte(text)))
textCarrier := opentracing.HTTPHeadersCarrier(httpReq.Header)
err := span.Tracer().Inject(span.Context(), opentracing.TextMap, textCarrier)
if err != nil {
panic(err)
}
resp, err := httpClient.Do(httpReq)
if err != nil {
span.LogFields(log.Error(err))
} else {
span.LogFields(log.Object("response", resp))
}
span.Finish()
}
}
func server() {
http.HandleFunc("/", func(w http.ResponseWriter, req *http.Request) {
textCarrier := opentracing.HTTPHeadersCarrier(req.Header)
wireSpanContext, err := opentracing.GlobalTracer().Extract(
opentracing.TextMap, textCarrier)
if err != nil {
panic(err)
}
serverSpan := opentracing.GlobalTracer().StartSpan(
"serverSpan",
ext.RPCServerOption(wireSpanContext))
serverSpan.SetTag("component", "server")
defer serverSpan.Finish()
fullBody, err := ioutil.ReadAll(req.Body)
if err != nil {
serverSpan.LogFields(log.Error(err))
}
serverSpan.LogFields(log.String("request body", string(fullBody)))
})
golog.Fatal(http.ListenAndServe(":8080", nil))
}
func main() {
opentracing.InitGlobalTracer(dapperish.NewTracer("dapperish_tester"))
go server()
go client()
runtime.Goexit()
}