call.go 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. /*
  2. *
  3. * Copyright 2014 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 grpc
  19. import (
  20. "context"
  21. )
  22. // Invoke sends the RPC request on the wire and returns after response is
  23. // received. This is typically called by generated code.
  24. //
  25. // All errors returned by Invoke are compatible with the status package.
  26. func (cc *ClientConn) Invoke(ctx context.Context, method string, args, reply interface{}, opts ...CallOption) error {
  27. // allow interceptor to see all applicable call options, which means those
  28. // configured as defaults from dial option as well as per-call options
  29. opts = combine(cc.dopts.callOptions, opts)
  30. if cc.dopts.unaryInt != nil {
  31. return cc.dopts.unaryInt(ctx, method, args, reply, cc, invoke, opts...)
  32. }
  33. return invoke(ctx, method, args, reply, cc, opts...)
  34. }
  35. func combine(o1 []CallOption, o2 []CallOption) []CallOption {
  36. // we don't use append because o1 could have extra capacity whose
  37. // elements would be overwritten, which could cause inadvertent
  38. // sharing (and race conditions) between concurrent calls
  39. if len(o1) == 0 {
  40. return o2
  41. } else if len(o2) == 0 {
  42. return o1
  43. }
  44. ret := make([]CallOption, len(o1)+len(o2))
  45. copy(ret, o1)
  46. copy(ret[len(o1):], o2)
  47. return ret
  48. }
  49. // Invoke sends the RPC request on the wire and returns after response is
  50. // received. This is typically called by generated code.
  51. //
  52. // DEPRECATED: Use ClientConn.Invoke instead.
  53. func Invoke(ctx context.Context, method string, args, reply interface{}, cc *ClientConn, opts ...CallOption) error {
  54. return cc.Invoke(ctx, method, args, reply, opts...)
  55. }
  56. var unaryStreamDesc = &StreamDesc{ServerStreams: false, ClientStreams: false}
  57. func invoke(ctx context.Context, method string, req, reply interface{}, cc *ClientConn, opts ...CallOption) error {
  58. cs, err := newClientStream(ctx, unaryStreamDesc, cc, method, opts...)
  59. if err != nil {
  60. return err
  61. }
  62. if err := cs.SendMsg(req); err != nil {
  63. return err
  64. }
  65. return cs.RecvMsg(reply)
  66. }