Added option to set timestamps to pb response

Author: kostz

internal/lightstep_pb/grpc/server.go

@@ -3,8 +3,10 @@ package grpc
 import (
 	"context"
 	"errors"
+	"google.golang.org/protobuf/types/known/timestamppb"
 	"net"
 	"sync"
+	"time"
 
 	"go.opentelemetry.io/collector/component/componentstatus"
 	"go.opentelemetry.io/collector/config/configgrpc"
@@ -93,15 +95,17 @@ func (s *ServerGRPC) Report(ctx context.Context, rq *pb.ReportRequest) (*pb.Repo
 		spanCount     int
 	)
 	ctx = client.NewContext(ctx, client.Info{})
-
+	receivedAt := time.Now()
 	ctx = s.obsreport.StartTracesOp(ctx)
 	spanCount = len(rq.Spans)
 	s.logger.Debug("report", zap.Any("incoming", rq))
 	lr := lightstep_pb.NewLightstepRequest(rq, s.telemetry, transport)
 	if projectTraces, err = lr.ToOtel(ctx); err != nil {
 		s.telemetry.IncrementFailed(transport, 1)
 		return &pb.ReportResponse{
-			Errors: []string{err.Error()},
+			Errors:            []string{err.Error()},
+			ReceiveTimestamp:  timestamppb.New(receivedAt),
+			TransmitTimestamp: timestamppb.Now(),
 		}, err
 	}
 	s.telemetry.IncrementProcessed(transport, 1)
@@ -118,11 +122,15 @@ func (s *ServerGRPC) Report(ctx context.Context, rq *pb.ReportRequest) (*pb.Repo
 
 	if err != nil {
 		return &pb.ReportResponse{
-			Errors: []string{err.Error()},
+			Errors:            []string{err.Error()},
+			ReceiveTimestamp:  timestamppb.New(receivedAt),
+			TransmitTimestamp: timestamppb.Now(),
 		}, err
 	}
 
 	return &pb.ReportResponse{
-		Errors: nil,
+		Errors:            nil,
+		ReceiveTimestamp:  timestamppb.New(receivedAt),
+		TransmitTimestamp: timestamppb.Now(),
 	}, nil
 }