Set span id to tag

demedia-app · Sep 3, 2023 · 61dfd44 · 61dfd44
1 parent 27f3012
commit 61dfd44
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/peer/bridge/bridge.go b/peer/bridge/bridge.go
@@ -8,6 +8,7 @@ import (
 	"github.com/nbd-wtf/go-nostr"
 	"github.com/sithumonline/demedia-nostr/relayer"
 	"github.com/sithumonline/demedia-nostr/relayer/ql"
+	"go.opentelemetry.io/otel/attribute"
 	"go.opentelemetry.io/otel/propagation"
 	"go.opentelemetry.io/otel/trace"
 )
@@ -29,12 +30,14 @@ func (t *BridgeService) Ql(ctx context.Context, argType ql.BridgeArgs, replyType
 	}
 	ctx = propagation.TraceContext{}.Extract(ctx, call.Carrier)
 	ctx, span := t.tracer.Start(ctx, "ql.method")
+	span.SetAttributes(attribute.String("span_id", span.SpanContext().SpanID().String()))
 	defer span.End()
 	log := relayer.DefaultLogger()
 	log.InfofWithContext(ctx, "Received a Ql call, method: %s", call.Method)
 	switch call.Method {
 	case "saveEvent":
 		ctx, span := t.tracer.Start(ctx, "ql.method.saveEvent")
+		span.SetAttributes(attribute.String("span_id", span.SpanContext().SpanID().String()))
 		defer span.End()
 		var d nostr.Event
 		err := json.Unmarshal(call.Body, &d)
@@ -45,6 +48,7 @@ func (t *BridgeService) Ql(ctx context.Context, argType ql.BridgeArgs, replyType
 		return t.relay.Storage().SaveEvent(&d)
 	case "queryEvents":
 		ctx, span := t.tracer.Start(ctx, "ql.method.queryEvents")
+		span.SetAttributes(attribute.String("span_id", span.SpanContext().SpanID().String()))
 		defer span.End()
 		var d nostr.Filter
 		err := json.Unmarshal(call.Body, &d)
@@ -65,6 +69,7 @@ func (t *BridgeService) Ql(ctx context.Context, argType ql.BridgeArgs, replyType
 		return nil
 	case "deleteEvent":
 		ctx, span := t.tracer.Start(ctx, "ql.method.deleteEvent")
+		span.SetAttributes(attribute.String("span_id", span.SpanContext().SpanID().String()))
 		defer span.End()
 		var d nostr.Event
 		err := json.Unmarshal(call.Body, &d)

diff --git a/relayer/handlers.go b/relayer/handlers.go
@@ -18,6 +18,7 @@ import (
 	"github.com/nbd-wtf/go-nostr/nip11"
 	"github.com/nbd-wtf/go-nostr/nip42"
 	"github.com/sithumonline/demedia-nostr/relayer/hashutil"
+	"go.opentelemetry.io/otel/attribute"
 	"golang.org/x/exp/slices"
 )
 
@@ -45,6 +46,7 @@ var upgrader = websocket.Upgrader{
 
 func (s *Server) handleWebsocket(w http.ResponseWriter, r *http.Request) {
 	ctx, span := s.tracer.Start(r.Context(), "handleWebsocket")
+	span.SetAttributes(attribute.String("span_id", span.SpanContext().SpanID().String()))
 	defer span.End()
 	s.Log.InfofWithContext(ctx, "handling websocket request from %s", r.RemoteAddr)
 	store := s.relay.Storage()
@@ -73,6 +75,7 @@ func (s *Server) handleWebsocket(w http.ResponseWriter, r *http.Request) {
 	// reader
 	go func() {
 		ctx, span := s.tracer.Start(ctx, "handleWebsocket.reader")
+		span.SetAttributes(attribute.String("span_id", span.SpanContext().SpanID().String()))
 		defer span.End()
 		defer func() {
 			ticker.Stop()
@@ -99,6 +102,7 @@ func (s *Server) handleWebsocket(w http.ResponseWriter, r *http.Request) {
 		s.Log.InfofWithContext(ctx, "auth challenge sent")
 		for {
 			ctx, span := s.tracer.Start(ctx, "handleWebsocket.reader.for")
+			span.SetAttributes(attribute.String("span_id", span.SpanContext().SpanID().String()))
 			s.Log.InfofWithContext(ctx, "inside for loop and waiting for message")
 			typ, message, err := conn.ReadMessage()
 			if err != nil {
@@ -120,6 +124,7 @@ func (s *Server) handleWebsocket(w http.ResponseWriter, r *http.Request) {
 
 			go func(message []byte) {
 				ctx, span := s.tracer.Start(ctx, "handleWebsocket.reader.for.go")
+				span.SetAttributes(attribute.String("span_id", span.SpanContext().SpanID().String()))
 				defer span.End()
 				s.Log.InfofWithContext(ctx, "initializing go routine for message")
 				var notice string