X-Git-Url: http://git.ieval.ro/?a=blobdiff_plain;f=lib%2FGruntmaster%2FData.pm;h=1b2492cc28c6a07dc927edf30d4039ec9fe1e1eb;hb=0b282e58119cc0dbce678ef56eab20cc352557b5;hp=544a6d060613d816a305c5e3ea6700a351523116;hpb=3da9c3c228106c819530a0b8e4c646981e6f71c9;p=plack-app-gruntmaster.git

diff --git a/lib/Gruntmaster/Data.pm b/lib/Gruntmaster/Data.pm
index 544a6d0..1b2492c 100644
--- a/lib/Gruntmaster/Data.pm
+++ b/lib/Gruntmaster/Data.pm
@@ -9,6 +9,7 @@ use Redis;
 
 our $contest;
 my $redis = Redis->new;
+my $pubsub = Redis->new;
 
 sub dynsub{
 	no strict 'refs';
@@ -16,8 +17,12 @@ sub dynsub{
 }
 
 BEGIN {
-	for my $cmd (qw/multi exec smembers get hget hset sadd srem incr hmset hsetnx/) {
-		dynsub uc $cmd, sub { say "Arguments to \U$cmd: ", join ', ', @_; $redis->$cmd(@_) };
+	for my $cmd (qw/multi exec smembers get hget hdel hset sadd srem incr hmset hsetnx publish del/) {
+		dynsub uc $cmd, sub { $redis->$cmd(@_) };
+	}
+
+	for my $cmd (qw/subscribe wait_for_messages/) {
+		dynsub uc $cmd, sub { $pubsub->$cmd(@_) };
 	}
 }
 
@@ -28,6 +33,7 @@ sub rexec				()		{ EXEC }
 
 sub problems			()		{ SMEMBERS cp . 'problem' }
 sub contests			()		{ SMEMBERS cp . 'contest' }
+sub users				()		{ SMEMBERS cp . 'user' }
 sub jobcard				()		{ GET cp . 'job' }
 
 sub job_results			(_)		{ decode_json HGET cp . "job.$_[0]", 'results' }
@@ -46,6 +52,11 @@ sub defhash{
 		dynsub "set_${name}_$key", sub ($$) { HSET cp . "$name.$_[0]", $key, $_[1] };
 	}
 
+	dynsub "edit_$name", sub {
+		my ($key, %values) = @_;
+		HMSET cp . "$name.$key", %values;
+	};
+
 	dynsub "insert_$name", sub {
 		my ($key, %values) = @_;
 		SADD cp . $name, $key or return;
@@ -54,19 +65,36 @@ sub defhash{
 	dynsub "remove_$name", sub (_) {
 		my $key = shift;
 		SREM cp . $name, $key;
+		DEL cp . "$name.$key";
 	};
 
 	dynsub "push_$name", sub {
 		my $nr = INCR cp . $name;
 		HMSET cp . "$name.$nr", @_;
+		$nr
 	};
 }
 
-defhash problem => qw/name level statement/;
+defhash problem => qw/name level statement owner author/;
 defhash contest => qw/start end name owner/;
-defhash job => qw/date file name private problem result result_text user/;
+defhash job => qw/date errors extension filesize private problem result result_text user/;
+defhash user => qw/name email town university level/;
+
+sub clean_job (_){
+	HDEL cp . "job.$_[0]", qw/result result_text results daemon/
+}
+
+sub mark_open {
+	my ($problem, $user) = @_;
+	HSETNX cp . 'open', "$problem.$user", time;
+}
+
+sub get_open {
+	my ($problem, $user) = @_;
+	HGET cp . 'open', "$problem.$user";
+}
 
-our @EXPORT_OK = do {
+our @EXPORT = do {
 	no strict 'refs';
 	grep { $_ =~ /^[a-zA-Z]/ and exists &$_ } keys %{__PACKAGE__ . '::'};
 };