mirror of https://github.com/grpc/grpc.git
commit
bc5d466c97
17 changed files with 292 additions and 958 deletions
@ -1,30 +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. |
||||
|
||||
|
@ -1,264 +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. |
||||
|
||||
"""Implementations for assembling RPC framework values.""" |
||||
|
||||
import threading |
||||
|
||||
# tickets_interfaces, face_interfaces, and activated are referenced from |
||||
# specification in this module. |
||||
from grpc.framework.assembly import interfaces |
||||
from grpc.framework.base import util as base_utilities |
||||
from grpc.framework.base.packets import implementations as tickets_implementations |
||||
from grpc.framework.base.packets import interfaces as tickets_interfaces # pylint: disable=unused-import |
||||
from grpc.framework.common import cardinality |
||||
from grpc.framework.common import style |
||||
from grpc.framework.face import implementations as face_implementations |
||||
from grpc.framework.face import interfaces as face_interfaces # pylint: disable=unused-import |
||||
from grpc.framework.face import utilities as face_utilities |
||||
from grpc.framework.foundation import activated # pylint: disable=unused-import |
||||
from grpc.framework.foundation import logging_pool |
||||
|
||||
_ONE_DAY_IN_SECONDS = 60 * 60 * 24 |
||||
_THREAD_POOL_SIZE = 100 |
||||
|
||||
|
||||
class _FaceStub(object): |
||||
|
||||
def __init__(self, rear_link): |
||||
self._rear_link = rear_link |
||||
self._lock = threading.Lock() |
||||
self._pool = None |
||||
self._front = None |
||||
self._under_stub = None |
||||
|
||||
def __enter__(self): |
||||
with self._lock: |
||||
self._pool = logging_pool.pool(_THREAD_POOL_SIZE) |
||||
self._front = tickets_implementations.front( |
||||
self._pool, self._pool, self._pool) |
||||
self._rear_link.start() |
||||
self._rear_link.join_fore_link(self._front) |
||||
self._front.join_rear_link(self._rear_link) |
||||
self._under_stub = face_implementations.generic_stub(self._front, self._pool) |
||||
|
||||
def __exit__(self, exc_type, exc_val, exc_tb): |
||||
with self._lock: |
||||
self._under_stub = None |
||||
self._rear_link.stop() |
||||
base_utilities.wait_for_idle(self._front) |
||||
self._front = None |
||||
self._pool.shutdown(wait=True) |
||||
self._pool = None |
||||
return False |
||||
|
||||
def __getattr__(self, attr): |
||||
with self._lock: |
||||
if self._under_stub is None: |
||||
raise ValueError('Called out of context!') |
||||
else: |
||||
return getattr(self._under_stub, attr) |
||||
|
||||
|
||||
def _behaviors(method_cardinalities, front, pool): |
||||
behaviors = {} |
||||
stub = face_implementations.generic_stub(front, pool) |
||||
for name, method_cardinality in method_cardinalities.iteritems(): |
||||
if method_cardinality is cardinality.Cardinality.UNARY_UNARY: |
||||
behaviors[name] = stub.unary_unary_multi_callable(name) |
||||
elif method_cardinality is cardinality.Cardinality.UNARY_STREAM: |
||||
behaviors[name] = lambda request, context, bound_name=name: ( |
||||
stub.inline_value_in_stream_out(bound_name, request, context)) |
||||
elif method_cardinality is cardinality.Cardinality.STREAM_UNARY: |
||||
behaviors[name] = stub.stream_unary_multi_callable(name) |
||||
elif method_cardinality is cardinality.Cardinality.STREAM_STREAM: |
||||
behaviors[name] = lambda request_iterator, context, bound_name=name: ( |
||||
stub.inline_stream_in_stream_out( |
||||
bound_name, request_iterator, context)) |
||||
return behaviors |
||||
|
||||
|
||||
class _DynamicInlineStub(object): |
||||
|
||||
def __init__(self, cardinalities, rear_link): |
||||
self._cardinalities = cardinalities |
||||
self._rear_link = rear_link |
||||
self._lock = threading.Lock() |
||||
self._pool = None |
||||
self._front = None |
||||
self._behaviors = None |
||||
|
||||
def __enter__(self): |
||||
with self._lock: |
||||
self._pool = logging_pool.pool(_THREAD_POOL_SIZE) |
||||
self._front = tickets_implementations.front( |
||||
self._pool, self._pool, self._pool) |
||||
self._rear_link.start() |
||||
self._rear_link.join_fore_link(self._front) |
||||
self._front.join_rear_link(self._rear_link) |
||||
self._behaviors = _behaviors( |
||||
self._cardinalities, self._front, self._pool) |
||||
return self |
||||
|
||||
def __exit__(self, exc_type, exc_val, exc_tb): |
||||
with self._lock: |
||||
self._behaviors = None |
||||
self._rear_link.stop() |
||||
base_utilities.wait_for_idle(self._front) |
||||
self._front = None |
||||
self._pool.shutdown(wait=True) |
||||
self._pool = None |
||||
return False |
||||
|
||||
def __getattr__(self, attr): |
||||
with self._lock: |
||||
behavior = self._behaviors.get(attr) |
||||
if behavior is None: |
||||
for name, behavior in self._behaviors.iteritems(): |
||||
last_slash_index = name.rfind('/') |
||||
if 0 <= last_slash_index and name[last_slash_index + 1:] == attr: |
||||
return behavior |
||||
else: |
||||
raise AttributeError( |
||||
'_DynamicInlineStub instance has no attribute "%s"!' % attr) |
||||
else: |
||||
return behavior |
||||
|
||||
|
||||
class _ServiceAssembly(interfaces.Server): |
||||
|
||||
def __init__(self, implementations, fore_link): |
||||
self._implementations = implementations |
||||
self._fore_link = fore_link |
||||
self._lock = threading.Lock() |
||||
self._pool = None |
||||
self._back = None |
||||
|
||||
def _start(self): |
||||
with self._lock: |
||||
self._pool = logging_pool.pool(_THREAD_POOL_SIZE) |
||||
servicer = face_implementations.servicer( |
||||
self._pool, self._implementations, None) |
||||
self._back = tickets_implementations.back( |
||||
servicer, self._pool, self._pool, self._pool, _ONE_DAY_IN_SECONDS, |
||||
_ONE_DAY_IN_SECONDS) |
||||
self._fore_link.start() |
||||
self._fore_link.join_rear_link(self._back) |
||||
self._back.join_fore_link(self._fore_link) |
||||
|
||||
def _stop(self): |
||||
with self._lock: |
||||
self._fore_link.stop() |
||||
base_utilities.wait_for_idle(self._back) |
||||
self._back = None |
||||
self._pool.shutdown(wait=True) |
||||
self._pool = None |
||||
|
||||
def __enter__(self): |
||||
self._start() |
||||
return self |
||||
|
||||
def __exit__(self, exc_type, exc_val, exc_tb): |
||||
self._stop() |
||||
return False |
||||
|
||||
def start(self): |
||||
return self._start() |
||||
|
||||
def stop(self): |
||||
self._stop() |
||||
|
||||
def port(self): |
||||
with self._lock: |
||||
return self._fore_link.port() |
||||
|
||||
|
||||
def assemble_face_stub(activated_rear_link): |
||||
"""Assembles a face_interfaces.GenericStub. |
||||
|
||||
The returned object is a context manager and may only be used in context to |
||||
invoke RPCs. |
||||
|
||||
Args: |
||||
activated_rear_link: An object that is both a tickets_interfaces.RearLink |
||||
and an activated.Activated. The object should be in the inactive state |
||||
when passed to this method. |
||||
|
||||
Returns: |
||||
A face_interfaces.GenericStub on which, in context, RPCs can be invoked. |
||||
""" |
||||
return _FaceStub(activated_rear_link) |
||||
|
||||
|
||||
def assemble_dynamic_inline_stub(cardinalities, activated_rear_link): |
||||
"""Assembles a stub with method names for attributes. |
||||
|
||||
The returned object is a context manager and may only be used in context to |
||||
invoke RPCs. |
||||
|
||||
The returned object, when used in context, will respond to attribute access |
||||
as follows: if the requested attribute is the name of a unary-unary RPC |
||||
method, the value of the attribute will be a |
||||
face_interfaces.UnaryUnaryMultiCallable with which to invoke the RPC method. |
||||
If the requested attribute is the name of a unary-stream RPC method, the |
||||
value of the attribute will be a face_interfaces.UnaryStreamMultiCallable |
||||
with which to invoke the RPC method. If the requested attribute is the name |
||||
of a stream-unary RPC method, the value of the attribute will be a |
||||
face_interfaces.StreamUnaryMultiCallable with which to invoke the RPC method. |
||||
If the requested attribute is the name of a stream-stream RPC method, the |
||||
value of the attribute will be a face_interfaces.StreamStreamMultiCallable |
||||
with which to invoke the RPC method. |
||||
|
||||
Args: |
||||
cardinalities: A dictionary from RPC method name to cardinality.Cardinality |
||||
value identifying the cardinality of the named RPC method. |
||||
activated_rear_link: An object that is both a tickets_interfaces.RearLink |
||||
and an activated.Activated. The object should be in the inactive state |
||||
when passed to this method. |
||||
|
||||
Returns: |
||||
A face_interfaces.DynamicStub on which, in context, RPCs can be invoked. |
||||
""" |
||||
return _DynamicInlineStub(cardinalities, activated_rear_link) |
||||
|
||||
|
||||
def assemble_service(implementations, activated_fore_link): |
||||
"""Assembles the service-side of the RPC Framework stack. |
||||
|
||||
Args: |
||||
implementations: A dictionary from RPC method name to |
||||
face_interfaces.MethodImplementation. |
||||
activated_fore_link: An object that is both a tickets_interfaces.ForeLink |
||||
and an activated.Activated. The object should be in the inactive state |
||||
when passed to this method. |
||||
|
||||
Returns: |
||||
An interfaces.Server encapsulating RPC service. |
||||
""" |
||||
return _ServiceAssembly(implementations, activated_fore_link) |
@ -1,288 +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. |
||||
|
||||
# TODO(nathaniel): Expand this test coverage. |
||||
|
||||
"""Test of the GRPC-backed ForeLink and RearLink.""" |
||||
|
||||
import threading |
||||
import unittest |
||||
|
||||
from grpc.framework.assembly import implementations |
||||
from grpc.framework.base import interfaces |
||||
from grpc.framework.base.packets import packets as tickets |
||||
from grpc.framework.base.packets import interfaces as tickets_interfaces |
||||
from grpc.framework.base.packets import null |
||||
from grpc.framework.face import utilities as face_utilities |
||||
from grpc.framework.foundation import logging_pool |
||||
from grpc._junkdrawer import math_pb2 |
||||
|
||||
DIV = 'Div' |
||||
DIV_MANY = 'DivMany' |
||||
FIB = 'Fib' |
||||
SUM = 'Sum' |
||||
|
||||
def _fibbonacci(limit): |
||||
left, right = 0, 1 |
||||
for _ in xrange(limit): |
||||
yield left |
||||
left, right = right, left + right |
||||
|
||||
|
||||
def _div(request, unused_context): |
||||
return math_pb2.DivReply( |
||||
quotient=request.dividend / request.divisor, |
||||
remainder=request.dividend % request.divisor) |
||||
|
||||
|
||||
def _div_many(request_iterator, unused_context): |
||||
for request in request_iterator: |
||||
yield math_pb2.DivReply( |
||||
quotient=request.dividend / request.divisor, |
||||
remainder=request.dividend % request.divisor) |
||||
|
||||
|
||||
def _fib(request, unused_context): |
||||
for number in _fibbonacci(request.limit): |
||||
yield math_pb2.Num(num=number) |
||||
|
||||
|
||||
def _sum(request_iterator, unused_context): |
||||
accumulation = 0 |
||||
for request in request_iterator: |
||||
accumulation += request.num |
||||
return math_pb2.Num(num=accumulation) |
||||
|
||||
|
||||
_IMPLEMENTATIONS = { |
||||
DIV: face_utilities.unary_unary_inline(_div), |
||||
DIV_MANY: face_utilities.stream_stream_inline(_div_many), |
||||
FIB: face_utilities.unary_stream_inline(_fib), |
||||
SUM: face_utilities.stream_unary_inline(_sum), |
||||
} |
||||
|
||||
_CARDINALITIES = { |
||||
name: implementation.cardinality |
||||
for name, implementation in _IMPLEMENTATIONS.iteritems()} |
||||
|
||||
_TIMEOUT = 10 |
||||
|
||||
|
||||
class PipeLink(tickets_interfaces.ForeLink, tickets_interfaces.RearLink): |
||||
|
||||
def __init__(self): |
||||
self._fore_lock = threading.Lock() |
||||
self._fore_link = null.NULL_FORE_LINK |
||||
self._rear_lock = threading.Lock() |
||||
self._rear_link = null.NULL_REAR_LINK |
||||
|
||||
def __enter__(self): |
||||
return self |
||||
|
||||
def __exit__(self, exc_type, exc_val, exc_tb): |
||||
return False |
||||
|
||||
def start(self): |
||||
pass |
||||
|
||||
def stop(self): |
||||
pass |
||||
|
||||
def accept_back_to_front_ticket(self, ticket): |
||||
with self._fore_lock: |
||||
self._fore_link.accept_back_to_front_ticket(ticket) |
||||
|
||||
def join_rear_link(self, rear_link): |
||||
with self._rear_lock: |
||||
self._rear_link = null.NULL_REAR_LINK if rear_link is None else rear_link |
||||
|
||||
def accept_front_to_back_ticket(self, ticket): |
||||
with self._rear_lock: |
||||
self._rear_link.accept_front_to_back_ticket(ticket) |
||||
|
||||
def join_fore_link(self, fore_link): |
||||
with self._fore_lock: |
||||
self._fore_link = null.NULL_FORE_LINK if fore_link is None else fore_link |
||||
|
||||
|
||||
class FaceStubTest(unittest.TestCase): |
||||
|
||||
def testUnaryUnary(self): |
||||
divisor = 7 |
||||
dividend = 13 |
||||
expected_quotient = 1 |
||||
expected_remainder = 6 |
||||
pipe = PipeLink() |
||||
service = implementations.assemble_service(_IMPLEMENTATIONS, pipe) |
||||
face_stub = implementations.assemble_face_stub(pipe) |
||||
|
||||
service.start() |
||||
try: |
||||
with face_stub: |
||||
response = face_stub.blocking_value_in_value_out( |
||||
DIV, math_pb2.DivArgs(divisor=divisor, dividend=dividend), |
||||
_TIMEOUT) |
||||
self.assertEqual(expected_quotient, response.quotient) |
||||
self.assertEqual(expected_remainder, response.remainder) |
||||
finally: |
||||
service.stop() |
||||
|
||||
def testUnaryStream(self): |
||||
stream_length = 29 |
||||
pipe = PipeLink() |
||||
service = implementations.assemble_service(_IMPLEMENTATIONS, pipe) |
||||
face_stub = implementations.assemble_face_stub(pipe) |
||||
|
||||
with service, face_stub: |
||||
responses = list( |
||||
face_stub.inline_value_in_stream_out( |
||||
FIB, math_pb2.FibArgs(limit=stream_length), _TIMEOUT)) |
||||
numbers = [response.num for response in responses] |
||||
for early, middle, later in zip(numbers, numbers[1:], numbers[2:]): |
||||
self.assertEqual(early + middle, later) |
||||
|
||||
def testStreamUnary(self): |
||||
stream_length = 13 |
||||
pipe = PipeLink() |
||||
service = implementations.assemble_service(_IMPLEMENTATIONS, pipe) |
||||
face_stub = implementations.assemble_face_stub(pipe) |
||||
|
||||
with service, face_stub: |
||||
multi_callable = face_stub.stream_unary_multi_callable(SUM) |
||||
response_future = multi_callable.future( |
||||
(math_pb2.Num(num=index) for index in range(stream_length)), |
||||
_TIMEOUT) |
||||
self.assertEqual( |
||||
(stream_length * (stream_length - 1)) / 2, |
||||
response_future.result().num) |
||||
|
||||
def testStreamStream(self): |
||||
stream_length = 17 |
||||
divisor_offset = 7 |
||||
dividend_offset = 17 |
||||
pipe = PipeLink() |
||||
service = implementations.assemble_service(_IMPLEMENTATIONS, pipe) |
||||
face_stub = implementations.assemble_face_stub(pipe) |
||||
|
||||
with service, face_stub: |
||||
response_iterator = face_stub.inline_stream_in_stream_out( |
||||
DIV_MANY, |
||||
(math_pb2.DivArgs( |
||||
divisor=divisor_offset + index, |
||||
dividend=dividend_offset + index) |
||||
for index in range(stream_length)), |
||||
_TIMEOUT) |
||||
for index, response in enumerate(response_iterator): |
||||
self.assertEqual( |
||||
(dividend_offset + index) / (divisor_offset + index), |
||||
response.quotient) |
||||
self.assertEqual( |
||||
(dividend_offset + index) % (divisor_offset + index), |
||||
response.remainder) |
||||
self.assertEqual(stream_length, index + 1) |
||||
|
||||
|
||||
class DynamicInlineStubTest(unittest.TestCase): |
||||
|
||||
def testUnaryUnary(self): |
||||
divisor = 59 |
||||
dividend = 973 |
||||
expected_quotient = dividend / divisor |
||||
expected_remainder = dividend % divisor |
||||
pipe = PipeLink() |
||||
service = implementations.assemble_service(_IMPLEMENTATIONS, pipe) |
||||
dynamic_stub = implementations.assemble_dynamic_inline_stub( |
||||
_CARDINALITIES, pipe) |
||||
|
||||
service.start() |
||||
with dynamic_stub: |
||||
response = dynamic_stub.Div( |
||||
math_pb2.DivArgs(divisor=divisor, dividend=dividend), _TIMEOUT) |
||||
self.assertEqual(expected_quotient, response.quotient) |
||||
self.assertEqual(expected_remainder, response.remainder) |
||||
service.stop() |
||||
|
||||
def testUnaryStream(self): |
||||
stream_length = 43 |
||||
pipe = PipeLink() |
||||
service = implementations.assemble_service(_IMPLEMENTATIONS, pipe) |
||||
dynamic_stub = implementations.assemble_dynamic_inline_stub( |
||||
_CARDINALITIES, pipe) |
||||
|
||||
with service, dynamic_stub: |
||||
response_iterator = dynamic_stub.Fib( |
||||
math_pb2.FibArgs(limit=stream_length), _TIMEOUT) |
||||
numbers = tuple(response.num for response in response_iterator) |
||||
for early, middle, later in zip(numbers, numbers[:1], numbers[:2]): |
||||
self.assertEqual(early + middle, later) |
||||
self.assertEqual(stream_length, len(numbers)) |
||||
|
||||
def testStreamUnary(self): |
||||
stream_length = 127 |
||||
pipe = PipeLink() |
||||
service = implementations.assemble_service(_IMPLEMENTATIONS, pipe) |
||||
dynamic_stub = implementations.assemble_dynamic_inline_stub( |
||||
_CARDINALITIES, pipe) |
||||
|
||||
with service, dynamic_stub: |
||||
response_future = dynamic_stub.Sum.future( |
||||
(math_pb2.Num(num=index) for index in range(stream_length)), |
||||
_TIMEOUT) |
||||
self.assertEqual( |
||||
(stream_length * (stream_length - 1)) / 2, |
||||
response_future.result().num) |
||||
|
||||
def testStreamStream(self): |
||||
stream_length = 179 |
||||
divisor_offset = 71 |
||||
dividend_offset = 1763 |
||||
pipe = PipeLink() |
||||
service = implementations.assemble_service(_IMPLEMENTATIONS, pipe) |
||||
dynamic_stub = implementations.assemble_dynamic_inline_stub( |
||||
_CARDINALITIES, pipe) |
||||
|
||||
with service, dynamic_stub: |
||||
response_iterator = dynamic_stub.DivMany( |
||||
(math_pb2.DivArgs( |
||||
divisor=divisor_offset + index, |
||||
dividend=dividend_offset + index) |
||||
for index in range(stream_length)), |
||||
_TIMEOUT) |
||||
for index, response in enumerate(response_iterator): |
||||
self.assertEqual( |
||||
(dividend_offset + index) / (divisor_offset + index), |
||||
response.quotient) |
||||
self.assertEqual( |
||||
(dividend_offset + index) % (divisor_offset + index), |
||||
response.remainder) |
||||
self.assertEqual(stream_length, index + 1) |
||||
|
||||
|
||||
if __name__ == '__main__': |
||||
unittest.main() |
@ -1,58 +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. |
||||
|
||||
# TODO(nathaniel): The assembly layer only exists to smooth out wrinkles in |
||||
# the face layer. The two should be squashed together as soon as manageable. |
||||
"""Interfaces for assembling RPC Framework values.""" |
||||
|
||||
import abc |
||||
|
||||
from grpc.framework.foundation import activated |
||||
|
||||
|
||||
class Server(activated.Activated): |
||||
"""The server interface. |
||||
|
||||
Aside from being able to be activated and deactivated, objects of this type |
||||
are able to report the port on which they are servicing RPCs. |
||||
""" |
||||
__metaclass__ = abc.ABCMeta |
||||
|
||||
# TODO(issue 726): This is an abstraction violation; not every Server is |
||||
# necessarily serving over a network at all. |
||||
@abc.abstractmethod |
||||
def port(self): |
||||
"""Identifies the port on which this Server is servicing RPCs. |
||||
|
||||
This method may only be called while the server is active. |
||||
|
||||
Returns: |
||||
The number of the port on which this Server is servicing RPCs. |
||||
""" |
||||
raise NotImplementedError() |
Loading…
Reference in new issue