1 // Copyright 2020 The Pigweed Authors 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); you may not 4 // use this file except in compliance with the License. You may obtain a copy of 5 // the License at 6 // 7 // https://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, WITHOUT 11 // WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the 12 // License for the specific language governing permissions and limitations under 13 // the License. 14 #pragma once 15 16 #include "pw_rpc/internal/method.h" 17 18 namespace pw::rpc::internal { 19 20 // Gets a Method object from a generated RPC service class. Getter functions are 21 // provided for each supported method implementation. The 22 // 23 // To ensure the MethodUnion actually holds the requested method type, the 24 // method ID is accessed in a static_assert. It is invalid to access an unset 25 // union member in a constant expression, so this results in a compiler error. 26 class MethodLookup { 27 public: 28 template <typename Service, uint32_t kMethodId> GetRawMethod()29 static constexpr const auto& GetRawMethod() { 30 const auto& method = GetMethodUnion<Service, kMethodId>().raw_method(); 31 static_assert(method.id() == kMethodId, "Incorrect method implementation"); 32 return method; 33 } 34 35 template <typename Service, uint32_t kMethodId> GetNanopbMethod()36 static constexpr const auto& GetNanopbMethod() { 37 const auto& method = GetMethodUnion<Service, kMethodId>().nanopb_method(); 38 static_assert(method.id() == kMethodId, "Incorrect method implementation"); 39 return method; 40 } 41 42 private: 43 template <typename Service, uint32_t kMethodId> GetMethodUnion()44 static constexpr const auto& GetMethodUnion() { 45 constexpr auto method = GetMethodUnionPointer<Service>(kMethodId); 46 static_assert(method != nullptr, 47 "The selected function is not an RPC service method"); 48 return *method; 49 } 50 51 template <typename Service> 52 static constexpr decltype(GeneratedService<Service>::kMethods)53 typename decltype(GeneratedService<Service>::kMethods)::const_pointer 54 GetMethodUnionPointer(uint32_t kMethodId) { 55 for (size_t i = 0; i < GeneratedService<Service>::kMethodIds.size(); ++i) { 56 if (GeneratedService<Service>::kMethodIds[i] == kMethodId) { 57 return &GeneratedService<Service>::kMethods[i]; 58 } 59 } 60 return nullptr; 61 } 62 }; 63 64 } // namespace pw::rpc::internal 65