mirror of https://github.com/grpc/grpc.git
Alex Polcyn
7 years ago
parent
2cfc216cfd
commit
92e0a70ec6
11 changed files with 127 additions and 17 deletions
@ -0,0 +1,45 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2016 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. |
||||
* |
||||
*/ |
||||
|
||||
#include "qps_server_builder.h" |
||||
|
||||
using grpc::ServerBuilder; |
||||
|
||||
namespace grpc { |
||||
namespace testing { |
||||
|
||||
namespace { |
||||
std::unique_ptr<ServerBuilder> DefaultCreateQpsServerBuilder() { |
||||
return std::unique_ptr<ServerBuilder>(new ServerBuilder()); |
||||
} |
||||
|
||||
std::function<std::unique_ptr<ServerBuilder>()> g_create_qps_server_builder = |
||||
DefaultCreateQpsServerBuilder; |
||||
} // namespace
|
||||
|
||||
std::unique_ptr<ServerBuilder> CreateQpsServerBuilder() { |
||||
return g_create_qps_server_builder(); |
||||
} |
||||
|
||||
void SetCreateQpsServerBuilderFunc( |
||||
std::function<std::unique_ptr<ServerBuilder>()> create_qps_server_builder) { |
||||
g_create_qps_server_builder = std::move(create_qps_server_builder); |
||||
} |
||||
|
||||
} // namespace testing
|
||||
} // namespace grpc
|
||||
@ -0,0 +1,46 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2016 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_QPS_SERVER_BUILDER_H |
||||
#define GRPC_QPS_SERVER_BUILDER_H |
||||
|
||||
#include <functional> |
||||
#include <memory> |
||||
|
||||
#include <grpcpp/server_builder.h> |
||||
|
||||
namespace grpc { |
||||
namespace testing { |
||||
|
||||
// CreateQpsServerBuilder creates a new ServerBuilder.
|
||||
// This uses the "create ServerBuilder" func that was set
|
||||
// in SetCreateQpsServerBuilderFunc if one has been set,
|
||||
// otherwise, this defaults to creating a new ServerBuilder
|
||||
// with only its default constructor.
|
||||
std::unique_ptr<ServerBuilder> CreateQpsServerBuilder(); |
||||
|
||||
// SetCreateQpsServerBuilderFunc sets a function to use to create new
|
||||
// ServerBuilders in "CreateQpsServerBuilder". It can be used to modify options
|
||||
// that the server is built with.
|
||||
void SetCreateQpsServerBuilderFunc( |
||||
std::function<std::unique_ptr<ServerBuilder>()>); |
||||
|
||||
} // namespace testing
|
||||
} // namespace grpc
|
||||
|
||||
#endif // GRPC_QPS_SERVER_BUILDER_H
|
||||
Loading…
Reference in new issue