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