utilities.html 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366
  1. <!DOCTYPE html>
  2. <!--[if IE 8]><html class="no-js lt-ie9" lang="en" > <![endif]-->
  3. <!--[if gt IE 8]><!--> <html class="no-js" lang="en" > <!--<![endif]-->
  4. <head>
  5. <meta charset="utf-8">
  6. <meta name="viewport" content="width=device-width, initial-scale=1.0">
  7. <title>grpc.beta.utilities &mdash; grpcio 1.0.4 documentation</title>
  8. <link rel="stylesheet" href="../../../_static/css/theme.css" type="text/css" />
  9. <link rel="index" title="Index"
  10. href="../../../genindex.html"/>
  11. <link rel="search" title="Search" href="../../../search.html"/>
  12. <link rel="top" title="grpcio 1.0.4 documentation" href="../../../index.html"/>
  13. <link rel="up" title="grpc" href="../../grpc.html"/>
  14. <script src="../../../_static/js/modernizr.min.js"></script>
  15. </head>
  16. <body class="wy-body-for-nav" role="document">
  17. <div class="wy-grid-for-nav">
  18. <nav data-toggle="wy-nav-shift" class="wy-nav-side">
  19. <div class="wy-side-scroll">
  20. <div class="wy-side-nav-search">
  21. <a href="../../../index.html" class="icon icon-home"> grpcio
  22. </a>
  23. <div class="version">
  24. 1.0.4
  25. </div>
  26. <div role="search">
  27. <form id="rtd-search-form" class="wy-form" action="../../../search.html" method="get">
  28. <input type="text" name="q" placeholder="Search docs" />
  29. <input type="hidden" name="check_keywords" value="yes" />
  30. <input type="hidden" name="area" value="default" />
  31. </form>
  32. </div>
  33. </div>
  34. <div class="wy-menu wy-menu-vertical" data-spy="affix" role="navigation" aria-label="main navigation">
  35. <p class="caption"><span class="caption-text">Contents:</span></p>
  36. <ul>
  37. <li class="toctree-l1"><a class="reference internal" href="../../../grpc.html">grpc package</a></li>
  38. <li class="toctree-l1"><a class="reference internal" href="../../../grpc.html#glossary">Glossary</a></li>
  39. </ul>
  40. </div>
  41. </div>
  42. </nav>
  43. <section data-toggle="wy-nav-shift" class="wy-nav-content-wrap">
  44. <nav class="wy-nav-top" role="navigation" aria-label="top navigation">
  45. <i data-toggle="wy-nav-top" class="fa fa-bars"></i>
  46. <a href="../../../index.html">grpcio</a>
  47. </nav>
  48. <div class="wy-nav-content">
  49. <div class="rst-content">
  50. <div role="navigation" aria-label="breadcrumbs navigation">
  51. <ul class="wy-breadcrumbs">
  52. <li><a href="../../../index.html">Docs</a> &raquo;</li>
  53. <li><a href="../../index.html">Module code</a> &raquo;</li>
  54. <li><a href="../../grpc.html">grpc</a> &raquo;</li>
  55. <li>grpc.beta.utilities</li>
  56. <li class="wy-breadcrumbs-aside">
  57. </li>
  58. </ul>
  59. <hr/>
  60. </div>
  61. <div role="main" class="document" itemscope="itemscope" itemtype="http://schema.org/Article">
  62. <div itemprop="articleBody">
  63. <h1>Source code for grpc.beta.utilities</h1><div class="highlight"><pre>
  64. <span></span><span class="c1"># Copyright 2015, Google Inc.</span>
  65. <span class="c1"># All rights reserved.</span>
  66. <span class="c1">#</span>
  67. <span class="c1"># Redistribution and use in source and binary forms, with or without</span>
  68. <span class="c1"># modification, are permitted provided that the following conditions are</span>
  69. <span class="c1"># met:</span>
  70. <span class="c1">#</span>
  71. <span class="c1"># * Redistributions of source code must retain the above copyright</span>
  72. <span class="c1"># notice, this list of conditions and the following disclaimer.</span>
  73. <span class="c1"># * Redistributions in binary form must reproduce the above</span>
  74. <span class="c1"># copyright notice, this list of conditions and the following disclaimer</span>
  75. <span class="c1"># in the documentation and/or other materials provided with the</span>
  76. <span class="c1"># distribution.</span>
  77. <span class="c1"># * Neither the name of Google Inc. nor the names of its</span>
  78. <span class="c1"># contributors may be used to endorse or promote products derived from</span>
  79. <span class="c1"># this software without specific prior written permission.</span>
  80. <span class="c1">#</span>
  81. <span class="c1"># THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS</span>
  82. <span class="c1"># &quot;AS IS&quot; AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT</span>
  83. <span class="c1"># LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR</span>
  84. <span class="c1"># A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT</span>
  85. <span class="c1"># OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,</span>
  86. <span class="c1"># SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT</span>
  87. <span class="c1"># LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,</span>
  88. <span class="c1"># DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY</span>
  89. <span class="c1"># THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT</span>
  90. <span class="c1"># (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE</span>
  91. <span class="c1"># OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.</span>
  92. <span class="sd">&quot;&quot;&quot;Utilities for the gRPC Python Beta API.&quot;&quot;&quot;</span>
  93. <span class="kn">import</span> <span class="nn">threading</span>
  94. <span class="kn">import</span> <span class="nn">time</span>
  95. <span class="c1"># implementations is referenced from specification in this module.</span>
  96. <span class="kn">from</span> <span class="nn">grpc.beta</span> <span class="k">import</span> <span class="n">implementations</span> <span class="c1"># pylint: disable=unused-import</span>
  97. <span class="kn">from</span> <span class="nn">grpc.beta</span> <span class="k">import</span> <span class="n">interfaces</span>
  98. <span class="kn">from</span> <span class="nn">grpc.framework.foundation</span> <span class="k">import</span> <span class="n">callable_util</span>
  99. <span class="kn">from</span> <span class="nn">grpc.framework.foundation</span> <span class="k">import</span> <span class="n">future</span>
  100. <span class="n">_DONE_CALLBACK_EXCEPTION_LOG_MESSAGE</span> <span class="o">=</span> <span class="p">(</span>
  101. <span class="s1">&#39;Exception calling connectivity future &quot;done&quot; callback!&#39;</span><span class="p">)</span>
  102. <span class="k">class</span> <span class="nc">_ChannelReadyFuture</span><span class="p">(</span><span class="n">future</span><span class="o">.</span><span class="n">Future</span><span class="p">):</span>
  103. <span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">channel</span><span class="p">):</span>
  104. <span class="bp">self</span><span class="o">.</span><span class="n">_condition</span> <span class="o">=</span> <span class="n">threading</span><span class="o">.</span><span class="n">Condition</span><span class="p">()</span>
  105. <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span> <span class="o">=</span> <span class="n">channel</span>
  106. <span class="bp">self</span><span class="o">.</span><span class="n">_matured</span> <span class="o">=</span> <span class="kc">False</span>
  107. <span class="bp">self</span><span class="o">.</span><span class="n">_cancelled</span> <span class="o">=</span> <span class="kc">False</span>
  108. <span class="bp">self</span><span class="o">.</span><span class="n">_done_callbacks</span> <span class="o">=</span> <span class="p">[]</span>
  109. <span class="k">def</span> <span class="nf">_block</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">timeout</span><span class="p">):</span>
  110. <span class="n">until</span> <span class="o">=</span> <span class="kc">None</span> <span class="k">if</span> <span class="n">timeout</span> <span class="ow">is</span> <span class="kc">None</span> <span class="k">else</span> <span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()</span> <span class="o">+</span> <span class="n">timeout</span>
  111. <span class="k">with</span> <span class="bp">self</span><span class="o">.</span><span class="n">_condition</span><span class="p">:</span>
  112. <span class="k">while</span> <span class="kc">True</span><span class="p">:</span>
  113. <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">_cancelled</span><span class="p">:</span>
  114. <span class="k">raise</span> <span class="n">future</span><span class="o">.</span><span class="n">CancelledError</span><span class="p">()</span>
  115. <span class="k">elif</span> <span class="bp">self</span><span class="o">.</span><span class="n">_matured</span><span class="p">:</span>
  116. <span class="k">return</span>
  117. <span class="k">else</span><span class="p">:</span>
  118. <span class="k">if</span> <span class="n">until</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
  119. <span class="bp">self</span><span class="o">.</span><span class="n">_condition</span><span class="o">.</span><span class="n">wait</span><span class="p">()</span>
  120. <span class="k">else</span><span class="p">:</span>
  121. <span class="n">remaining</span> <span class="o">=</span> <span class="n">until</span> <span class="o">-</span> <span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()</span>
  122. <span class="k">if</span> <span class="n">remaining</span> <span class="o">&lt;</span> <span class="mi">0</span><span class="p">:</span>
  123. <span class="k">raise</span> <span class="n">future</span><span class="o">.</span><span class="n">TimeoutError</span><span class="p">()</span>
  124. <span class="k">else</span><span class="p">:</span>
  125. <span class="bp">self</span><span class="o">.</span><span class="n">_condition</span><span class="o">.</span><span class="n">wait</span><span class="p">(</span><span class="n">timeout</span><span class="o">=</span><span class="n">remaining</span><span class="p">)</span>
  126. <span class="k">def</span> <span class="nf">_update</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">connectivity</span><span class="p">):</span>
  127. <span class="k">with</span> <span class="bp">self</span><span class="o">.</span><span class="n">_condition</span><span class="p">:</span>
  128. <span class="k">if</span> <span class="p">(</span><span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_cancelled</span> <span class="ow">and</span>
  129. <span class="n">connectivity</span> <span class="ow">is</span> <span class="n">interfaces</span><span class="o">.</span><span class="n">ChannelConnectivity</span><span class="o">.</span><span class="n">READY</span><span class="p">):</span>
  130. <span class="bp">self</span><span class="o">.</span><span class="n">_matured</span> <span class="o">=</span> <span class="kc">True</span>
  131. <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="o">.</span><span class="n">unsubscribe</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_update</span><span class="p">)</span>
  132. <span class="bp">self</span><span class="o">.</span><span class="n">_condition</span><span class="o">.</span><span class="n">notify_all</span><span class="p">()</span>
  133. <span class="n">done_callbacks</span> <span class="o">=</span> <span class="nb">tuple</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_done_callbacks</span><span class="p">)</span>
  134. <span class="bp">self</span><span class="o">.</span><span class="n">_done_callbacks</span> <span class="o">=</span> <span class="kc">None</span>
  135. <span class="k">else</span><span class="p">:</span>
  136. <span class="k">return</span>
  137. <span class="k">for</span> <span class="n">done_callback</span> <span class="ow">in</span> <span class="n">done_callbacks</span><span class="p">:</span>
  138. <span class="n">callable_util</span><span class="o">.</span><span class="n">call_logging_exceptions</span><span class="p">(</span>
  139. <span class="n">done_callback</span><span class="p">,</span> <span class="n">_DONE_CALLBACK_EXCEPTION_LOG_MESSAGE</span><span class="p">,</span> <span class="bp">self</span><span class="p">)</span>
  140. <span class="k">def</span> <span class="nf">cancel</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  141. <span class="k">with</span> <span class="bp">self</span><span class="o">.</span><span class="n">_condition</span><span class="p">:</span>
  142. <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_matured</span><span class="p">:</span>
  143. <span class="bp">self</span><span class="o">.</span><span class="n">_cancelled</span> <span class="o">=</span> <span class="kc">True</span>
  144. <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="o">.</span><span class="n">unsubscribe</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_update</span><span class="p">)</span>
  145. <span class="bp">self</span><span class="o">.</span><span class="n">_condition</span><span class="o">.</span><span class="n">notify_all</span><span class="p">()</span>
  146. <span class="n">done_callbacks</span> <span class="o">=</span> <span class="nb">tuple</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_done_callbacks</span><span class="p">)</span>
  147. <span class="bp">self</span><span class="o">.</span><span class="n">_done_callbacks</span> <span class="o">=</span> <span class="kc">None</span>
  148. <span class="k">else</span><span class="p">:</span>
  149. <span class="k">return</span> <span class="kc">False</span>
  150. <span class="k">for</span> <span class="n">done_callback</span> <span class="ow">in</span> <span class="n">done_callbacks</span><span class="p">:</span>
  151. <span class="n">callable_util</span><span class="o">.</span><span class="n">call_logging_exceptions</span><span class="p">(</span>
  152. <span class="n">done_callback</span><span class="p">,</span> <span class="n">_DONE_CALLBACK_EXCEPTION_LOG_MESSAGE</span><span class="p">,</span> <span class="bp">self</span><span class="p">)</span>
  153. <span class="k">def</span> <span class="nf">cancelled</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  154. <span class="k">with</span> <span class="bp">self</span><span class="o">.</span><span class="n">_condition</span><span class="p">:</span>
  155. <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_cancelled</span>
  156. <span class="k">def</span> <span class="nf">running</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  157. <span class="k">with</span> <span class="bp">self</span><span class="o">.</span><span class="n">_condition</span><span class="p">:</span>
  158. <span class="k">return</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_cancelled</span> <span class="ow">and</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_matured</span>
  159. <span class="k">def</span> <span class="nf">done</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  160. <span class="k">with</span> <span class="bp">self</span><span class="o">.</span><span class="n">_condition</span><span class="p">:</span>
  161. <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_cancelled</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">_matured</span>
  162. <span class="k">def</span> <span class="nf">result</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">timeout</span><span class="o">=</span><span class="kc">None</span><span class="p">):</span>
  163. <span class="bp">self</span><span class="o">.</span><span class="n">_block</span><span class="p">(</span><span class="n">timeout</span><span class="p">)</span>
  164. <span class="k">return</span> <span class="kc">None</span>
  165. <span class="k">def</span> <span class="nf">exception</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">timeout</span><span class="o">=</span><span class="kc">None</span><span class="p">):</span>
  166. <span class="bp">self</span><span class="o">.</span><span class="n">_block</span><span class="p">(</span><span class="n">timeout</span><span class="p">)</span>
  167. <span class="k">return</span> <span class="kc">None</span>
  168. <span class="k">def</span> <span class="nf">traceback</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">timeout</span><span class="o">=</span><span class="kc">None</span><span class="p">):</span>
  169. <span class="bp">self</span><span class="o">.</span><span class="n">_block</span><span class="p">(</span><span class="n">timeout</span><span class="p">)</span>
  170. <span class="k">return</span> <span class="kc">None</span>
  171. <span class="k">def</span> <span class="nf">add_done_callback</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">fn</span><span class="p">):</span>
  172. <span class="k">with</span> <span class="bp">self</span><span class="o">.</span><span class="n">_condition</span><span class="p">:</span>
  173. <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_cancelled</span> <span class="ow">and</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_matured</span><span class="p">:</span>
  174. <span class="bp">self</span><span class="o">.</span><span class="n">_done_callbacks</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">fn</span><span class="p">)</span>
  175. <span class="k">return</span>
  176. <span class="n">fn</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span>
  177. <span class="k">def</span> <span class="nf">start</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  178. <span class="k">with</span> <span class="bp">self</span><span class="o">.</span><span class="n">_condition</span><span class="p">:</span>
  179. <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="o">.</span><span class="n">subscribe</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_update</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>
  180. <span class="k">def</span> <span class="nf">__del__</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  181. <span class="k">with</span> <span class="bp">self</span><span class="o">.</span><span class="n">_condition</span><span class="p">:</span>
  182. <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_cancelled</span> <span class="ow">and</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_matured</span><span class="p">:</span>
  183. <span class="bp">self</span><span class="o">.</span><span class="n">_channel</span><span class="o">.</span><span class="n">unsubscribe</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_update</span><span class="p">)</span>
  184. <div class="viewcode-block" id="channel_ready_future"><a class="viewcode-back" href="../../../grpc.beta.html#grpc.beta.utilities.channel_ready_future">[docs]</a><span class="k">def</span> <span class="nf">channel_ready_future</span><span class="p">(</span><span class="n">channel</span><span class="p">):</span>
  185. <span class="sd">&quot;&quot;&quot;Creates a future.Future tracking when an implementations.Channel is ready.</span>
  186. <span class="sd"> Cancelling the returned future.Future does not tell the given</span>
  187. <span class="sd"> implementations.Channel to abandon attempts it may have been making to</span>
  188. <span class="sd"> connect; cancelling merely deactivates the return future.Future&#39;s</span>
  189. <span class="sd"> subscription to the given implementations.Channel&#39;s connectivity.</span>
  190. <span class="sd"> Args:</span>
  191. <span class="sd"> channel: An implementations.Channel.</span>
  192. <span class="sd"> Returns:</span>
  193. <span class="sd"> A future.Future that matures when the given Channel has connectivity</span>
  194. <span class="sd"> interfaces.ChannelConnectivity.READY.</span>
  195. <span class="sd"> &quot;&quot;&quot;</span>
  196. <span class="n">ready_future</span> <span class="o">=</span> <span class="n">_ChannelReadyFuture</span><span class="p">(</span><span class="n">channel</span><span class="p">)</span>
  197. <span class="n">ready_future</span><span class="o">.</span><span class="n">start</span><span class="p">()</span>
  198. <span class="k">return</span> <span class="n">ready_future</span></div>
  199. </pre></div>
  200. </div>
  201. </div>
  202. <footer>
  203. <hr/>
  204. <div role="contentinfo">
  205. <p>
  206. &copy; Copyright 2016, Author.
  207. </p>
  208. </div>
  209. Built with <a href="http://sphinx-doc.org/">Sphinx</a> using a <a href="https://github.com/snide/sphinx_rtd_theme">theme</a> provided by <a href="https://readthedocs.org">Read the Docs</a>.
  210. </footer>
  211. </div>
  212. </div>
  213. </section>
  214. </div>
  215. <script type="text/javascript">
  216. var DOCUMENTATION_OPTIONS = {
  217. URL_ROOT:'../../../',
  218. VERSION:'1.0.4',
  219. COLLAPSE_INDEX:false,
  220. FILE_SUFFIX:'.html',
  221. HAS_SOURCE: true
  222. };
  223. </script>
  224. <script type="text/javascript" src="../../../_static/jquery.js"></script>
  225. <script type="text/javascript" src="../../../_static/underscore.js"></script>
  226. <script type="text/javascript" src="../../../_static/doctools.js"></script>
  227. <script type="text/javascript" src="../../../_static/js/theme.js"></script>
  228. <script type="text/javascript">
  229. jQuery(function () {
  230. SphinxRtdTheme.StickyNav.enable();
  231. });
  232. </script>
  233. </body>
  234. </html>