vbatts/microservices-demo
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

frontend/checkoutservice: add tracing

Browse source 
Signed-off-by: Ahmet Alp Balkan <ahmetb@google.com>
This commit is contained in:
Ahmet Alp Balkan 2018-07-12 10:46:43 -07:00
parent 89618dbeb6
commit 555c8a4c05
4 changed files with 39 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/checkoutservice/Dockerfile
View file

@ -7,7 +7,9 @@ RUN go get -d github.com/google/uuid \
google.golang.org/grpc \
google.golang.org/grpc/codes \
google.golang.org/grpc/status \
go.opencensus.io/plugin/ocgrpc
go.opencensus.io/plugin/ocgrpc \
go.opencensus.io/exporter/stackdriver \
go.opencensus.io/trace
COPY . .
# get remaining dependencies
RUN go get -d ./...

16
src/checkoutservice/main.go
View file

@ -8,7 +8,9 @@ import (
"os"
"github.com/google/uuid"
"go.opencensus.io/exporter/stackdriver"
"go.opencensus.io/plugin/ocgrpc"
"go.opencensus.io/trace"
"google.golang.org/grpc"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
@ -37,6 +39,8 @@ func main() {
port = os.Getenv("PORT")
}
initTracing()
svc := new(checkoutService)
mustMapEnv(&svc.shippingSvcAddr, "SHIPPING_SERVICE_ADDR")
mustMapEnv(&svc.productCatalogSvcAddr, "PRODUCT_CATALOG_SERVICE_ADDR")
@ -57,6 +61,18 @@ func main() {
log.Fatal(srv.Serve(lis))
}
func initTracing() {
exporter, err := stackdriver.NewExporter(stackdriver.Options{})
if err != nil {
log.Printf("failed to initialize stackdriver exporter: %+v", err)
log.Println("skipping uploading traces to stackdriver")
} else {
trace.RegisterExporter(exporter)
trace.ApplyConfig(trace.Config{DefaultSampler: trace.AlwaysSample()})
log.Println("registered stackdriver tracing")
}
}
func mustMapEnv(target *string, envKey string) {
v := os.Getenv(envKey)
if v == "" {