_channel.html 71 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593
  1. <!DOCTYPE html>
  2. <html lang="en">
  3. <head>
  4. <meta charset="utf-8" />
  5. <meta name="viewport" content="width=device-width, initial-scale=1.0">
  6. <title>grpc.experimental.aio._channel &#8212; gRPC Python 1.31.0 documentation</title>
  7. <link rel="stylesheet" href="../../../../_static/alabaster.css" type="text/css" />
  8. <link rel="stylesheet" href="../../../../_static/pygments.css" type="text/css" />
  9. <script id="documentation_options" data-url_root="../../../../" src="../../../../_static/documentation_options.js"></script>
  10. <script src="../../../../_static/jquery.js"></script>
  11. <script src="../../../../_static/underscore.js"></script>
  12. <script src="../../../../_static/doctools.js"></script>
  13. <script src="../../../../_static/language_data.js"></script>
  14. <link rel="index" title="Index" href="../../../../genindex.html" />
  15. <link rel="search" title="Search" href="../../../../search.html" />
  16. <link rel="stylesheet" href="../../../../_static/custom.css" type="text/css" />
  17. <meta name="viewport" content="width=device-width, initial-scale=0.9, maximum-scale=0.9" />
  18. </head><body>
  19. <div class="document">
  20. <div class="sphinxsidebar" role="navigation" aria-label="main navigation">
  21. <div class="sphinxsidebarwrapper">
  22. <h1 class="logo"><a href="../../../../index.html">gRPC Python</a></h1>
  23. <p class="blurb">1.31.0</p>
  24. <h3>Navigation</h3>
  25. <p class="caption"><span class="caption-text">Contents:</span></p>
  26. <ul>
  27. <li class="toctree-l1"><a class="reference internal" href="../../../../grpc.html">gRPC</a></li>
  28. <li class="toctree-l1"><a class="reference internal" href="../../../../grpc_asyncio.html">gRPC AsyncIO API</a></li>
  29. <li class="toctree-l1"><a class="reference internal" href="../../../../grpc_channelz.html">gRPC Channelz</a></li>
  30. <li class="toctree-l1"><a class="reference internal" href="../../../../grpc_health_checking.html">gRPC Health Checking</a></li>
  31. <li class="toctree-l1"><a class="reference internal" href="../../../../grpc_reflection.html">gRPC Reflection</a></li>
  32. <li class="toctree-l1"><a class="reference internal" href="../../../../grpc_status.html">gRPC Status</a></li>
  33. <li class="toctree-l1"><a class="reference internal" href="../../../../grpc_testing.html">gRPC Testing</a></li>
  34. <li class="toctree-l1"><a class="reference internal" href="../../../../glossary.html">Glossary</a></li>
  35. </ul>
  36. <div class="relations">
  37. <h3>Related Topics</h3>
  38. <ul>
  39. <li><a href="../../../../index.html">Documentation overview</a><ul>
  40. <li><a href="../../../index.html">Module code</a><ul>
  41. <li><a href="../../../grpc.html">grpc</a><ul>
  42. </ul></li>
  43. </ul></li>
  44. </ul></li>
  45. </ul>
  46. </div>
  47. <div id="searchbox" style="display: none" role="search">
  48. <h3 id="searchlabel">Quick search</h3>
  49. <div class="searchformwrapper">
  50. <form class="search" action="../../../../search.html" method="get">
  51. <input type="text" name="q" aria-labelledby="searchlabel" />
  52. <input type="submit" value="Go" />
  53. </form>
  54. </div>
  55. </div>
  56. <script>$('#searchbox').show(0);</script>
  57. </div>
  58. </div>
  59. <div class="documentwrapper">
  60. <div class="bodywrapper">
  61. <div class="body" role="main">
  62. <h1>Source code for grpc.experimental.aio._channel</h1><div class="highlight"><pre>
  63. <span></span><span class="c1"># Copyright 2019 gRPC authors.</span>
  64. <span class="c1">#</span>
  65. <span class="c1"># Licensed under the Apache License, Version 2.0 (the &quot;License&quot;);</span>
  66. <span class="c1"># you may not use this file except in compliance with the License.</span>
  67. <span class="c1"># You may obtain a copy of the License at</span>
  68. <span class="c1">#</span>
  69. <span class="c1"># http://www.apache.org/licenses/LICENSE-2.0</span>
  70. <span class="c1">#</span>
  71. <span class="c1"># Unless required by applicable law or agreed to in writing, software</span>
  72. <span class="c1"># distributed under the License is distributed on an &quot;AS IS&quot; BASIS,</span>
  73. <span class="c1"># WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.</span>
  74. <span class="c1"># See the License for the specific language governing permissions and</span>
  75. <span class="c1"># limitations under the License.</span>
  76. <span class="sd">&quot;&quot;&quot;Invocation-side implementation of gRPC Asyncio Python.&quot;&quot;&quot;</span>
  77. <span class="kn">import</span> <span class="nn">asyncio</span>
  78. <span class="kn">import</span> <span class="nn">sys</span>
  79. <span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span class="n">Any</span><span class="p">,</span> <span class="n">Iterable</span><span class="p">,</span> <span class="n">Optional</span><span class="p">,</span> <span class="n">Sequence</span><span class="p">,</span> <span class="n">List</span>
  80. <span class="kn">import</span> <span class="nn">grpc</span>
  81. <span class="kn">from</span> <span class="nn">grpc</span> <span class="kn">import</span> <span class="n">_common</span><span class="p">,</span> <span class="n">_compression</span><span class="p">,</span> <span class="n">_grpcio_metadata</span>
  82. <span class="kn">from</span> <span class="nn">grpc._cython</span> <span class="kn">import</span> <span class="n">cygrpc</span>
  83. <span class="kn">from</span> <span class="nn">.</span> <span class="kn">import</span> <span class="n">_base_call</span><span class="p">,</span> <span class="n">_base_channel</span>
  84. <span class="kn">from</span> <span class="nn">._call</span> <span class="kn">import</span> <span class="p">(</span><span class="n">StreamStreamCall</span><span class="p">,</span> <span class="n">StreamUnaryCall</span><span class="p">,</span> <span class="n">UnaryStreamCall</span><span class="p">,</span>
  85. <span class="n">UnaryUnaryCall</span><span class="p">)</span>
  86. <span class="kn">from</span> <span class="nn">._interceptor</span> <span class="kn">import</span> <span class="p">(</span>
  87. <span class="n">InterceptedUnaryUnaryCall</span><span class="p">,</span> <span class="n">InterceptedUnaryStreamCall</span><span class="p">,</span>
  88. <span class="n">InterceptedStreamUnaryCall</span><span class="p">,</span> <span class="n">InterceptedStreamStreamCall</span><span class="p">,</span> <span class="n">ClientInterceptor</span><span class="p">,</span>
  89. <span class="n">UnaryUnaryClientInterceptor</span><span class="p">,</span> <span class="n">UnaryStreamClientInterceptor</span><span class="p">,</span>
  90. <span class="n">StreamUnaryClientInterceptor</span><span class="p">,</span> <span class="n">StreamStreamClientInterceptor</span><span class="p">)</span>
  91. <span class="kn">from</span> <span class="nn">._metadata</span> <span class="kn">import</span> <span class="n">Metadata</span>
  92. <span class="kn">from</span> <span class="nn">._typing</span> <span class="kn">import</span> <span class="p">(</span><span class="n">ChannelArgumentType</span><span class="p">,</span> <span class="n">DeserializingFunction</span><span class="p">,</span>
  93. <span class="n">SerializingFunction</span><span class="p">,</span> <span class="n">RequestIterableType</span><span class="p">)</span>
  94. <span class="kn">from</span> <span class="nn">._utils</span> <span class="kn">import</span> <span class="n">_timeout_to_deadline</span>
  95. <span class="n">_USER_AGENT</span> <span class="o">=</span> <span class="s1">&#39;grpc-python-asyncio/</span><span class="si">{}</span><span class="s1">&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">_grpcio_metadata</span><span class="o">.</span><span class="n">__version__</span><span class="p">)</span>
  96. <span class="k">if</span> <span class="n">sys</span><span class="o">.</span><span class="n">version_info</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span> <span class="o">&lt;</span> <span class="mi">7</span><span class="p">:</span>
  97. <span class="k">def</span> <span class="nf">_all_tasks</span><span class="p">()</span> <span class="o">-&gt;</span> <span class="n">Iterable</span><span class="p">[</span><span class="n">asyncio</span><span class="o">.</span><span class="n">Task</span><span class="p">]:</span>
  98. <span class="k">return</span> <span class="n">asyncio</span><span class="o">.</span><span class="n">Task</span><span class="o">.</span><span class="n">all_tasks</span><span class="p">()</span>
  99. <span class="k">else</span><span class="p">:</span>
  100. <span class="k">def</span> <span class="nf">_all_tasks</span><span class="p">()</span> <span class="o">-&gt;</span> <span class="n">Iterable</span><span class="p">[</span><span class="n">asyncio</span><span class="o">.</span><span class="n">Task</span><span class="p">]:</span>
  101. <span class="k">return</span> <span class="n">asyncio</span><span class="o">.</span><span class="n">all_tasks</span><span class="p">()</span>
  102. <span class="k">def</span> <span class="nf">_augment_channel_arguments</span><span class="p">(</span><span class="n">base_options</span><span class="p">:</span> <span class="n">ChannelArgumentType</span><span class="p">,</span>
  103. <span class="n">compression</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">grpc</span><span class="o">.</span><span class="n">Compression</span><span class="p">]):</span>
  104. <span class="n">compression_channel_argument</span> <span class="o">=</span> <span class="n">_compression</span><span class="o">.</span><span class="n">create_channel_option</span><span class="p">(</span>
  105. <span class="n">compression</span><span class="p">)</span>
  106. <span class="n">user_agent_channel_argument</span> <span class="o">=</span> <span class="p">((</span>
  107. <span class="n">cygrpc</span><span class="o">.</span><span class="n">ChannelArgKey</span><span class="o">.</span><span class="n">primary_user_agent_string</span><span class="p">,</span>
  108. <span class="n">_USER_AGENT</span><span class="p">,</span>
  109. <span class="p">),)</span>
  110. <span class="k">return</span> <span class="nb">tuple</span><span class="p">(</span><span class="n">base_options</span>
  111. <span class="p">)</span> <span class="o">+</span> <span class="n">compression_channel_argument</span> <span class="o">+</span> <span class="n">user_agent_channel_argument</span>
  112. <span class="k">class</span> <span class="nc">_BaseMultiCallable</span><span class="p">:</span>
  113. <span class="sd">&quot;&quot;&quot;Base class of all multi callable objects.</span>
  114. <span class="sd"> Handles the initialization logic and stores common attributes.</span>
  115. <span class="sd"> &quot;&quot;&quot;</span>
  116. <span class="n">_loop</span><span class="p">:</span> <span class="n">asyncio</span><span class="o">.</span><span class="n">AbstractEventLoop</span>
  117. <span class="n">_channel</span><span class="p">:</span> <span class="n">cygrpc</span><span class="o">.</span><span class="n">AioChannel</span>
  118. <span class="n">_method</span><span class="p">:</span> <span class="nb">bytes</span>
  119. <span class="n">_request_serializer</span><span class="p">:</span> <span class="n">SerializingFunction</span>
  120. <span class="n">_response_deserializer</span><span class="p">:</span> <span class="n">DeserializingFunction</span>
  121. <span class="n">_interceptors</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Sequence</span><span class="p">[</span><span class="n">ClientInterceptor</span><span class="p">]]</span>
  122. <span class="n">_loop</span><span class="p">:</span> <span class="n">asyncio</span><span class="o">.</span><span class="n">AbstractEventLoop</span>
  123. <span class="c1"># pylint: disable=too-many-arguments</span>
  124. <span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
  125. <span class="bp">self</span><span class="p">,</span>
  126. <span class="n">channel</span><span class="p">:</span> <span class="n">cygrpc</span><span class="o">.</span><span class="n">AioChannel</span><span class="p">,</span>
  127. <span class="n">method</span><span class="p">:</span> <span class="nb">bytes</span><span class="p">,</span>
  128. <span class="n">request_serializer</span><span class="p">:</span> <span class="n">SerializingFunction</span><span class="p">,</span>
  129. <span class="n">response_deserializer</span><span class="p">:</span> <span class="n">DeserializingFunction</span><span class="p">,</span>
  130. <span class="n">interceptors</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Sequence</span><span class="p">[</span><span class="n">ClientInterceptor</span><span class="p">]],</span>
  131. <span class="n">loop</span><span class="p">:</span> <span class="n">asyncio</span><span class="o">.</span><span class="n">AbstractEventLoop</span><span class="p">,</span>
  132. <span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
  133. <span class="bp">self</span><span class="o">.</span><span class="n">_loop</span> <span class="o">=</span> <span class="n">loop</span>
  134. <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span> <span class="o">=</span> <span class="n">channel</span>
  135. <span class="bp">self</span><span class="o">.</span><span class="n">_method</span> <span class="o">=</span> <span class="n">method</span>
  136. <span class="bp">self</span><span class="o">.</span><span class="n">_request_serializer</span> <span class="o">=</span> <span class="n">request_serializer</span>
  137. <span class="bp">self</span><span class="o">.</span><span class="n">_response_deserializer</span> <span class="o">=</span> <span class="n">response_deserializer</span>
  138. <span class="bp">self</span><span class="o">.</span><span class="n">_interceptors</span> <span class="o">=</span> <span class="n">interceptors</span>
  139. <span class="nd">@staticmethod</span>
  140. <span class="k">def</span> <span class="nf">_init_metadata</span><span class="p">(</span><span class="n">metadata</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Metadata</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  141. <span class="n">compression</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">grpc</span><span class="o">.</span><span class="n">Compression</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span>
  142. <span class="p">)</span> <span class="o">-&gt;</span> <span class="n">Metadata</span><span class="p">:</span>
  143. <span class="sd">&quot;&quot;&quot;Based on the provided values for &lt;metadata&gt; or &lt;compression&gt; initialise the final</span>
  144. <span class="sd"> metadata, as it should be used for the current call.</span>
  145. <span class="sd"> &quot;&quot;&quot;</span>
  146. <span class="n">metadata</span> <span class="o">=</span> <span class="n">metadata</span> <span class="ow">or</span> <span class="n">Metadata</span><span class="p">()</span>
  147. <span class="k">if</span> <span class="n">compression</span><span class="p">:</span>
  148. <span class="n">metadata</span> <span class="o">=</span> <span class="n">Metadata</span><span class="p">(</span>
  149. <span class="o">*</span><span class="n">_compression</span><span class="o">.</span><span class="n">augment_metadata</span><span class="p">(</span><span class="n">metadata</span><span class="p">,</span> <span class="n">compression</span><span class="p">))</span>
  150. <span class="k">return</span> <span class="n">metadata</span>
  151. <span class="k">class</span> <span class="nc">UnaryUnaryMultiCallable</span><span class="p">(</span><span class="n">_BaseMultiCallable</span><span class="p">,</span>
  152. <span class="n">_base_channel</span><span class="o">.</span><span class="n">UnaryUnaryMultiCallable</span><span class="p">):</span>
  153. <span class="k">def</span> <span class="fm">__call__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span>
  154. <span class="n">request</span><span class="p">:</span> <span class="n">Any</span><span class="p">,</span>
  155. <span class="o">*</span><span class="p">,</span>
  156. <span class="n">timeout</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="nb">float</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  157. <span class="n">metadata</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Metadata</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  158. <span class="n">credentials</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">grpc</span><span class="o">.</span><span class="n">CallCredentials</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  159. <span class="n">wait_for_ready</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="nb">bool</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  160. <span class="n">compression</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">grpc</span><span class="o">.</span><span class="n">Compression</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span>
  161. <span class="p">)</span> <span class="o">-&gt;</span> <span class="n">_base_call</span><span class="o">.</span><span class="n">UnaryUnaryCall</span><span class="p">:</span>
  162. <span class="n">metadata</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_init_metadata</span><span class="p">(</span><span class="n">metadata</span><span class="p">,</span> <span class="n">compression</span><span class="p">)</span>
  163. <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_interceptors</span><span class="p">:</span>
  164. <span class="n">call</span> <span class="o">=</span> <span class="n">UnaryUnaryCall</span><span class="p">(</span><span class="n">request</span><span class="p">,</span> <span class="n">_timeout_to_deadline</span><span class="p">(</span><span class="n">timeout</span><span class="p">),</span>
  165. <span class="n">metadata</span><span class="p">,</span> <span class="n">credentials</span><span class="p">,</span> <span class="n">wait_for_ready</span><span class="p">,</span>
  166. <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_method</span><span class="p">,</span>
  167. <span class="bp">self</span><span class="o">.</span><span class="n">_request_serializer</span><span class="p">,</span>
  168. <span class="bp">self</span><span class="o">.</span><span class="n">_response_deserializer</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_loop</span><span class="p">)</span>
  169. <span class="k">else</span><span class="p">:</span>
  170. <span class="n">call</span> <span class="o">=</span> <span class="n">InterceptedUnaryUnaryCall</span><span class="p">(</span>
  171. <span class="bp">self</span><span class="o">.</span><span class="n">_interceptors</span><span class="p">,</span> <span class="n">request</span><span class="p">,</span> <span class="n">timeout</span><span class="p">,</span> <span class="n">metadata</span><span class="p">,</span> <span class="n">credentials</span><span class="p">,</span>
  172. <span class="n">wait_for_ready</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_method</span><span class="p">,</span>
  173. <span class="bp">self</span><span class="o">.</span><span class="n">_request_serializer</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_response_deserializer</span><span class="p">,</span>
  174. <span class="bp">self</span><span class="o">.</span><span class="n">_loop</span><span class="p">)</span>
  175. <span class="k">return</span> <span class="n">call</span>
  176. <span class="k">class</span> <span class="nc">UnaryStreamMultiCallable</span><span class="p">(</span><span class="n">_BaseMultiCallable</span><span class="p">,</span>
  177. <span class="n">_base_channel</span><span class="o">.</span><span class="n">UnaryStreamMultiCallable</span><span class="p">):</span>
  178. <span class="k">def</span> <span class="fm">__call__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span>
  179. <span class="n">request</span><span class="p">:</span> <span class="n">Any</span><span class="p">,</span>
  180. <span class="o">*</span><span class="p">,</span>
  181. <span class="n">timeout</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="nb">float</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  182. <span class="n">metadata</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Metadata</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  183. <span class="n">credentials</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">grpc</span><span class="o">.</span><span class="n">CallCredentials</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  184. <span class="n">wait_for_ready</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="nb">bool</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  185. <span class="n">compression</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">grpc</span><span class="o">.</span><span class="n">Compression</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span>
  186. <span class="p">)</span> <span class="o">-&gt;</span> <span class="n">_base_call</span><span class="o">.</span><span class="n">UnaryStreamCall</span><span class="p">:</span>
  187. <span class="n">metadata</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_init_metadata</span><span class="p">(</span><span class="n">metadata</span><span class="p">,</span> <span class="n">compression</span><span class="p">)</span>
  188. <span class="n">deadline</span> <span class="o">=</span> <span class="n">_timeout_to_deadline</span><span class="p">(</span><span class="n">timeout</span><span class="p">)</span>
  189. <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_interceptors</span><span class="p">:</span>
  190. <span class="n">call</span> <span class="o">=</span> <span class="n">UnaryStreamCall</span><span class="p">(</span><span class="n">request</span><span class="p">,</span> <span class="n">deadline</span><span class="p">,</span> <span class="n">metadata</span><span class="p">,</span> <span class="n">credentials</span><span class="p">,</span>
  191. <span class="n">wait_for_ready</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_method</span><span class="p">,</span>
  192. <span class="bp">self</span><span class="o">.</span><span class="n">_request_serializer</span><span class="p">,</span>
  193. <span class="bp">self</span><span class="o">.</span><span class="n">_response_deserializer</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_loop</span><span class="p">)</span>
  194. <span class="k">else</span><span class="p">:</span>
  195. <span class="n">call</span> <span class="o">=</span> <span class="n">InterceptedUnaryStreamCall</span><span class="p">(</span>
  196. <span class="bp">self</span><span class="o">.</span><span class="n">_interceptors</span><span class="p">,</span> <span class="n">request</span><span class="p">,</span> <span class="n">deadline</span><span class="p">,</span> <span class="n">metadata</span><span class="p">,</span> <span class="n">credentials</span><span class="p">,</span>
  197. <span class="n">wait_for_ready</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_method</span><span class="p">,</span>
  198. <span class="bp">self</span><span class="o">.</span><span class="n">_request_serializer</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_response_deserializer</span><span class="p">,</span>
  199. <span class="bp">self</span><span class="o">.</span><span class="n">_loop</span><span class="p">)</span>
  200. <span class="k">return</span> <span class="n">call</span>
  201. <span class="k">class</span> <span class="nc">StreamUnaryMultiCallable</span><span class="p">(</span><span class="n">_BaseMultiCallable</span><span class="p">,</span>
  202. <span class="n">_base_channel</span><span class="o">.</span><span class="n">StreamUnaryMultiCallable</span><span class="p">):</span>
  203. <span class="k">def</span> <span class="fm">__call__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span>
  204. <span class="n">request_iterator</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">RequestIterableType</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  205. <span class="n">timeout</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="nb">float</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  206. <span class="n">metadata</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Metadata</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  207. <span class="n">credentials</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">grpc</span><span class="o">.</span><span class="n">CallCredentials</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  208. <span class="n">wait_for_ready</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="nb">bool</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  209. <span class="n">compression</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">grpc</span><span class="o">.</span><span class="n">Compression</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span>
  210. <span class="p">)</span> <span class="o">-&gt;</span> <span class="n">_base_call</span><span class="o">.</span><span class="n">StreamUnaryCall</span><span class="p">:</span>
  211. <span class="n">metadata</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_init_metadata</span><span class="p">(</span><span class="n">metadata</span><span class="p">,</span> <span class="n">compression</span><span class="p">)</span>
  212. <span class="n">deadline</span> <span class="o">=</span> <span class="n">_timeout_to_deadline</span><span class="p">(</span><span class="n">timeout</span><span class="p">)</span>
  213. <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_interceptors</span><span class="p">:</span>
  214. <span class="n">call</span> <span class="o">=</span> <span class="n">StreamUnaryCall</span><span class="p">(</span><span class="n">request_iterator</span><span class="p">,</span> <span class="n">deadline</span><span class="p">,</span> <span class="n">metadata</span><span class="p">,</span>
  215. <span class="n">credentials</span><span class="p">,</span> <span class="n">wait_for_ready</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="p">,</span>
  216. <span class="bp">self</span><span class="o">.</span><span class="n">_method</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_request_serializer</span><span class="p">,</span>
  217. <span class="bp">self</span><span class="o">.</span><span class="n">_response_deserializer</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_loop</span><span class="p">)</span>
  218. <span class="k">else</span><span class="p">:</span>
  219. <span class="n">call</span> <span class="o">=</span> <span class="n">InterceptedStreamUnaryCall</span><span class="p">(</span>
  220. <span class="bp">self</span><span class="o">.</span><span class="n">_interceptors</span><span class="p">,</span> <span class="n">request_iterator</span><span class="p">,</span> <span class="n">deadline</span><span class="p">,</span> <span class="n">metadata</span><span class="p">,</span>
  221. <span class="n">credentials</span><span class="p">,</span> <span class="n">wait_for_ready</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_method</span><span class="p">,</span>
  222. <span class="bp">self</span><span class="o">.</span><span class="n">_request_serializer</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_response_deserializer</span><span class="p">,</span>
  223. <span class="bp">self</span><span class="o">.</span><span class="n">_loop</span><span class="p">)</span>
  224. <span class="k">return</span> <span class="n">call</span>
  225. <span class="k">class</span> <span class="nc">StreamStreamMultiCallable</span><span class="p">(</span><span class="n">_BaseMultiCallable</span><span class="p">,</span>
  226. <span class="n">_base_channel</span><span class="o">.</span><span class="n">StreamStreamMultiCallable</span><span class="p">):</span>
  227. <span class="k">def</span> <span class="fm">__call__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span>
  228. <span class="n">request_iterator</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">RequestIterableType</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  229. <span class="n">timeout</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="nb">float</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  230. <span class="n">metadata</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Metadata</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  231. <span class="n">credentials</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">grpc</span><span class="o">.</span><span class="n">CallCredentials</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  232. <span class="n">wait_for_ready</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="nb">bool</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  233. <span class="n">compression</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">grpc</span><span class="o">.</span><span class="n">Compression</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span>
  234. <span class="p">)</span> <span class="o">-&gt;</span> <span class="n">_base_call</span><span class="o">.</span><span class="n">StreamStreamCall</span><span class="p">:</span>
  235. <span class="n">metadata</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_init_metadata</span><span class="p">(</span><span class="n">metadata</span><span class="p">,</span> <span class="n">compression</span><span class="p">)</span>
  236. <span class="n">deadline</span> <span class="o">=</span> <span class="n">_timeout_to_deadline</span><span class="p">(</span><span class="n">timeout</span><span class="p">)</span>
  237. <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_interceptors</span><span class="p">:</span>
  238. <span class="n">call</span> <span class="o">=</span> <span class="n">StreamStreamCall</span><span class="p">(</span><span class="n">request_iterator</span><span class="p">,</span> <span class="n">deadline</span><span class="p">,</span> <span class="n">metadata</span><span class="p">,</span>
  239. <span class="n">credentials</span><span class="p">,</span> <span class="n">wait_for_ready</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="p">,</span>
  240. <span class="bp">self</span><span class="o">.</span><span class="n">_method</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_request_serializer</span><span class="p">,</span>
  241. <span class="bp">self</span><span class="o">.</span><span class="n">_response_deserializer</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_loop</span><span class="p">)</span>
  242. <span class="k">else</span><span class="p">:</span>
  243. <span class="n">call</span> <span class="o">=</span> <span class="n">InterceptedStreamStreamCall</span><span class="p">(</span>
  244. <span class="bp">self</span><span class="o">.</span><span class="n">_interceptors</span><span class="p">,</span> <span class="n">request_iterator</span><span class="p">,</span> <span class="n">deadline</span><span class="p">,</span> <span class="n">metadata</span><span class="p">,</span>
  245. <span class="n">credentials</span><span class="p">,</span> <span class="n">wait_for_ready</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_method</span><span class="p">,</span>
  246. <span class="bp">self</span><span class="o">.</span><span class="n">_request_serializer</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_response_deserializer</span><span class="p">,</span>
  247. <span class="bp">self</span><span class="o">.</span><span class="n">_loop</span><span class="p">)</span>
  248. <span class="k">return</span> <span class="n">call</span>
  249. <span class="k">class</span> <span class="nc">Channel</span><span class="p">(</span><span class="n">_base_channel</span><span class="o">.</span><span class="n">Channel</span><span class="p">):</span>
  250. <span class="n">_loop</span><span class="p">:</span> <span class="n">asyncio</span><span class="o">.</span><span class="n">AbstractEventLoop</span>
  251. <span class="n">_channel</span><span class="p">:</span> <span class="n">cygrpc</span><span class="o">.</span><span class="n">AioChannel</span>
  252. <span class="n">_unary_unary_interceptors</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="n">UnaryUnaryClientInterceptor</span><span class="p">]</span>
  253. <span class="n">_unary_stream_interceptors</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="n">UnaryStreamClientInterceptor</span><span class="p">]</span>
  254. <span class="n">_stream_unary_interceptors</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="n">StreamUnaryClientInterceptor</span><span class="p">]</span>
  255. <span class="n">_stream_stream_interceptors</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="n">StreamStreamClientInterceptor</span><span class="p">]</span>
  256. <span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">target</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">options</span><span class="p">:</span> <span class="n">ChannelArgumentType</span><span class="p">,</span>
  257. <span class="n">credentials</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">grpc</span><span class="o">.</span><span class="n">ChannelCredentials</span><span class="p">],</span>
  258. <span class="n">compression</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">grpc</span><span class="o">.</span><span class="n">Compression</span><span class="p">],</span>
  259. <span class="n">interceptors</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Sequence</span><span class="p">[</span><span class="n">ClientInterceptor</span><span class="p">]]):</span>
  260. <span class="sd">&quot;&quot;&quot;Constructor.</span>
  261. <span class="sd"> Args:</span>
  262. <span class="sd"> target: The target to which to connect.</span>
  263. <span class="sd"> options: Configuration options for the channel.</span>
  264. <span class="sd"> credentials: A cygrpc.ChannelCredentials or None.</span>
  265. <span class="sd"> compression: An optional value indicating the compression method to be</span>
  266. <span class="sd"> used over the lifetime of the channel.</span>
  267. <span class="sd"> interceptors: An optional list of interceptors that would be used for</span>
  268. <span class="sd"> intercepting any RPC executed with that channel.</span>
  269. <span class="sd"> &quot;&quot;&quot;</span>
  270. <span class="bp">self</span><span class="o">.</span><span class="n">_unary_unary_interceptors</span> <span class="o">=</span> <span class="p">[]</span>
  271. <span class="bp">self</span><span class="o">.</span><span class="n">_unary_stream_interceptors</span> <span class="o">=</span> <span class="p">[]</span>
  272. <span class="bp">self</span><span class="o">.</span><span class="n">_stream_unary_interceptors</span> <span class="o">=</span> <span class="p">[]</span>
  273. <span class="bp">self</span><span class="o">.</span><span class="n">_stream_stream_interceptors</span> <span class="o">=</span> <span class="p">[]</span>
  274. <span class="k">if</span> <span class="n">interceptors</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span><span class="p">:</span>
  275. <span class="k">for</span> <span class="n">interceptor</span> <span class="ow">in</span> <span class="n">interceptors</span><span class="p">:</span>
  276. <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">interceptor</span><span class="p">,</span> <span class="n">UnaryUnaryClientInterceptor</span><span class="p">):</span>
  277. <span class="bp">self</span><span class="o">.</span><span class="n">_unary_unary_interceptors</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">interceptor</span><span class="p">)</span>
  278. <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">interceptor</span><span class="p">,</span> <span class="n">UnaryStreamClientInterceptor</span><span class="p">):</span>
  279. <span class="bp">self</span><span class="o">.</span><span class="n">_unary_stream_interceptors</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">interceptor</span><span class="p">)</span>
  280. <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">interceptor</span><span class="p">,</span> <span class="n">StreamUnaryClientInterceptor</span><span class="p">):</span>
  281. <span class="bp">self</span><span class="o">.</span><span class="n">_stream_unary_interceptors</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">interceptor</span><span class="p">)</span>
  282. <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">interceptor</span><span class="p">,</span> <span class="n">StreamStreamClientInterceptor</span><span class="p">):</span>
  283. <span class="bp">self</span><span class="o">.</span><span class="n">_stream_stream_interceptors</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">interceptor</span><span class="p">)</span>
  284. <span class="k">else</span><span class="p">:</span>
  285. <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span>
  286. <span class="s2">&quot;Interceptor </span><span class="si">{}</span><span class="s2"> must be &quot;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">interceptor</span><span class="p">)</span> <span class="o">+</span>
  287. <span class="s2">&quot;</span><span class="si">{}</span><span class="s2"> or &quot;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">UnaryUnaryClientInterceptor</span><span class="o">.</span><span class="vm">__name__</span><span class="p">)</span> <span class="o">+</span>
  288. <span class="s2">&quot;</span><span class="si">{}</span><span class="s2"> or &quot;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">UnaryStreamClientInterceptor</span><span class="o">.</span><span class="vm">__name__</span><span class="p">)</span> <span class="o">+</span>
  289. <span class="s2">&quot;</span><span class="si">{}</span><span class="s2"> or &quot;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">StreamUnaryClientInterceptor</span><span class="o">.</span><span class="vm">__name__</span><span class="p">)</span> <span class="o">+</span>
  290. <span class="s2">&quot;</span><span class="si">{}</span><span class="s2">. &quot;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">StreamStreamClientInterceptor</span><span class="o">.</span><span class="vm">__name__</span><span class="p">))</span>
  291. <span class="bp">self</span><span class="o">.</span><span class="n">_loop</span> <span class="o">=</span> <span class="n">cygrpc</span><span class="o">.</span><span class="n">get_working_loop</span><span class="p">()</span>
  292. <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span> <span class="o">=</span> <span class="n">cygrpc</span><span class="o">.</span><span class="n">AioChannel</span><span class="p">(</span>
  293. <span class="n">_common</span><span class="o">.</span><span class="n">encode</span><span class="p">(</span><span class="n">target</span><span class="p">),</span>
  294. <span class="n">_augment_channel_arguments</span><span class="p">(</span><span class="n">options</span><span class="p">,</span> <span class="n">compression</span><span class="p">),</span> <span class="n">credentials</span><span class="p">,</span>
  295. <span class="bp">self</span><span class="o">.</span><span class="n">_loop</span><span class="p">)</span>
  296. <span class="k">async</span> <span class="k">def</span> <span class="fm">__aenter__</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  297. <span class="k">return</span> <span class="bp">self</span>
  298. <span class="k">async</span> <span class="k">def</span> <span class="fm">__aexit__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">exc_type</span><span class="p">,</span> <span class="n">exc_val</span><span class="p">,</span> <span class="n">exc_tb</span><span class="p">):</span>
  299. <span class="k">await</span> <span class="bp">self</span><span class="o">.</span><span class="n">_close</span><span class="p">(</span><span class="kc">None</span><span class="p">)</span>
  300. <span class="k">async</span> <span class="k">def</span> <span class="nf">_close</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">grace</span><span class="p">):</span> <span class="c1"># pylint: disable=too-many-branches</span>
  301. <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="o">.</span><span class="n">closed</span><span class="p">():</span>
  302. <span class="k">return</span>
  303. <span class="c1"># No new calls will be accepted by the Cython channel.</span>
  304. <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="o">.</span><span class="n">closing</span><span class="p">()</span>
  305. <span class="c1"># Iterate through running tasks</span>
  306. <span class="n">tasks</span> <span class="o">=</span> <span class="n">_all_tasks</span><span class="p">()</span>
  307. <span class="n">calls</span> <span class="o">=</span> <span class="p">[]</span>
  308. <span class="n">call_tasks</span> <span class="o">=</span> <span class="p">[]</span>
  309. <span class="k">for</span> <span class="n">task</span> <span class="ow">in</span> <span class="n">tasks</span><span class="p">:</span>
  310. <span class="k">try</span><span class="p">:</span>
  311. <span class="n">stack</span> <span class="o">=</span> <span class="n">task</span><span class="o">.</span><span class="n">get_stack</span><span class="p">(</span><span class="n">limit</span><span class="o">=</span><span class="mi">1</span><span class="p">)</span>
  312. <span class="k">except</span> <span class="ne">AttributeError</span> <span class="k">as</span> <span class="n">attribute_error</span><span class="p">:</span>
  313. <span class="c1"># NOTE(lidiz) tl;dr: If the Task is created with a CPython</span>
  314. <span class="c1"># object, it will trigger AttributeError.</span>
  315. <span class="c1">#</span>
  316. <span class="c1"># In the global finalizer, the event loop schedules</span>
  317. <span class="c1"># a CPython PyAsyncGenAThrow object.</span>
  318. <span class="c1"># https://github.com/python/cpython/blob/00e45877e33d32bb61aa13a2033e3bba370bda4d/Lib/asyncio/base_events.py#L484</span>
  319. <span class="c1">#</span>
  320. <span class="c1"># However, the PyAsyncGenAThrow object is written in C and</span>
  321. <span class="c1"># failed to include the normal Python frame objects. Hence,</span>
  322. <span class="c1"># this exception is a false negative, and it is safe to ignore</span>
  323. <span class="c1"># the failure. It is fixed by https://github.com/python/cpython/pull/18669,</span>
  324. <span class="c1"># but not available until 3.9 or 3.8.3. So, we have to keep it</span>
  325. <span class="c1"># for a while.</span>
  326. <span class="c1"># TODO(lidiz) drop this hack after 3.8 deprecation</span>
  327. <span class="k">if</span> <span class="s1">&#39;frame&#39;</span> <span class="ow">in</span> <span class="nb">str</span><span class="p">(</span><span class="n">attribute_error</span><span class="p">):</span>
  328. <span class="k">continue</span>
  329. <span class="k">else</span><span class="p">:</span>
  330. <span class="k">raise</span>
  331. <span class="c1"># If the Task is created by a C-extension, the stack will be empty.</span>
  332. <span class="k">if</span> <span class="ow">not</span> <span class="n">stack</span><span class="p">:</span>
  333. <span class="k">continue</span>
  334. <span class="c1"># Locate ones created by `aio.Call`.</span>
  335. <span class="n">frame</span> <span class="o">=</span> <span class="n">stack</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span>
  336. <span class="n">candidate</span> <span class="o">=</span> <span class="n">frame</span><span class="o">.</span><span class="n">f_locals</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s1">&#39;self&#39;</span><span class="p">)</span>
  337. <span class="k">if</span> <span class="n">candidate</span><span class="p">:</span>
  338. <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">candidate</span><span class="p">,</span> <span class="n">_base_call</span><span class="o">.</span><span class="n">Call</span><span class="p">):</span>
  339. <span class="k">if</span> <span class="nb">hasattr</span><span class="p">(</span><span class="n">candidate</span><span class="p">,</span> <span class="s1">&#39;_channel&#39;</span><span class="p">):</span>
  340. <span class="c1"># For intercepted Call object</span>
  341. <span class="k">if</span> <span class="n">candidate</span><span class="o">.</span><span class="n">_channel</span> <span class="ow">is</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="p">:</span>
  342. <span class="k">continue</span>
  343. <span class="k">elif</span> <span class="nb">hasattr</span><span class="p">(</span><span class="n">candidate</span><span class="p">,</span> <span class="s1">&#39;_cython_call&#39;</span><span class="p">):</span>
  344. <span class="c1"># For normal Call object</span>
  345. <span class="k">if</span> <span class="n">candidate</span><span class="o">.</span><span class="n">_cython_call</span><span class="o">.</span><span class="n">_channel</span> <span class="ow">is</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="p">:</span>
  346. <span class="k">continue</span>
  347. <span class="k">else</span><span class="p">:</span>
  348. <span class="c1"># Unidentified Call object</span>
  349. <span class="k">raise</span> <span class="n">cygrpc</span><span class="o">.</span><span class="n">InternalError</span><span class="p">(</span>
  350. <span class="sa">f</span><span class="s1">&#39;Unrecognized call object: </span><span class="si">{</span><span class="n">candidate</span><span class="si">}</span><span class="s1">&#39;</span><span class="p">)</span>
  351. <span class="n">calls</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">candidate</span><span class="p">)</span>
  352. <span class="n">call_tasks</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">task</span><span class="p">)</span>
  353. <span class="c1"># If needed, try to wait for them to finish.</span>
  354. <span class="c1"># Call objects are not always awaitables.</span>
  355. <span class="k">if</span> <span class="n">grace</span> <span class="ow">and</span> <span class="n">call_tasks</span><span class="p">:</span>
  356. <span class="k">await</span> <span class="n">asyncio</span><span class="o">.</span><span class="n">wait</span><span class="p">(</span><span class="n">call_tasks</span><span class="p">,</span> <span class="n">timeout</span><span class="o">=</span><span class="n">grace</span><span class="p">,</span> <span class="n">loop</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">_loop</span><span class="p">)</span>
  357. <span class="c1"># Time to cancel existing calls.</span>
  358. <span class="k">for</span> <span class="n">call</span> <span class="ow">in</span> <span class="n">calls</span><span class="p">:</span>
  359. <span class="n">call</span><span class="o">.</span><span class="n">cancel</span><span class="p">()</span>
  360. <span class="c1"># Destroy the channel</span>
  361. <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="o">.</span><span class="n">close</span><span class="p">()</span>
  362. <span class="k">async</span> <span class="k">def</span> <span class="nf">close</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">grace</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="nb">float</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">):</span>
  363. <span class="k">await</span> <span class="bp">self</span><span class="o">.</span><span class="n">_close</span><span class="p">(</span><span class="n">grace</span><span class="p">)</span>
  364. <span class="k">def</span> <span class="nf">get_state</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span>
  365. <span class="n">try_to_connect</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">grpc</span><span class="o">.</span><span class="n">ChannelConnectivity</span><span class="p">:</span>
  366. <span class="n">result</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="o">.</span><span class="n">check_connectivity_state</span><span class="p">(</span><span class="n">try_to_connect</span><span class="p">)</span>
  367. <span class="k">return</span> <span class="n">_common</span><span class="o">.</span><span class="n">CYGRPC_CONNECTIVITY_STATE_TO_CHANNEL_CONNECTIVITY</span><span class="p">[</span><span class="n">result</span><span class="p">]</span>
  368. <span class="k">async</span> <span class="k">def</span> <span class="nf">wait_for_state_change</span><span class="p">(</span>
  369. <span class="bp">self</span><span class="p">,</span>
  370. <span class="n">last_observed_state</span><span class="p">:</span> <span class="n">grpc</span><span class="o">.</span><span class="n">ChannelConnectivity</span><span class="p">,</span>
  371. <span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
  372. <span class="k">assert</span> <span class="k">await</span> <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="o">.</span><span class="n">watch_connectivity_state</span><span class="p">(</span>
  373. <span class="n">last_observed_state</span><span class="o">.</span><span class="n">value</span><span class="p">[</span><span class="mi">0</span><span class="p">],</span> <span class="kc">None</span><span class="p">)</span>
  374. <span class="k">async</span> <span class="k">def</span> <span class="nf">channel_ready</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
  375. <span class="n">state</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_state</span><span class="p">(</span><span class="n">try_to_connect</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span>
  376. <span class="k">while</span> <span class="n">state</span> <span class="o">!=</span> <span class="n">grpc</span><span class="o">.</span><span class="n">ChannelConnectivity</span><span class="o">.</span><span class="n">READY</span><span class="p">:</span>
  377. <span class="k">await</span> <span class="bp">self</span><span class="o">.</span><span class="n">wait_for_state_change</span><span class="p">(</span><span class="n">state</span><span class="p">)</span>
  378. <span class="n">state</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_state</span><span class="p">(</span><span class="n">try_to_connect</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span>
  379. <span class="k">def</span> <span class="nf">unary_unary</span><span class="p">(</span>
  380. <span class="bp">self</span><span class="p">,</span>
  381. <span class="n">method</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
  382. <span class="n">request_serializer</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">SerializingFunction</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  383. <span class="n">response_deserializer</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">DeserializingFunction</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span>
  384. <span class="p">)</span> <span class="o">-&gt;</span> <span class="n">UnaryUnaryMultiCallable</span><span class="p">:</span>
  385. <span class="k">return</span> <span class="n">UnaryUnaryMultiCallable</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="p">,</span> <span class="n">_common</span><span class="o">.</span><span class="n">encode</span><span class="p">(</span><span class="n">method</span><span class="p">),</span>
  386. <span class="n">request_serializer</span><span class="p">,</span>
  387. <span class="n">response_deserializer</span><span class="p">,</span>
  388. <span class="bp">self</span><span class="o">.</span><span class="n">_unary_unary_interceptors</span><span class="p">,</span>
  389. <span class="bp">self</span><span class="o">.</span><span class="n">_loop</span><span class="p">)</span>
  390. <span class="k">def</span> <span class="nf">unary_stream</span><span class="p">(</span>
  391. <span class="bp">self</span><span class="p">,</span>
  392. <span class="n">method</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
  393. <span class="n">request_serializer</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">SerializingFunction</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  394. <span class="n">response_deserializer</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">DeserializingFunction</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span>
  395. <span class="p">)</span> <span class="o">-&gt;</span> <span class="n">UnaryStreamMultiCallable</span><span class="p">:</span>
  396. <span class="k">return</span> <span class="n">UnaryStreamMultiCallable</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="p">,</span> <span class="n">_common</span><span class="o">.</span><span class="n">encode</span><span class="p">(</span><span class="n">method</span><span class="p">),</span>
  397. <span class="n">request_serializer</span><span class="p">,</span>
  398. <span class="n">response_deserializer</span><span class="p">,</span>
  399. <span class="bp">self</span><span class="o">.</span><span class="n">_unary_stream_interceptors</span><span class="p">,</span>
  400. <span class="bp">self</span><span class="o">.</span><span class="n">_loop</span><span class="p">)</span>
  401. <span class="k">def</span> <span class="nf">stream_unary</span><span class="p">(</span>
  402. <span class="bp">self</span><span class="p">,</span>
  403. <span class="n">method</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
  404. <span class="n">request_serializer</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">SerializingFunction</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  405. <span class="n">response_deserializer</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">DeserializingFunction</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span>
  406. <span class="p">)</span> <span class="o">-&gt;</span> <span class="n">StreamUnaryMultiCallable</span><span class="p">:</span>
  407. <span class="k">return</span> <span class="n">StreamUnaryMultiCallable</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="p">,</span> <span class="n">_common</span><span class="o">.</span><span class="n">encode</span><span class="p">(</span><span class="n">method</span><span class="p">),</span>
  408. <span class="n">request_serializer</span><span class="p">,</span>
  409. <span class="n">response_deserializer</span><span class="p">,</span>
  410. <span class="bp">self</span><span class="o">.</span><span class="n">_stream_unary_interceptors</span><span class="p">,</span>
  411. <span class="bp">self</span><span class="o">.</span><span class="n">_loop</span><span class="p">)</span>
  412. <span class="k">def</span> <span class="nf">stream_stream</span><span class="p">(</span>
  413. <span class="bp">self</span><span class="p">,</span>
  414. <span class="n">method</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
  415. <span class="n">request_serializer</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">SerializingFunction</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  416. <span class="n">response_deserializer</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">DeserializingFunction</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span>
  417. <span class="p">)</span> <span class="o">-&gt;</span> <span class="n">StreamStreamMultiCallable</span><span class="p">:</span>
  418. <span class="k">return</span> <span class="n">StreamStreamMultiCallable</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="p">,</span> <span class="n">_common</span><span class="o">.</span><span class="n">encode</span><span class="p">(</span><span class="n">method</span><span class="p">),</span>
  419. <span class="n">request_serializer</span><span class="p">,</span>
  420. <span class="n">response_deserializer</span><span class="p">,</span>
  421. <span class="bp">self</span><span class="o">.</span><span class="n">_stream_stream_interceptors</span><span class="p">,</span>
  422. <span class="bp">self</span><span class="o">.</span><span class="n">_loop</span><span class="p">)</span>
  423. <div class="viewcode-block" id="insecure_channel"><a class="viewcode-back" href="../../../../grpc_asyncio.html#grpc.experimental.aio.insecure_channel">[docs]</a><span class="k">def</span> <span class="nf">insecure_channel</span><span class="p">(</span>
  424. <span class="n">target</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
  425. <span class="n">options</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">ChannelArgumentType</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  426. <span class="n">compression</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">grpc</span><span class="o">.</span><span class="n">Compression</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  427. <span class="n">interceptors</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Sequence</span><span class="p">[</span><span class="n">ClientInterceptor</span><span class="p">]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">):</span>
  428. <span class="sd">&quot;&quot;&quot;Creates an insecure asynchronous Channel to a server.</span>
  429. <span class="sd"> Args:</span>
  430. <span class="sd"> target: The server address</span>
  431. <span class="sd"> options: An optional list of key-value pairs (:term:`channel_arguments`</span>
  432. <span class="sd"> in gRPC Core runtime) to configure the channel.</span>
  433. <span class="sd"> compression: An optional value indicating the compression method to be</span>
  434. <span class="sd"> used over the lifetime of the channel. This is an EXPERIMENTAL option.</span>
  435. <span class="sd"> interceptors: An optional sequence of interceptors that will be executed for</span>
  436. <span class="sd"> any call executed with this channel.</span>
  437. <span class="sd"> Returns:</span>
  438. <span class="sd"> A Channel.</span>
  439. <span class="sd"> &quot;&quot;&quot;</span>
  440. <span class="k">return</span> <span class="n">Channel</span><span class="p">(</span><span class="n">target</span><span class="p">,</span> <span class="p">()</span> <span class="k">if</span> <span class="n">options</span> <span class="ow">is</span> <span class="kc">None</span> <span class="k">else</span> <span class="n">options</span><span class="p">,</span> <span class="kc">None</span><span class="p">,</span>
  441. <span class="n">compression</span><span class="p">,</span> <span class="n">interceptors</span><span class="p">)</span></div>
  442. <div class="viewcode-block" id="secure_channel"><a class="viewcode-back" href="../../../../grpc_asyncio.html#grpc.experimental.aio.secure_channel">[docs]</a><span class="k">def</span> <span class="nf">secure_channel</span><span class="p">(</span><span class="n">target</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
  443. <span class="n">credentials</span><span class="p">:</span> <span class="n">grpc</span><span class="o">.</span><span class="n">ChannelCredentials</span><span class="p">,</span>
  444. <span class="n">options</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">ChannelArgumentType</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  445. <span class="n">compression</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">grpc</span><span class="o">.</span><span class="n">Compression</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
  446. <span class="n">interceptors</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Sequence</span><span class="p">[</span><span class="n">ClientInterceptor</span><span class="p">]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">):</span>
  447. <span class="sd">&quot;&quot;&quot;Creates a secure asynchronous Channel to a server.</span>
  448. <span class="sd"> Args:</span>
  449. <span class="sd"> target: The server address.</span>
  450. <span class="sd"> credentials: A ChannelCredentials instance.</span>
  451. <span class="sd"> options: An optional list of key-value pairs (:term:`channel_arguments`</span>
  452. <span class="sd"> in gRPC Core runtime) to configure the channel.</span>
  453. <span class="sd"> compression: An optional value indicating the compression method to be</span>
  454. <span class="sd"> used over the lifetime of the channel. This is an EXPERIMENTAL option.</span>
  455. <span class="sd"> interceptors: An optional sequence of interceptors that will be executed for</span>
  456. <span class="sd"> any call executed with this channel.</span>
  457. <span class="sd"> Returns:</span>
  458. <span class="sd"> An aio.Channel.</span>
  459. <span class="sd"> &quot;&quot;&quot;</span>
  460. <span class="k">return</span> <span class="n">Channel</span><span class="p">(</span><span class="n">target</span><span class="p">,</span> <span class="p">()</span> <span class="k">if</span> <span class="n">options</span> <span class="ow">is</span> <span class="kc">None</span> <span class="k">else</span> <span class="n">options</span><span class="p">,</span>
  461. <span class="n">credentials</span><span class="o">.</span><span class="n">_credentials</span><span class="p">,</span> <span class="n">compression</span><span class="p">,</span> <span class="n">interceptors</span><span class="p">)</span></div>
  462. </pre></div>
  463. </div>
  464. </div>
  465. </div>
  466. <div class="clearer"></div>
  467. </div>
  468. <div class="footer">
  469. &copy;2020, The gRPC Authors.
  470. </div>
  471. <script type="text/javascript">
  472. var _gaq = _gaq || [];
  473. _gaq.push(['_setAccount', 'UA-60127042-1']);
  474. _gaq.push(['_setDomainName', 'none']);
  475. _gaq.push(['_setAllowLinker', true]);
  476. _gaq.push(['_trackPageview']);
  477. (function() {
  478. var ga = document.createElement('script'); ga.type = 'text/javascript'; ga.async = true;
  479. ga.src = ('https:' == document.location.protocol ? 'https://ssl' : 'http://www') + '.google-analytics.com/ga.js';
  480. var s = document.getElementsByTagName('script')[0]; s.parentNode.insertBefore(ga, s);
  481. })();
  482. </script>
  483. </body>
  484. </html>