dinislage/perl-anyevent-nsq
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve consumer.pl:

Browse Source 
* use Getopt::Long to enable/disable help, verbose and print mode;
* add proper usage message;
* keep track of total messages received.

Signed-off-by: Pedro Melo <melo@simplicidade.org>
This commit is contained in:
Pedro Melo
2014-07-21 05:10:39 +01:00
parent 71c97d864c
commit 5bd01740ee
1 changed files with 36 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
examples/consumer.pl
View File

@@ -6,24 +6,53 @@ use FindBin;
use lib "$FindBin::Bin/../lib"; use lib "$FindBin::Bin/../lib";
use AnyEvent; use AnyEvent;
use AnyEvent::NSQ::Reader; use AnyEvent::NSQ::Reader;
use Getopt::Long;
my ($topic, $channel) = @ARGV; my ($topic, $channel, $help, $verbose, $print);
die "Usage: consumer.pl topic channel\n" unless $topic and $channel; GetOptions('help' => \$help, 'verbose' => \$verbose, 'print' => \$print) or usage();
usage() if $help;
($topic, $channel) = @ARGV;
usage("topic and channel are required parameters") unless $topic and $channel;
my $cv = AE::cv; my $cv = AE::cv;
my $c = 1; ## return undef => mark_as_done_msg()
my $message_cb = $print ? sub { print "$_[1]{message}\n"; return } : sub {return};
my $t = 0;
my $r = AnyEvent::NSQ::Reader->new( my $r = AnyEvent::NSQ::Reader->new(
topic => $topic, topic => $topic,
channel => $channel, channel => $channel,
nsqd_tcp_addresses => '127.0.0.1', nsqd_tcp_addresses => '127.0.0.1',
client_id => "${channel}_consumer/pid_$$", client_id => "${channel}_consumer/pid_$$",
# message_cb => sub { print STDERR "$c: $_[1]{message}\n"; $c++; return }, ## return undef => mark_as_done_msg() message_cb => sub { $t++; $message_cb->(@_, $t) },
message_cb => sub {return}, ## return undef => mark_as_done_msg()
error_cb => sub { warn "$_[1]\n" }, error_cb => sub { warn "$_[1]\n" if $verbose },
disconnect_cb => sub { warn "Got disconnected... exiting...\n"; $cv->send }, disconnect_cb => sub { warn "Got disconnected after $t total messages... exiting...\n" if $verbose; $cv->send },
); );
$cv->recv; $cv->recv;
sub usage {
print "Error: @_\n" if @_;
print <<" EOU";
Usage: consumer.pl [--help|-h] [--print|-p] topic channel
Consume messages from channel <channel> linked to topic <topic>.
Topic and channel parameters are mandatory.
Options:
--help or -h Prints this message and exits
--verbose or -v Writes debug information about connections,
disconnections and errors
--print or -p Prints each received message
EOU
exit(1);
}