interceptor.go 3.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. /*
  2. *
  3. * Copyright 2016, Google Inc.
  4. * All rights reserved.
  5. *
  6. * Redistribution and use in source and binary forms, with or without
  7. * modification, are permitted provided that the following conditions are
  8. * met:
  9. *
  10. * * Redistributions of source code must retain the above copyright
  11. * notice, this list of conditions and the following disclaimer.
  12. * * Redistributions in binary form must reproduce the above
  13. * copyright notice, this list of conditions and the following disclaimer
  14. * in the documentation and/or other materials provided with the
  15. * distribution.
  16. * * Neither the name of Google Inc. nor the names of its
  17. * contributors may be used to endorse or promote products derived from
  18. * this software without specific prior written permission.
  19. *
  20. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  21. * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  22. * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  23. * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  24. * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  25. * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  26. * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  27. * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  28. * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  29. * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  30. * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  31. *
  32. */
  33. package grpc
  34. import (
  35. "golang.org/x/net/context"
  36. )
  37. // UnaryServerInfo consists of various information about a unary RPC on
  38. // server side. All per-rpc information may be mutated by the interceptor.
  39. type UnaryServerInfo struct {
  40. // Server is the service implementation the user provides. This is read-only.
  41. Server interface{}
  42. // FullMethod is the full RPC method string, i.e., /package.service/method.
  43. FullMethod string
  44. }
  45. // UnaryHandler defines the handler invoked by UnaryServerInterceptor to complete the normal
  46. // execution of a unary RPC.
  47. type UnaryHandler func(ctx context.Context, req interface{}) (interface{}, error)
  48. // UnaryServerInterceptor provides a hook to intercept the execution of a unary RPC on the server. info
  49. // contains all the information of this RPC the interceptor can operate on. And handler is the wrapper
  50. // of the service method implementation. It is the responsibility of the interceptor to invoke handler
  51. // to complete the RPC.
  52. type UnaryServerInterceptor func(ctx context.Context, req interface{}, info *UnaryServerInfo, handler UnaryHandler) (resp interface{}, err error)
  53. // StreamServerInfo consists of various information about a streaming RPC on
  54. // server side. All per-rpc information may be mutated by the interceptor.
  55. type StreamServerInfo struct {
  56. // FullMethod is the full RPC method string, i.e., /package.service/method.
  57. FullMethod string
  58. // IsClientStream indicates whether the RPC is a client streaming RPC.
  59. IsClientStream bool
  60. // IsServerStream indicates whether the RPC is a server streaming RPC.
  61. IsServerStream bool
  62. }
  63. // StreamServerInterceptor provides a hook to intercept the execution of a streaming RPC on the server.
  64. // info contains all the information of this RPC the interceptor can operate on. And handler is the
  65. // service method implementation. It is the responsibility of the interceptor to invoke handler to
  66. // complete the RPC.
  67. type StreamServerInterceptor func(srv interface{}, ss ServerStream, info *StreamServerInfo, handler StreamHandler) error