File: dapperish.go

package info (click to toggle)
golang-github-opentracing-basictracer-go 1.1.0-3
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, forky, sid, trixie
  • size: 240 kB
  • sloc: makefile: 24
file content (89 lines) | stat: -rw-r--r-- 2,219 bytes parent folder | download
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()
}