1 // 2 // Copyright 2020 gRPC authors. 3 // 4 // Licensed under the Apache License, Version 2.0 (the "License"); 5 // you may not use this file except in compliance with the License. 6 // You may obtain a copy of the License at 7 // 8 // http://www.apache.org/licenses/LICENSE-2.0 9 // 10 // Unless required by applicable law or agreed to in writing, software 11 // distributed under the License is distributed on an "AS IS" BASIS, 12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 // See the License for the specific language governing permissions and 14 // limitations under the License. 15 // 16 17 #ifndef GRPC_TEST_CPP_END2END_RLS_SERVER_H 18 #define GRPC_TEST_CPP_END2END_RLS_SERVER_H 19 20 #include "absl/types/optional.h" 21 22 #include "src/core/lib/gprpp/time.h" 23 #include "src/proto/grpc/lookup/v1/rls.grpc.pb.h" 24 #include "src/proto/grpc/lookup/v1/rls.pb.h" 25 #include "test/cpp/end2end/counted_service.h" 26 27 namespace grpc { 28 namespace testing { 29 30 using RlsService = 31 CountedService<grpc::lookup::v1::RouteLookupService::Service>; 32 33 class RlsServiceImpl : public RlsService { 34 public: 35 using ContextProcessingFunc = std::function<void(grpc::ServerContext*)>; 36 37 explicit RlsServiceImpl(ContextProcessingFunc context_proc = nullptr) context_proc_(std::move (context_proc))38 : context_proc_(std::move(context_proc)) {} 39 40 grpc::Status RouteLookup( 41 grpc::ServerContext* context, 42 const grpc::lookup::v1::RouteLookupRequest* request, 43 grpc::lookup::v1::RouteLookupResponse* response) override; 44 Start()45 void Start() {} 46 Shutdown()47 void Shutdown() {} 48 49 void SetResponse(grpc::lookup::v1::RouteLookupRequest request, 50 grpc::lookup::v1::RouteLookupResponse response, 51 grpc_core::Duration response_delay = grpc_core::Duration()); 52 53 void RemoveResponse(const grpc::lookup::v1::RouteLookupRequest& request); 54 55 std::vector<grpc::lookup::v1::RouteLookupRequest> GetUnmatchedRequests(); 56 57 private: 58 // Sorting thunk for RouteLookupRequest. 59 struct RlsRequestLessThan { operatorRlsRequestLessThan60 bool operator()(const grpc::lookup::v1::RouteLookupRequest& req1, 61 const grpc::lookup::v1::RouteLookupRequest& req2) const { 62 std::map<absl::string_view, absl::string_view> key_map1( 63 req1.key_map().begin(), req1.key_map().end()); 64 std::map<absl::string_view, absl::string_view> key_map2( 65 req2.key_map().begin(), req2.key_map().end()); 66 if (key_map1 < key_map2) return true; 67 if (req1.reason() < req2.reason()) return true; 68 if (req1.stale_header_data() < req2.stale_header_data()) return true; 69 return false; 70 } 71 }; 72 73 struct ResponseData { 74 grpc::lookup::v1::RouteLookupResponse response; 75 grpc_core::Duration response_delay; 76 }; 77 78 ContextProcessingFunc context_proc_; 79 grpc::internal::Mutex mu_; 80 std::map<grpc::lookup::v1::RouteLookupRequest, ResponseData, 81 RlsRequestLessThan> 82 responses_ ABSL_GUARDED_BY(&mu_); 83 std::vector<grpc::lookup::v1::RouteLookupRequest> unmatched_requests_ 84 ABSL_GUARDED_BY(&mu_); 85 }; 86 87 grpc::lookup::v1::RouteLookupRequest BuildRlsRequest( 88 std::map<std::string, std::string> key, 89 grpc::lookup::v1::RouteLookupRequest::Reason reason = 90 grpc::lookup::v1::RouteLookupRequest::REASON_MISS, 91 const char* stale_header_data = ""); 92 93 grpc::lookup::v1::RouteLookupResponse BuildRlsResponse( 94 std::vector<std::string> targets, const char* header_data = ""); 95 96 } // namespace testing 97 } // namespace grpc 98 99 #endif // GRPC_TEST_CPP_END2END_RLS_SERVER_H 100