rpush, refactor _sock_send_bulk to expect +OK
[perl-Redis.git] / lib / Redis.pm
1 package Redis;
2
3 use warnings;
4 use strict;
5
6 use IO::Socket::INET;
7 use Data::Dump qw/dump/;
8 use Carp qw/confess/;
9
10 =head1 NAME
11
12 Redis - The great new Redis!
13
14 =cut
15
16 our $VERSION = '0.01';
17
18
19 =head1 SYNOPSIS
20
21 Pure perl bindings for L<http://code.google.com/p/redis/>
22
23     use Redis;
24
25     my $r = Redis->new();
26
27
28
29
30 =head1 FUNCTIONS
31
32 =head2 new
33
34 =cut
35
36 our $sock;
37 my $server = '127.0.0.1:6379';
38
39 sub new {
40         my $class = shift;
41         my $self = {};
42         bless($self, $class);
43
44         warn "# opening socket to $server";
45
46         $sock ||= IO::Socket::INET->new(
47                 PeerAddr => $server,
48                 Proto => 'tcp',
49         ) || die $!;
50
51         $self;
52 }
53
54 sub _sock_result {
55         my $result = <$sock>;
56         warn "# result: ",dump( $result );
57         $result =~ s{\r\n$}{} || warn "can't find cr/lf";
58         return $result;
59 }
60
61 sub _sock_result_bulk {
62         my $len = <$sock>;
63         warn "# len: ",dump($len);
64         return undef if $len eq "nil\r\n";
65         my $v;
66         read($sock, $v, $len) || die $!;
67         warn "# v: ",dump($v);
68         my $crlf;
69         read($sock, $crlf, 2); # skip cr/lf
70         return $v;
71 }
72
73 sub _sock_ok {
74         my $ok = <$sock>;
75         confess dump($ok) unless $ok eq "+OK\r\n";
76 }
77
78 sub _sock_send_bulk {
79         my ( $self, $command, $key, $value ) = @_;
80         print $sock "$command $key " . length($value) . "\r\n$value\r\n";
81         _sock_ok();
82 }
83
84
85 =head1 Connection Handling
86
87 =head2 quit
88
89   $r->quit;
90
91 =cut
92
93 sub quit {
94         my $self = shift;
95
96         close( $sock ) || warn $!;
97 }
98
99 =head2 ping
100
101   $r->ping || die "no server?";
102
103 =cut
104
105 sub ping {
106         print $sock "PING\r\n";
107         my $pong = <$sock>;
108         die "ping failed, got ", dump($pong) unless $pong eq "+PONG\r\n";
109 }
110
111 =head1 Commands operating on string values
112
113 =head2 set
114
115   $r->set( foo => 'bar', $new );
116
117 =cut
118
119 sub set {
120         my ( $self, $key, $value, $new ) = @_;
121         $self->_sock_send_bulk( "SET" . ( $new ? 'NX' : '' ), $key, $value );
122 }
123
124 =head2 get
125
126   my $value = $r->get( 'foo' );
127
128 =cut
129
130 sub get {
131         my ( $self, $k ) = @_;
132         print $sock "GET $k\r\n";
133         _sock_result_bulk();
134 }
135
136 =head2 incr
137
138   $r->incr('counter');
139   $r->incr('tripplets', 3);
140
141 =cut
142
143         
144
145 sub incr {
146         my ( $self, $key, $value ) = @_;
147         if ( defined $value ) {
148                 print $sock "INCRBY $key $value\r\n";
149         } else {
150                 print $sock "INCR $key\r\n";
151         }
152         _sock_result();
153 }
154
155 =head2 decr
156
157   $r->decr('counter');
158   $r->decr('tripplets', 3);
159
160 =cut
161
162 sub decr {
163         my ( $self, $key, $value ) = @_;
164         if ( defined $value ) {
165                 print $sock "DECRBY $key $value\r\n";
166         } else {
167                 print $sock "DECR $key\r\n";
168         }
169         _sock_result();
170 }
171
172 =head2 exists
173
174   $r->exists( 'key' ) && print "got key!";
175
176 =cut
177
178 sub exists {
179         my ( $self, $key ) = @_;
180         print $sock "EXISTS $key\r\n";
181         _sock_result();
182 }
183
184 =head2 del
185
186   $r->del( 'key' ) || warn "key doesn't exist";
187
188 =cut
189
190 sub del {
191         my ( $self, $key ) = @_;
192         print $sock "DEL $key\r\n";
193         _sock_result();
194 }
195
196 =head2 type
197
198   $r->type( 'key' ); # = string
199
200 =cut
201
202 sub type {
203         my ( $self, $key ) = @_;
204         print $sock "TYPE $key\r\n";
205         _sock_result();
206 }
207
208 =head1 Commands operating on the key space
209
210 =head2 keys
211
212   my @keys = $r->keys( '*glob_pattern*' );
213
214 =cut
215
216 sub keys {
217         my ( $self, $glob ) = @_;
218         print $sock "KEYS $glob\r\n";
219         return split(/\s/, _sock_result_bulk());
220 }
221
222 =head2 randomkey
223
224   my $key = $r->randomkey;
225
226 =cut
227
228 sub randomkey {
229         my ( $self ) = @_;
230         print $sock "RANDOMKEY\r\n";
231         _sock_result();
232 }
233
234 =head2 rename
235
236   my $ok = $r->rename( 'old-key', 'new-key', $new );
237
238 =cut
239
240 sub rename {
241         my ( $self, $old, $new, $nx ) = @_;
242         print $sock "RENAME" . ( $nx ? 'NX' : '' ) . " $old $new\r\n";
243         _sock_ok();
244 }
245
246 =head2 dbsize
247
248   my $nr_keys = $r->dbsize;
249
250 =cut
251
252 sub dbsize {
253         my ( $self ) = @_;
254         print $sock "DBSIZE\r\n";
255         _sock_result();
256 }
257
258 =head1 Commands operating on lists
259
260 =head2 rpush
261
262   $r->rpush( $key, $value );
263
264 =cut
265
266 sub rpush {
267         my ( $self, $key, $value ) = @_;
268         $self->_sock_send_bulk('RPUSH', $key, $value);
269 }
270
271 =head1 AUTHOR
272
273 Dobrica Pavlinusic, C<< <dpavlin at rot13.org> >>
274
275 =head1 BUGS
276
277 Please report any bugs or feature requests to C<bug-redis at rt.cpan.org>, or through
278 the web interface at L<http://rt.cpan.org/NoAuth/ReportBug.html?Queue=Redis>.  I will be notified, and then you'll
279 automatically be notified of progress on your bug as I make changes.
280
281
282
283
284 =head1 SUPPORT
285
286 You can find documentation for this module with the perldoc command.
287
288     perldoc Redis
289
290
291 You can also look for information at:
292
293 =over 4
294
295 =item * RT: CPAN's request tracker
296
297 L<http://rt.cpan.org/NoAuth/Bugs.html?Dist=Redis>
298
299 =item * AnnoCPAN: Annotated CPAN documentation
300
301 L<http://annocpan.org/dist/Redis>
302
303 =item * CPAN Ratings
304
305 L<http://cpanratings.perl.org/d/Redis>
306
307 =item * Search CPAN
308
309 L<http://search.cpan.org/dist/Redis>
310
311 =back
312
313
314 =head1 ACKNOWLEDGEMENTS
315
316
317 =head1 COPYRIGHT & LICENSE
318
319 Copyright 2009 Dobrica Pavlinusic, all rights reserved.
320
321 This program is free software; you can redistribute it and/or modify it
322 under the same terms as Perl itself.
323
324
325 =cut
326
327 1; # End of Redis