shared.go 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. package otgrpc
  2. import (
  3. "strings"
  4. opentracing "github.com/opentracing/opentracing-go"
  5. "github.com/opentracing/opentracing-go/ext"
  6. "google.golang.org/grpc/metadata"
  7. )
  8. var (
  9. // Morally a const:
  10. gRPCComponentTag = opentracing.Tag{string(ext.Component), "gRPC"}
  11. )
  12. // metadataReaderWriter satisfies both the opentracing.TextMapReader and
  13. // opentracing.TextMapWriter interfaces.
  14. type metadataReaderWriter struct {
  15. metadata.MD
  16. }
  17. func (w metadataReaderWriter) Set(key, val string) {
  18. // The GRPC HPACK implementation rejects any uppercase keys here.
  19. //
  20. // As such, since the HTTP_HEADERS format is case-insensitive anyway, we
  21. // blindly lowercase the key (which is guaranteed to work in the
  22. // Inject/Extract sense per the OpenTracing spec).
  23. key = strings.ToLower(key)
  24. w.MD[key] = append(w.MD[key], val)
  25. }
  26. func (w metadataReaderWriter) ForeachKey(handler func(key, val string) error) error {
  27. for k, vals := range w.MD {
  28. for _, v := range vals {
  29. if err := handler(k, v); err != nil {
  30. return err
  31. }
  32. }
  33. }
  34. return nil
  35. }