This repository has been archived by the owner on Dec 19, 2023. It is now read-only.
/
CounterFile.pm
325 lines (239 loc) · 7.48 KB
/
CounterFile.pm
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
# This -*-perl -*- module implements a persistent counter class.
#
# $Id: CounterFile.pm,v 1.1 1999/11/12 08:46:23 mike Exp $
#
package File::CounterFile;
=head1 NAME
File::CounterFile - Persistent counter class
=head1 SYNOPSIS
use File::CounterFile;
$c = new File::CounterFile "COUNTER", "aa00";
$id = $c->inc;
open(F, ">F$id");
=head1 DESCRIPTION
This module implements a persistent counter class. Each counter is
represented by a separate file in the file system. File locking is
applied, so multiple processes might try to access the same counters
at the same time without risk of counter destruction.
You give the file name as the first parameter to the object
constructor (C<new>). The file is created if it does not exist.
If the file name does not start with "/" or ".", then it is
interpreted as a file relative to C<$File::CounterFile::DEFAULT_DIR>.
The default value for this variable is initialized from the
environment variable C<TMPDIR>, or F</usr/tmp> is no environment
variable is defined. You may want to assign a different value to this
variable before creating counters.
If you pass a second parameter to the constructor, that sets the
initial value for a new counter. This parameter only takes effect
when the file is created (i.e. it does not exist before the call).
When you call the C<inc()> method, you increment the counter value by
one. When you call C<dec()> the counter value is decrementd. In both
cases the new value is returned. The C<dec()> method only works for
numerical counters (digits only).
You can peek at the value of the counter (without incrementing it) by
using the C<value()> method.
The counter can be locked and unlocked with the C<lock()> and
C<unlock()> methods. Incrementing and value retrieval is faster when
the counter is locked, because we do not have to update the counter
file all the time. You can query whether the counter is locked with
the C<locked()> method.
There is also an operator overloading interface to the
File::CounterFile object. This means that you might use the C<++>
operator for incrementing the counter, C<--> operator for decrementing
and you can interpolate counters diretly into strings.
=head1 BUGS
It uses flock(2) to lock the counter file. This does not work on all
systems. Perhaps we should use the File::Lock module?
=head1 INSTALLATION
Copy this file to the F<File> subdirectory of your Perl 5 library
directory (often F</usr/local/lib/perl5>).
=head1 COPYRIGHT
Copyright (c) 1995-1996 Gisle Aas. All rights reserved.
This library is free software; you can redistribute it and/or
modify it under the same terms as Perl itself.
=head1 AUTHOR
Gisle Aas <aas@sn.no>
=cut
require 5.002;
use Carp qw(croak);
use Symbol qw(gensym);
sub Version { $VERSION; }
$VERSION = sprintf("%d.%02d", q$Revision: 1.1 $ =~ /(\d+)\.(\d+)/);
$MAGIC = "#COUNTER-1.0\n"; # first line in counter files
$DEFAULT_INITIAL = 0; # default initial counter value
# default location for counter files
$DEFAULT_DIR = $ENV{TMPDIR} || "/usr/tmp";
# Experimental overloading.
%OVERLOAD = ('++' => \&inc,
'--' => \&dec,
'""' => \&value,
fallback => 1,
);
sub new
{
my($class, $file, $initial) = @_;
croak "No file specified\n" unless defined $file;
$file = "$DEFAULT_DIR/$file" unless $file =~ /^[\.\/]/;
$initial = $DEFAULT_INITIAL unless defined $initial;
my $value;
if (-e $file) {
croak "Specified file is a directory" if -d _;
open(F, $file) or croak "Can't open $file: $!";
my $first_line = <F>;
$value = <F>;
close(F);
croak "Bad counter magic in $file" unless $first_line eq $MAGIC;
chomp($value);
} else {
open(F, ">$file") or croak "Can't create $file: $!";
print F $MAGIC;
print F $initial, "\n";
close(F);
$value = $initial;
}
bless { file => $file, # the filename for the counter
value => $value, # the current value
updated => 0, # flag indicating if value has changed
# handle => XXX, # file handle symbol. Only present when locked
};
}
sub locked
{
exists shift->{handle};
}
sub lock
{
my($self) = @_;
# MODIFICATION FOR WIN32, UGGH
return $self if $ =~ /win32/i;
$self->unlock if $self->locked;
my $fh = gensym();
my $file = $self->{file};
open($fh, "+<$file") or croak "Can't open $file: $!";
flock($fh, 2) or croak "Can't flock: $!"; # 2 = exlusive lock
my $magic = <$fh>;
if ($magic ne $MAGIC) {
$self->unlock;
croak("Bad counter magic '$magic' in $file");
}
chomp($self->{value} = <$fh>);
$self->{handle} = $fh;
$self->{updated} = 0;
$self;
}
sub unlock
{
my($self) = @_;
return unless $self->locked;
my $fh = $self->{handle};
if ($self->{updated}) {
# write back new value
seek($fh, 0, 0) or croak "Can't seek to beginning: $!";
print $fh $MAGIC;
print $fh "$self->{value}\n";
}
close($fh) or warn "Can't close: $!";
delete $self->{handle};
$self;
}
sub inc
{
my($self) = @_;
if ($self->locked) {
$self->{value}++;
$self->{updated} = 1;
} else {
$self->lock;
$self->{value}++;
$self->{updated} = 1;
$self->unlock;
}
$self->{value}; # return value
}
sub dec
{
my($self) = @_;
if ($self->locked) {
croak "Autodecrement is not magical in perl"
unless $self->{value} =~ /^\d+$/;
$self->{value}--;
$self->{updated} = 1;
} else {
$self->lock;
croak "Autodecrement is not magical in perl"
unless $self->{value} =~ /^\d+$/;
$self->{value}--;
$self->{updated} = 1;
$self->unlock;
}
$self->{value}; # return value
}
sub value
{
my($self) = @_;
my $value;
if ($self->locked) {
$value = $self->{value};
} else {
$self->lock;
$value = $self->{value};
$self->unlock;
}
$value;
}
sub DESTROY
{
my $self = shift;
$self->unlock;
}
####################################################################
#
# S E L F T E S T S E C T I O N
#
#####################################################################
#
# If we're not use'd or require'd execute self-test.
#
# Test is kept behind __END__ so it doesn't take uptime
# and memory unless explicitly required. If you're working
# on the code you might find it easier to comment out the
# eval and __END__ so that error line numbers make more sense.
package main;
eval join('',<DATA>) || die $@ unless caller();
1;
__END__
$cf = "./zz-counter-$$"; # the name for out temprary counter
# Test normal object creation and increment
$c = new File::CounterFile $cf;
$id1 = $c->inc;
$id2 = $c->inc;
$c = new File::CounterFile $cf;
$id3 = $c->inc;
$id4 = $c->dec;
die "test failed" unless ($id1 == 1 && $id2 == 2 && $id3 == 3 && $id4 == 2);
unlink $cf;
# Test magic increment
$id1 = (new File::CounterFile $cf, "aa98")->inc;
$id2 = (new File::CounterFile $cf)->inc;
$id3 = (new File::CounterFile $cf)->inc;
eval {
# This should now work because "Decrement is not magical in perl"
$c = new File::CounterFile $cf; $id4 = $c->dec; $c = undef;
};
die "test failed (No exception to catch)" unless $@;
#print "$id1 $id2 $id3\n";
die "test failed" unless ($id1 eq "aa99" && $id2 eq "ab00" && $id3 eq "ab01");
unlink $cf;
# Test operator overloading
$c = new File::CounterFile $cf, "100";
$c->lock;
$c++; # counter is now 101
$c++; # counter is now 102
$c++; # counter is now 103
$c--; # counter is now 102 again
$id1 = "$c";
$id2 = ++$c;
$c = undef; # destroy object
unlink $cf;
die "test failed" unless $id1 == 102 && $id2 == 103;
print "Selftest for File::CounterFile $File::CounterFile::VERSION ok\n";