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

/*
 *
 * Copyright 2019 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_DELEGATING_CHANNEL_H
#define GRPCPP_IMPL_CODEGEN_DELEGATING_CHANNEL_H

// IWYU pragma: private

#include <memory>

#include <grpcpp/impl/codegen/channel_interface.h>

namespace grpc
{
    namespace experimental
    {

        class DelegatingChannel : public grpc::ChannelInterface
        {
        public:
            ~DelegatingChannel() override
            {
            }

            explicit DelegatingChannel(
                std::shared_ptr<grpc::ChannelInterface> delegate_channel
            ) :
                delegate_channel_(delegate_channel)
            {
            }

            grpc_connectivity_state GetState(bool try_to_connect) override
            {
                return delegate_channel()->GetState(try_to_connect);
            }

            std::shared_ptr<grpc::ChannelInterface> delegate_channel()
            {
                return delegate_channel_;
            }

        private:
            internal::Call CreateCall(const internal::RpcMethod& method, ClientContext* context, grpc::CompletionQueue* cq) final
            {
                return delegate_channel()->CreateCall(method, context, cq);
            }

            void PerformOpsOnCall(internal::CallOpSetInterface* ops, internal::Call* call) final
            {
                delegate_channel()->PerformOpsOnCall(ops, call);
            }

            void* RegisterMethod(const char* method) final
            {
                return delegate_channel()->RegisterMethod(method);
            }

            void NotifyOnStateChangeImpl(grpc_connectivity_state last_observed, gpr_timespec deadline, grpc::CompletionQueue* cq, void* tag) override
            {
                delegate_channel()->NotifyOnStateChangeImpl(last_observed, deadline, cq, tag);
            }

            bool WaitForStateChangeImpl(grpc_connectivity_state last_observed, gpr_timespec deadline) override
            {
                return delegate_channel()->WaitForStateChangeImpl(last_observed, deadline);
            }

            internal::Call CreateCallInternal(const internal::RpcMethod& method, ClientContext* context, grpc::CompletionQueue* cq, size_t interceptor_pos) final
            {
                return delegate_channel()->CreateCallInternal(method, context, cq, interceptor_pos);
            }

            grpc::CompletionQueue* CallbackCQ() final
            {
                return delegate_channel()->CallbackCQ();
            }

            std::shared_ptr<grpc::ChannelInterface> delegate_channel_;
        };

    }  // namespace experimental
}  // namespace grpc

#endif  // GRPCPP_IMPL_CODEGEN_DELEGATING_CHANNEL_H