1 // Copyright 2024 gRPC authors. 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 15 #ifndef GRPC_SRC_CORE_CLIENT_CHANNEL_LOAD_BALANCED_CALL_DESTINATION_H 16 #define GRPC_SRC_CORE_CLIENT_CHANNEL_LOAD_BALANCED_CALL_DESTINATION_H 17 18 #include "absl/functional/any_invocable.h" 19 #include "src/core/client_channel/client_channel.h" 20 #include "src/core/lib/promise/context.h" 21 #include "src/core/lib/transport/call_destination.h" 22 #include "src/core/load_balancing/lb_policy.h" 23 24 namespace grpc_core { 25 26 // Context type for LB on_commit callback. 27 // TODO(ctiller): make this a struct, so we don't accidentally alias context 28 // types 29 using LbOnCommit = absl::AnyInvocable<void()>; 30 template <> 31 struct ContextType<LbOnCommit> {}; 32 33 class LoadBalancedCallDestination final : public UnstartedCallDestination { 34 public: 35 explicit LoadBalancedCallDestination(ClientChannel::PickerObservable picker) 36 : picker_(std::move(picker)) {} 37 38 void Orphaned() override {} 39 40 void StartCall(UnstartedCallHandler unstarted_handler) override; 41 42 private: 43 ClientChannel::PickerObservable picker_; 44 }; 45 46 } // namespace grpc_core 47 48 #endif // GRPC_SRC_CORE_CLIENT_CHANNEL_LOAD_BALANCED_CALL_DESTINATION_H 49