1 |
package YAMLDB; |
2 |
use YAML::Error; |
3 |
use Carp; |
4 |
use YAML qw/LoadFile DumpFile Dump/; |
5 |
use POSIX qw/strftime/; |
6 |
|
7 |
sub new { |
8 |
my ($class, %arg) = @_; |
9 |
|
10 |
my $self = bless { |
11 |
%arg, |
12 |
}, $class; |
13 |
|
14 |
$self->{last_check} = (-M $self->{db_file}) + 1; |
15 |
unless (-x $self->{db_file}) { |
16 |
DumpFile ($self->{db_file}, {}); |
17 |
} |
18 |
$self->check_maildb (); |
19 |
|
20 |
$self; |
21 |
} |
22 |
|
23 |
sub check_maildb { |
24 |
my ($self) = @_; |
25 |
|
26 |
if ($self->{last_check} > (-M $self->{db_file})) { |
27 |
$self->{last_check} = -M $self->{db_file}; |
28 |
|
29 |
my $maildb = eval { my $m = LoadFile ($self->{db_file}); return $m }; |
30 |
if ($@) { |
31 |
warn "ERROR when reading mail database $self->{db_file}: $@\n"; |
32 |
$self->{db_file} = $self->{db_file} . ".after_failure"; |
33 |
} else { |
34 |
$self->{maildb} = $maildb; |
35 |
} |
36 |
} |
37 |
} |
38 |
|
39 |
# This is not thread or multiprocess safe! |
40 |
# XXX: A second process will (of course) terribly damage the maildatabase |
41 |
# but i assume there wont be any second process. |
42 |
sub sync_maildb { |
43 |
my ($self) = @_; |
44 |
|
45 |
eval { |
46 |
DumpFile ("$self->{db_file}.tmp", $self->{maildb}); |
47 |
}; |
48 |
if ($@) { |
49 |
warn "ERROR when writing mail database $self->{db_file}.tmp: $@\n"; |
50 |
} else { |
51 |
rename "$self->{db_file}.tmp", $self->{db_file}; |
52 |
$self->{last_check} = -M $self->{db_file}; |
53 |
} |
54 |
} |
55 |
|
56 |
sub get { |
57 |
my ($self, $key) = @_; |
58 |
$self->check_maildb; |
59 |
$self->{maildb}->{$key}; |
60 |
} |
61 |
|
62 |
sub clear { |
63 |
my ($self, $key) = @_; |
64 |
$self->check_maildb; |
65 |
delete $self->{maildb}->{$key}; |
66 |
$self->sync_maildb; |
67 |
} |
68 |
|
69 |
sub set { |
70 |
my ($self, $key, $value) = @_; |
71 |
$self->check_maildb; |
72 |
$self->{maildb}->{$key} = $value; |
73 |
$self->sync_maildb; |
74 |
} |
75 |
1 |