Shortcuts on this page
r m x toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
1# -*- coding: utf-8 -*-
2# Copyright 2025 Google LLC
3#
4# Licensed under the Apache License, Version 2.0 (the "License");
5# you may not use this file except in compliance with the License.
6# You may obtain a copy of the License at
7#
8# http://www.apache.org/licenses/LICENSE-2.0
9#
10# Unless required by applicable law or agreed to in writing, software
11# distributed under the License is distributed on an "AS IS" BASIS,
12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13# See the License for the specific language governing permissions and
14# limitations under the License.
15#
16import json
17import logging as std_logging
18import pickle
19import warnings
20from typing import Callable, Dict, Optional, Sequence, Tuple, Union
22from google.api_core import grpc_helpers
23from google.api_core import gapic_v1
24import google.auth # type: ignore
25from google.auth import credentials as ga_credentials # type: ignore
26from google.auth.transport.grpc import SslCredentials # type: ignore
27from google.protobuf.json_format import MessageToJson
28import google.protobuf.message
30import grpc # type: ignore
31import proto # type: ignore
33from google.cloud.errorreporting_v1beta1.types import error_stats_service
34from .base import ErrorStatsServiceTransport, DEFAULT_CLIENT_INFO
36try:
37 from google.api_core import client_logging # type: ignore
39 CLIENT_LOGGING_SUPPORTED = True # pragma: NO COVER
40except ImportError: # pragma: NO COVER
41 CLIENT_LOGGING_SUPPORTED = False
43_LOGGER = std_logging.getLogger(__name__)
46class _LoggingClientInterceptor(grpc.UnaryUnaryClientInterceptor): # pragma: NO COVER
47 def intercept_unary_unary(self, continuation, client_call_details, request):
48 logging_enabled = CLIENT_LOGGING_SUPPORTED and _LOGGER.isEnabledFor(
49 std_logging.DEBUG
50 )
51 if logging_enabled: # pragma: NO COVER
52 request_metadata = client_call_details.metadata
53 if isinstance(request, proto.Message):
54 request_payload = type(request).to_json(request)
55 elif isinstance(request, google.protobuf.message.Message):
56 request_payload = MessageToJson(request)
57 else:
58 request_payload = f"{type(request).__name__}: {pickle.dumps(request)}"
60 request_metadata = {
61 key: value.decode("utf-8") if isinstance(value, bytes) else value
62 for key, value in request_metadata
63 }
64 grpc_request = {
65 "payload": request_payload,
66 "requestMethod": "grpc",
67 "metadata": dict(request_metadata),
68 }
69 _LOGGER.debug(
70 f"Sending request for {client_call_details.method}",
71 extra={
72 "serviceName": "google.devtools.clouderrorreporting.v1beta1.ErrorStatsService",
73 "rpcName": str(client_call_details.method),
74 "request": grpc_request,
75 "metadata": grpc_request["metadata"],
76 },
77 )
78 response = continuation(client_call_details, request)
79 if logging_enabled: # pragma: NO COVER
80 response_metadata = response.trailing_metadata()
81 # Convert gRPC metadata `<class 'grpc.aio._metadata.Metadata'>` to list of tuples
82 metadata = (
83 dict([(k, str(v)) for k, v in response_metadata])
84 if response_metadata
85 else None
86 )
87 result = response.result()
88 if isinstance(result, proto.Message):
89 response_payload = type(result).to_json(result)
90 elif isinstance(result, google.protobuf.message.Message):
91 response_payload = MessageToJson(result)
92 else:
93 response_payload = f"{type(result).__name__}: {pickle.dumps(result)}"
94 grpc_response = {
95 "payload": response_payload,
96 "metadata": metadata,
97 "status": "OK",
98 }
99 _LOGGER.debug(
100 f"Received response for {client_call_details.method}.",
101 extra={
102 "serviceName": "google.devtools.clouderrorreporting.v1beta1.ErrorStatsService",
103 "rpcName": client_call_details.method,
104 "response": grpc_response,
105 "metadata": grpc_response["metadata"],
106 },
107 )
108 return response
111class ErrorStatsServiceGrpcTransport(ErrorStatsServiceTransport):
112 """gRPC backend transport for ErrorStatsService.
114 An API for retrieving and managing error statistics as well
115 as data for individual events.
117 This class defines the same methods as the primary client, so the
118 primary client can load the underlying transport implementation
119 and call it.
121 It sends protocol buffers over the wire using gRPC (which is built on
122 top of HTTP/2); the ``grpcio`` package must be installed.
123 """
125 _stubs: Dict[str, Callable]
127 def __init__(
128 self,
129 *,
130 host: str = "clouderrorreporting.googleapis.com",
131 credentials: Optional[ga_credentials.Credentials] = None,
132 credentials_file: Optional[str] = None,
133 scopes: Optional[Sequence[str]] = None,
134 channel: Optional[Union[grpc.Channel, Callable[..., grpc.Channel]]] = None,
135 api_mtls_endpoint: Optional[str] = None,
136 client_cert_source: Optional[Callable[[], Tuple[bytes, bytes]]] = None,
137 ssl_channel_credentials: Optional[grpc.ChannelCredentials] = None,
138 client_cert_source_for_mtls: Optional[Callable[[], Tuple[bytes, bytes]]] = None,
139 quota_project_id: Optional[str] = None,
140 client_info: gapic_v1.client_info.ClientInfo = DEFAULT_CLIENT_INFO,
141 always_use_jwt_access: Optional[bool] = False,
142 api_audience: Optional[str] = None,
143 ) -> None:
144 """Instantiate the transport.
146 Args:
147 host (Optional[str]):
148 The hostname to connect to (default: 'clouderrorreporting.googleapis.com').
149 credentials (Optional[google.auth.credentials.Credentials]): The
150 authorization credentials to attach to requests. These
151 credentials identify the application to the service; if none
152 are specified, the client will attempt to ascertain the
153 credentials from the environment.
154 This argument is ignored if a ``channel`` instance is provided.
155 credentials_file (Optional[str]): A file with credentials that can
156 be loaded with :func:`google.auth.load_credentials_from_file`.
157 This argument is ignored if a ``channel`` instance is provided.
158 scopes (Optional(Sequence[str])): A list of scopes. This argument is
159 ignored if a ``channel`` instance is provided.
160 channel (Optional[Union[grpc.Channel, Callable[..., grpc.Channel]]]):
161 A ``Channel`` instance through which to make calls, or a Callable
162 that constructs and returns one. If set to None, ``self.create_channel``
163 is used to create the channel. If a Callable is given, it will be called
164 with the same arguments as used in ``self.create_channel``.
165 api_mtls_endpoint (Optional[str]): Deprecated. The mutual TLS endpoint.
166 If provided, it overrides the ``host`` argument and tries to create
167 a mutual TLS channel with client SSL credentials from
168 ``client_cert_source`` or application default SSL credentials.
169 client_cert_source (Optional[Callable[[], Tuple[bytes, bytes]]]):
170 Deprecated. A callback to provide client SSL certificate bytes and
171 private key bytes, both in PEM format. It is ignored if
172 ``api_mtls_endpoint`` is None.
173 ssl_channel_credentials (grpc.ChannelCredentials): SSL credentials
174 for the grpc channel. It is ignored if a ``channel`` instance is provided.
175 client_cert_source_for_mtls (Optional[Callable[[], Tuple[bytes, bytes]]]):
176 A callback to provide client certificate bytes and private key bytes,
177 both in PEM format. It is used to configure a mutual TLS channel. It is
178 ignored if a ``channel`` instance or ``ssl_channel_credentials`` is provided.
179 quota_project_id (Optional[str]): An optional project to use for billing
180 and quota.
181 client_info (google.api_core.gapic_v1.client_info.ClientInfo):
182 The client info used to send a user-agent string along with
183 API requests. If ``None``, then default info will be used.
184 Generally, you only need to set this if you're developing
185 your own client library.
186 always_use_jwt_access (Optional[bool]): Whether self signed JWT should
187 be used for service account credentials.
189 Raises:
190 google.auth.exceptions.MutualTLSChannelError: If mutual TLS transport
191 creation failed for any reason.
192 google.api_core.exceptions.DuplicateCredentialArgs: If both ``credentials``
193 and ``credentials_file`` are passed.
194 """
195 self._grpc_channel = None
196 self._ssl_channel_credentials = ssl_channel_credentials
197 self._stubs: Dict[str, Callable] = {}
199 if api_mtls_endpoint:
200 warnings.warn("api_mtls_endpoint is deprecated", DeprecationWarning)
201 if client_cert_source:
202 warnings.warn("client_cert_source is deprecated", DeprecationWarning)
204 if isinstance(channel, grpc.Channel):
205 # Ignore credentials if a channel was passed.
206 credentials = None
207 self._ignore_credentials = True
208 # If a channel was explicitly provided, set it.
209 self._grpc_channel = channel
210 self._ssl_channel_credentials = None
212 else:
213 if api_mtls_endpoint:
214 host = api_mtls_endpoint
216 # Create SSL credentials with client_cert_source or application
217 # default SSL credentials.
218 if client_cert_source:
219 cert, key = client_cert_source()
220 self._ssl_channel_credentials = grpc.ssl_channel_credentials(
221 certificate_chain=cert, private_key=key
222 )
223 else:
224 self._ssl_channel_credentials = SslCredentials().ssl_credentials
226 else:
227 if client_cert_source_for_mtls and not ssl_channel_credentials:
228 cert, key = client_cert_source_for_mtls()
229 self._ssl_channel_credentials = grpc.ssl_channel_credentials(
230 certificate_chain=cert, private_key=key
231 )
233 # The base transport sets the host, credentials and scopes
234 super().__init__(
235 host=host,
236 credentials=credentials,
237 credentials_file=credentials_file,
238 scopes=scopes,
239 quota_project_id=quota_project_id,
240 client_info=client_info,
241 always_use_jwt_access=always_use_jwt_access,
242 api_audience=api_audience,
243 )
245 if not self._grpc_channel:
246 # initialize with the provided callable or the default channel
247 channel_init = channel or type(self).create_channel
248 self._grpc_channel = channel_init(
249 self._host,
250 # use the credentials which are saved
251 credentials=self._credentials,
252 # Set ``credentials_file`` to ``None`` here as
253 # the credentials that we saved earlier should be used.
254 credentials_file=None,
255 scopes=self._scopes,
256 ssl_credentials=self._ssl_channel_credentials,
257 quota_project_id=quota_project_id,
258 options=[
259 ("grpc.max_send_message_length", -1),
260 ("grpc.max_receive_message_length", -1),
261 ],
262 )
264 self._interceptor = _LoggingClientInterceptor()
265 self._logged_channel = grpc.intercept_channel(
266 self._grpc_channel, self._interceptor
267 )
269 # Wrap messages. This must be done after self._logged_channel exists
270 self._prep_wrapped_messages(client_info)
272 @classmethod
273 def create_channel(
274 cls,
275 host: str = "clouderrorreporting.googleapis.com",
276 credentials: Optional[ga_credentials.Credentials] = None,
277 credentials_file: Optional[str] = None,
278 scopes: Optional[Sequence[str]] = None,
279 quota_project_id: Optional[str] = None,
280 **kwargs,
281 ) -> grpc.Channel:
282 """Create and return a gRPC channel object.
283 Args:
284 host (Optional[str]): The host for the channel to use.
285 credentials (Optional[~.Credentials]): The
286 authorization credentials to attach to requests. These
287 credentials identify this application to the service. If
288 none are specified, the client will attempt to ascertain
289 the credentials from the environment.
290 credentials_file (Optional[str]): A file with credentials that can
291 be loaded with :func:`google.auth.load_credentials_from_file`.
292 This argument is mutually exclusive with credentials.
293 scopes (Optional[Sequence[str]]): A optional list of scopes needed for this
294 service. These are only used when credentials are not specified and
295 are passed to :func:`google.auth.default`.
296 quota_project_id (Optional[str]): An optional project to use for billing
297 and quota.
298 kwargs (Optional[dict]): Keyword arguments, which are passed to the
299 channel creation.
300 Returns:
301 grpc.Channel: A gRPC channel object.
303 Raises:
304 google.api_core.exceptions.DuplicateCredentialArgs: If both ``credentials``
305 and ``credentials_file`` are passed.
306 """
308 return grpc_helpers.create_channel(
309 host,
310 credentials=credentials,
311 credentials_file=credentials_file,
312 quota_project_id=quota_project_id,
313 default_scopes=cls.AUTH_SCOPES,
314 scopes=scopes,
315 default_host=cls.DEFAULT_HOST,
316 **kwargs,
317 )
319 @property
320 def grpc_channel(self) -> grpc.Channel:
321 """Return the channel designed to connect to this service."""
322 return self._grpc_channel
324 @property
325 def list_group_stats(
326 self,
327 ) -> Callable[
328 [error_stats_service.ListGroupStatsRequest],
329 error_stats_service.ListGroupStatsResponse,
330 ]:
331 r"""Return a callable for the list group stats method over gRPC.
333 Lists the specified groups.
335 Returns:
336 Callable[[~.ListGroupStatsRequest],
337 ~.ListGroupStatsResponse]:
338 A function that, when called, will call the underlying RPC
339 on the server.
340 """
341 # Generate a "stub function" on-the-fly which will actually make
342 # the request.
343 # gRPC handles serialization and deserialization, so we just need
344 # to pass in the functions for each.
345 if "list_group_stats" not in self._stubs:
346 self._stubs["list_group_stats"] = self._logged_channel.unary_unary(
347 "/google.devtools.clouderrorreporting.v1beta1.ErrorStatsService/ListGroupStats",
348 request_serializer=error_stats_service.ListGroupStatsRequest.serialize,
349 response_deserializer=error_stats_service.ListGroupStatsResponse.deserialize,
350 )
351 return self._stubs["list_group_stats"]
353 @property
354 def list_events(
355 self,
356 ) -> Callable[
357 [error_stats_service.ListEventsRequest], error_stats_service.ListEventsResponse
358 ]:
359 r"""Return a callable for the list events method over gRPC.
361 Lists the specified events.
363 Returns:
364 Callable[[~.ListEventsRequest],
365 ~.ListEventsResponse]:
366 A function that, when called, will call the underlying RPC
367 on the server.
368 """
369 # Generate a "stub function" on-the-fly which will actually make
370 # the request.
371 # gRPC handles serialization and deserialization, so we just need
372 # to pass in the functions for each.
373 if "list_events" not in self._stubs:
374 self._stubs["list_events"] = self._logged_channel.unary_unary(
375 "/google.devtools.clouderrorreporting.v1beta1.ErrorStatsService/ListEvents",
376 request_serializer=error_stats_service.ListEventsRequest.serialize,
377 response_deserializer=error_stats_service.ListEventsResponse.deserialize,
378 )
379 return self._stubs["list_events"]
381 @property
382 def delete_events(
383 self,
384 ) -> Callable[
385 [error_stats_service.DeleteEventsRequest],
386 error_stats_service.DeleteEventsResponse,
387 ]:
388 r"""Return a callable for the delete events method over gRPC.
390 Deletes all error events of a given project.
392 Returns:
393 Callable[[~.DeleteEventsRequest],
394 ~.DeleteEventsResponse]:
395 A function that, when called, will call the underlying RPC
396 on the server.
397 """
398 # Generate a "stub function" on-the-fly which will actually make
399 # the request.
400 # gRPC handles serialization and deserialization, so we just need
401 # to pass in the functions for each.
402 if "delete_events" not in self._stubs:
403 self._stubs["delete_events"] = self._logged_channel.unary_unary(
404 "/google.devtools.clouderrorreporting.v1beta1.ErrorStatsService/DeleteEvents",
405 request_serializer=error_stats_service.DeleteEventsRequest.serialize,
406 response_deserializer=error_stats_service.DeleteEventsResponse.deserialize,
407 )
408 return self._stubs["delete_events"]
410 def close(self):
411 self._logged_channel.close()
413 @property
414 def kind(self) -> str:
415 return "grpc"
418__all__ = ("ErrorStatsServiceGrpcTransport",)