David Garcia Quintas
9 years ago
commit
8f7365021f
79 changed files with 1380 additions and 664 deletions
@ -0,0 +1,151 @@ |
||||
// Copyright 2016, Google Inc. |
||||
// All rights reserved. |
||||
// |
||||
// Redistribution and use in source and binary forms, with or without |
||||
// modification, are permitted provided that the following conditions are |
||||
// met: |
||||
// |
||||
// * Redistributions of source code must retain the above copyright |
||||
// notice, this list of conditions and the following disclaimer. |
||||
// * Redistributions in binary form must reproduce the above |
||||
// copyright notice, this list of conditions and the following disclaimer |
||||
// in the documentation and/or other materials provided with the |
||||
// distribution. |
||||
// * Neither the name of Google Inc. nor the names of its |
||||
// contributors may be used to endorse or promote products derived from |
||||
// this software without specific prior written permission. |
||||
// |
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
|
||||
// Service exported by server reflection |
||||
|
||||
syntax = "proto3"; |
||||
|
||||
package grpc.reflection.v1alpha; |
||||
|
||||
service ServerReflection { |
||||
// The reflection service is structured as a bidirectional stream, ensuring |
||||
// all related requests go to a single server. |
||||
rpc ServerReflectionInfo(stream ServerReflectionRequest) |
||||
returns (stream ServerReflectionResponse); |
||||
} |
||||
|
||||
// The message sent by the client when calling ServerReflectionInfo method. |
||||
message ServerReflectionRequest { |
||||
string host = 1; |
||||
// To use reflection service, the client should set one of the following |
||||
// fields in message_request. The server distinguishes requests by their |
||||
// defined field and then handles them using corresponding methods. |
||||
oneof message_request { |
||||
// Find a proto file by the file name. |
||||
string file_by_filename = 3; |
||||
|
||||
// Find the proto file that declares the given fully-qualified symbol name. |
||||
// This field should be a fully-qualified symbol name |
||||
// (e.g. <package>.<service>[.<method>] or <package>.<type>). |
||||
string file_containing_symbol = 4; |
||||
|
||||
// Find the proto file which defines an extension extending the given |
||||
// message type with the given field number. |
||||
ExtensionRequest file_containing_extension = 5; |
||||
|
||||
// Finds the tag numbers used by all known extensions of extendee_type, and |
||||
// appends them to ExtensionNumberResponse in an undefined order. |
||||
// Its corresponding method is best-effort: it's not guaranteed that the |
||||
// reflection service will implement this method, and it's not guaranteed |
||||
// that this method will provide all extensions. Returns |
||||
// StatusCode::UNIMPLEMENTED if it's not implemented. |
||||
// This field should be a fully-qualified type name. The format is |
||||
// <package>.<type> |
||||
string all_extension_numbers_of_type = 6; |
||||
|
||||
// List the full names of registered services. The content will not be |
||||
// checked. |
||||
string list_services = 7; |
||||
} |
||||
} |
||||
|
||||
// The type name and extension number sent by the client when requesting |
||||
// file_containing_extension. |
||||
message ExtensionRequest { |
||||
// Fully-qualified type name. The format should be <package>.<type> |
||||
string containing_type = 1; |
||||
int32 extension_number = 2; |
||||
} |
||||
|
||||
// The message sent by the server to answer ServerReflectionInfo method. |
||||
message ServerReflectionResponse { |
||||
string valid_host = 1; |
||||
ServerReflectionRequest original_request = 2; |
||||
// The server set one of the following fields accroding to the message_request |
||||
// in the request. |
||||
oneof message_response { |
||||
// This message is used to answer file_by_filename, file_containing_symbol, |
||||
// file_containing_extension requests with transitive dependencies. As |
||||
// the repeated label is not allowed in oneof fields, we use a |
||||
// FileDescriptorResponse message to encapsulate the repeated fields. |
||||
// The reflection service is allowed to avoid sending FileDescriptorProtos |
||||
// that were previously sent in response to earlier requests in the stream. |
||||
FileDescriptorResponse file_descriptor_response = 4; |
||||
|
||||
// This message is used to answer all_extension_numbers_of_type requst. |
||||
ExtensionNumberResponse all_extension_numbers_response = 5; |
||||
|
||||
// This message is used to answer list_services request. |
||||
ListServiceResponse list_services_response = 6; |
||||
|
||||
// This message is used when an error occurs. |
||||
ErrorResponse error_response = 7; |
||||
} |
||||
} |
||||
|
||||
// Serialized FileDescriptorProto messages sent by the server answering |
||||
// a file_by_filename, file_containing_symbol, or file_containing_extension |
||||
// request. |
||||
message FileDescriptorResponse { |
||||
// Serialized FileDescriptorProto messages. We avoid taking a dependency on |
||||
// descriptor.proto, which uses proto2 only features, by making them opaque |
||||
// bytes instead. |
||||
repeated bytes file_descriptor_proto = 1; |
||||
} |
||||
|
||||
// A list of extension numbers sent by the server answering |
||||
// all_extension_numbers_of_type request. |
||||
message ExtensionNumberResponse { |
||||
// Full name of the base type, including the package name. The format |
||||
// is <package>.<type> |
||||
string base_type_name = 1; |
||||
repeated int32 extension_number = 2; |
||||
} |
||||
|
||||
// A list of ServiceResponse sent by the server answering list_services request. |
||||
message ListServiceResponse { |
||||
// The information of each service may be expanded in the future, so we use |
||||
// ServiceResponse message to encapsulate it. |
||||
repeated ServiceResponse service = 1; |
||||
} |
||||
|
||||
// The information of a single service used by ListServiceResponse to answer |
||||
// list_services request. |
||||
message ServiceResponse { |
||||
// Full name of a registered service, including its package name. The format |
||||
// is <package>.<service> |
||||
string name = 1; |
||||
} |
||||
|
||||
// The error code and error message sent by the server when an error occurs. |
||||
message ErrorResponse { |
||||
// This field uses the error codes defined in grpc::StatusCode. |
||||
int32 error_code = 1; |
||||
string error_message = 2; |
||||
} |
||||
@ -0,0 +1,28 @@ |
||||
# Copyright 2016, Google Inc. |
||||
# All rights reserved. |
||||
# |
||||
# Redistribution and use in source and binary forms, with or without |
||||
# modification, are permitted provided that the following conditions are |
||||
# met: |
||||
# |
||||
# * Redistributions of source code must retain the above copyright |
||||
# notice, this list of conditions and the following disclaimer. |
||||
# * Redistributions in binary form must reproduce the above |
||||
# copyright notice, this list of conditions and the following disclaimer |
||||
# in the documentation and/or other materials provided with the |
||||
# distribution. |
||||
# * Neither the name of Google Inc. nor the names of its |
||||
# contributors may be used to endorse or promote products derived from |
||||
# this software without specific prior written permission. |
||||
# |
||||
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
@ -0,0 +1,75 @@ |
||||
# Copyright 2016, Google Inc. |
||||
# All rights reserved. |
||||
# |
||||
# Redistribution and use in source and binary forms, with or without |
||||
# modification, are permitted provided that the following conditions are |
||||
# met: |
||||
# |
||||
# * Redistributions of source code must retain the above copyright |
||||
# notice, this list of conditions and the following disclaimer. |
||||
# * Redistributions in binary form must reproduce the above |
||||
# copyright notice, this list of conditions and the following disclaimer |
||||
# in the documentation and/or other materials provided with the |
||||
# distribution. |
||||
# * Neither the name of Google Inc. nor the names of its |
||||
# contributors may be used to endorse or promote products derived from |
||||
# this software without specific prior written permission. |
||||
# |
||||
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
|
||||
"""Tests of grpc_health.health.v1.health.""" |
||||
|
||||
import unittest |
||||
|
||||
from grpc_health.health.v1 import health |
||||
from grpc_health.health.v1 import health_pb2 |
||||
|
||||
|
||||
class HealthServicerTest(unittest.TestCase): |
||||
|
||||
def setUp(self): |
||||
self.servicer = health.HealthServicer() |
||||
self.servicer.set('', health_pb2.HealthCheckResponse.SERVING) |
||||
self.servicer.set('grpc.test.TestServiceServing', |
||||
health_pb2.HealthCheckResponse.SERVING) |
||||
self.servicer.set('grpc.test.TestServiceUnknown', |
||||
health_pb2.HealthCheckResponse.UNKNOWN) |
||||
self.servicer.set('grpc.test.TestServiceNotServing', |
||||
health_pb2.HealthCheckResponse.NOT_SERVING) |
||||
|
||||
def test_empty_service(self): |
||||
request = health_pb2.HealthCheckRequest() |
||||
resp = self.servicer.Check(request, None) |
||||
self.assertEqual(resp.status, health_pb2.HealthCheckResponse.SERVING) |
||||
|
||||
def test_serving_service(self): |
||||
request = health_pb2.HealthCheckRequest( |
||||
service='grpc.test.TestServiceServing') |
||||
resp = self.servicer.Check(request, None) |
||||
self.assertEqual(resp.status, health_pb2.HealthCheckResponse.SERVING) |
||||
|
||||
def test_unknown_serivce(self): |
||||
request = health_pb2.HealthCheckRequest( |
||||
service='grpc.test.TestServiceUnknown') |
||||
resp = self.servicer.Check(request, None) |
||||
self.assertEqual(resp.status, health_pb2.HealthCheckResponse.UNKNOWN) |
||||
|
||||
def test_not_serving_service(self): |
||||
request = health_pb2.HealthCheckRequest( |
||||
service='grpc.test.TestServiceNotServing') |
||||
resp = self.servicer.Check(request, None) |
||||
self.assertEqual(resp.status, health_pb2.HealthCheckResponse.NOT_SERVING) |
||||
|
||||
|
||||
if __name__ == '__main__': |
||||
unittest.main(verbosity=2) |
||||
@ -0,0 +1,5 @@ |
||||
*.proto |
||||
*_pb2.py |
||||
build/ |
||||
grpcio_health_checking.egg-info/ |
||||
dist/ |
||||
@ -1,2 +1,3 @@ |
||||
graft grpc |
||||
include commands.py |
||||
include health_commands.py |
||||
graft grpc_health |
||||
global-exclude *.pyc |
||||
|
||||
@ -1,49 +0,0 @@ |
||||
// Copyright 2015, Google Inc. |
||||
// All rights reserved. |
||||
// |
||||
// Redistribution and use in source and binary forms, with or without |
||||
// modification, are permitted provided that the following conditions are |
||||
// met: |
||||
// |
||||
// * Redistributions of source code must retain the above copyright |
||||
// notice, this list of conditions and the following disclaimer. |
||||
// * Redistributions in binary form must reproduce the above |
||||
// copyright notice, this list of conditions and the following disclaimer |
||||
// in the documentation and/or other materials provided with the |
||||
// distribution. |
||||
// * Neither the name of Google Inc. nor the names of its |
||||
// contributors may be used to endorse or promote products derived from |
||||
// this software without specific prior written permission. |
||||
// |
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
|
||||
syntax = "proto3"; |
||||
|
||||
package grpc.health.v1; |
||||
|
||||
message HealthCheckRequest { |
||||
string service = 1; |
||||
} |
||||
|
||||
message HealthCheckResponse { |
||||
enum ServingStatus { |
||||
UNKNOWN = 0; |
||||
SERVING = 1; |
||||
NOT_SERVING = 2; |
||||
} |
||||
ServingStatus status = 1; |
||||
} |
||||
|
||||
service Health { |
||||
rpc Check(HealthCheckRequest) returns (HealthCheckResponse); |
||||
} |
||||
@ -1,129 +0,0 @@ |
||||
# Copyright 2015, Google Inc. |
||||
# All rights reserved. |
||||
# |
||||
# Redistribution and use in source and binary forms, with or without |
||||
# modification, are permitted provided that the following conditions are |
||||
# met: |
||||
# |
||||
# * Redistributions of source code must retain the above copyright |
||||
# notice, this list of conditions and the following disclaimer. |
||||
# * Redistributions in binary form must reproduce the above |
||||
# copyright notice, this list of conditions and the following disclaimer |
||||
# in the documentation and/or other materials provided with the |
||||
# distribution. |
||||
# * Neither the name of Google Inc. nor the names of its |
||||
# contributors may be used to endorse or promote products derived from |
||||
# this software without specific prior written permission. |
||||
# |
||||
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
|
||||
"""Reference implementation for health checking in gRPC Python.""" |
||||
|
||||
import abc |
||||
import enum |
||||
import threading |
||||
|
||||
from grpc.health.v1 import health_pb2 |
||||
|
||||
|
||||
@enum.unique |
||||
class HealthStatus(enum.Enum): |
||||
"""Statuses for a service mirroring the reference health.proto's values.""" |
||||
UNKNOWN = health_pb2.HealthCheckResponse.UNKNOWN |
||||
SERVING = health_pb2.HealthCheckResponse.SERVING |
||||
NOT_SERVING = health_pb2.HealthCheckResponse.NOT_SERVING |
||||
|
||||
|
||||
class _HealthServicer(health_pb2.EarlyAdopterHealthServicer): |
||||
"""Servicer handling RPCs for service statuses.""" |
||||
|
||||
def __init__(self): |
||||
self._server_status_lock = threading.Lock() |
||||
self._server_status = {} |
||||
|
||||
def Check(self, request, context): |
||||
with self._server_status_lock: |
||||
if request.service not in self._server_status: |
||||
# TODO(atash): once the Python API has a way of setting the server |
||||
# status, bring us into conformance with the health check spec by |
||||
# returning the NOT_FOUND status here. |
||||
raise NotImplementedError() |
||||
else: |
||||
return health_pb2.HealthCheckResponse( |
||||
status=self._server_status[request.service].value) |
||||
|
||||
def set(service, status): |
||||
if not isinstance(status, HealthStatus): |
||||
raise TypeError('expected grpc.health.v1.health.HealthStatus ' |
||||
'for argument `status` but got {}'.format(status)) |
||||
with self._server_status_lock: |
||||
self._server_status[service] = status |
||||
|
||||
|
||||
class HealthServer(health_pb2.EarlyAdopterHealthServer): |
||||
"""Interface for the reference gRPC Python health server.""" |
||||
__metaclass__ = abc.ABCMeta |
||||
|
||||
@abc.abstractmethod |
||||
def start(self): |
||||
raise NotImplementedError() |
||||
|
||||
@abc.abstractmethod |
||||
def stop(self): |
||||
raise NotImplementedError() |
||||
|
||||
@abc.abstractmethod |
||||
def set(self, service, status): |
||||
"""Set the status of the given service. |
||||
|
||||
Args: |
||||
service (str): service name of the service to set the reported status of |
||||
status (HealthStatus): status to set for the specified service |
||||
""" |
||||
raise NotImplementedError() |
||||
|
||||
|
||||
class _HealthServerImplementation(HealthServer): |
||||
"""Implementation for the reference gRPC Python health server.""" |
||||
|
||||
def __init__(self, server, servicer): |
||||
self._server = server |
||||
self._servicer = servicer |
||||
|
||||
def start(self): |
||||
self._server.start() |
||||
|
||||
def stop(self): |
||||
self._server.stop() |
||||
|
||||
def set(self, service, status): |
||||
self._servicer.set(service, status) |
||||
|
||||
|
||||
def create_Health_server(port, private_key=None, certificate_chain=None): |
||||
"""Get a HealthServer instance. |
||||
|
||||
Args: |
||||
port (int): port number passed through to health_pb2 server creation |
||||
routine. |
||||
private_key (str): to-be-created server's desired private key |
||||
certificate_chain (str): to-be-created server's desired certificate chain |
||||
|
||||
Returns: |
||||
An instance of HealthServer (conforming thus to |
||||
EarlyAdopterHealthServer and providing a method to set server status).""" |
||||
servicer = _HealthServicer() |
||||
server = health_pb2.early_adopter_create_Health_server( |
||||
servicer, port=port, private_key=private_key, |
||||
certificate_chain=certificate_chain) |
||||
return _HealthServerImplementation(server, servicer) |
||||
@ -0,0 +1,66 @@ |
||||
# Copyright 2015, Google Inc. |
||||
# All rights reserved. |
||||
# |
||||
# Redistribution and use in source and binary forms, with or without |
||||
# modification, are permitted provided that the following conditions are |
||||
# met: |
||||
# |
||||
# * Redistributions of source code must retain the above copyright |
||||
# notice, this list of conditions and the following disclaimer. |
||||
# * Redistributions in binary form must reproduce the above |
||||
# copyright notice, this list of conditions and the following disclaimer |
||||
# in the documentation and/or other materials provided with the |
||||
# distribution. |
||||
# * Neither the name of Google Inc. nor the names of its |
||||
# contributors may be used to endorse or promote products derived from |
||||
# this software without specific prior written permission. |
||||
# |
||||
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
|
||||
"""Reference implementation for health checking in gRPC Python.""" |
||||
|
||||
import threading |
||||
|
||||
from grpc_health.health.v1 import health_pb2 |
||||
|
||||
|
||||
class HealthServicer(health_pb2.BetaHealthServicer): |
||||
"""Servicer handling RPCs for service statuses.""" |
||||
|
||||
def __init__(self): |
||||
self._server_status_lock = threading.Lock() |
||||
self._server_status = {} |
||||
|
||||
def Check(self, request, context): |
||||
with self._server_status_lock: |
||||
if request.service not in self._server_status: |
||||
# TODO(atash): once the Python API has a way of setting the server |
||||
# status, bring us into conformance with the health check spec by |
||||
# returning the NOT_FOUND status here. |
||||
raise NotImplementedError() |
||||
else: |
||||
return health_pb2.HealthCheckResponse( |
||||
status=self._server_status[request.service]) |
||||
|
||||
def set(self, service, status): |
||||
"""Sets the status of a service. |
||||
|
||||
Args: |
||||
service: string, the name of the service. |
||||
NOTE, '' must be set. |
||||
status: HealthCheckResponse.status enum value indicating |
||||
the status of the service |
||||
""" |
||||
with self._server_status_lock: |
||||
self._server_status[service] = status |
||||
|
||||
Loading…
Reference in new issue