mirror of https://github.com/grpc/grpc.git
Merge pull request #668 from nathanielmanistaatgoogle/assembly
The framework.assembly package.pull/640/head
commit
04608c4134
16 changed files with 1298 additions and 24 deletions
@ -0,0 +1,30 @@ |
||||
# 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. |
||||
|
||||
|
@ -0,0 +1,305 @@ |
||||
# 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 |
||||
|
||||
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 |
||||
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 |
||||
from grpc.framework.face import utilities as face_utilities |
||||
from grpc.framework.foundation import activated |
||||
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.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(implementations, front, pool): |
||||
behaviors = {} |
||||
stub = face_implementations.stub(front, pool) |
||||
for name, implementation in implementations.iteritems(): |
||||
if implementation.cardinality is cardinality.Cardinality.UNARY_UNARY: |
||||
behaviors[name] = stub.unary_unary_sync_async(name) |
||||
elif implementation.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 implementation.cardinality is cardinality.Cardinality.STREAM_UNARY: |
||||
behaviors[name] = stub.stream_unary_sync_async(name) |
||||
elif implementation.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, implementations, rear_link): |
||||
self._implementations = implementations |
||||
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._implementations, 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: |
||||
raise AttributeError(attr) |
||||
else: |
||||
return behavior |
||||
|
||||
|
||||
def _servicer(implementations, pool): |
||||
inline_value_in_value_out_methods = {} |
||||
inline_value_in_stream_out_methods = {} |
||||
inline_stream_in_value_out_methods = {} |
||||
inline_stream_in_stream_out_methods = {} |
||||
event_value_in_value_out_methods = {} |
||||
event_value_in_stream_out_methods = {} |
||||
event_stream_in_value_out_methods = {} |
||||
event_stream_in_stream_out_methods = {} |
||||
|
||||
for name, implementation in implementations.iteritems(): |
||||
if implementation.cardinality is cardinality.Cardinality.UNARY_UNARY: |
||||
if implementation.style is style.Service.INLINE: |
||||
inline_value_in_value_out_methods[name] = ( |
||||
face_utilities.inline_unary_unary_method(implementation.unary_unary_inline)) |
||||
elif implementation.style is style.Service.EVENT: |
||||
event_value_in_value_out_methods[name] = ( |
||||
face_utilities.event_unary_unary_method(implementation.unary_unary_event)) |
||||
elif implementation.cardinality is cardinality.Cardinality.UNARY_STREAM: |
||||
if implementation.style is style.Service.INLINE: |
||||
inline_value_in_stream_out_methods[name] = ( |
||||
face_utilities.inline_unary_stream_method(implementation.unary_stream_inline)) |
||||
elif implementation.style is style.Service.EVENT: |
||||
event_value_in_stream_out_methods[name] = ( |
||||
face_utilities.event_unary_stream_method(implementation.unary_stream_event)) |
||||
if implementation.cardinality is cardinality.Cardinality.STREAM_UNARY: |
||||
if implementation.style is style.Service.INLINE: |
||||
inline_stream_in_value_out_methods[name] = ( |
||||
face_utilities.inline_stream_unary_method(implementation.stream_unary_inline)) |
||||
elif implementation.style is style.Service.EVENT: |
||||
event_stream_in_value_out_methods[name] = ( |
||||
face_utilities.event_stream_unary_method(implementation.stream_unary_event)) |
||||
elif implementation.cardinality is cardinality.Cardinality.STREAM_STREAM: |
||||
if implementation.style is style.Service.INLINE: |
||||
inline_stream_in_stream_out_methods[name] = ( |
||||
face_utilities.inline_stream_stream_method(implementation.stream_stream_inline)) |
||||
elif implementation.style is style.Service.EVENT: |
||||
event_stream_in_stream_out_methods[name] = ( |
||||
face_utilities.event_stream_stream_method(implementation.stream_stream_event)) |
||||
|
||||
return face_implementations.servicer( |
||||
pool, |
||||
inline_value_in_value_out_methods=inline_value_in_value_out_methods, |
||||
inline_value_in_stream_out_methods=inline_value_in_stream_out_methods, |
||||
inline_stream_in_value_out_methods=inline_stream_in_value_out_methods, |
||||
inline_stream_in_stream_out_methods=inline_stream_in_stream_out_methods, |
||||
event_value_in_value_out_methods=event_value_in_value_out_methods, |
||||
event_value_in_stream_out_methods=event_value_in_stream_out_methods, |
||||
event_stream_in_value_out_methods=event_stream_in_value_out_methods, |
||||
event_stream_in_stream_out_methods=event_stream_in_stream_out_methods) |
||||
|
||||
|
||||
class _ServiceAssembly(activated.Activated): |
||||
|
||||
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 = _servicer(self._implementations, self._pool) |
||||
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 assemble_face_stub(activated_rear_link): |
||||
"""Assembles a face_interfaces.Stub. |
||||
|
||||
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.Stub on which, in context, RPCs can be invoked. |
||||
""" |
||||
return _FaceStub(activated_rear_link) |
||||
|
||||
|
||||
def assemble_dynamic_inline_stub(implementations, 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.UnaryUnarySyncAsync 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 callable with the semantics of |
||||
face_interfaces.Stub.inline_value_in_stream_out, minus the "name" parameter, |
||||
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.StreamUnarySyncAsync 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 callable with the semantics of |
||||
face_interfaces.Stub.inline_stream_in_stream_out, minus the "name" parameter, |
||||
with which to invoke the RPC method. |
||||
|
||||
Args: |
||||
implementations: A dictionary from RPC method name to |
||||
interfaces.MethodImplementation. |
||||
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 stub on which, in context, RPCs can be invoked. |
||||
""" |
||||
return _DynamicInlineStub(implementations, 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 |
||||
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 activated.Activated value encapsulating RPC service. |
||||
""" |
||||
return _ServiceAssembly(implementations, activated_fore_link) |
@ -0,0 +1,284 @@ |
||||
# 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.assembly import utilities |
||||
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.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: utilities.unary_unary_inline(_div), |
||||
DIV_MANY: utilities.stream_stream_inline(_div_many), |
||||
FIB: utilities.unary_stream_inline(_fib), |
||||
SUM: utilities.stream_unary_inline(_sum), |
||||
} |
||||
|
||||
_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: |
||||
sync_async = face_stub.stream_unary_sync_async(SUM) |
||||
response_future = sync_async.async( |
||||
(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( |
||||
_IMPLEMENTATIONS, 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( |
||||
_IMPLEMENTATIONS, 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( |
||||
_IMPLEMENTATIONS, pipe) |
||||
|
||||
with service, dynamic_stub: |
||||
response_future = dynamic_stub.Sum.async( |
||||
(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( |
||||
_IMPLEMENTATIONS, 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() |
@ -0,0 +1,91 @@ |
||||
# 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 |
||||
|
||||
# cardinality, style, and stream are referenced from specification in this |
||||
# module. |
||||
from grpc.framework.common import cardinality # pylint: disable=unused-import |
||||
from grpc.framework.common import style # pylint: disable=unused-import |
||||
from grpc.framework.foundation import stream # pylint: disable=unused-import |
||||
|
||||
|
||||
class MethodImplementation(object): |
||||
"""A sum type that describes an RPC method implementation. |
||||
|
||||
Attributes: |
||||
cardinality: A cardinality.Cardinality value. |
||||
style: A style.Service value. |
||||
unary_unary_inline: The implementation of the RPC method as a callable |
||||
value that takes a request value and a face_interfaces.RpcContext object |
||||
and returns a response value. Only non-None if cardinality is |
||||
cardinality.Cardinality.UNARY_UNARY and style is style.Service.INLINE. |
||||
unary_stream_inline: The implementation of the RPC method as a callable |
||||
value that takes a request value and a face_interfaces.RpcContext object |
||||
and returns an iterator of response values. Only non-None if cardinality |
||||
is cardinality.Cardinality.UNARY_STREAM and style is |
||||
style.Service.INLINE. |
||||
stream_unary_inline: The implementation of the RPC method as a callable |
||||
value that takes an iterator of request values and a |
||||
face_interfaces.RpcContext object and returns a response value. Only |
||||
non-None if cardinality is cardinality.Cardinality.STREAM_UNARY and style |
||||
is style.Service.INLINE. |
||||
stream_stream_inline: The implementation of the RPC method as a callable |
||||
value that takes an iterator of request values and a |
||||
face_interfaces.RpcContext object and returns an iterator of response |
||||
values. Only non-None if cardinality is |
||||
cardinality.Cardinality.STREAM_STREAM and style is style.Service.INLINE. |
||||
unary_unary_event: The implementation of the RPC method as a callable value |
||||
that takes a request value, a response callback to which to pass the |
||||
response value of the RPC, and a face_interfaces.RpcContext. Only |
||||
non-None if cardinality is cardinality.Cardinality.UNARY_UNARY and style |
||||
is style.Service.EVENT. |
||||
unary_stream_event: The implementation of the RPC method as a callable |
||||
value that takes a request value, a stream.Consumer to which to pass the |
||||
the response values of the RPC, and a face_interfaces.RpcContext. Only |
||||
non-None if cardinality is cardinality.Cardinality.UNARY_STREAM and style |
||||
is style.Service.EVENT. |
||||
stream_unary_event: The implementation of the RPC method as a callable |
||||
value that takes a response callback to which to pass the response value |
||||
of the RPC and a face_interfaces.RpcContext and returns a stream.Consumer |
||||
to which the request values of the RPC should be passed. Only non-None if |
||||
cardinality is cardinality.Cardinality.STREAM_UNARY and style is |
||||
style.Service.EVENT. |
||||
stream_stream_event: The implementation of the RPC method as a callable |
||||
value that takes a stream.Consumer to which to pass the response values |
||||
of the RPC and a face_interfaces.RpcContext and returns a stream.Consumer |
||||
to which the request values of the RPC should be passed. Only non-None if |
||||
cardinality is cardinality.Cardinality.STREAM_STREAM and style is |
||||
style.Service.EVENT. |
||||
""" |
||||
__metaclass__ = abc.ABCMeta |
@ -0,0 +1,179 @@ |
||||
# 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. |
||||
|
||||
"""Utilities for assembling RPC framework values.""" |
||||
|
||||
import collections |
||||
|
||||
from grpc.framework.assembly import interfaces |
||||
from grpc.framework.common import cardinality |
||||
from grpc.framework.common import style |
||||
from grpc.framework.face import interfaces as face_interfaces |
||||
from grpc.framework.foundation import stream |
||||
|
||||
|
||||
class _MethodImplementation( |
||||
interfaces.MethodImplementation, |
||||
collections.namedtuple( |
||||
'_MethodImplementation', |
||||
['cardinality', 'style', 'unary_unary_inline', 'unary_stream_inline', |
||||
'stream_unary_inline', 'stream_stream_inline', 'unary_unary_event', |
||||
'unary_stream_event', 'stream_unary_event', 'stream_stream_event',])): |
||||
pass |
||||
|
||||
|
||||
def unary_unary_inline(behavior): |
||||
"""Creates an interfaces.MethodImplementation for the given behavior. |
||||
|
||||
Args: |
||||
behavior: The implementation of a unary-unary RPC method as a callable value |
||||
that takes a request value and a face_interfaces.RpcContext object and |
||||
returns a response value. |
||||
|
||||
Returns: |
||||
An interfaces.MethodImplementation derived from the given behavior. |
||||
""" |
||||
return _MethodImplementation( |
||||
cardinality.Cardinality.UNARY_UNARY, style.Service.INLINE, behavior, |
||||
None, None, None, None, None, None, None) |
||||
|
||||
|
||||
def unary_stream_inline(behavior): |
||||
"""Creates an interfaces.MethodImplementation for the given behavior. |
||||
|
||||
Args: |
||||
behavior: The implementation of a unary-stream RPC method as a callable |
||||
value that takes a request value and a face_interfaces.RpcContext object |
||||
and returns an iterator of response values. |
||||
|
||||
Returns: |
||||
An interfaces.MethodImplementation derived from the given behavior. |
||||
""" |
||||
return _MethodImplementation( |
||||
cardinality.Cardinality.UNARY_STREAM, style.Service.INLINE, None, |
||||
behavior, None, None, None, None, None, None) |
||||
|
||||
|
||||
def stream_unary_inline(behavior): |
||||
"""Creates an interfaces.MethodImplementation for the given behavior. |
||||
|
||||
Args: |
||||
behavior: The implementation of a stream-unary RPC method as a callable |
||||
value that takes an iterator of request values and a |
||||
face_interfaces.RpcContext object and returns a response value. |
||||
|
||||
Returns: |
||||
An interfaces.MethodImplementation derived from the given behavior. |
||||
""" |
||||
return _MethodImplementation( |
||||
cardinality.Cardinality.STREAM_UNARY, style.Service.INLINE, None, None, |
||||
behavior, None, None, None, None, None) |
||||
|
||||
|
||||
def stream_stream_inline(behavior): |
||||
"""Creates an interfaces.MethodImplementation for the given behavior. |
||||
|
||||
Args: |
||||
behavior: The implementation of a stream-stream RPC method as a callable |
||||
value that takes an iterator of request values and a |
||||
face_interfaces.RpcContext object and returns an iterator of response |
||||
values. |
||||
|
||||
Returns: |
||||
An interfaces.MethodImplementation derived from the given behavior. |
||||
""" |
||||
return _MethodImplementation( |
||||
cardinality.Cardinality.STREAM_STREAM, style.Service.INLINE, None, None, |
||||
None, behavior, None, None, None, None) |
||||
|
||||
|
||||
def unary_unary_event(behavior): |
||||
"""Creates an interfaces.MethodImplementation for the given behavior. |
||||
|
||||
Args: |
||||
behavior: The implementation of a unary-unary RPC method as a callable |
||||
value that takes a request value, a response callback to which to pass |
||||
the response value of the RPC, and a face_interfaces.RpcContext. |
||||
|
||||
Returns: |
||||
An interfaces.MethodImplementation derived from the given behavior. |
||||
""" |
||||
return _MethodImplementation( |
||||
cardinality.Cardinality.UNARY_UNARY, style.Service.EVENT, None, None, |
||||
None, None, behavior, None, None, None) |
||||
|
||||
|
||||
def unary_stream_event(behavior): |
||||
"""Creates an interfaces.MethodImplementation for the given behavior. |
||||
|
||||
Args: |
||||
behavior: The implementation of a unary-stream RPC method as a callable |
||||
value that takes a request value, a stream.Consumer to which to pass the |
||||
the response values of the RPC, and a face_interfaces.RpcContext. |
||||
|
||||
Returns: |
||||
An interfaces.MethodImplementation derived from the given behavior. |
||||
""" |
||||
return _MethodImplementation( |
||||
cardinality.Cardinality.UNARY_STREAM, style.Service.EVENT, None, None, |
||||
None, None, None, behavior, None, None) |
||||
|
||||
|
||||
def stream_unary_event(behavior): |
||||
"""Creates an interfaces.MethodImplementation for the given behavior. |
||||
|
||||
Args: |
||||
behavior: The implementation of a stream-unary RPC method as a callable |
||||
value that takes a response callback to which to pass the response value |
||||
of the RPC and a face_interfaces.RpcContext and returns a stream.Consumer |
||||
to which the request values of the RPC should be passed. |
||||
|
||||
Returns: |
||||
An interfaces.MethodImplementation derived from the given behavior. |
||||
""" |
||||
return _MethodImplementation( |
||||
cardinality.Cardinality.STREAM_UNARY, style.Service.EVENT, None, None, |
||||
None, None, None, None, behavior, None) |
||||
|
||||
|
||||
def stream_stream_event(behavior): |
||||
"""Creates an interfaces.MethodImplementation for the given behavior. |
||||
|
||||
Args: |
||||
behavior: The implementation of a stream-stream RPC method as a callable |
||||
value that takes a stream.Consumer to which to pass the response values |
||||
of the RPC and a face_interfaces.RpcContext and returns a stream.Consumer |
||||
to which the request values of the RPC should be passed. |
||||
|
||||
Returns: |
||||
An interfaces.MethodImplementation derived from the given behavior. |
||||
""" |
||||
return _MethodImplementation( |
||||
cardinality.Cardinality.STREAM_STREAM, style.Service.EVENT, None, None, |
||||
None, None, None, None, None, behavior) |
@ -0,0 +1,40 @@ |
||||
# 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. |
||||
|
||||
"""Defines an enum for classifying RPC methods by control flow semantics.""" |
||||
|
||||
import enum |
||||
|
||||
|
||||
@enum.unique |
||||
class Service(enum.Enum): |
||||
"""Describes the control flow style of RPC method implementation.""" |
||||
|
||||
INLINE = 'inline' |
||||
EVENT = 'event' |
@ -0,0 +1,221 @@ |
||||
# 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. |
||||
|
||||
"""Utilities for the face layer of RPC Framework.""" |
||||
|
||||
# stream is referenced from specification in this module. |
||||
from grpc.framework.face import interfaces |
||||
from grpc.framework.foundation import stream # pylint: disable=unused-import |
||||
|
||||
|
||||
class _InlineUnaryUnaryMethod(interfaces.InlineValueInValueOutMethod): |
||||
|
||||
def __init__(self, behavior): |
||||
self._behavior = behavior |
||||
|
||||
def service(self, request, context): |
||||
return self._behavior(request, context) |
||||
|
||||
|
||||
class _InlineUnaryStreamMethod(interfaces.InlineValueInStreamOutMethod): |
||||
|
||||
def __init__(self, behavior): |
||||
self._behavior = behavior |
||||
|
||||
def service(self, request, context): |
||||
return self._behavior(request, context) |
||||
|
||||
|
||||
class _InlineStreamUnaryMethod(interfaces.InlineStreamInValueOutMethod): |
||||
|
||||
def __init__(self, behavior): |
||||
self._behavior = behavior |
||||
|
||||
def service(self, request_iterator, context): |
||||
return self._behavior(request_iterator, context) |
||||
|
||||
|
||||
class _InlineStreamStreamMethod(interfaces.InlineStreamInStreamOutMethod): |
||||
|
||||
def __init__(self, behavior): |
||||
self._behavior = behavior |
||||
|
||||
def service(self, request_iterator, context): |
||||
return self._behavior(request_iterator, context) |
||||
|
||||
|
||||
class _EventUnaryUnaryMethod(interfaces.EventValueInValueOutMethod): |
||||
|
||||
def __init__(self, behavior): |
||||
self._behavior = behavior |
||||
|
||||
def service(self, request, response_callback, context): |
||||
return self._behavior(request, response_callback, context) |
||||
|
||||
|
||||
class _EventUnaryStreamMethod(interfaces.EventValueInStreamOutMethod): |
||||
|
||||
def __init__(self, behavior): |
||||
self._behavior = behavior |
||||
|
||||
def service(self, request, response_consumer, context): |
||||
return self._behavior(request, response_consumer, context) |
||||
|
||||
|
||||
class _EventStreamUnaryMethod(interfaces.EventStreamInValueOutMethod): |
||||
|
||||
def __init__(self, behavior): |
||||
self._behavior = behavior |
||||
|
||||
def service(self, response_callback, context): |
||||
return self._behavior(response_callback, context) |
||||
|
||||
|
||||
class _EventStreamStreamMethod(interfaces.EventStreamInStreamOutMethod): |
||||
|
||||
def __init__(self, behavior): |
||||
self._behavior = behavior |
||||
|
||||
def service(self, response_consumer, context): |
||||
return self._behavior(response_consumer, context) |
||||
|
||||
|
||||
def inline_unary_unary_method(behavior): |
||||
"""Creates an interfaces.InlineValueInValueOutMethod from a behavior. |
||||
|
||||
Args: |
||||
behavior: The implementation of a unary-unary RPC method as a callable |
||||
value that takes a request value and an interfaces.RpcContext object and |
||||
returns a response value. |
||||
|
||||
Returns: |
||||
An interfaces.InlineValueInValueOutMethod derived from the given behavior. |
||||
""" |
||||
return _InlineUnaryUnaryMethod(behavior) |
||||
|
||||
|
||||
def inline_unary_stream_method(behavior): |
||||
"""Creates an interfaces.InlineValueInStreamOutMethod from a behavior. |
||||
|
||||
Args: |
||||
behavior: The implementation of a unary-stream RPC method as a callable |
||||
value that takes a request value and an interfaces.RpcContext object and |
||||
returns an iterator of response values. |
||||
|
||||
Returns: |
||||
An interfaces.InlineValueInStreamOutMethod derived from the given behavior. |
||||
""" |
||||
return _InlineUnaryStreamMethod(behavior) |
||||
|
||||
|
||||
def inline_stream_unary_method(behavior): |
||||
"""Creates an interfaces.InlineStreamInValueOutMethod from a behavior. |
||||
|
||||
Args: |
||||
behavior: The implementation of a stream-unary RPC method as a callable |
||||
value that takes an iterator of request values and an |
||||
interfaces.RpcContext object and returns a response value. |
||||
|
||||
Returns: |
||||
An interfaces.InlineStreamInValueOutMethod derived from the given behavior. |
||||
""" |
||||
return _InlineStreamUnaryMethod(behavior) |
||||
|
||||
|
||||
def inline_stream_stream_method(behavior): |
||||
"""Creates an interfaces.InlineStreamInStreamOutMethod from a behavior. |
||||
|
||||
Args: |
||||
behavior: The implementation of a stream-stream RPC method as a callable |
||||
value that takes an iterator of request values and an |
||||
interfaces.RpcContext object and returns an iterator of response values. |
||||
|
||||
Returns: |
||||
An interfaces.InlineStreamInStreamOutMethod derived from the given |
||||
behavior. |
||||
""" |
||||
return _InlineStreamStreamMethod(behavior) |
||||
|
||||
|
||||
def event_unary_unary_method(behavior): |
||||
"""Creates an interfaces.EventValueInValueOutMethod from a behavior. |
||||
|
||||
Args: |
||||
behavior: The implementation of a unary-unary RPC method as a callable |
||||
value that takes a request value, a response callback to which to pass |
||||
the response value of the RPC, and an interfaces.RpcContext. |
||||
|
||||
Returns: |
||||
An interfaces.EventValueInValueOutMethod derived from the given behavior. |
||||
""" |
||||
return _EventUnaryUnaryMethod(behavior) |
||||
|
||||
|
||||
def event_unary_stream_method(behavior): |
||||
"""Creates an interfaces.EventValueInStreamOutMethod from a behavior. |
||||
|
||||
Args: |
||||
behavior: The implementation of a unary-stream RPC method as a callable |
||||
value that takes a request value, a stream.Consumer to which to pass the |
||||
response values of the RPC, and an interfaces.RpcContext. |
||||
|
||||
Returns: |
||||
An interfaces.EventValueInStreamOutMethod derived from the given behavior. |
||||
""" |
||||
return _EventUnaryStreamMethod(behavior) |
||||
|
||||
|
||||
def event_stream_unary_method(behavior): |
||||
"""Creates an interfaces.EventStreamInValueOutMethod from a behavior. |
||||
|
||||
Args: |
||||
behavior: The implementation of a stream-unary RPC method as a callable |
||||
value that takes a response callback to which to pass the response value |
||||
of the RPC and an interfaces.RpcContext and returns a stream.Consumer to |
||||
which the request values of the RPC should be passed. |
||||
|
||||
Returns: |
||||
An interfaces.EventStreamInValueOutMethod derived from the given behavior. |
||||
""" |
||||
return _EventStreamUnaryMethod(behavior) |
||||
|
||||
|
||||
def event_stream_stream_method(behavior): |
||||
"""Creates an interfaces.EventStreamInStreamOutMethod from a behavior. |
||||
|
||||
Args: |
||||
behavior: The implementation of a stream-stream RPC method as a callable |
||||
value that takes a stream.Consumer to which to pass the response values |
||||
of the RPC and an interfaces.RpcContext and returns a stream.Consumer to |
||||
which the request values of the RPC should be passed. |
||||
|
||||
Returns: |
||||
An interfaces.EventStreamInStreamOutMethod derived from the given behavior. |
||||
""" |
||||
return _EventStreamStreamMethod(behavior) |
@ -0,0 +1,65 @@ |
||||
# 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. |
||||
|
||||
"""Interfaces related to streams of values or objects.""" |
||||
|
||||
import abc |
||||
|
||||
|
||||
class Activated(object): |
||||
"""Interface for objects that may be started and stopped. |
||||
|
||||
Values implementing this type must also implement the context manager |
||||
protocol. |
||||
""" |
||||
__metaclass__ = abc.ABCMeta |
||||
|
||||
@abc.abstractmethod |
||||
def __enter__(self): |
||||
"""See the context manager protocol for specification.""" |
||||
raise NotImplementedError() |
||||
|
||||
@abc.abstractmethod |
||||
def __exit__(self, exc_type, exc_val, exc_tb): |
||||
"""See the context manager protocol for specification.""" |
||||
raise NotImplementedError() |
||||
|
||||
@abc.abstractmethod |
||||
def start(self): |
||||
"""Activates this object. |
||||
|
||||
Returns: |
||||
A value equal to the value returned by this object's __enter__ method. |
||||
""" |
||||
raise NotImplementedError() |
||||
|
||||
@abc.abstractmethod |
||||
def stop(self): |
||||
"""Deactivates this object.""" |
||||
raise NotImplementedError() |
Loading…
Reference in new issue