| |
| <h1>server.rb</h1> |
| <div class="highlight"><pre><span></span><span class="nb">require</span> <span class="s1">'qpid_proton'</span> |
| <span class="nb">require</span> <span class="s1">'optparse'</span> |
| |
| <span class="k">class</span> <span class="nc">Server</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">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="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">@senders</span> <span class="o">=</span> <span class="p">{}</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="n">c</span><span class="o">.</span><span class="n">open_receiver</span><span class="p">(</span><span class="vi">@address</span><span class="p">)</span> |
| <span class="vi">@relay</span> <span class="o">=</span> <span class="kp">nil</span> |
| <span class="k">end</span> |
| |
| <span class="k">def</span> <span class="nf">on_connection_open</span><span class="p">(</span><span class="n">connection</span><span class="p">)</span> |
| <span class="k">if</span> <span class="n">connection</span><span class="o">.</span><span class="n">offered_capabilities</span> <span class="o">&&</span> |
| <span class="n">connection</span><span class="o">.</span><span class="n">offered_capabilities</span><span class="o">.</span><span class="n">include?</span><span class="p">(</span><span class="ss">:"ANONYMOUS-RELAY"</span><span class="p">)</span> |
| <span class="vi">@relay</span> <span class="o">=</span> <span class="n">connection</span><span class="o">.</span><span class="n">open_sender</span><span class="p">({</span><span class="ss">:target</span> <span class="o">=></span> <span class="kp">nil</span><span class="p">})</span> |
| <span class="k">end</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="k">return</span> <span class="k">unless</span> <span class="n">message</span><span class="o">.</span><span class="n">reply_to</span> <span class="c1"># Not a request message</span> |
| <span class="nb">puts</span> <span class="s2">"<- </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">"</span> |
| <span class="k">unless</span> <span class="p">(</span><span class="n">sender</span> <span class="o">=</span> <span class="vi">@relay</span><span class="p">)</span> |
| <span class="n">sender</span> <span class="o">=</span> <span class="p">(</span><span class="vi">@senders</span><span class="o">[</span><span class="n">message</span><span class="o">.</span><span class="n">reply_to</span><span class="o">]</span> <span class="o">||=</span> <span class="n">delivery</span><span class="o">.</span><span class="n">connection</span><span class="o">.</span><span class="n">open_sender</span><span class="p">(</span><span class="n">message</span><span class="o">.</span><span class="n">reply_to</span><span class="p">))</span> |
| <span class="k">end</span> |
| <span class="n">reply</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">reply</span><span class="o">.</span><span class="n">address</span> <span class="o">=</span> <span class="n">message</span><span class="o">.</span><span class="n">reply_to</span> |
| <span class="n">reply</span><span class="o">.</span><span class="n">body</span> <span class="o">=</span> <span class="n">message</span><span class="o">.</span><span class="n">body</span><span class="o">.</span><span class="n">upcase</span> |
| <span class="nb">puts</span> <span class="s2">"-> </span><span class="si">#{</span><span class="n">reply</span><span class="o">.</span><span class="n">body</span><span class="si">}</span><span class="s2">"</span> |
| <span class="n">reply</span><span class="o">.</span><span class="n">correlation_id</span> <span class="o">=</span> <span class="n">message</span><span class="o">.</span><span class="n">correlation_id</span> |
| <span class="n">sender</span><span class="o">.</span><span class="n">send</span><span class="p">(</span><span class="n">reply</span><span class="p">)</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">"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">"</span> |
| <span class="k">end</span> |
| <span class="k">end</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">"Usage: </span><span class="si">#{</span><span class="bp">__FILE__</span><span class="si">}</span><span class="s2"> URL ADDRESS</span> |
| <span class="s2">Server listening on URL, reply to messages to ADDRESS"</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">Server</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="o">.</span><span class="n">run</span> |
| </pre></div> |
| |
| <p><a href="server.rb">Download this file</a></p> |