mirror of https://github.com/grpc/grpc.git
Mark D. Roth
5 years ago
parent
ec8ec630ac
commit
aed6574023
28 changed files with 423 additions and 240 deletions
@ -0,0 +1,68 @@ |
||||
//
|
||||
// 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_CORE_EXT_FILTERS_CLIENT_CHANNEL_SERVICE_CONFIG_CALL_DATA_H |
||||
#define GRPC_CORE_EXT_FILTERS_CLIENT_CHANNEL_SERVICE_CONFIG_CALL_DATA_H |
||||
|
||||
#include <grpc/support/port_platform.h> |
||||
|
||||
#include "src/core/ext/filters/client_channel/service_config.h" |
||||
#include "src/core/ext/filters/client_channel/service_config_parser.h" |
||||
#include "src/core/lib/channel/context.h" |
||||
#include "src/core/lib/gprpp/ref_counted_ptr.h" |
||||
|
||||
namespace grpc_core { |
||||
|
||||
/// When a service config is applied to a call in the client_channel_filter,
|
||||
/// we create an instance of this object on the arena. A pointer to this
|
||||
/// object is also stored in the call_context, so that future filters can
|
||||
/// easily access method and global parameters for the call.
|
||||
class ServiceConfigCallData { |
||||
public: |
||||
ServiceConfigCallData( |
||||
RefCountedPtr<ServiceConfig> service_config, |
||||
const ServiceConfigParser::ParsedConfigVector* method_configs, |
||||
grpc_call_context_element* call_context) |
||||
: service_config_(std::move(service_config)), |
||||
method_configs_(method_configs) { |
||||
call_context[GRPC_CONTEXT_SERVICE_CONFIG_CALL_DATA].value = this; |
||||
call_context[GRPC_CONTEXT_SERVICE_CONFIG_CALL_DATA].destroy = Destroy; |
||||
} |
||||
|
||||
ServiceConfig* service_config() { return service_config_.get(); } |
||||
|
||||
ServiceConfigParser::ParsedConfig* GetMethodParsedConfig(size_t index) const { |
||||
return method_configs_ != nullptr ? (*method_configs_)[index].get() |
||||
: nullptr; |
||||
} |
||||
|
||||
ServiceConfigParser::ParsedConfig* GetGlobalParsedConfig(size_t index) const { |
||||
return service_config_->GetGlobalParsedConfig(index); |
||||
} |
||||
|
||||
private: |
||||
static void Destroy(void* ptr) { |
||||
ServiceConfigCallData* self = static_cast<ServiceConfigCallData*>(ptr); |
||||
self->~ServiceConfigCallData(); |
||||
} |
||||
|
||||
RefCountedPtr<ServiceConfig> service_config_; |
||||
const ServiceConfigParser::ParsedConfigVector* method_configs_ = nullptr; |
||||
}; |
||||
|
||||
} // namespace grpc_core
|
||||
|
||||
#endif /* GRPC_CORE_EXT_FILTERS_CLIENT_CHANNEL_SERVICE_CONFIG_CALL_DATA_H */ |
||||
@ -0,0 +1,87 @@ |
||||
//
|
||||
// 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 <grpc/support/port_platform.h> |
||||
|
||||
#include "src/core/ext/filters/client_channel/service_config_parser.h" |
||||
|
||||
#include <grpc/support/log.h> |
||||
|
||||
namespace grpc_core { |
||||
|
||||
namespace { |
||||
typedef absl::InlinedVector<std::unique_ptr<ServiceConfigParser::Parser>, |
||||
ServiceConfigParser::kNumPreallocatedParsers> |
||||
ServiceConfigParserList; |
||||
ServiceConfigParserList* g_registered_parsers; |
||||
} // namespace
|
||||
|
||||
void ServiceConfigParser::Init() { |
||||
GPR_ASSERT(g_registered_parsers == nullptr); |
||||
g_registered_parsers = new ServiceConfigParserList(); |
||||
} |
||||
|
||||
void ServiceConfigParser::Shutdown() { |
||||
delete g_registered_parsers; |
||||
g_registered_parsers = nullptr; |
||||
} |
||||
|
||||
size_t ServiceConfigParser::RegisterParser(std::unique_ptr<Parser> parser) { |
||||
g_registered_parsers->push_back(std::move(parser)); |
||||
return g_registered_parsers->size() - 1; |
||||
} |
||||
|
||||
ServiceConfigParser::ParsedConfigVector |
||||
ServiceConfigParser::ParseGlobalParameters(const Json& json, |
||||
grpc_error** error) { |
||||
ParsedConfigVector parsed_global_configs; |
||||
std::vector<grpc_error*> error_list; |
||||
for (size_t i = 0; i < g_registered_parsers->size(); i++) { |
||||
grpc_error* parser_error = GRPC_ERROR_NONE; |
||||
auto parsed_config = |
||||
(*g_registered_parsers)[i]->ParseGlobalParams(json, &parser_error); |
||||
if (parser_error != GRPC_ERROR_NONE) { |
||||
error_list.push_back(parser_error); |
||||
} |
||||
parsed_global_configs.push_back(std::move(parsed_config)); |
||||
} |
||||
if (!error_list.empty()) { |
||||
*error = GRPC_ERROR_CREATE_FROM_VECTOR("Global Params", &error_list); |
||||
} |
||||
return parsed_global_configs; |
||||
} |
||||
|
||||
ServiceConfigParser::ParsedConfigVector |
||||
ServiceConfigParser::ParsePerMethodParameters(const Json& json, |
||||
grpc_error** error) { |
||||
ParsedConfigVector parsed_method_configs; |
||||
std::vector<grpc_error*> error_list; |
||||
for (size_t i = 0; i < g_registered_parsers->size(); i++) { |
||||
grpc_error* parser_error = GRPC_ERROR_NONE; |
||||
auto parsed_config = |
||||
(*g_registered_parsers)[i]->ParsePerMethodParams(json, &parser_error); |
||||
if (parser_error != GRPC_ERROR_NONE) { |
||||
error_list.push_back(parser_error); |
||||
} |
||||
parsed_method_configs.push_back(std::move(parsed_config)); |
||||
} |
||||
if (!error_list.empty()) { |
||||
*error = GRPC_ERROR_CREATE_FROM_VECTOR("methodConfig", &error_list); |
||||
} |
||||
return parsed_method_configs; |
||||
} |
||||
|
||||
} // namespace grpc_core
|
||||
@ -0,0 +1,89 @@ |
||||
//
|
||||
// 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_CORE_EXT_FILTERS_CLIENT_CHANNEL_SERVICE_CONFIG_PARSER_H |
||||
#define GRPC_CORE_EXT_FILTERS_CLIENT_CHANNEL_SERVICE_CONFIG_PARSER_H |
||||
|
||||
#include <grpc/support/port_platform.h> |
||||
|
||||
#include <memory> |
||||
|
||||
#include "absl/container/inlined_vector.h" |
||||
|
||||
#include "src/core/lib/iomgr/error.h" |
||||
#include "src/core/lib/json/json.h" |
||||
|
||||
namespace grpc_core { |
||||
|
||||
// Service config parser registry.
|
||||
// See service_config.h for more information.
|
||||
class ServiceConfigParser { |
||||
public: |
||||
/// This is the base class that all service config parsers MUST use to store
|
||||
/// parsed service config data.
|
||||
class ParsedConfig { |
||||
public: |
||||
virtual ~ParsedConfig() = default; |
||||
}; |
||||
|
||||
/// This is the base class that all service config parsers should derive from.
|
||||
class Parser { |
||||
public: |
||||
virtual ~Parser() = default; |
||||
|
||||
virtual std::unique_ptr<ParsedConfig> ParseGlobalParams( |
||||
const Json& /* json */, grpc_error** error) { |
||||
// Avoid unused parameter warning on debug-only parameter
|
||||
(void)error; |
||||
GPR_DEBUG_ASSERT(error != nullptr); |
||||
return nullptr; |
||||
} |
||||
|
||||
virtual std::unique_ptr<ParsedConfig> ParsePerMethodParams( |
||||
const Json& /* json */, grpc_error** error) { |
||||
// Avoid unused parameter warning on debug-only parameter
|
||||
(void)error; |
||||
GPR_DEBUG_ASSERT(error != nullptr); |
||||
return nullptr; |
||||
} |
||||
}; |
||||
|
||||
static constexpr int kNumPreallocatedParsers = 4; |
||||
typedef absl::InlinedVector<std::unique_ptr<ParsedConfig>, |
||||
kNumPreallocatedParsers> |
||||
ParsedConfigVector; |
||||
|
||||
static void Init(); |
||||
static void Shutdown(); |
||||
|
||||
/// Globally register a service config parser. On successful registration, it
|
||||
/// returns the index at which the parser was registered. On failure, -1 is
|
||||
/// returned. Each new service config update will go through all the
|
||||
/// registered parser. Each parser is responsible for reading the service
|
||||
/// config json and returning a parsed config. This parsed config can later be
|
||||
/// retrieved using the same index that was returned at registration time.
|
||||
static size_t RegisterParser(std::unique_ptr<Parser> parser); |
||||
|
||||
static ParsedConfigVector ParseGlobalParameters(const Json& json, |
||||
grpc_error** error); |
||||
|
||||
static ParsedConfigVector ParsePerMethodParameters(const Json& json, |
||||
grpc_error** error); |
||||
}; |
||||
|
||||
} // namespace grpc_core
|
||||
|
||||
#endif /* GRPC_CORE_EXT_FILTERS_CLIENT_CHANNEL_SERVICE_CONFIG_PARSER_H */ |
||||
Loading…
Reference in new issue