| /* |
| * |
| * 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 |
| |
| #include <grpcpp/impl/codegen/channel_interface.h> |
| |
| namespace grpc { |
| |
| 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: |
| virtual ~InterceptedChannel() { 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, ClientContext* context, |
| 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, 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); |
| } |
| |
| 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 |