| /* |
| * |
| * Copyright 2015 gRPC authors. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| * |
| */ |
| |
| #include <grpc++/channel.h> |
| |
| #include <chrono> |
| #include <condition_variable> |
| #include <memory> |
| #include <mutex> |
| |
| #include <grpc++/client_context.h> |
| #include <grpc++/completion_queue.h> |
| #include <grpc++/impl/call.h> |
| #include <grpc++/impl/codegen/completion_queue_tag.h> |
| #include <grpc++/impl/grpc_library.h> |
| #include <grpc++/impl/rpc_method.h> |
| #include <grpc++/security/credentials.h> |
| #include <grpc++/support/channel_arguments.h> |
| #include <grpc++/support/config.h> |
| #include <grpc++/support/status.h> |
| #include <grpc++/support/time.h> |
| #include <grpc/grpc.h> |
| #include <grpc/slice.h> |
| #include <grpc/support/alloc.h> |
| #include <grpc/support/log.h> |
| #include <grpc/support/sync.h> |
| #include <grpc/support/thd.h> |
| #include <grpc/support/time.h> |
| #include <grpc/support/useful.h> |
| #include "src/core/lib/profiling/timers.h" |
| #include "src/core/lib/support/env.h" |
| #include "src/core/lib/support/string.h" |
| |
| namespace grpc { |
| |
| static internal::GrpcLibraryInitializer g_gli_initializer; |
| Channel::Channel(const grpc::string& host, grpc_channel* channel) |
| : host_(host), c_channel_(channel) { |
| g_gli_initializer.summon(); |
| } |
| |
| Channel::~Channel() { grpc_channel_destroy(c_channel_); } |
| |
| namespace { |
| |
| grpc::string GetChannelInfoField(grpc_channel* channel, |
| grpc_channel_info* channel_info, |
| char*** channel_info_field) { |
| char* value = NULL; |
| memset(channel_info, 0, sizeof(*channel_info)); |
| *channel_info_field = &value; |
| grpc_channel_get_info(channel, channel_info); |
| if (value == NULL) return ""; |
| grpc::string result = value; |
| gpr_free(value); |
| return result; |
| } |
| |
| } // namespace |
| |
| grpc::string Channel::GetLoadBalancingPolicyName() const { |
| grpc_channel_info channel_info; |
| return GetChannelInfoField(c_channel_, &channel_info, |
| &channel_info.lb_policy_name); |
| } |
| |
| grpc::string Channel::GetServiceConfigJSON() const { |
| grpc_channel_info channel_info; |
| return GetChannelInfoField(c_channel_, &channel_info, |
| &channel_info.service_config_json); |
| } |
| |
| internal::Call Channel::CreateCall(const internal::RpcMethod& method, |
| ClientContext* context, |
| CompletionQueue* cq) { |
| const bool kRegistered = method.channel_tag() && context->authority().empty(); |
| grpc_call* c_call = NULL; |
| if (kRegistered) { |
| c_call = grpc_channel_create_registered_call( |
| c_channel_, context->propagate_from_call_, |
| context->propagation_options_.c_bitmask(), cq->cq(), |
| method.channel_tag(), context->raw_deadline(), nullptr); |
| } else { |
| const char* host_str = NULL; |
| if (!context->authority().empty()) { |
| host_str = context->authority_.c_str(); |
| } else if (!host_.empty()) { |
| host_str = host_.c_str(); |
| } |
| grpc_slice method_slice = SliceFromCopiedString(method.name()); |
| grpc_slice host_slice; |
| if (host_str != nullptr) { |
| host_slice = SliceFromCopiedString(host_str); |
| } |
| c_call = grpc_channel_create_call( |
| c_channel_, context->propagate_from_call_, |
| context->propagation_options_.c_bitmask(), cq->cq(), method_slice, |
| host_str == nullptr ? nullptr : &host_slice, context->raw_deadline(), |
| nullptr); |
| grpc_slice_unref(method_slice); |
| if (host_str != nullptr) { |
| grpc_slice_unref(host_slice); |
| } |
| } |
| grpc_census_call_set_context(c_call, context->census_context()); |
| context->set_call(c_call, shared_from_this()); |
| return internal::Call(c_call, this, cq); |
| } |
| |
| void Channel::PerformOpsOnCall(internal::CallOpSetInterface* ops, |
| internal::Call* call) { |
| static const size_t MAX_OPS = 8; |
| size_t nops = 0; |
| grpc_op cops[MAX_OPS]; |
| ops->FillOps(call->call(), cops, &nops); |
| GPR_ASSERT(GRPC_CALL_OK == |
| grpc_call_start_batch(call->call(), cops, nops, ops, nullptr)); |
| } |
| |
| void* Channel::RegisterMethod(const char* method) { |
| return grpc_channel_register_call( |
| c_channel_, method, host_.empty() ? NULL : host_.c_str(), nullptr); |
| } |
| |
| grpc_connectivity_state Channel::GetState(bool try_to_connect) { |
| return grpc_channel_check_connectivity_state(c_channel_, try_to_connect); |
| } |
| |
| namespace { |
| |
| class TagSaver final : public internal::CompletionQueueTag { |
| public: |
| explicit TagSaver(void* tag) : tag_(tag) {} |
| ~TagSaver() override {} |
| bool FinalizeResult(void** tag, bool* status) override { |
| *tag = tag_; |
| delete this; |
| return true; |
| } |
| |
| private: |
| void* tag_; |
| }; |
| |
| } // namespace |
| |
| void Channel::NotifyOnStateChangeImpl(grpc_connectivity_state last_observed, |
| gpr_timespec deadline, |
| CompletionQueue* cq, void* tag) { |
| TagSaver* tag_saver = new TagSaver(tag); |
| grpc_channel_watch_connectivity_state(c_channel_, last_observed, deadline, |
| cq->cq(), tag_saver); |
| } |
| |
| bool Channel::WaitForStateChangeImpl(grpc_connectivity_state last_observed, |
| gpr_timespec deadline) { |
| CompletionQueue cq; |
| bool ok = false; |
| void* tag = NULL; |
| NotifyOnStateChangeImpl(last_observed, deadline, &cq, NULL); |
| cq.Next(&tag, &ok); |
| GPR_ASSERT(tag == NULL); |
| return ok; |
| } |
| |
| } // namespace grpc |