The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#) https://grpc.io/
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

290 lines
9.2 KiB

//
//
// 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_FIXTURES_H
#define GRPC_TEST_CPP_MICROBENCHMARKS_FULLSTACK_FIXTURES_H
#include <grpc/grpc.h>
#include <grpc/support/atm.h>
#include <grpc/support/log.h>
#include <grpcpp/channel.h>
#include <grpcpp/create_channel.h>
#include <grpcpp/security/credentials.h>
#include <grpcpp/security/server_credentials.h>
#include <grpcpp/server.h>
#include <grpcpp/server_builder.h>
#include "src/core/ext/transport/chttp2/transport/chttp2_transport.h"
#include "src/core/lib/channel/channel_args.h"
#include "src/core/lib/config/core_configuration.h"
#include "src/core/lib/gprpp/crash.h"
#include "src/core/lib/iomgr/endpoint.h"
#include "src/core/lib/iomgr/endpoint_pair.h"
#include "src/core/lib/iomgr/exec_ctx.h"
#include "src/core/lib/iomgr/tcp_posix.h"
#include "src/core/lib/surface/channel.h"
#include "src/core/lib/surface/channel_create.h"
#include "src/core/lib/surface/completion_queue.h"
#include "src/core/lib/surface/server.h"
#include "src/cpp/client/create_channel_internal.h"
#include "test/core/util/port.h"
#include "test/core/util/test_config.h"
#include "test/cpp/microbenchmarks/helpers.h"
namespace grpc {
namespace testing {
class FixtureConfiguration {
public:
virtual ~FixtureConfiguration() {}
virtual void ApplyCommonChannelArguments(ChannelArguments* c) const {
c->SetInt(GRPC_ARG_MAX_RECEIVE_MESSAGE_LENGTH, INT_MAX);
c->SetInt(GRPC_ARG_MAX_SEND_MESSAGE_LENGTH, INT_MAX);
c->SetInt(GRPC_ARG_ENABLE_RETRIES, 0);
c->SetResourceQuota(ResourceQuota());
}
virtual void ApplyCommonServerBuilderConfig(ServerBuilder* b) const {
b->SetMaxReceiveMessageSize(INT_MAX);
b->SetMaxSendMessageSize(INT_MAX);
}
};
class BaseFixture {
public:
virtual ~BaseFixture() = default;
};
class FullstackFixture : public BaseFixture {
public:
FullstackFixture(Service* service, const FixtureConfiguration& config,
const std::string& address) {
ServerBuilder b;
if (address.length() > 0) {
b.AddListeningPort(address, InsecureServerCredentials());
}
cq_ = b.AddCompletionQueue(true);
b.RegisterService(service);
config.ApplyCommonServerBuilderConfig(&b);
server_ = b.BuildAndStart();
ChannelArguments args;
config.ApplyCommonChannelArguments(&args);
if (address.length() > 0) {
channel_ = grpc::CreateCustomChannel(address,
InsecureChannelCredentials(), args);
} else {
channel_ = server_->InProcessChannel(args);
}
}
~FullstackFixture() override {
server_->Shutdown(grpc_timeout_milliseconds_to_deadline(0));
cq_->Shutdown();
void* tag;
bool ok;
while (cq_->Next(&tag, &ok)) {
}
}
ServerCompletionQueue* cq() { return cq_.get(); }
std::shared_ptr<Channel> channel() { return channel_; }
private:
std::unique_ptr<Server> server_;
std::unique_ptr<ServerCompletionQueue> cq_;
std::shared_ptr<Channel> channel_;
};
class TCP : public FullstackFixture {
public:
explicit TCP(Service* service,
const FixtureConfiguration& fixture_configuration =
FixtureConfiguration())
8 years ago
: FullstackFixture(service, fixture_configuration, MakeAddress(&port_)) {}
~TCP() override { grpc_recycle_unused_port(port_); }
private:
8 years ago
int port_;
static std::string MakeAddress(int* port) {
8 years ago
*port = grpc_pick_unused_port_or_die();
std::stringstream addr;
8 years ago
addr << "localhost:" << *port;
return addr.str();
}
};
class UDS : public FullstackFixture {
public:
explicit UDS(Service* service,
const FixtureConfiguration& fixture_configuration =
FixtureConfiguration())
8 years ago
: FullstackFixture(service, fixture_configuration, MakeAddress(&port_)) {}
~UDS() override { grpc_recycle_unused_port(port_); }
private:
8 years ago
int port_;
static std::string MakeAddress(int* port) {
8 years ago
*port = grpc_pick_unused_port_or_die(); // just for a unique id - not a
// real port
std::stringstream addr;
8 years ago
addr << "unix:/tmp/bm_fullstack." << *port;
return addr.str();
}
};
class InProcess : public FullstackFixture {
public:
explicit InProcess(Service* service,
const FixtureConfiguration& fixture_configuration =
FixtureConfiguration())
: FullstackFixture(service, fixture_configuration, "") {}
~InProcess() override {}
};
class EndpointPairFixture : public BaseFixture {
public:
EndpointPairFixture(Service* service, grpc_endpoint_pair endpoints,
const FixtureConfiguration& fixture_configuration)
8 years ago
: endpoint_pair_(endpoints) {
ServerBuilder b;
cq_ = b.AddCompletionQueue(true);
b.RegisterService(service);
fixture_configuration.ApplyCommonServerBuilderConfig(&b);
server_ = b.BuildAndStart();
grpc_core::ExecCtx exec_ctx;
// add server endpoint to server_
//
{
grpc_core::Server* core_server =
grpc_core::Server::FromC(server_->c_server());
grpc_core::ChannelArgs server_args = core_server->channel_args();
8 years ago
server_transport_ = grpc_create_chttp2_transport(
server_args, endpoints.server, false /* is_client */);
for (grpc_pollset* pollset : core_server->pollsets()) {
grpc_endpoint_add_to_pollset(endpoints.server, pollset);
}
GPR_ASSERT(GRPC_LOG_IF_ERROR(
"SetupTransport",
core_server->SetupTransport(server_transport_, nullptr, server_args,
nullptr)));
grpc_chttp2_transport_start_reading(server_transport_, nullptr, nullptr,
nullptr);
}
// create channel
{
grpc_core::ChannelArgs c_args;
{
ChannelArguments args;
args.SetString(GRPC_ARG_DEFAULT_AUTHORITY, "test.authority");
fixture_configuration.ApplyCommonChannelArguments(&args);
// precondition
grpc_channel_args tmp_args;
args.SetChannelArgs(&tmp_args);
c_args = grpc_core::CoreConfiguration::Get()
.channel_args_preconditioning()
.PreconditionChannelArgs(&tmp_args);
}
client_transport_ =
grpc_create_chttp2_transport(c_args, endpoints.client, true);
8 years ago
GPR_ASSERT(client_transport_);
Convert grpc_channel to C++ (#29266) * begin * tests * fix * http * Filter fuzzer * progress * basics * progress * Automated change: Fix sanity tests * Automated change: Fix sanity tests * Revert "Revert "HTTP Client Filter --> promises (#29031)" (#29181)" This reverts commit 6ee276f67219d5b72bf48d327fc3dd6b6509258d. * stuff * debug * minimal reproduction * progress * progress * create call * progress * recv trailing metadata * wakeups * corpus * Automated change: Fix sanity tests * Automated change: Fix sanity tests * debug * fix state machine for c# * Revert "minimal reproduction" This reverts commit 4d02d2e7301172410d7243de70e993f5275c732b. * Revert "debug" This reverts commit 7960842f48742575eb0c60aa5e983ee13793bbb5. * Revert "debug" This reverts commit a6f224e4a1fe8376d7dfc019c8c37074a4fd3de4. * no-logging * initial-metadata * Revert "Revert "debug"" This reverts commit 951844e8573caacf1061ac8f5e9c2ce73e747d2b. * Better int conversion * debug * Fix for Cronet * Revert "debug" This reverts commit 4d641c428142d60aa6133b71a5b78643ec9c929a. * Revert "Better int conversion" This reverts commit 4001b957cb775ca148b3f6f28a3faed3f087f9be. * Revert "Revert "Revert "debug""" This reverts commit d135c610432227393eedb954bab058f151ab0bc6. * progress * progress * Automated change: Fix sanity tests * fix, c++ize * handle transport, use objects * enable more stuffs * remove placeholder * contexts * fix * fix * Automated change: Fix sanity tests * Automated change: Fix sanity tests * x * exceptional toast * include idle filters, time * fix * namespace * fixes * final info * progress * cleanup * progress * progress * progress * Automated change: Fix sanity tests * Automated change: Fix sanity tests * progress * Set int * Set int * Automated change: Fix sanity tests * fix * fix * fix * fixes * fixes * fixes * fixes * fix * fix race * fix race * mac fix * review feedback * getgetget * fix ios Co-authored-by: ctiller <ctiller@users.noreply.github.com> Co-authored-by: Jan Tattermusch <jtattermusch@google.com>
3 years ago
grpc_channel* channel =
grpc_core::ChannelCreate("target", c_args, GRPC_CLIENT_DIRECT_CHANNEL,
client_transport_)
Convert grpc_channel to C++ (#29266) * begin * tests * fix * http * Filter fuzzer * progress * basics * progress * Automated change: Fix sanity tests * Automated change: Fix sanity tests * Revert "Revert "HTTP Client Filter --> promises (#29031)" (#29181)" This reverts commit 6ee276f67219d5b72bf48d327fc3dd6b6509258d. * stuff * debug * minimal reproduction * progress * progress * create call * progress * recv trailing metadata * wakeups * corpus * Automated change: Fix sanity tests * Automated change: Fix sanity tests * debug * fix state machine for c# * Revert "minimal reproduction" This reverts commit 4d02d2e7301172410d7243de70e993f5275c732b. * Revert "debug" This reverts commit 7960842f48742575eb0c60aa5e983ee13793bbb5. * Revert "debug" This reverts commit a6f224e4a1fe8376d7dfc019c8c37074a4fd3de4. * no-logging * initial-metadata * Revert "Revert "debug"" This reverts commit 951844e8573caacf1061ac8f5e9c2ce73e747d2b. * Better int conversion * debug * Fix for Cronet * Revert "debug" This reverts commit 4d641c428142d60aa6133b71a5b78643ec9c929a. * Revert "Better int conversion" This reverts commit 4001b957cb775ca148b3f6f28a3faed3f087f9be. * Revert "Revert "Revert "debug""" This reverts commit d135c610432227393eedb954bab058f151ab0bc6. * progress * progress * Automated change: Fix sanity tests * fix, c++ize * handle transport, use objects * enable more stuffs * remove placeholder * contexts * fix * fix * Automated change: Fix sanity tests * Automated change: Fix sanity tests * x * exceptional toast * include idle filters, time * fix * namespace * fixes * final info * progress * cleanup * progress * progress * progress * Automated change: Fix sanity tests * Automated change: Fix sanity tests * progress * Set int * Set int * Automated change: Fix sanity tests * fix * fix * fix * fixes * fixes * fixes * fixes * fix * fix race * fix race * mac fix * review feedback * getgetget * fix ios Co-authored-by: ctiller <ctiller@users.noreply.github.com> Co-authored-by: Jan Tattermusch <jtattermusch@google.com>
3 years ago
->release()
->c_ptr();
grpc_chttp2_transport_start_reading(client_transport_, nullptr, nullptr,
nullptr);
channel_ = grpc::CreateChannelInternal(
"", channel,
std::vector<std::unique_ptr<
experimental::ClientInterceptorFactoryInterface>>());
}
}
~EndpointPairFixture() override {
server_->Shutdown(grpc_timeout_milliseconds_to_deadline(0));
cq_->Shutdown();
void* tag;
bool ok;
while (cq_->Next(&tag, &ok)) {
}
}
ServerCompletionQueue* cq() { return cq_.get(); }
std::shared_ptr<Channel> channel() { return channel_; }
8 years ago
protected:
grpc_endpoint_pair endpoint_pair_;
grpc_core::Transport* client_transport_;
grpc_core::Transport* server_transport_;
8 years ago
private:
std::unique_ptr<Server> server_;
std::unique_ptr<ServerCompletionQueue> cq_;
std::shared_ptr<Channel> channel_;
};
class SockPair : public EndpointPairFixture {
public:
explicit SockPair(Service* service,
const FixtureConfiguration& fixture_configuration =
FixtureConfiguration())
8 years ago
: EndpointPairFixture(service,
grpc_iomgr_create_endpoint_pair("test", nullptr),
fixture_configuration) {}
};
////////////////////////////////////////////////////////////////////////////////
// Minimal stack fixtures
class MinStackConfiguration : public FixtureConfiguration {
void ApplyCommonChannelArguments(ChannelArguments* a) const override {
a->SetInt(GRPC_ARG_MINIMAL_STACK, 1);
FixtureConfiguration::ApplyCommonChannelArguments(a);
}
void ApplyCommonServerBuilderConfig(ServerBuilder* b) const override {
b->AddChannelArgument(GRPC_ARG_MINIMAL_STACK, 1);
FixtureConfiguration::ApplyCommonServerBuilderConfig(b);
}
};
template <class Base>
class MinStackize : public Base {
public:
explicit MinStackize(Service* service)
: Base(service, MinStackConfiguration()) {}
};
typedef MinStackize<TCP> MinTCP;
typedef MinStackize<UDS> MinUDS;
typedef MinStackize<InProcess> MinInProcess;
typedef MinStackize<SockPair> MinSockPair;
} // namespace testing
} // namespace grpc
#endif // GRPC_TEST_CPP_MICROBENCHMARKS_FULLSTACK_FIXTURES_H