Add further attributes to span in tracing package

hramazani · Dec 8, 2022 · b8bbe44 · b8bbe44
1 parent 2a142f3
commit b8bbe44
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/tracing/cogs/cogs.go b/tracing/cogs/cogs.go
@@ -0,0 +1,8 @@
+package cogs
+
+// const values used in setting span attributes for COGS: Cost Of Goods Sold.
+const (
+	CostOfGoodsKey       = "cogs"
+	CostOfGoodsMethodKey = "cogs.method"
+	CostOfGoodsAccountID = "cogs.accountID"
+)
diff --git a/tracing/http.go b/tracing/http.go
@@ -11,6 +11,7 @@ import (
 	"go.opencensus.io/trace"
 
 	"github.com/infobloxopen/atlas-app-toolkit/auth"
+	"github.com/infobloxopen/atlas-app-toolkit/tracing/cogs"
 )
 
 const (
@@ -137,6 +138,10 @@ type Handler struct {
 func (h *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	span := trace.FromContext(r.Context())
 
+	// Add accountID to span attributes
+	accountID, _ := auth.GetAccountID(r.Context(), nil)
+	span.AddAttributes(trace.StringAttribute(cogs.CostOfGoodsAccountID, accountID))
+
 	withHeaders := h.options.spanWithHeaders != nil && h.options.spanWithHeaders(r)
 	withPayload := h.options.spanWithPayload != nil && h.options.spanWithPayload(r)