Chiebot-Mirror
/
grpc
mirror of https://github.com/grpc/grpc.git
8
0
Fork 0
Code Issues Projects Releases Wiki Activity

Logging: Move filter to Core (#32467)

Browse Source 
This filter was originally written only for the C++ wrapped layer, but
we have plans to use this for Python (and maybe other wrapped languages
too in the future.)

<!--

If you know who should review your pull request, please assign it to
that
person, otherwise the pull request would get assigned randomly.

If your pull request is for a specific language, please add the
appropriate
lang label.

-->
pull/23431/head
Yash Tibrewal 2 years ago committed by GitHub
parent b3fb266937
commit 7af4bc7f1a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
13 changed files with 189 additions and 243 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 63
      src/core/BUILD
  2. 200
      src/core/ext/filters/logging/logging_filter.cc
  3. 15
      src/core/ext/filters/logging/logging_filter.h
  4. 16
      src/core/ext/filters/logging/logging_sink.h
  5. 92
      src/cpp/ext/filters/logging/BUILD
  6. 4
      src/cpp/ext/gcp/BUILD
  7. 4
      src/cpp/ext/gcp/observability.cc
  8. 2
      src/cpp/ext/gcp/observability_logging_sink.cc
  9. 6
      src/cpp/ext/gcp/observability_logging_sink.h
  10. 2
      test/cpp/ext/filters/logging/BUILD
  11. 25
      test/cpp/ext/filters/logging/logging_test.cc
  12. 2
      test/cpp/ext/gcp/observability_logging_sink_test.cc
  13. 1
      tools/distrib/fix_build_deps.py

63
src/core/BUILD
Unescape View File

@ -5350,6 +5350,69 @@ grpc_cc_library(
],
)
grpc_cc_library(
name = "logging_sink",
hdrs = [
"ext/filters/logging/logging_sink.h",
],
external_deps = ["absl/strings"],
language = "c++",
visibility = [
"//src/cpp/ext/gcp:__subpackages__",
"//test:__subpackages__",
],
deps = [
"time",
"//:gpr_platform",
],
)
grpc_cc_library(
name = "logging_filter",
srcs = [
"ext/filters/logging/logging_filter.cc",
],
hdrs = [
"ext/filters/logging/logging_filter.h",
],
external_deps = [
"absl/random",
"absl/status:statusor",
"absl/strings",
"absl/types:optional",
],
language = "c++",
visibility = [
"//src/cpp/ext/gcp:__subpackages__",
"//test:__subpackages__",
],
deps = [
"arena",
"arena_promise",
"cancel_callback",
"channel_args",
"channel_fwd",
"channel_stack_type",
"context",
"logging_sink",
"map",
"pipe",
"poll",
"slice",
"slice_buffer",
"time",
"//:channel_stack_builder",
"//:config",
"//:gpr",
"//:gpr_platform",
"//:grpc_base",
"//:grpc_client_channel",
"//:grpc_public_hdrs",
"//:grpc_resolver",
"//:uri_parser",
],
)
### UPB Targets
grpc_upb_proto_library(

200
src/cpp/ext/filters/logging/logging_filter.cc → src/core/ext/filters/logging/logging_filter.cc
Unescape View File

@ -18,7 +18,7 @@
#include <grpc/support/port_platform.h>
#include "src/cpp/ext/filters/logging/logging_filter.h"
#include "src/core/ext/filters/logging/logging_filter.h"
#include <inttypes.h>
#include <limits.h>
@ -48,6 +48,7 @@
#include <grpc/support/log.h>
#include "src/core/ext/filters/client_channel/client_channel.h"
#include "src/core/ext/filters/logging/logging_sink.h"
#include "src/core/lib/channel/channel_args.h"
#include "src/core/lib/channel/channel_fwd.h"
#include "src/core/lib/channel/channel_stack.h"
@ -70,10 +71,8 @@
#include "src/core/lib/transport/metadata_batch.h"
#include "src/core/lib/transport/transport.h"
#include "src/core/lib/uri/uri_parser.h"
#include "src/cpp/ext/filters/logging/logging_sink.h"
namespace grpc {
namespace internal {
namespace grpc_core {
namespace {
@ -92,8 +91,7 @@ class MetadataEncoder {
status_details_bin_(status_details_bin),
log_len_(log_len) {}
void Encode(const grpc_core::Slice& key_slice,
const grpc_core::Slice& value_slice) {
void Encode(const Slice& key_slice, const Slice& value_slice) {
auto key = key_slice.as_string_view();
auto value = value_slice.as_string_view();
if (status_details_bin_ != nullptr && key == "grpc-status-details-bin") {
@ -122,12 +120,11 @@ class MetadataEncoder {
template <typename Which>
void Encode(Which, const typename Which::ValueType&) {}
void Encode(grpc_core::GrpcStatusMetadata, grpc_status_code status) {
void Encode(GrpcStatusMetadata, grpc_status_code status) {
payload_->status_code = status;
}
void Encode(grpc_core::GrpcMessageMetadata,
const grpc_core::Slice& status_message) {
void Encode(GrpcMessageMetadata, const Slice& status_message) {
payload_->status_message = std::string(status_message.as_string_view());
}
@ -143,8 +140,8 @@ class MetadataEncoder {
void SetIpPort(absl::string_view s, LoggingSink::Entry::Address* peer) {
absl::string_view host;
absl::string_view port;
if (grpc_core::SplitHostPort(absl::string_view(s.data(), s.length()), &host,
&port) == 1) {
if (SplitHostPort(absl::string_view(s.data(), s.length()), &host, &port) ==
1) {
if (!host.empty()) {
peer->address = std::string(host);
}
@ -157,11 +154,9 @@ void SetIpPort(absl::string_view s, LoggingSink::Entry::Address* peer) {
}
}
LoggingSink::Entry::Address PeerStringToAddress(
const grpc_core::Slice& peer_string) {
LoggingSink::Entry::Address PeerStringToAddress(const Slice& peer_string) {
LoggingSink::Entry::Address address;
absl::StatusOr<grpc_core::URI> uri =
grpc_core::URI::Parse(peer_string.as_string_view());
absl::StatusOr<URI> uri = URI::Parse(peer_string.as_string_view());
if (!uri.ok()) {
gpr_log(GPR_DEBUG, "peer_string is in invalid format and cannot be logged");
return address;
@ -181,8 +176,8 @@ LoggingSink::Entry::Address PeerStringToAddress(
return address;
}
void EncodeMessageToPayload(const grpc_core::SliceBuffer* message,
uint32_t log_len, LoggingSink::Entry* entry) {
void EncodeMessageToPayload(const SliceBuffer* message, uint32_t log_len,
LoggingSink::Entry* entry) {
auto* sb = message->c_slice_buffer();
entry->payload.message_length = sb->length;
// Log the message to a max of the configured message length
@ -203,12 +198,12 @@ void EncodeMessageToPayload(const grpc_core::SliceBuffer* message,
class CallData {
public:
CallData(bool is_client, const grpc_core::CallArgs& call_args,
CallData(bool is_client, const CallArgs& call_args,
const std::string& authority)
: call_id_(GetCallId()) {
absl::string_view path;
if (auto* value = call_args.client_initial_metadata->get_pointer(
grpc_core::HttpPathMetadata())) {
HttpPathMetadata())) {
path = value->as_string_view();
}
std::vector<std::string> parts =
@ -220,7 +215,7 @@ class CallData {
config_ = g_logging_sink->FindMatch(is_client, service_name_, method_name_);
if (config_.ShouldLog()) {
if (auto* value = call_args.client_initial_metadata->get_pointer(
grpc_core::HttpAuthorityMetadata())) {
HttpAuthorityMetadata())) {
authority_ = std::string(value->as_string_view());
} else {
authority_ = authority;
@ -230,8 +225,7 @@ class CallData {
bool ShouldLog() { return config_.ShouldLog(); }
void LogClientHeader(bool is_client,
const grpc_core::ClientMetadataHandle& metadata) {
void LogClientHeader(bool is_client, const ClientMetadataHandle& metadata) {
LoggingSink::Entry entry;
SetCommonEntryFields(&entry, is_client,
LoggingSink::Entry::EventType::kClientHeader);
@ -240,7 +234,7 @@ class CallData {
metadata->Encode(&encoder);
entry.payload_truncated = encoder.truncated();
if (!is_client) {
if (auto* value = metadata->get_pointer(grpc_core::PeerString())) {
if (auto* value = metadata->get_pointer(PeerString())) {
peer_ = PeerStringToAddress(*value);
}
}
@ -254,8 +248,7 @@ class CallData {
g_logging_sink->LogEntry(std::move(entry));
}
void LogServerHeader(bool is_client,
const grpc_core::ServerMetadata* metadata) {
void LogServerHeader(bool is_client, const ServerMetadata* metadata) {
LoggingSink::Entry entry;
SetCommonEntryFields(&entry, is_client,
LoggingSink::Entry::EventType::kServerHeader);
@ -265,7 +258,7 @@ class CallData {
metadata->Encode(&encoder);
entry.payload_truncated = encoder.truncated();
if (is_client) {
if (auto* value = metadata->get_pointer(grpc_core::PeerString())) {
if (auto* value = metadata->get_pointer(PeerString())) {
peer_ = PeerStringToAddress(*value);
}
}
@ -273,8 +266,7 @@ class CallData {
g_logging_sink->LogEntry(std::move(entry));
}
void LogServerTrailer(bool is_client,
const grpc_core::ServerMetadata* metadata) {
void LogServerTrailer(bool is_client, const ServerMetadata* metadata) {
LoggingSink::Entry entry;
SetCommonEntryFields(&entry, is_client,
LoggingSink::Entry::EventType::kServerTrailer);
@ -287,7 +279,7 @@ class CallData {
g_logging_sink->LogEntry(std::move(entry));
}
void LogClientMessage(bool is_client, const grpc_core::SliceBuffer* message) {
void LogClientMessage(bool is_client, const SliceBuffer* message) {
LoggingSink::Entry entry;
SetCommonEntryFields(&entry, is_client,
LoggingSink::Entry::EventType::kClientMessage);
@ -295,7 +287,7 @@ class CallData {
g_logging_sink->LogEntry(std::move(entry));
}
void LogServerMessage(bool is_client, const grpc_core::SliceBuffer* message) {
void LogServerMessage(bool is_client, const SliceBuffer* message) {
LoggingSink::Entry entry;
SetCommonEntryFields(&entry, is_client,
LoggingSink::Entry::EventType::kServerMessage);
@ -322,7 +314,7 @@ class CallData {
entry->peer = peer_;
entry->service_name = service_name_;
entry->method_name = method_name_;
entry->timestamp = grpc_core::Timestamp::Now();
entry->timestamp = Timestamp::Now();
}
uint64_t call_id_;
uint32_t sequence_id_ = 0;
@ -333,12 +325,12 @@ class CallData {
LoggingSink::Config config_;
};
class ClientLoggingFilter final : public grpc_core::ChannelFilter {
class ClientLoggingFilter final : public ChannelFilter {
public:
static const grpc_channel_filter kFilter;
static absl::StatusOr<ClientLoggingFilter> Create(
const grpc_core::ChannelArgs& args, ChannelFilter::Args /*filter_args*/) {
const ChannelArgs& args, ChannelFilter::Args /*filter_args*/) {
absl::optional<absl::string_view> default_authority =
args.GetString(GRPC_ARG_DEFAULT_AUTHORITY);
if (default_authority.has_value()) {
@ -347,48 +339,45 @@ class ClientLoggingFilter final : public grpc_core::ChannelFilter {
absl::optional<std::string> server_uri =
args.GetOwnedString(GRPC_ARG_SERVER_URI);
if (server_uri.has_value()) {
return ClientLoggingFilter(grpc_core::CoreConfiguration::Get()
.resolver_registry()
.GetDefaultAuthority(*server_uri));
return ClientLoggingFilter(
CoreConfiguration::Get().resolver_registry().GetDefaultAuthority(
*server_uri));
}
return ClientLoggingFilter("");
}
// Construct a promise for one call.
grpc_core::ArenaPromise<grpc_core::ServerMetadataHandle> MakeCallPromise(
grpc_core::CallArgs call_args,
grpc_core::NextPromiseFactory next_promise_factory) override {
CallData* calld =
grpc_core::GetContext<grpc_core::Arena>()->ManagedNew<CallData>(
true, call_args, default_authority_);
ArenaPromise<ServerMetadataHandle> MakeCallPromise(
CallArgs call_args, NextPromiseFactory next_promise_factory) override {
CallData* calld = GetContext<Arena>()->ManagedNew<CallData>(
true, call_args, default_authority_);
if (!calld->ShouldLog()) {
return next_promise_factory(std::move(call_args));
}
calld->LogClientHeader(/*is_client=*/true,
call_args.client_initial_metadata);
call_args.server_initial_metadata->InterceptAndMap(
[calld](grpc_core::ServerMetadataHandle metadata) {
[calld](ServerMetadataHandle metadata) {
calld->LogServerHeader(/*is_client=*/true, metadata.get());
return metadata;
});
call_args.client_to_server_messages->InterceptAndMapWithHalfClose(
[calld](grpc_core::MessageHandle message) {
[calld](MessageHandle message) {
calld->LogClientMessage(/*is_client=*/true, message->payload());
return message;
},
[calld] { calld->LogClientHalfClose(/*is_client=*/true); });
call_args.server_to_client_messages->InterceptAndMap(
[calld](grpc_core::MessageHandle message) {
[calld](MessageHandle message) {
calld->LogServerMessage(/*is_client=*/true, message->payload());
return message;
});
return grpc_core::OnCancel(
Map(next_promise_factory(std::move(call_args)),
[calld](grpc_core::ServerMetadataHandle md) {
calld->LogServerTrailer(/*is_client=*/true, md.get());
return md;
}),
[calld]() { calld->LogCancel(/*is_client=*/true); });
return OnCancel(Map(next_promise_factory(std::move(call_args)),
[calld](ServerMetadataHandle md) {
calld->LogServerTrailer(/*is_client=*/true, md.get());
return md;
}),
[calld]() { calld->LogCancel(/*is_client=*/true); });
}
private:
@ -398,99 +387,90 @@ class ClientLoggingFilter final : public grpc_core::ChannelFilter {
};
const grpc_channel_filter ClientLoggingFilter::kFilter =
grpc_core::MakePromiseBasedFilter<
ClientLoggingFilter, grpc_core::FilterEndpoint::kClient,
grpc_core::kFilterExaminesServerInitialMetadata |
grpc_core::kFilterExaminesInboundMessages |
grpc_core::kFilterExaminesOutboundMessages>("logging");
MakePromiseBasedFilter<ClientLoggingFilter, FilterEndpoint::kClient,
kFilterExaminesServerInitialMetadata |
kFilterExaminesInboundMessages |
kFilterExaminesOutboundMessages>("logging");
class ServerLoggingFilter final : public grpc_core::ChannelFilter {
class ServerLoggingFilter final : public ChannelFilter {
public:
static const grpc_channel_filter kFilter;
static absl::StatusOr<ServerLoggingFilter> Create(
const grpc_core::ChannelArgs& /*args*/,
ChannelFilter::Args /*filter_args*/) {
const ChannelArgs& /*args*/, ChannelFilter::Args /*filter_args*/) {
return ServerLoggingFilter();
}
// Construct a promise for one call.
grpc_core::ArenaPromise<grpc_core::ServerMetadataHandle> MakeCallPromise(
grpc_core::CallArgs call_args,
grpc_core::NextPromiseFactory next_promise_factory) override {
CallData* calld =
grpc_core::GetContext<grpc_core::Arena>()->ManagedNew<CallData>(
false, call_args, /*default_authority=*/"");
ArenaPromise<ServerMetadataHandle> MakeCallPromise(
CallArgs call_args, NextPromiseFactory next_promise_factory) override {
CallData* calld = GetContext<Arena>()->ManagedNew<CallData>(
false, call_args, /*default_authority=*/"");
if (!calld->ShouldLog()) {
return next_promise_factory(std::move(call_args));
}
calld->LogClientHeader(/*is_client=*/false,
call_args.client_initial_metadata);
call_args.server_initial_metadata->InterceptAndMap(
[calld](grpc_core::ServerMetadataHandle metadata) {
[calld](ServerMetadataHandle metadata) {
calld->LogServerHeader(/*is_client=*/false, metadata.get());
return metadata;
});
call_args.client_to_server_messages->InterceptAndMapWithHalfClose(
[calld](grpc_core::MessageHandle message) {
[calld](MessageHandle message) {
calld->LogClientMessage(/*is_client=*/false, message->payload());
return message;
},
[calld] { calld->LogClientHalfClose(/*is_client=*/false); });
call_args.server_to_client_messages->InterceptAndMap(
[calld](grpc_core::MessageHandle message) {
[calld](MessageHandle message) {
calld->LogServerMessage(/*is_client=*/false, message->payload());
return message;
});
return grpc_core::OnCancel(
Map(next_promise_factory(std::move(call_args)),
[calld](grpc_core::ServerMetadataHandle md) {
calld->LogServerTrailer(/*is_client=*/false, md.get());
return md;
}),
[calld]() { calld->LogCancel(/*is_client=*/false); });
return OnCancel(Map(next_promise_factory(std::move(call_args)),
[calld](ServerMetadataHandle md) {
calld->LogServerTrailer(/*is_client=*/false,
md.get());
return md;
}),
[calld]() { calld->LogCancel(/*is_client=*/false); });
}
};
const grpc_channel_filter ServerLoggingFilter::kFilter =
grpc_core::MakePromiseBasedFilter<
ServerLoggingFilter, grpc_core::FilterEndpoint::kServer,
grpc_core::kFilterExaminesServerInitialMetadata |
grpc_core::kFilterExaminesInboundMessages |
grpc_core::kFilterExaminesOutboundMessages>("logging");
MakePromiseBasedFilter<ServerLoggingFilter, FilterEndpoint::kServer,
kFilterExaminesServerInitialMetadata |
kFilterExaminesInboundMessages |
kFilterExaminesOutboundMessages>("logging");
} // namespace
void RegisterLoggingFilter(LoggingSink* sink) {
g_logging_sink = sink;
grpc_core::CoreConfiguration::RegisterBuilder(
[](grpc_core::CoreConfiguration::Builder* builder) {
builder->channel_init()->RegisterStage(
GRPC_SERVER_CHANNEL, INT_MAX,
[](grpc_core::ChannelStackBuilder* builder) {
// TODO(yashykt) : Figure out a good place to place this channel
// arg
if (builder->channel_args()
.GetInt("grpc.experimental.enable_observability")
.value_or(true)) {
builder->PrependFilter(&ServerLoggingFilter::kFilter);
}
return true;
});
builder->channel_init()->RegisterStage(
GRPC_CLIENT_CHANNEL, INT_MAX,
[](grpc_core::ChannelStackBuilder* builder) {
// TODO(yashykt) : Figure out a good place to place this channel
// arg
if (builder->channel_args()
.GetInt("grpc.experimental.enable_observability")
.value_or(true)) {
builder->PrependFilter(&ClientLoggingFilter::kFilter);
}
return true;
});
});
CoreConfiguration::RegisterBuilder([](CoreConfiguration::Builder* builder) {
builder->channel_init()->RegisterStage(
GRPC_SERVER_CHANNEL, INT_MAX, [](ChannelStackBuilder* builder) {
// TODO(yashykt) : Figure out a good place to place this channel
// arg
if (builder->channel_args()
.GetInt("grpc.experimental.enable_observability")
.value_or(true)) {
builder->PrependFilter(&ServerLoggingFilter::kFilter);
}
return true;
});
builder->channel_init()->RegisterStage(
GRPC_CLIENT_CHANNEL, INT_MAX, [](ChannelStackBuilder* builder) {
// TODO(yashykt) : Figure out a good place to place this channel
// arg
if (builder->channel_args()
.GetInt("grpc.experimental.enable_observability")
.value_or(true)) {
builder->PrependFilter(&ClientLoggingFilter::kFilter);
}
return true;
});
});
}
} // namespace internal
} // namespace grpc
} // namespace grpc_core

15
src/cpp/ext/filters/logging/logging_filter.h → src/core/ext/filters/logging/logging_filter.h
Unescape View File

@ -16,20 +16,17 @@
//
//
#ifndef GRPC_SRC_CPP_EXT_FILTERS_LOGGING_LOGGING_FILTER_H
#define GRPC_SRC_CPP_EXT_FILTERS_LOGGING_LOGGING_FILTER_H
#ifndef GRPC_SRC_CORE_EXT_FILTERS_LOGGING_LOGGING_FILTER_H
#define GRPC_SRC_CORE_EXT_FILTERS_LOGGING_LOGGING_FILTER_H
#include <grpc/support/port_platform.h>
#include "src/cpp/ext/filters/logging/logging_sink.h"
#include "src/core/ext/filters/logging/logging_sink.h"
namespace grpc {
namespace internal {
namespace grpc_core {
// TODO(yashykt): Add logging sink registration
void RegisterLoggingFilter(LoggingSink* sink);
} // namespace internal
} // namespace grpc
} // namespace grpc_core
#endif // GRPC_SRC_CPP_EXT_FILTERS_LOGGING_LOGGING_FILTER_H
#endif // GRPC_SRC_CORE_EXT_FILTERS_LOGGING_LOGGING_FILTER_H

16
src/cpp/ext/filters/logging/logging_sink.h → src/core/ext/filters/logging/logging_sink.h
Unescape View File

@ -16,8 +16,8 @@
//
//
#ifndef GRPC_SRC_CPP_EXT_FILTERS_LOGGING_LOGGING_SINK_H
#define GRPC_SRC_CPP_EXT_FILTERS_LOGGING_LOGGING_SINK_H
#ifndef GRPC_SRC_CORE_EXT_FILTERS_LOGGING_LOGGING_SINK_H
#define GRPC_SRC_CORE_EXT_FILTERS_LOGGING_LOGGING_SINK_H
#include <grpc/support/port_platform.h>
@ -30,8 +30,7 @@
#include "src/core/lib/gprpp/time.h"
namespace grpc {
namespace internal {
namespace grpc_core {
// Interface for a logging sink that will be used by the logging filter.
class LoggingSink {
@ -77,7 +76,7 @@ class LoggingSink {
struct Payload {
std::map<std::string, std::string> metadata;
grpc_core::Duration timeout;
Duration timeout;
uint32_t status_code = 0;
std::string status_message;
std::string status_details;
@ -102,7 +101,7 @@ class LoggingSink {
std::string authority;
std::string service_name;
std::string method_name;
grpc_core::Timestamp timestamp;
Timestamp timestamp;
};
virtual ~LoggingSink() = default;
@ -113,7 +112,6 @@ class LoggingSink {
virtual void LogEntry(Entry entry) = 0;
};
} // namespace internal
} // namespace grpc
} // namespace grpc_core
#endif // GRPC_SRC_CPP_EXT_FILTERS_LOGGING_LOGGING_SINK_H
#endif // GRPC_SRC_CORE_EXT_FILTERS_LOGGING_LOGGING_SINK_H

92
src/cpp/ext/filters/logging/BUILD
Unescape View File

@ -1,92 +0,0 @@
# gRPC Bazel BUILD file.
#
# Copyright 2022 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.
load(
"//bazel:grpc_build_system.bzl",
"grpc_cc_library",
)
licenses(["reciprocal"])
package(
default_visibility = ["//visibility:public"],
features = [
"layering_check",
],
)
grpc_cc_library(
name = "logging_sink",
hdrs = [
"logging_sink.h",
],
external_deps = ["absl/strings"],
language = "c++",
visibility = [
"//src/cpp/ext/gcp:__subpackages__",
"//test:__subpackages__",
],
deps = [
"//:gpr_platform",
"//src/core:time",
],
)
grpc_cc_library(
name = "logging_filter",
srcs = [
"logging_filter.cc",
],
hdrs = [
"logging_filter.h",
],
external_deps = [
"absl/random",
"absl/status:statusor",
"absl/strings",
"absl/types:optional",
],
language = "c++",
visibility = [
"//src/cpp/ext/gcp:__subpackages__",
"//test:__subpackages__",
],
deps = [
"logging_sink",
"//:channel_stack_builder",
"//:config",
"//:gpr",
"//:gpr_platform",
"//:grpc_base",
"//:grpc_client_channel",
"//:grpc_public_hdrs",
"//:grpc_resolver",
"//:uri_parser",
"//src/core:arena",
"//src/core:arena_promise",
"//src/core:cancel_callback",
"//src/core:channel_args",
"//src/core:channel_fwd",
"//src/core:channel_stack_type",
"//src/core:context",
"//src/core:map",
"//src/core:pipe",
"//src/core:poll",
"//src/core:slice",
"//src/core:slice_buffer",
"//src/core:time",
],
)

4
src/cpp/ext/gcp/BUILD
Unescape View File

@ -58,7 +58,7 @@ grpc_cc_library(
"//:gpr",
"//:grpc++",
"//:grpc_opencensus_plugin",
"//src/cpp/ext/filters/logging:logging_filter",
"//src/core:logging_filter",
],
)
@ -125,8 +125,8 @@ grpc_cc_library(
"//src/core:default_event_engine",
"//src/core:env",
"//src/core:json",
"//src/core:logging_sink",
"//src/core:time",
"//src/cpp/ext/filters/logging:logging_sink",
],
)

4
src/cpp/ext/gcp/observability.cc
Unescape View File

@ -43,9 +43,9 @@
#include <grpcpp/security/credentials.h>
#include <grpcpp/support/channel_arguments.h>
#include "src/core/ext/filters/logging/logging_filter.h"
#include "src/cpp/ext/filters/census/grpc_plugin.h"
#include "src/cpp/ext/filters/census/open_census_call_tracer.h"
#include "src/cpp/ext/filters/logging/logging_filter.h"
#include "src/cpp/ext/gcp/environment_autodetect.h"
#include "src/cpp/ext/gcp/observability_config.h"
#include "src/cpp/ext/gcp/observability_logging_sink.h"
@ -172,7 +172,7 @@ absl::Status GcpObservabilityInit() {
if (config->cloud_logging.has_value()) {
g_logging_sink = new grpc::internal::ObservabilityLoggingSink(
config->cloud_logging.value(), config->project_id, config->labels);
grpc::internal::RegisterLoggingFilter(g_logging_sink);
grpc_core::RegisterLoggingFilter(g_logging_sink);
}
return absl::OkStatus();
}

2
src/cpp/ext/gcp/observability_logging_sink.cc
Unescape View File

@ -54,6 +54,8 @@
namespace grpc {
namespace internal {
using grpc_core::LoggingSink;
ObservabilityLoggingSink::ObservabilityLoggingSink(
GcpObservabilityConfig::CloudLogging logging_config, std::string project_id,
std::map<std::string, std::string> labels)

6
src/cpp/ext/gcp/observability_logging_sink.h
Unescape View File

@ -37,8 +37,8 @@
#include <grpc/event_engine/event_engine.h>
#include "src/core/ext/filters/logging/logging_sink.h"
#include "src/core/lib/gprpp/sync.h"
#include "src/cpp/ext/filters/logging/logging_sink.h"
#include "src/cpp/ext/gcp/environment_autodetect.h"
#include "src/cpp/ext/gcp/observability_config.h"
@ -46,7 +46,7 @@ namespace grpc {
namespace internal {
// Interface for a logging sink that will be used by the logging filter.
class ObservabilityLoggingSink : public LoggingSink {
class ObservabilityLoggingSink : public grpc_core::LoggingSink {
public:
ObservabilityLoggingSink(GcpObservabilityConfig::CloudLogging logging_config,
std::string project_id,
@ -115,7 +115,7 @@ class ObservabilityLoggingSink : public LoggingSink {
};
// Exposed for just for testing purposes
void EntryToJsonStructProto(LoggingSink::Entry entry,
void EntryToJsonStructProto(grpc_core::LoggingSink::Entry entry,
::google::protobuf::Struct* json_payload);
} // namespace internal

2
test/cpp/ext/filters/logging/BUILD
Unescape View File

@ -34,7 +34,7 @@ grpc_cc_test(
],
deps = [
"//:grpc++",
"//src/cpp/ext/filters/logging:logging_filter",
"//src/core:logging_filter",
"//src/cpp/ext/gcp:observability_logging_sink",
"//src/proto/grpc/testing:echo_proto",
"//test/core/util:grpc_test_util",

25
test/cpp/ext/filters/logging/logging_test.cc
Unescape View File

@ -28,8 +28,8 @@
#include <grpc++/grpc++.h>
#include <grpcpp/support/status.h>
#include "src/core/ext/filters/logging/logging_filter.h"
#include "src/core/lib/gprpp/sync.h"
#include "src/cpp/ext/filters/logging/logging_filter.h"
#include "src/cpp/ext/gcp/observability_logging_sink.h"
#include "src/proto/grpc/testing/echo.grpc.pb.h"
#include "src/proto/grpc/testing/echo_messages.pb.h"
@ -42,7 +42,7 @@ namespace testing {
namespace {
using grpc::internal::LoggingSink;
using grpc_core::LoggingSink;
using ::testing::AllOf;
using ::testing::Eq;
@ -60,7 +60,7 @@ class MyTestServiceImpl : public TestServiceImpl {
}
};
class TestLoggingSink : public grpc::internal::LoggingSink {
class TestLoggingSink : public LoggingSink {
public:
Config FindMatch(bool /* is_client */, absl::string_view /* service */,
absl::string_view /* method */) override {
@ -105,7 +105,7 @@ class LoggingTest : public ::testing::Test {
protected:
static void SetUpTestSuite() {
g_test_logging_sink = new TestLoggingSink;
grpc::internal::RegisterLoggingFilter(g_test_logging_sink);
grpc_core::RegisterLoggingFilter(g_test_logging_sink);
}
void SetUp() override {
@ -149,8 +149,7 @@ class LoggingTest : public ::testing::Test {
};
TEST_F(LoggingTest, SimpleRpc) {
g_test_logging_sink->SetConfig(
grpc::internal::LoggingSink::Config(4096, 4096));
g_test_logging_sink->SetConfig(LoggingSink::Config(4096, 4096));
EchoRequest request;
request.set_message("foo");
EchoResponse response;
@ -306,7 +305,7 @@ TEST_F(LoggingTest, SimpleRpc) {
}
TEST_F(LoggingTest, LoggingDisabled) {
g_test_logging_sink->SetConfig(grpc::internal::LoggingSink::Config());
g_test_logging_sink->SetConfig(LoggingSink::Config());
EchoRequest request;
request.set_message("foo");
EchoResponse response;
@ -318,8 +317,8 @@ TEST_F(LoggingTest, LoggingDisabled) {
}
TEST_F(LoggingTest, MetadataTruncated) {
g_test_logging_sink->SetConfig(grpc::internal::LoggingSink::Config(
40 /* expect truncated metadata*/, 4096));
g_test_logging_sink->SetConfig(
LoggingSink::Config(40 /* expect truncated metadata*/, 4096));
EchoRequest request;
request.set_message("foo");
EchoResponse response;
@ -476,7 +475,7 @@ TEST_F(LoggingTest, MetadataTruncated) {
}
TEST_F(LoggingTest, PayloadTruncated) {
g_test_logging_sink->SetConfig(grpc::internal::LoggingSink::Config(4096, 10));
g_test_logging_sink->SetConfig(LoggingSink::Config(4096, 10));
EchoRequest request;
// The following message should get truncated
request.set_message("Hello World");
@ -637,8 +636,7 @@ TEST_F(LoggingTest, PayloadTruncated) {
}
TEST_F(LoggingTest, CancelledRpc) {
g_test_logging_sink->SetConfig(
grpc::internal::LoggingSink::Config(4096, 4096));
g_test_logging_sink->SetConfig(LoggingSink::Config(4096, 4096));
EchoRequest request;
request.set_message("foo");
const int kCancelDelayUs = 10 * 1000;
@ -679,8 +677,7 @@ TEST_F(LoggingTest, CancelledRpc) {
}
TEST_F(LoggingTest, ServerCancelsRpc) {
g_test_logging_sink->SetConfig(
grpc::internal::LoggingSink::Config(4096, 4096));
g_test_logging_sink->SetConfig(LoggingSink::Config(4096, 4096));
EchoRequest request;
request.set_message("foo");
auto* error = request.mutable_param()->mutable_expected_error();

2
test/cpp/ext/gcp/observability_logging_sink_test.cc
Unescape View File

@ -29,6 +29,8 @@ namespace internal {
namespace {
using grpc_core::LoggingSink;
TEST(GcpObservabilityLoggingSinkTest, LoggingConfigEmpty) {
const char* json_str = R"json({
"cloud_logging": {

1
tools/distrib/fix_build_deps.py
Unescape View File

@ -425,7 +425,6 @@ for dirname in [
"",
"src/core",
"src/cpp/ext/gcp",
"src/cpp/ext/filters/logging",
"test/core/backoff",
"test/core/uri",
"test/core/util",

Write Preview
Loading…
Cancel
Save