blob: 3e5888249ca0dd9ba8578648e5e372c34d582e25 [file] [log] [blame]
<h1>client.rb</h1>
<div class="highlight"><pre><span></span><span class="nb">require</span> <span class="s1">&#39;qpid_proton&#39;</span>
<span class="nb">require</span> <span class="s1">&#39;optparse&#39;</span>
<span class="k">class</span> <span class="nc">Client</span> <span class="o">&lt;</span> <span class="no">Qpid</span><span class="o">::</span><span class="no">Proton</span><span class="o">::</span><span class="no">MessagingHandler</span>
<span class="k">def</span> <span class="nf">initialize</span><span class="p">(</span><span class="n">url</span><span class="p">,</span> <span class="n">address</span><span class="p">,</span> <span class="n">requests</span><span class="p">)</span>
<span class="k">super</span><span class="p">()</span>
<span class="vi">@url</span> <span class="o">=</span> <span class="n">url</span>
<span class="vi">@address</span> <span class="o">=</span> <span class="n">address</span>
<span class="vi">@requests</span> <span class="o">=</span> <span class="n">requests</span>
<span class="k">end</span>
<span class="k">def</span> <span class="nf">on_container_start</span><span class="p">(</span><span class="n">container</span><span class="p">)</span>
<span class="n">c</span> <span class="o">=</span> <span class="n">container</span><span class="o">.</span><span class="n">connect</span><span class="p">(</span><span class="vi">@url</span><span class="p">)</span>
<span class="vi">@sender</span> <span class="o">=</span> <span class="n">c</span><span class="o">.</span><span class="n">open_sender</span><span class="p">(</span><span class="vi">@address</span><span class="p">)</span>
<span class="vi">@receiver</span> <span class="o">=</span> <span class="n">c</span><span class="o">.</span><span class="n">open_receiver</span><span class="p">({</span><span class="ss">:dynamic</span> <span class="o">=&gt;</span> <span class="kp">true</span><span class="p">})</span>
<span class="k">end</span>
<span class="k">def</span> <span class="nf">next_request</span>
<span class="k">if</span> <span class="vi">@receiver</span><span class="o">.</span><span class="n">remote_source</span><span class="o">.</span><span class="n">address</span>
<span class="n">req</span> <span class="o">=</span> <span class="no">Qpid</span><span class="o">::</span><span class="no">Proton</span><span class="o">::</span><span class="no">Message</span><span class="o">.</span><span class="n">new</span>
<span class="n">req</span><span class="o">.</span><span class="n">reply_to</span> <span class="o">=</span> <span class="vi">@receiver</span><span class="o">.</span><span class="n">remote_source</span><span class="o">.</span><span class="n">address</span>
<span class="n">req</span><span class="o">.</span><span class="n">body</span> <span class="o">=</span> <span class="vi">@requests</span><span class="o">.</span><span class="n">first</span>
<span class="nb">puts</span> <span class="s2">&quot;-&gt; </span><span class="si">#{</span><span class="n">req</span><span class="o">.</span><span class="n">body</span><span class="si">}</span><span class="s2">&quot;</span>
<span class="vi">@sender</span><span class="o">.</span><span class="n">send</span><span class="p">(</span><span class="n">req</span><span class="p">)</span>
<span class="k">end</span>
<span class="k">end</span>
<span class="k">def</span> <span class="nf">on_receiver_open</span><span class="p">(</span><span class="n">receiver</span><span class="p">)</span>
<span class="n">next_request</span>
<span class="k">end</span>
<span class="k">def</span> <span class="nf">on_message</span><span class="p">(</span><span class="n">delivery</span><span class="p">,</span> <span class="n">message</span><span class="p">)</span>
<span class="nb">puts</span> <span class="s2">&quot;&lt;- </span><span class="si">#{</span><span class="n">message</span><span class="o">.</span><span class="n">body</span><span class="si">}</span><span class="s2">&quot;</span>
<span class="vi">@requests</span><span class="o">.</span><span class="n">delete_at</span><span class="p">(</span><span class="mi">0</span><span class="p">)</span>
<span class="k">if</span> <span class="o">!</span><span class="vi">@requests</span><span class="o">.</span><span class="n">empty?</span>
<span class="n">next_request</span>
<span class="k">else</span>
<span class="n">delivery</span><span class="o">.</span><span class="n">connection</span><span class="o">.</span><span class="n">close</span>
<span class="k">end</span>
<span class="k">end</span>
<span class="k">def</span> <span class="nf">on_transport_error</span><span class="p">(</span><span class="n">transport</span><span class="p">)</span>
<span class="k">raise</span> <span class="s2">&quot;Connection error: </span><span class="si">#{</span><span class="n">transport</span><span class="o">.</span><span class="n">condition</span><span class="si">}</span><span class="s2">&quot;</span>
<span class="k">end</span>
<span class="k">end</span>
<span class="no">REQUESTS</span> <span class="o">=</span> <span class="o">[</span><span class="s2">&quot;Twas brillig, and the slithy toves&quot;</span><span class="p">,</span>
<span class="s2">&quot;Did gire and gymble in the wabe.&quot;</span><span class="p">,</span>
<span class="s2">&quot;All mimsy were the borogroves,&quot;</span><span class="p">,</span>
<span class="s2">&quot;And the mome raths outgrabe.&quot;</span><span class="o">]</span>
<span class="k">if</span> <span class="no">ARGV</span><span class="o">.</span><span class="n">size</span> <span class="o">!=</span> <span class="mi">2</span>
<span class="no">STDERR</span><span class="o">.</span><span class="n">puts</span> <span class="s2">&quot;Usage: </span><span class="si">#{</span><span class="bp">__FILE__</span><span class="si">}</span><span class="s2"> URL ADDRESS</span>
<span class="s2">Connect to URL and send messages to ADDRESS&quot;</span>
<span class="k">return</span> <span class="mi">1</span>
<span class="k">end</span>
<span class="n">url</span><span class="p">,</span> <span class="n">address</span> <span class="o">=</span> <span class="no">ARGV</span>
<span class="no">Qpid</span><span class="o">::</span><span class="no">Proton</span><span class="o">::</span><span class="no">Container</span><span class="o">.</span><span class="n">new</span><span class="p">(</span><span class="no">Client</span><span class="o">.</span><span class="n">new</span><span class="p">(</span><span class="n">url</span><span class="p">,</span> <span class="n">address</span><span class="p">,</span> <span class="no">REQUESTS</span><span class="p">))</span><span class="o">.</span><span class="n">run</span>
</pre></div>
<p><a href="client.rb">Download this file</a></p>