/* * * Copyright 2018 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. * */ #ifndef GRPCPP_IMPL_CODEGEN_INTERCEPTED_CHANNEL_H #define GRPCPP_IMPL_CODEGEN_INTERCEPTED_CHANNEL_H // IWYU pragma: private #include <grpcpp/impl/codegen/channel_interface.h> namespace grpc { class CompletionQueue; namespace internal { class InterceptorBatchMethodsImpl; /// An InterceptedChannel is available to client Interceptors. An /// InterceptedChannel is unique to an interceptor, and when an RPC is started /// on this channel, only those interceptors that come after this interceptor /// see the RPC. class InterceptedChannel : public ChannelInterface { public: ~InterceptedChannel() override { channel_ = nullptr; } /// Get the current channel state. If the channel is in IDLE and /// \a try_to_connect is set to true, try to connect. grpc_connectivity_state GetState(bool try_to_connect) override { return channel_->GetState(try_to_connect); } private: InterceptedChannel(ChannelInterface* channel, size_t pos) : channel_(channel), interceptor_pos_(pos) {} Call CreateCall(const RpcMethod& method, grpc::ClientContext* context, grpc::CompletionQueue* cq) override { return channel_->CreateCallInternal(method, context, cq, interceptor_pos_); } void PerformOpsOnCall(CallOpSetInterface* ops, Call* call) override { return channel_->PerformOpsOnCall(ops, call); } void* RegisterMethod(const char* method) override { return channel_->RegisterMethod(method); } void NotifyOnStateChangeImpl(grpc_connectivity_state last_observed, gpr_timespec deadline, grpc::CompletionQueue* cq, void* tag) override { return channel_->NotifyOnStateChangeImpl(last_observed, deadline, cq, tag); } bool WaitForStateChangeImpl(grpc_connectivity_state last_observed, gpr_timespec deadline) override { return channel_->WaitForStateChangeImpl(last_observed, deadline); } grpc::CompletionQueue* CallbackCQ() override { return channel_->CallbackCQ(); } ChannelInterface* channel_; size_t interceptor_pos_; friend class InterceptorBatchMethodsImpl; }; } // namespace internal } // namespace grpc #endif // GRPCPP_IMPL_CODEGEN_INTERCEPTED_CHANNEL_H