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 MethodLookup() = delete; 29 30 template <typename Service, uint32_t kMethodId> GetRawMethod()31 static constexpr const auto& GetRawMethod() { 32 const auto& method = GetMethodUnion<Service, kMethodId>().raw_method(); 33 static_assert(method.id() == kMethodId, "Incorrect method implementation"); 34 return method; 35 } 36 37 template <typename Service, uint32_t kMethodId> GetPwpbMethod()38 static constexpr const auto& GetPwpbMethod() { 39 const auto& method = GetMethodUnion<Service, kMethodId>().pwpb_method(); 40 static_assert(method.id() == kMethodId, "Incorrect method implementation"); 41 return method; 42 } 43 44 template <typename Service, uint32_t kMethodId> GetNanopbMethod()45 static constexpr const auto& GetNanopbMethod() { 46 const auto& method = GetMethodUnion<Service, kMethodId>().nanopb_method(); 47 static_assert(method.id() == kMethodId, "Incorrect method implementation"); 48 return method; 49 } 50 51 private: 52 template <typename Service, uint32_t kMethodId> GetMethodUnion()53 static constexpr const auto& GetMethodUnion() { 54 constexpr auto method = GetMethodUnionPointer<Service>(kMethodId); 55 static_assert(method != nullptr, 56 "The selected function is not an RPC service method"); 57 return *method; 58 } 59 60 template <typename Service> decltype(Service::kPwRpcMethods)61 static constexpr typename decltype(Service::kPwRpcMethods)::const_pointer 62 GetMethodUnionPointer(uint32_t kMethodId) { 63 for (size_t i = 0; i < Service::kPwRpcMethodIds.size(); ++i) { 64 if (Service::kPwRpcMethodIds[i] == kMethodId) { 65 return &Service::kPwRpcMethods[i]; 66 } 67 } 68 return nullptr; 69 } 70 }; 71 72 } // namespace pw::rpc::internal 73