From 48a96d8bc8f6442bb92551302dcec151e56cc5ea Mon Sep 17 00:00:00 2001 From: Caleb Gilmour Date: Wed, 17 Jun 2020 00:15:28 +0000 Subject: [PATCH] contrib/go.mongodb.org/mongo-driver: add span type --- contrib/go.mongodb.org/mongo-driver/mongo/mongo.go | 1 + contrib/go.mongodb.org/mongo-driver/mongo/mongo_test.go | 1 + 2 files changed, 2 insertions(+) diff --git a/contrib/go.mongodb.org/mongo-driver/mongo/mongo.go b/contrib/go.mongodb.org/mongo-driver/mongo/mongo.go index 132774c5cf..d5b7647bf6 100644 --- a/contrib/go.mongodb.org/mongo-driver/mongo/mongo.go +++ b/contrib/go.mongodb.org/mongo-driver/mongo/mongo.go @@ -39,6 +39,7 @@ func (m *monitor) Started(ctx context.Context, evt *event.CommandStartedEvent) { hostname, port := peerInfo(evt) b, _ := bson.MarshalExtJSON(evt.Command, false, false) opts := []ddtrace.StartSpanOption{ + tracer.SpanType(ext.SpanTypeMongoDB), tracer.ServiceName(m.cfg.serviceName), tracer.ResourceName("mongo." + evt.CommandName), tracer.Tag(ext.DBInstance, evt.DatabaseName), diff --git a/contrib/go.mongodb.org/mongo-driver/mongo/mongo_test.go b/contrib/go.mongodb.org/mongo-driver/mongo/mongo_test.go index 41507e84c9..39bbbf868a 100644 --- a/contrib/go.mongodb.org/mongo-driver/mongo/mongo_test.go +++ b/contrib/go.mongodb.org/mongo-driver/mongo/mongo_test.go @@ -65,6 +65,7 @@ func Test(t *testing.T) { assert.Equal(t, spans[0].TraceID(), spans[1].TraceID()) s := spans[0] + assert.Equal(t, ext.SpanTypeMongoDB, s.Tag(ext.SpanType)) assert.Equal(t, "mongo", s.Tag(ext.ServiceName)) assert.Equal(t, "mongo.insert", s.Tag(ext.ResourceName)) assert.Equal(t, hostname, s.Tag(ext.PeerHostname))