THUAI6/CAPI/cpp/grpc/include/grpcpp/impl/codegen/intercepted_channel.h

/*
 *
 * 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