lib/IndexData/Utils/PersistentCounter.pm

   1 package IndexData::Utils::PersistentCounter;
   2
   3 use 5.018002;
   4 use strict;
   5 use warnings;
   6
   7 use IO::File;
   8
   9
  10 sub new {
  11     my $class = shift();
  12     my($file, $create) = @_;
  13
  14     if (! -f $file) {
  15         return undef if !$create;
  16         #   ### There is a bit of a race condition here, but it's not
  17         #       something that's going to crop up in real life.
  18         my $fh = new IO::File(">$file") || return undef;
  19         $fh->print("1\n");
  20         $fh->close() or return undef;
  21     }
  22
  23     my $this = bless {
  24         file => $file,
  25     }, $class;
  26
  27     return $this;
  28 }
  29
  30
  31 sub next {
  32     my $this = shift();
  33
  34     my $fh = new IO::File('+<' . $this->{file}) || return undef;
  35     flock($fh, 2) || die "can't lock file";
  36     my $n = <$fh>;
  37     $fh->seek(0, 0);
  38     $fh->print($n+1, "\n");
  39     $fh->close() or return undef;
  40     return $n+0;
  41 }
  42
  43
  44 sub delete {
  45     my $this = shift();
  46
  47     unlink $this->{file} or return 0;
  48     return 1;
  49 }
  50
  51
  52 1;