tap.go 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  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. Everything here is EXPERIMENTAL.
  20. package tap
  21. import (
  22. "golang.org/x/net/context"
  23. )
  24. // Info defines the relevant information needed by the handles.
  25. type Info struct {
  26. // FullMethodName is the string of grpc method (in the format of
  27. // /package.service/method).
  28. FullMethodName string
  29. // TODO: More to be added.
  30. }
  31. // ServerInHandle defines the function which runs before a new stream is created
  32. // on the server side. If it returns a non-nil error, the stream will not be
  33. // created and a RST_STREAM will be sent back to the client with REFUSED_STREAM.
  34. // The client will receive an RPC error "code = Unavailable, desc = stream
  35. // terminated by RST_STREAM with error code: REFUSED_STREAM".
  36. //
  37. // It's intended to be used in situations where you don't want to waste the
  38. // resources to accept the new stream (e.g. rate-limiting). And the content of
  39. // the error will be ignored and won't be sent back to the client. For other
  40. // general usages, please use interceptors.
  41. //
  42. // Note that it is executed in the per-connection I/O goroutine(s) instead of
  43. // per-RPC goroutine. Therefore, users should NOT have any
  44. // blocking/time-consuming work in this handle. Otherwise all the RPCs would
  45. // slow down. Also, for the same reason, this handle won't be called
  46. // concurrently by gRPC.
  47. type ServerInHandle func(ctx context.Context, info *Info) (context.Context, error)