Craig Tiller | c496575 | 2015-02-09 09:51:00 -0800 | [diff] [blame] | 1 | /* |
| 2 | * |
Craig Tiller | 0605995 | 2015-02-18 08:34:56 -0800 | [diff] [blame] | 3 | * Copyright 2015, Google Inc. |
Craig Tiller | c496575 | 2015-02-09 09:51:00 -0800 | [diff] [blame] | 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 | |
Craig Tiller | a5c0e7b | 2015-02-11 13:14:49 -0800 | [diff] [blame] | 34 | #include <grpc++/impl/call.h> |
Yang Gao | 5f4539f | 2015-03-06 16:11:16 -0800 | [diff] [blame] | 35 | |
| 36 | #include <grpc/support/alloc.h> |
yang-g | 8c2be9f | 2015-08-19 16:28:09 -0700 | [diff] [blame] | 37 | #include <grpc++/channel.h> |
yang-g | 9e2f90c | 2015-08-21 15:35:03 -0700 | [diff] [blame] | 38 | #include <grpc++/client_context.h> |
| 39 | #include <grpc++/support/byte_buffer.h> |
Vijay Pai | fad38ba | 2015-04-15 01:06:58 -0700 | [diff] [blame] | 40 | #include "src/core/profiling/timers.h" |
Yang Gao | d5a04bd | 2015-02-11 00:04:32 -0800 | [diff] [blame] | 41 | |
Craig Tiller | c496575 | 2015-02-09 09:51:00 -0800 | [diff] [blame] | 42 | namespace grpc { |
| 43 | |
yang-g | e21908f | 2015-08-25 13:47:51 -0700 | [diff] [blame] | 44 | void FillMetadataMap( |
| 45 | grpc_metadata_array* arr, |
| 46 | std::multimap<grpc::string_ref, grpc::string_ref>* metadata) { |
Craig Tiller | 789471c | 2015-06-04 16:19:22 -0700 | [diff] [blame] | 47 | for (size_t i = 0; i < arr->count; i++) { |
| 48 | // TODO(yangg) handle duplicates? |
yang-g | e21908f | 2015-08-25 13:47:51 -0700 | [diff] [blame] | 49 | metadata->insert(std::pair<grpc::string_ref, grpc::string_ref>( |
| 50 | arr->metadata[i].key, grpc::string_ref(arr->metadata[i].value, |
| 51 | arr->metadata[i].value_length))); |
Craig Tiller | 789471c | 2015-06-04 16:19:22 -0700 | [diff] [blame] | 52 | } |
| 53 | grpc_metadata_array_destroy(arr); |
| 54 | grpc_metadata_array_init(arr); |
| 55 | } |
| 56 | |
| 57 | // TODO(yangg) if the map is changed before we send, the pointers will be a |
| 58 | // mess. Make sure it does not happen. |
| 59 | grpc_metadata* FillMetadataArray( |
| 60 | const std::multimap<grpc::string, grpc::string>& metadata) { |
| 61 | if (metadata.empty()) { |
| 62 | return nullptr; |
| 63 | } |
| 64 | grpc_metadata* metadata_array = |
| 65 | (grpc_metadata*)gpr_malloc(metadata.size() * sizeof(grpc_metadata)); |
| 66 | size_t i = 0; |
| 67 | for (auto iter = metadata.cbegin(); iter != metadata.cend(); ++iter, ++i) { |
| 68 | metadata_array[i].key = iter->first.c_str(); |
| 69 | metadata_array[i].value = iter->second.c_str(); |
| 70 | metadata_array[i].value_length = iter->second.size(); |
| 71 | } |
| 72 | return metadata_array; |
| 73 | } |
| 74 | |
Craig Tiller | 573523f | 2015-02-17 07:38:26 -0800 | [diff] [blame] | 75 | Call::Call(grpc_call* call, CallHook* call_hook, CompletionQueue* cq) |
Yang Gao | 3921c56 | 2015-04-30 16:07:06 -0700 | [diff] [blame] | 76 | : call_hook_(call_hook), cq_(cq), call_(call), max_message_size_(-1) {} |
| 77 | |
| 78 | Call::Call(grpc_call* call, CallHook* call_hook, CompletionQueue* cq, |
| 79 | int max_message_size) |
Yang Gao | c71a9d2 | 2015-05-04 00:22:12 -0700 | [diff] [blame] | 80 | : call_hook_(call_hook), |
| 81 | cq_(cq), |
| 82 | call_(call), |
Yang Gao | 3921c56 | 2015-04-30 16:07:06 -0700 | [diff] [blame] | 83 | max_message_size_(max_message_size) {} |
Yang Gao | 7eb7d75 | 2015-02-10 14:23:15 -0800 | [diff] [blame] | 84 | |
Craig Tiller | 50a7a68 | 2015-06-04 12:53:40 -0700 | [diff] [blame] | 85 | void Call::PerformOps(CallOpSetInterface* ops) { |
Yang Gao | 3921c56 | 2015-04-30 16:07:06 -0700 | [diff] [blame] | 86 | if (max_message_size_ > 0) { |
Craig Tiller | 50a7a68 | 2015-06-04 12:53:40 -0700 | [diff] [blame] | 87 | ops->set_max_message_size(max_message_size_); |
Yang Gao | 3921c56 | 2015-04-30 16:07:06 -0700 | [diff] [blame] | 88 | } |
Craig Tiller | 50a7a68 | 2015-06-04 12:53:40 -0700 | [diff] [blame] | 89 | call_hook_->PerformOpsOnCall(ops, this); |
Yang Gao | a52ea7b | 2015-02-11 10:31:29 -0800 | [diff] [blame] | 90 | } |
Craig Tiller | c496575 | 2015-02-09 09:51:00 -0800 | [diff] [blame] | 91 | |
| 92 | } // namespace grpc |