contrib/go.mongodb.org/mongo-driver: add span type (#683)

DataDog · Jun 29, 2020 · 057f9e8 · 057f9e8
1 parent 99f8eaa
commit 057f9e8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -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))