blob: c012e31f9d4a4b075a491f271c75ac7dac93a00f [file] [log] [blame]
#
# 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.
#
$:.unshift File.join(File.dirname(__FILE__), "..", "lib")
require 'qpid'
require 'optparse'
options = {
:broker => "127.0.0.1",
:address => "",
:timeout => 0,
:count => 1,
:properties => {},
:content => nil,
:mapped => {}
}
opts = OptionParser.new do |opts|
opts.banner = "Usage: spout.rb [OPTIONS] ADDRESS"
opts.on("-h", "--help",
"show this message") do |help|
puts opts
exit
end
opts.on("-b","--broker VALUE",
"url of broker to connect to ") do |broker|
options[:broker] = broker
end
opts.on("-t", "--timeout VALUE", Integer,
"exit after the specified time") do |timeout|
options[:timeout] = Qpid::Messaging::Duration.new timeout * 1000
end
opts.on("-c", "--count VALUE", Integer,
"stop after count messages have been sent, zero disables") do |count|
options[:count] = count
end
opts.on("-i", "--id VALUE",
"use the supplied id instead of generating one") do |id|
options[:id] = id
end
opts.on("--reply-to VALUE",
"specify reply-to address") do |replyto|
options[:replyto] = replyto
end
opts.on("-P", "--property VALUE",
"specify message property") do |property|
name = property.split(/=/)[0]
value = property.split(/=/)[1]
options[:properties][name] = value
end
opts.on("-M", "--map VALUE",
"specify entry for map content") do |mapped|
name = mapped.split(/=/)[0]
value = mapped.split(/=/)[1]
options[:mapped][name] = value
end
opts.on("--content VALUE",
"specify textual content") do |content|
options[:content] = content
end
opts.on(nil, "--connection-options VALUE",
"connection options string in the form {name1:value1, name2:value2}") do |conopts|
options[:connection_options] = conopts
end
end
begin
opts.parse!(ARGV)
rescue => error
opts.parse(["-h"])
end
# now get the non-arg options
options[:address] = ARGV[0] unless ARGV[0].nil?
connection = Qpid::Messaging::Connection.new options[:broker], options[:connection_options]
connection.open
session = connection.create_session
sender = session.create_sender options[:address]
message = Qpid::Messaging::Message.new
options[:properties].each_key {|key| message.properties[key] = options[:properties][key]}
(1..options[:count]).each do |count|
if !options[:mapped].keys.empty?
message.content = options[:mapped]
elsif options[:content]
message.content = options[:content]
end
message.content = options[:content] unless options[:content].nil?
message.properties["spout-id"] = "#{count}"
message.reply_to = options[:replyto] unless options[:replyto].nil? || options[:replyto].empty?
sender.send message
end
# session.sync
connection.close