| #!/usr/bin/ruby |
| # ------------------------------------------------------------------------ |
| # Licensed to the Apache Software Foundation (ASF) under one or more |
| # contributor license agreements. See the NOTICE file distributed with |
| # this work for additional information regarding copyright ownership. |
| # The ASF licenses this file to You under the Apache License, Version 2.0 |
| # (the "License"); you may not use this file except in compliance with |
| # the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| # ------------------------------------------------------------------------ |
| |
| require 'rubygems' |
| require 'stomp' |
| |
| @conn = Stomp::Connection.open '', '', 'localhost', 61613, false |
| @count = 0 |
| |
| @conn.subscribe '/topic/event', { :ack =>"auto" } |
| while true |
| @msg = @conn.receive |
| if @msg.command == "MESSAGE" |
| if @msg.body == "SHUTDOWN" |
| exit 0 |
| |
| elsif @msg.body == "REPORT" |
| @diff = Time.now - @start |
| @body = "Received #{@count} in #{@diff} seconds"; |
| @conn.send '/queue/response', @body, {'persistent'=>'false'} |
| @count = 0; |
| else |
| if @count == 0 |
| @start = Time.now |
| end |
| |
| @count += 1; |
| if @count % 1000 == 0 |
| $stdout.print "Received #{@count} messages.\n" |
| end |
| end |
| else |
| $stdout.print "#{@msg.command}: #{@msg.body}\n" |
| end |
| end |
| @conn.disconnect |