Move QUIT command out of the common case
[perl-Redis.git] / lib / Redis.pm
index 4c46a66..ca065b9 100644 (file)
@@ -4,78 +4,391 @@ use warnings;
 use strict;
 
 use IO::Socket::INET;
-use Data::Dump qw/dump/;
+use Data::Dumper;
+use Carp qw/confess/;
+use Encode;
 
 =head1 NAME
 
-Redis - The great new Redis!
+Redis - perl binding for Redis database
 
 =cut
 
-our $VERSION = '0.01';
+our $VERSION = '1.2001';
 
 
-=head1 SYNOPSIS
+=head1 DESCRIPTION
 
 Pure perl bindings for L<http://code.google.com/p/redis/>
 
-    use Redis;
-
-    my $r = Redis->new();
-
+This version supports protocol 1.2 or later of Redis available at
 
+L<git://github.com/antirez/redis>
 
+This documentation
+lists commands which are exercised in test suite, but
+additinal commands will work correctly since protocol
+specifies enough information to support almost all commands
+with same peace of code with a little help of C<AUTOLOAD>.
 
 =head1 FUNCTIONS
 
 =head2 new
 
-=cut
+  my $r = Redis->new; # $ENV{REDIS_SERVER} or 127.0.0.1:6379
+
+  my $r = Redis->new( server => '192.168.0.1:6379', debug = 0 );
 
-our $sock;
-my $server = '127.0.0.1:6379';
+=cut
 
 sub new {
        my $class = shift;
-       my $self = {};
-       bless($self, $class);
+       my $self  = {@_};
 
-       warn "# opening socket to $server";
+       $self->{debug} ||= $ENV{REDIS_DEBUG};
+       $self->{encoding} ||= 'utf8';    ## default to lax utf8
 
-       $sock ||= IO::Socket::INET->new(
-               PeerAddr => $server,
+       $self->{server} ||= $ENV{REDIS_SERVER} || '127.0.0.1:6379';
+       $self->{sock} = IO::Socket::INET->new(
+               PeerAddr => $self->{server},
                Proto => 'tcp',
-       ) || die $!;
+       ) || confess("Could not connect to Redis server at $self->{server}: $!");
 
-       $self;
+       return bless($self, $class);
 }
 
-=head1 Connection Handling
+# we don't want DESTROY to fallback into AUTOLOAD
+sub DESTROY {}
 
-=head2 quit
 
-  $r->quit;
+### Deal with common, general case, Redis commands
+our $AUTOLOAD;
+sub AUTOLOAD {
+       my $self = shift;
+       my $sock = $self->{sock} || confess("Not connected to any server");
+       my $enc = $self->{encoding};
+       my $deb = $self->{debug};
+
+       my $command = $AUTOLOAD;
+       $command =~ s/.*://;
+
+       $self->__send_command($command, @_);
+
+       my $result = <$sock> || confess("Can't read socket: $!");
+       my $type = substr($result,0,1);
+       $result = substr($result,1,-2);
+
+       $result = decode($enc, $result) if $enc;
+       warn "[RECV] '$type$result'" if $deb;
+
+       if ( $command eq 'info' ) {
+               my $hash;
+               foreach my $l ( split(/\r\n/, $self->__read_bulk($result) ) ) {
+                       my ($n,$v) = split(/:/, $l, 2);
+                       $hash->{$n} = $v;
+               }
+               return $hash;
+       } elsif ( $command eq 'keys' ) {
+               return $self->__read_multi_bulk($result)
+                       if $type eq '*';
+               my $keys = $self->__read_bulk($result);
+               return split(/\s/, $keys) if $keys;
+               return;
+       }
+
+       if ( $type eq '-' ) {
+               confess "[$command] $result";
+       } elsif ( $type eq '+' ) {
+               return $result;
+       } elsif ( $type eq '$' ) {
+               return $self->__read_bulk($result);
+       } elsif ( $type eq '*' ) {
+               return $self->__read_multi_bulk($result);
+       } elsif ( $type eq ':' ) {
+               return $result; # FIXME check if int?
+       } else {
+               confess "unknown type: $type", $self->__read_line();
+       }
+}
 
-=cut
+
+### Commands with extra logic
 
 sub quit {
-       my $self = shift;
+  my ($self) = @_;
+
+  $self->__send_command('QUIT');
 
-       close( $sock ) || warn $!;
+  close(delete $self->{sock}) || confess("Can't close socket: $!");
+  return 1;
 }
 
-=head2 ping
 
-       $r->ping || die "no server?";
+### Socket operations
 
-=cut
+sub __send_command {
+  my $self = shift;
+  my $cmd  = uc(shift);
+  my $enc  = $self->{encoding};
+  my $deb  = $self->{debug};
 
-sub ping {
-       print $sock "PING\r\n";
-       my $pong = <$sock>;
-       die "ping failed, got ", dump($pong) unless $pong eq "+PONG\r\n";
+  warn "[SEND] $cmd ", Dumper([@_]) if $deb;
+
+  ## Encode command using multi-bulk format
+  my $n_elems = scalar(@_) + 1;
+  my $buf     = "\*$n_elems\r\n";
+  for my $elem ($cmd, @_) {
+    my $bin = $enc ? encode($enc, $elem) : $elem;
+    $buf .= defined($bin) ? '$' . length($bin) . "\r\n$bin\r\n" : "\$-1\r\n";
+  }
+
+  ## Send command, take care for partial writes
+  warn "[SEND RAW] $buf" if $deb;
+  my $sock = $self->{sock} || confess("Not connected to any server");
+  while ($buf) {
+    my $len = syswrite $sock, $buf, length $buf;
+    confess("Could not write to Redis server: $!")
+      unless $len;
+    substr $buf, 0, $len, "";
+  }
+
+  return;
+}
+
+sub __read_bulk {
+       my ($self,$len) = @_;
+       return if $len < 0;
+
+       my $enc = $self->{encoding};
+       my $v = '';
+       if ( $len > 0 ) {
+               read($self->{sock}, $v, $len) || confess("Could not read from sock: $!");
+               $v = decode($enc, $v) if $enc;
+       }
+       my $crlf;
+       read($self->{sock}, $crlf, 2); # skip cr/lf
+
+       warn "[PARSE] read_bulk ".Dumper($v) if $self->{debug};
+       return $v;
+}
+
+sub __read_multi_bulk {
+       my ($self,$size) = @_;
+       return if $size <= 0;
+
+       my $sock = $self->{sock};
+       my $deb = $self->{debug};
+       my $enc = $self->{encoding};
+  my @list;    
+       while ($size--) {
+               my $v = $self->__read_bulk( substr(<$sock>,1,-2) );
+               $v = decode($enc, $v) if $enc;
+               warn "  [PARSE] read_multi_bulk ($size) ".Dumper($v) if $deb;
+               push @list, $v;
+       }
+
+       warn "[PARSE] multi_bulk ".Dumper( \@list ) if $deb;
+       return @list;
 }
 
+1;
+
+__END__
+
+=head1 Connection Handling
+
+=head2 quit
+
+  $r->quit;
+
+=head2 ping
+
+  $r->ping || die "no server?";
+
+=head1 Commands operating on string values
+
+=head2 set
+
+  $r->set( foo => 'bar' );
+
+  $r->setnx( foo => 42 );
+
+=head2 get
+
+  my $value = $r->get( 'foo' );
+
+=head2 mget
+
+  my @values = $r->mget( 'foo', 'bar', 'baz' );
+
+=head2 incr
+
+  $r->incr('counter');
+
+  $r->incrby('tripplets', 3);
+
+=head2 decr
+
+  $r->decr('counter');
+
+  $r->decrby('tripplets', 3);
+
+=head2 exists
+
+  $r->exists( 'key' ) && print "got key!";
+
+=head2 del
+
+  $r->del( 'key' ) || warn "key doesn't exist";
+
+=head2 type
+
+  $r->type( 'key' ); # = string
+
+=head1 Commands operating on the key space
+
+=head2 keys
+
+  my @keys = $r->keys( '*glob_pattern*' );
+
+=head2 randomkey
+
+  my $key = $r->randomkey;
+
+=head2 rename
+
+  my $ok = $r->rename( 'old-key', 'new-key', $new );
+
+=head2 dbsize
+
+  my $nr_keys = $r->dbsize;
+
+=head1 Commands operating on lists
+
+See also L<Redis::List> for tie interface.
+
+=head2 rpush
+
+  $r->rpush( $key, $value );
+
+=head2 lpush
+
+  $r->lpush( $key, $value );
+
+=head2 llen
+
+  $r->llen( $key );
+
+=head2 lrange
+
+  my @list = $r->lrange( $key, $start, $end );
+
+=head2 ltrim
+
+  my $ok = $r->ltrim( $key, $start, $end );
+
+=head2 lindex
+
+  $r->lindex( $key, $index );
+
+=head2 lset
+
+  $r->lset( $key, $index, $value );
+
+=head2 lrem
+
+  my $modified_count = $r->lrem( $key, $count, $value );
+
+=head2 lpop
+
+  my $value = $r->lpop( $key );
+
+=head2 rpop
+
+  my $value = $r->rpop( $key );
+
+=head1 Commands operating on sets
+
+=head2 sadd
+
+  $r->sadd( $key, $member );
+
+=head2 srem
+
+  $r->srem( $key, $member );
+
+=head2 scard
+
+  my $elements = $r->scard( $key );
+
+=head2 sismember
+
+  $r->sismember( $key, $member );
+
+=head2 sinter
+
+  $r->sinter( $key1, $key2, ... );
+
+=head2 sinterstore
+
+  my $ok = $r->sinterstore( $dstkey, $key1, $key2, ... );
+
+=head1 Multiple databases handling commands
+
+=head2 select
+
+  $r->select( $dbindex ); # 0 for new clients
+
+=head2 move
+
+  $r->move( $key, $dbindex );
+
+=head2 flushdb
+
+  $r->flushdb;
+
+=head2 flushall
+
+  $r->flushall;
+
+=head1 Sorting
+
+=head2 sort
+
+  $r->sort("key BY pattern LIMIT start end GET pattern ASC|DESC ALPHA');
+
+=head1 Persistence control commands
+
+=head2 save
+
+  $r->save;
+
+=head2 bgsave
+
+  $r->bgsave;
+
+=head2 lastsave
+
+  $r->lastsave;
+
+=head2 shutdown
+
+  $r->shutdown;
+
+=head1 Remote server control commands
+
+=head2 info
+
+  my $info_hash = $r->info;
+
+=head1 ENCODING
+
+Since Redis knows nothing about encoding, we are forcing utf-8 flag on all data received from Redis.
+This change is introduced in 1.2001 version.
+
+This allows us to round-trip utf-8 encoded characters correctly, but might be problem if you push
+binary junk into Redis and expect to get it back without utf-8 flag turned on.
+
 =head1 AUTHOR
 
 Dobrica Pavlinusic, C<< <dpavlin at rot13.org> >>
@@ -94,6 +407,8 @@ automatically be notified of progress on your bug as I make changes.
 You can find documentation for this module with the perldoc command.
 
     perldoc Redis
+       perldoc Redis::List
+       perldoc Redis::Hash
 
 
 You can also look for information at:
@@ -124,7 +439,7 @@ L<http://search.cpan.org/dist/Redis>
 
 =head1 COPYRIGHT & LICENSE
 
-Copyright 2009 Dobrica Pavlinusic, all rights reserved.
+Copyright 2009-2010 Dobrica Pavlinusic, all rights reserved.
 
 This program is free software; you can redistribute it and/or modify it
 under the same terms as Perl itself.