tap.go 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  1. /*
  2. *
  3. * Copyright 2016 gRPC authors.
  4. *
  5. * Licensed under the Apache License, Version 2.0 (the "License");
  6. * you may not use this file except in compliance with the License.
  7. * You may obtain a copy of the License at
  8. *
  9. * http://www.apache.org/licenses/LICENSE-2.0
  10. *
  11. * Unless required by applicable law or agreed to in writing, software
  12. * distributed under the License is distributed on an "AS IS" BASIS,
  13. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14. * See the License for the specific language governing permissions and
  15. * limitations under the License.
  16. *
  17. */
  18. // Package tap defines the function handles which are executed on the transport
  19. // layer of gRPC-Go and related information.
  20. //
  21. // # Experimental
  22. //
  23. // Notice: This API is EXPERIMENTAL and may be changed or removed in a
  24. // later release.
  25. package tap
  26. import (
  27. "context"
  28. "google.golang.org/grpc/metadata"
  29. )
  30. // Info defines the relevant information needed by the handles.
  31. type Info struct {
  32. // FullMethodName is the string of grpc method (in the format of
  33. // /package.service/method).
  34. FullMethodName string
  35. // Header contains the header metadata received.
  36. Header metadata.MD
  37. // TODO: More to be added.
  38. }
  39. // ServerInHandle defines the function which runs before a new stream is
  40. // created on the server side. If it returns a non-nil error, the stream will
  41. // not be created and an error will be returned to the client. If the error
  42. // returned is a status error, that status code and message will be used,
  43. // otherwise PermissionDenied will be the code and err.Error() will be the
  44. // message.
  45. //
  46. // It's intended to be used in situations where you don't want to waste the
  47. // resources to accept the new stream (e.g. rate-limiting). For other general
  48. // usages, please use interceptors.
  49. //
  50. // Note that it is executed in the per-connection I/O goroutine(s) instead of
  51. // per-RPC goroutine. Therefore, users should NOT have any
  52. // blocking/time-consuming work in this handle. Otherwise all the RPCs would
  53. // slow down. Also, for the same reason, this handle won't be called
  54. // concurrently by gRPC.
  55. type ServerInHandle func(ctx context.Context, info *Info) (context.Context, error)