// // // Copyright 2017 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 GRPC_TEST_CPP_MICROBENCHMARKS_FULLSTACK_CONTEXT_MUTATORS_H #define GRPC_TEST_CPP_MICROBENCHMARKS_FULLSTACK_CONTEXT_MUTATORS_H #include #include #include #include #include #include #include #include "src/core/lib/gprpp/crash.h" #include "test/cpp/microbenchmarks/helpers.h" namespace grpc { namespace testing { //****************************************************************************** // CONTEXT MUTATORS // static const int kPregenerateKeyCount = 100000; template auto MakeVector(size_t length, F f) -> std::vector { std::vector out; out.reserve(length); for (size_t i = 0; i < length; i++) { out.push_back(f()); } return out; } class NoOpMutator { public: template explicit NoOpMutator(ContextType* /*context*/) {} }; template class RandomBinaryMetadata { public: static const std::string& Key() { return kKey; } static const std::string& Value() { return kValues[rand() % kValues.size()]; } private: static const std::string kKey; static const std::vector kValues; static std::string GenerateOneString() { std::string s; s.reserve(length + 1); for (int i = 0; i < length; i++) { s += static_cast(rand()); } return s; } }; template class RandomAsciiMetadata { public: static const std::string& Key() { return kKey; } static const std::string& Value() { return kValues[rand() % kValues.size()]; } private: static const std::string kKey; static const std::vector kValues; static std::string GenerateOneString() { std::string s; s.reserve(length + 1); for (int i = 0; i < length; i++) { s += static_cast(rand() % 26 + 'a'); } return s; } }; template class Client_AddMetadata : public NoOpMutator { public: explicit Client_AddMetadata(ClientContext* context) : NoOpMutator(context) { for (int i = 0; i < kNumKeys; i++) { context->AddMetadata(Generator::Key(), Generator::Value()); } } }; template class Server_AddInitialMetadata : public NoOpMutator { public: explicit Server_AddInitialMetadata(ServerContext* context) : NoOpMutator(context) { for (int i = 0; i < kNumKeys; i++) { context->AddInitialMetadata(Generator::Key(), Generator::Value()); } } }; // static initialization template const std::string RandomBinaryMetadata::kKey = "foo-bin"; template const std::vector RandomBinaryMetadata::kValues = MakeVector(kPregenerateKeyCount, GenerateOneString); template const std::string RandomAsciiMetadata::kKey = "foo"; template const std::vector RandomAsciiMetadata::kValues = MakeVector(kPregenerateKeyCount, GenerateOneString); } // namespace testing } // namespace grpc #endif // GRPC_TEST_CPP_MICROBENCHMARKS_FULLSTACK_CONTEXT_MUTATORS_H