module/apmgrpc: set span destination context (#861)

Update the apmgrpc client interceptor to record
span destination address/port and context for
service maps.

CHANGELOG.asciidoc

@@ -26,6 +26,7 @@ https://github.com/elastic/apm-agent-go/compare/v1.9.0...master[View commits]
 - module/apmsql: add tracingDriver.Unwrap method to get underlying driver {pull}#849[#(849)]
 - module/apmgopgv10: add support for github.com/go-pg/pg/v10 {pull}857[(#857)]
 - Enable central configuration of "sanitize_field_names" {pull}856[(#856)]
+- module/apmgrpc: set span destination context {pull}861[(#861)]
 
 [[release-notes-1.x]]
 === Go Agent version 1.x

module/apmgrpc/client.go

@@ -20,10 +20,13 @@
 package apmgrpc // import "go.elastic.co/apm/module/apmgrpc"
 
 import (
+	"net"
+
 	"golang.org/x/net/context"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/metadata"
+	"google.golang.org/grpc/peer"
 
 	"go.elastic.co/apm"
 	"go.elastic.co/apm/module/apmhttp"
@@ -48,13 +51,25 @@ func NewUnaryClientInterceptor(o ...ClientOption) grpc.UnaryClientInterceptor {
 		invoker grpc.UnaryInvoker,
 		opts ...grpc.CallOption,
 	) error {
+		var peer peer.Peer // maybe set after call if span != nil
 		span, ctx := startSpan(ctx, method)
 		if span != nil {
 			defer span.End()
+			opts = append(opts, grpc.Peer(&peer))
 		}
 		err := invoker(ctx, method, req, resp, cc, opts...)
 		if span != nil {
 			setSpanOutcome(span, err)
+			if peer.Addr != nil {
+				if tcpAddr, ok := peer.Addr.(*net.TCPAddr); ok {
+					span.Context.SetDestinationAddress(tcpAddr.IP.String(), tcpAddr.Port)
+				}
+				addrString := peer.Addr.String()
+				span.Context.SetDestinationService(apm.DestinationServiceSpanContext{
+					Name:     addrString,
+					Resource: addrString,
+				})
+			}
 		}
 		return err
 	}

module/apmgrpc/client_test.go

@@ -20,6 +20,7 @@
 package apmgrpc_test
 
 import (
+	"net"
 	"os"
 	"testing"
 
@@ -56,6 +57,7 @@ func testClientSpan(t *testing.T, traceparentHeaders ...string) {
 	defer serverTracer.Close()
 	s, _, addr := newServer(t, serverTracer)
 	defer s.GracefulStop()
+	tcpAddr := addr.(*net.TCPAddr)
 
 	conn, client := newClient(t, addr)
 	defer conn.Close()
@@ -84,6 +86,15 @@ func testClientSpan(t *testing.T, traceparentHeaders ...string) {
 	assert.Equal(t, "/helloworld.Greeter/SayHello", clientSpans[0].Name)
 	assert.Equal(t, "external", clientSpans[0].Type)
 	assert.Equal(t, "grpc", clientSpans[0].Subtype)
+	assert.Equal(t, &model.DestinationSpanContext{
+		Address: tcpAddr.IP.String(),
+		Port:    tcpAddr.Port,
+		Service: &model.DestinationServiceSpanContext{
+			Type:     "external",
+			Name:     tcpAddr.String(),
+			Resource: tcpAddr.String(),
+		},
+	}, clientSpans[0].Context.Destination)
 
 	serverTracer.Flush(nil)
 	serverTransactions := serverTransport.Payloads().Transactions