84 lines
2.5 KiB
C++
84 lines
2.5 KiB
C++
/*
|
|
*
|
|
* Copyright 2015 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 <grpcpp/ext/proto_server_reflection_plugin.h>
|
|
#include <grpcpp/impl/server_builder_plugin.h>
|
|
#include <grpcpp/impl/server_initializer.h>
|
|
#include <grpcpp/server.h>
|
|
|
|
#include "src/cpp/ext/proto_server_reflection.h"
|
|
|
|
namespace grpc {
|
|
namespace reflection {
|
|
|
|
ProtoServerReflectionPlugin::ProtoServerReflectionPlugin()
|
|
: reflection_service_(new grpc::ProtoServerReflection()) {}
|
|
|
|
std::string ProtoServerReflectionPlugin::name() {
|
|
return "proto_server_reflection";
|
|
}
|
|
|
|
void ProtoServerReflectionPlugin::InitServer(grpc::ServerInitializer* si) {
|
|
si->RegisterService(reflection_service_);
|
|
}
|
|
|
|
void ProtoServerReflectionPlugin::Finish(grpc::ServerInitializer* si) {
|
|
reflection_service_->SetServiceList(si->GetServiceList());
|
|
}
|
|
|
|
void ProtoServerReflectionPlugin::ChangeArguments(const std::string& /*name*/,
|
|
void* /*value*/) {}
|
|
|
|
bool ProtoServerReflectionPlugin::has_sync_methods() const {
|
|
if (reflection_service_) {
|
|
return reflection_service_->has_synchronous_methods();
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool ProtoServerReflectionPlugin::has_async_methods() const {
|
|
if (reflection_service_) {
|
|
return reflection_service_->has_async_methods();
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static std::unique_ptr< ::grpc::ServerBuilderPlugin> CreateProtoReflection() {
|
|
return std::unique_ptr< ::grpc::ServerBuilderPlugin>(
|
|
new ProtoServerReflectionPlugin());
|
|
}
|
|
|
|
void InitProtoReflectionServerBuilderPlugin() {
|
|
static struct Initialize {
|
|
Initialize() {
|
|
::grpc::ServerBuilder::InternalAddPluginFactory(&CreateProtoReflection);
|
|
}
|
|
} initializer;
|
|
}
|
|
|
|
// Force InitProtoReflectionServerBuilderPlugin() to be called at static
|
|
// initialization time.
|
|
struct StaticProtoReflectionPluginInitializer {
|
|
StaticProtoReflectionPluginInitializer() {
|
|
InitProtoReflectionServerBuilderPlugin();
|
|
}
|
|
} static_proto_reflection_plugin_initializer;
|
|
|
|
} // namespace reflection
|
|
} // namespace grpc
|