From 2944cb07be67d87e1a93ca6abf058dd6ce23d276 Mon Sep 17 00:00:00 2001
From: Niko Tyni <ntyni@iki.fi>
Date: Wed, 19 Mar 2008 12:36:25 +0000
Subject: unrevert the locking change, LOCK_SH should work

---
 lib/Smokeping/Master.pm | 32 +++++++++++++++++++-------------
 1 file changed, 19 insertions(+), 13 deletions(-)

(limited to 'lib/Smokeping')

diff --git a/lib/Smokeping/Master.pm b/lib/Smokeping/Master.pm
index bbe5a24..7114b00 100644
--- a/lib/Smokeping/Master.pm
+++ b/lib/Smokeping/Master.pm
@@ -1,7 +1,7 @@
 # -*- perl -*-
 package Smokeping::Master;
 use Data::Dumper;
-use Storable qw(nstore dclone retrieve);
+use Storable qw(nstore_fd dclone fd_retrieve);
 use strict;
 use warnings;
 use Fcntl qw(:flock);
@@ -118,13 +118,16 @@ sub save_updates {
             warn "Skipping update for ${name}.${slave}.slave_cache since ".
                  "you seem to try todo some directory magic here. Don't!";
         } 
-        elsif ( open (my $lock, '>>' , "$file.lock") ) {
+        else {
 
             for (my $i = 3; $i > 0; $i--){
-                if ( flock($lock, LOCK_EX) ){
+                my $fh;
+                if ( open ($fh, '+>>' , $file) and flock($fh, LOCK_EX) ){
                     my $existing = [];
-	            if ( -r $file ){
-                        my $in = eval { retrieve $file };
+                    next if ! -e $file; # the reader unlinked it from under us
+                    seek $fh, 0, 0;
+                    if ( -s _ ){
+                        my $in = eval { fd_retrieve $fh };
                         if ($@) { #error
                             warn "Loading $file: $@";
                         } else {
@@ -132,16 +135,17 @@ sub save_updates {
 			};
                     };
                     push @{$existing}, [ $slave, $time, $updatestring ];
-                    nstore($existing, $file);		    
+                    nstore_fd($existing, $fh);		    
+                    flock($fh, LOCK_UN);
+                    close $fh;
                     last;
                 } else {
                     warn "Could not lock $file ($!). Trying again for $i rounds.\n";
                     sleep rand(2);
                 }
+                warn "Could not update $file, giving up for now.";
+                close $fh;
             }
-            close $lock;
-        } else {
-            warn "Could not update $file: $!";
         }
     }         
 };
@@ -164,10 +168,12 @@ sub get_slaveupdates {
     my $dir = slavedatadir($cfg);
     $file =~ s/^\Q$datadir\E/$dir/;
 
-    if ( -r $file and open (my $lock, '>>', "$file.lock") ) {
-        if ( flock $lock, LOCK_EX ){
-            eval { $data = retrieve $file };
+    my $fh;
+    if ( open ($fh, '<', $file) ) {
+        if ( flock $fh, LOCK_SH ){
+            eval { $data = fd_retrieve $fh };
             unlink $file;
+            flock $fh, LOCK_UN;
             if ($@) { #error
                 warn "Loading $file: $@";  
                 return $empty;
@@ -175,7 +181,7 @@ sub get_slaveupdates {
         } else {
             warn "Could not lock $file. Will skip and try again in the next round. No harm done!\n";
         }
-        close $lock;        
+        close $fh;        
         return $data;
     }
     return $empty;
-- 
cgit v1.2.3-24-g4f1b