del, small refacture into sock_result

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

diff --git a/lib/Redis.pm b/lib/Redis.pm
index bc65da9..c5597df 100644 (file)
--- a/lib/Redis.pm
+++ b/lib/Redis.pm
@@ -5,6 +5,7 @@ use strict;
 
 use IO::Socket::INET;
 use Data::Dump qw/dump/;
+use Carp qw/confess/;
 
 =head1 NAME
 
@@ -80,15 +81,15 @@ sub ping {
 
 =head2 set
 
-  $r->set( foo => 'bar' );
+  $r->set( foo => 'bar', $new );
 
 =cut
 
 sub set {
-       my ( $self, $k, $v ) = @_;
-       print $sock "SET $k " . length($v) . "\r\n$v\r\n";
+       my ( $self, $k, $v, $new ) = @_;
+       print $sock ( $new ? "SETNX" : "SET" ) . " $k " . length($v) . "\r\n$v\r\n";
        my $ok = <$sock>;
-       die dump($ok) unless $ok eq "+OK\r\n";
+       confess dump($ok) unless $ok eq "+OK\r\n";
 }
 
 =head2 get
@@ -111,7 +112,71 @@ sub get {
        return $v;
 }
 
+=head2 incr
 
+  $r->incr('counter');
+  $r->incr('tripplets', 3);
+
+=cut
+
+sub sock_result {
+       my $result = <$sock>;
+       warn "# result: ",dump( $result );
+       $result =~ s{\r\n$}{} || warn "can't find cr/lf";
+       return $result;
+}
+       
+
+sub incr {
+       my ( $self, $key, $value ) = @_;
+       if ( defined $value ) {
+               print $sock "INCRBY $key $value\r\n";
+       } else {
+               print $sock "INCR $key\r\n";
+       }
+       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();
+}
 
 =head1 AUTHOR