llen, _sock_send

[perl-Redis.git] / lib / Redis.pm

diff --git a/lib/Redis.pm b/lib/Redis.pm
index 952b80b..5a62fbf 100644 (file)
--- a/lib/Redis.pm
+++ b/lib/Redis.pm
@@ -51,6 +51,43 @@ sub new {
        $self;
 }
 
+sub _sock_result {
+       my $result = <$sock>;
+       warn "# result: ",dump( $result );
+       $result =~ s{\r\n$}{} || warn "can't find cr/lf";
+       return $result;
+}
+
+sub _sock_result_bulk {
+       my $len = <$sock>;
+       warn "# len: ",dump($len);
+       return undef if $len eq "nil\r\n";
+       my $v;
+       read($sock, $v, $len) || die $!;
+       warn "# v: ",dump($v);
+       my $crlf;
+       read($sock, $crlf, 2); # skip cr/lf
+       return $v;
+}
+
+sub _sock_ok {
+       my $ok = <$sock>;
+       confess dump($ok) unless $ok eq "+OK\r\n";
+}
+
+sub _sock_send {
+       my $self = shift;
+       print $sock join(' ',@_) . "\r\n";
+       _sock_result();
+}
+
+sub _sock_send_bulk {
+       my ( $self, $command, $key, $value ) = @_;
+       print $sock "$command $key " . length($value) . "\r\n$value\r\n";
+       _sock_ok();
+}
+
+
 =head1 Connection Handling
 
 =head2 quit
@@ -86,10 +123,8 @@ sub ping {
 =cut
 
 sub set {
-       my ( $self, $k, $v, $new ) = @_;
-       print $sock ( $new ? "SETNX" : "SET" ) . " $k " . length($v) . "\r\n$v\r\n";
-       my $ok = <$sock>;
-       confess dump($ok) unless $ok eq "+OK\r\n";
+       my ( $self, $key, $value, $new ) = @_;
+       $self->_sock_send_bulk( "SET" . ( $new ? 'NX' : '' ), $key, $value );
 }
 
 =head2 get
@@ -101,15 +136,7 @@ sub set {
 sub get {
        my ( $self, $k ) = @_;
        print $sock "GET $k\r\n";
-       my $len = <$sock>;
-#      warn "# len: ",dump($len);
-       return undef if $len eq "nil\r\n";
-       my $v;
-       read($sock, $v, $len) || die $!;
-#      warn "# v: ",dump($v);
-       my $crlf;
-       read($sock, $crlf, 2); # skip cr/lf
-       return $v;
+       _sock_result_bulk();
 }
 
 =head2 incr
@@ -119,6 +146,8 @@ sub get {
 
 =cut
 
+       
+
 sub incr {
        my ( $self, $key, $value ) = @_;
        if ( defined $value ) {
@@ -126,9 +155,145 @@ sub incr {
        } else {
                print $sock "INCR $key\r\n";
        }
-       my $count = <$sock>;
-       warn "# $key = $count";
-       return $count;
+       _sock_result();
+}
+
+=head2 decr
+
+  $r->decr('counter');
+  $r->decr('tripplets', 3);
+
+=cut
+
+sub decr {
+       my ( $self, $key, $value ) = @_;
+       if ( defined $value ) {
+               print $sock "DECRBY $key $value\r\n";
+       } else {
+               print $sock "DECR $key\r\n";
+       }
+       _sock_result();
+}
+
+=head2 exists
+
+  $r->exists( 'key' ) && print "got key!";
+
+=cut
+
+sub exists {
+       my ( $self, $key ) = @_;
+       print $sock "EXISTS $key\r\n";
+       _sock_result();
+}
+
+=head2 del
+
+  $r->del( 'key' ) || warn "key doesn't exist";
+
+=cut
+
+sub del {
+       my ( $self, $key ) = @_;
+       print $sock "DEL $key\r\n";
+       _sock_result();
+}
+
+=head2 type
+
+  $r->type( 'key' ); # = string
+
+=cut
+
+sub type {
+       my ( $self, $key ) = @_;
+       print $sock "TYPE $key\r\n";
+       _sock_result();
+}
+
+=head1 Commands operating on the key space
+
+=head2 keys
+
+  my @keys = $r->keys( '*glob_pattern*' );
+
+=cut
+
+sub keys {
+       my ( $self, $glob ) = @_;
+       print $sock "KEYS $glob\r\n";
+       return split(/\s/, _sock_result_bulk());
+}
+
+=head2 randomkey
+
+  my $key = $r->randomkey;
+
+=cut
+
+sub randomkey {
+       my ( $self ) = @_;
+       print $sock "RANDOMKEY\r\n";
+       _sock_result();
+}
+
+=head2 rename
+
+  my $ok = $r->rename( 'old-key', 'new-key', $new );
+
+=cut
+
+sub rename {
+       my ( $self, $old, $new, $nx ) = @_;
+       print $sock "RENAME" . ( $nx ? 'NX' : '' ) . " $old $new\r\n";
+       _sock_ok();
+}
+
+=head2 dbsize
+
+  my $nr_keys = $r->dbsize;
+
+=cut
+
+sub dbsize {
+       my ( $self ) = @_;
+       print $sock "DBSIZE\r\n";
+       _sock_result();
+}
+
+=head1 Commands operating on lists
+
+=head2 rpush
+
+  $r->rpush( $key, $value );
+
+=cut
+
+sub rpush {
+       my ( $self, $key, $value ) = @_;
+       $self->_sock_send_bulk('RPUSH', $key, $value);
+}
+
+=head2 lpush
+
+  $r->lpush( $key, $value );
+
+=cut
+
+sub lpush {
+       my ( $self, $key, $value ) = @_;
+       $self->_sock_send_bulk('LPUSH', $key, $value);
+}
+
+=head2 llen
+
+  $r->llen( $key );
+
+=cut
+
+sub llen {
+       my ( $self, $key ) = @_;
+       $self->_sock_send( 'llen', $key );
 }
 
 =head1 AUTHOR