Added a comment as to where an allocation error is occurring that crashes perl
[dbsrgits/DBM-Deep.git] / lib / DBM / Deep / Engine.pm
CommitLineData
a20d9a3f 1package DBM::Deep::Engine;
2
2120a181 3use 5.006_000;
460b1067 4
a20d9a3f 5use strict;
065b45be 6use warnings FATAL => 'all';
a20d9a3f 7
75a6a379 8# Never import symbols into our namespace. We are a class, not a library.
9# -RobK, 2008-05-27
10use Scalar::Util ();
d6d8e27e 11
12#use Data::Dumper ();
75a6a379 13
21838116 14# File-wide notes:
2120a181 15# * Every method in here assumes that the storage has been appropriately
c3aafc14 16# safeguarded. This can be anything from flock() to some sort of manual
17# mutex. But, it's the caller's responsability to make sure that this has
18# been done.
21838116 19
8db25060 20# Setup file and tag signatures. These should never change.
8db25060 21sub SIG_FILE () { 'DPDB' }
460b1067 22sub SIG_HEADER () { 'h' }
8db25060 23sub SIG_HASH () { 'H' }
24sub SIG_ARRAY () { 'A' }
8db25060 25sub SIG_NULL () { 'N' }
26sub SIG_DATA () { 'D' }
27sub SIG_INDEX () { 'I' }
28sub SIG_BLIST () { 'B' }
7b1e1aa1 29sub SIG_FREE () { 'F' }
8db25060 30sub SIG_SIZE () { 1 }
e9b0b5f0 31
065b45be 32our $STALE_SIZE = 2;
8db25060 33
2120a181 34# Please refer to the pack() documentation for further information
35my %StP = (
e9b0b5f0 36 1 => 'C', # Unsigned char value (no order needed as it's just one byte)
2120a181 37 2 => 'n', # Unsigned short in "network" (big-endian) order
38 4 => 'N', # Unsigned long in "network" (big-endian) order
39 8 => 'Q', # Usigned quad (no order specified, presumably machine-dependent)
40);
065b45be 41sub StP { $StP{$_[1]} }
83371fe3 42
00d9bd0b 43# Import these after the SIG_* definitions because those definitions are used
44# in the headers of these classes. -RobK, 2008-06-20
45use DBM::Deep::Engine::Sector::BucketList;
46use DBM::Deep::Engine::Sector::FileHeader;
47use DBM::Deep::Engine::Sector::Index;
48use DBM::Deep::Engine::Sector::Null;
49use DBM::Deep::Engine::Sector::Reference;
50use DBM::Deep::Engine::Sector::Scalar;
51use DBM::Deep::Iterator;
52
c3aafc14 53################################################################################
c3aafc14 54
612969fb 55sub new {
56 my $class = shift;
57 my ($args) = @_;
58
f1879fdc 59 $args->{storage} = DBM::Deep::File->new( $args )
60 unless exists $args->{storage};
61
612969fb 62 my $self = bless {
2120a181 63 byte_size => 4,
64
65 digest => undef,
66 hash_size => 16, # In bytes
67 hash_chars => 256, # Number of chars the algorithm uses per byte
612969fb 68 max_buckets => 16,
e9b0b5f0 69 num_txns => 1, # The HEAD
2120a181 70 trans_id => 0, # Default to the HEAD
460b1067 71
e9b0b5f0 72 data_sector_size => 64, # Size in bytes of each data sector
73
2120a181 74 entries => {}, # This is the list of entries for transactions
83371fe3 75 storage => undef,
612969fb 76 }, $class;
77
e9b0b5f0 78 # Never allow byte_size to be set directly.
79 delete $args->{byte_size};
e0098e7f 80 if ( defined $args->{pack_size} ) {
81 if ( lc $args->{pack_size} eq 'small' ) {
2120a181 82 $args->{byte_size} = 2;
e0098e7f 83 }
84 elsif ( lc $args->{pack_size} eq 'medium' ) {
2120a181 85 $args->{byte_size} = 4;
e0098e7f 86 }
87 elsif ( lc $args->{pack_size} eq 'large' ) {
2120a181 88 $args->{byte_size} = 8;
e0098e7f 89 }
90 else {
2120a181 91 DBM::Deep->_throw_error( "Unknown pack_size value: '$args->{pack_size}'" );
e0098e7f 92 }
93 }
94
fde3db1a 95 # Grab the parameters we want to use
96 foreach my $param ( keys %$self ) {
97 next unless exists $args->{$param};
3e9498a1 98 $self->{$param} = $args->{$param};
fde3db1a 99 }
100
e9b0b5f0 101 my %validations = (
102 max_buckets => { floor => 16, ceil => 256 },
103 num_txns => { floor => 1, ceil => 255 },
104 data_sector_size => { floor => 32, ceil => 256 },
105 );
106
107 while ( my ($attr, $c) = each %validations ) {
108 if ( !defined $self->{$attr}
109 || !length $self->{$attr}
110 || $self->{$attr} =~ /\D/
111 || $self->{$attr} < $c->{floor}
112 ) {
113 $self->{$attr} = '(undef)' if !defined $self->{$attr};
114 warn "Floor of $attr is $c->{floor}. Setting it to $c->{floor} from '$self->{$attr}'\n";
115 $self->{$attr} = $c->{floor};
116 }
117 elsif ( $self->{$attr} > $c->{ceil} ) {
118 warn "Ceiling of $attr is $c->{ceil}. Setting it to $c->{ceil} from '$self->{$attr}'\n";
119 $self->{$attr} = $c->{ceil};
120 }
e0098e7f 121 }
122
2120a181 123 if ( !$self->{digest} ) {
124 require Digest::MD5;
125 $self->{digest} = \&Digest::MD5::md5;
126 }
127
260a80b4 128 return $self;
129}
130
2120a181 131################################################################################
460b1067 132
2120a181 133sub read_value {
c3aafc14 134 my $self = shift;
2120a181 135 my ($obj, $key) = @_;
136
137 # This will be a Reference sector
138 my $sector = $self->_load_sector( $obj->_base_offset )
139 or return;
140
141 if ( $sector->staleness != $obj->_staleness ) {
142 return;
143 }
144
145 my $key_md5 = $self->_apply_digest( $key );
146
147 my $value_sector = $sector->get_data_for({
148 key_md5 => $key_md5,
149 allow_head => 1,
150 });
151
152 unless ( $value_sector ) {
153 $value_sector = DBM::Deep::Engine::Sector::Null->new({
154 engine => $self,
155 data => undef,
156 });
157
158 $sector->write_data({
159 key_md5 => $key_md5,
160 key => $key,
161 value => $value_sector,
162 });
163 }
164
165 return $value_sector->data;
c3aafc14 166}
167
2120a181 168sub get_classname {
260a80b4 169 my $self = shift;
2120a181 170 my ($obj) = @_;
260a80b4 171
2120a181 172 # This will be a Reference sector
173 my $sector = $self->_load_sector( $obj->_base_offset )
174 or DBM::Deep->_throw_error( "How did get_classname fail (no sector for '$obj')?!" );
612969fb 175
2120a181 176 if ( $sector->staleness != $obj->_staleness ) {
177 return;
178 }
ea2f6d67 179
2120a181 180 return $sector->get_classname;
1bf65be7 181}
182
1cff45d7 183sub make_reference {
184 my $self = shift;
185 my ($obj, $old_key, $new_key) = @_;
186
187 # This will be a Reference sector
188 my $sector = $self->_load_sector( $obj->_base_offset )
00d9bd0b 189 or DBM::Deep->_throw_error( "How did make_reference fail (no sector for '$obj')?!" );
1cff45d7 190
191 if ( $sector->staleness != $obj->_staleness ) {
192 return;
193 }
194
195 my $old_md5 = $self->_apply_digest( $old_key );
196
197 my $value_sector = $sector->get_data_for({
198 key_md5 => $old_md5,
199 allow_head => 1,
200 });
201
202 unless ( $value_sector ) {
203 $value_sector = DBM::Deep::Engine::Sector::Null->new({
204 engine => $self,
205 data => undef,
206 });
207
208 $sector->write_data({
209 key_md5 => $old_md5,
210 key => $old_key,
211 value => $value_sector,
212 });
213 }
214
215 if ( $value_sector->isa( 'DBM::Deep::Engine::Sector::Reference' ) ) {
216 $sector->write_data({
217 key => $new_key,
218 key_md5 => $self->_apply_digest( $new_key ),
219 value => $value_sector,
220 });
221 $value_sector->increment_refcount;
222 }
223 else {
224 $sector->write_data({
225 key => $new_key,
226 key_md5 => $self->_apply_digest( $new_key ),
227 value => $value_sector->clone,
228 });
229 }
230}
231
2120a181 232sub key_exists {
0d0f3d5d 233 my $self = shift;
2120a181 234 my ($obj, $key) = @_;
0d0f3d5d 235
2120a181 236 # This will be a Reference sector
237 my $sector = $self->_load_sector( $obj->_base_offset )
238 or return '';
0d0f3d5d 239
2120a181 240 if ( $sector->staleness != $obj->_staleness ) {
241 return '';
242 }
0d0f3d5d 243
2120a181 244 my $data = $sector->get_data_for({
245 key_md5 => $self->_apply_digest( $key ),
246 allow_head => 1,
247 });
20b7f047 248
2120a181 249 # exists() returns 1 or '' for true/false.
250 return $data ? 1 : '';
0d0f3d5d 251}
252
2120a181 253sub delete_key {
e064ccd1 254 my $self = shift;
2120a181 255 my ($obj, $key) = @_;
e064ccd1 256
2120a181 257 my $sector = $self->_load_sector( $obj->_base_offset )
258 or return;
460b1067 259
2120a181 260 if ( $sector->staleness != $obj->_staleness ) {
261 return;
262 }
263
264 return $sector->delete_key({
265 key_md5 => $self->_apply_digest( $key ),
266 allow_head => 0,
267 });
268}
269
270sub write_value {
271 my $self = shift;
272 my ($obj, $key, $value) = @_;
273
274 my $r = Scalar::Util::reftype( $value ) || '';
275 {
276 last if $r eq '';
277 last if $r eq 'HASH';
278 last if $r eq 'ARRAY';
e064ccd1 279
2120a181 280 DBM::Deep->_throw_error(
281 "Storage of references of type '$r' is not supported."
282 );
460b1067 283 }
260a80b4 284
1cff45d7 285 # This will be a Reference sector
286 my $sector = $self->_load_sector( $obj->_base_offset )
00d9bd0b 287 or DBM::Deep->_throw_error( "1: Cannot write to a deleted spot in DBM::Deep." );
1cff45d7 288
289 if ( $sector->staleness != $obj->_staleness ) {
00d9bd0b 290 DBM::Deep->_throw_error( "2: Cannot write to a deleted spot in DBM::Deep." );
1cff45d7 291 }
292
2120a181 293 my ($class, $type);
294 if ( !defined $value ) {
295 $class = 'DBM::Deep::Engine::Sector::Null';
296 }
297 elsif ( $r eq 'ARRAY' || $r eq 'HASH' ) {
75a6a379 298 my $tmpvar;
299 if ( $r eq 'ARRAY' ) {
300 $tmpvar = tied @$value;
301 } elsif ( $r eq 'HASH' ) {
302 $tmpvar = tied %$value;
303 }
5ef7542f 304
edd45134 305 if ( $tmpvar ) {
306 my $is_dbm_deep = eval { local $SIG{'__DIE__'}; $tmpvar->isa( 'DBM::Deep' ); };
307
308 unless ( $is_dbm_deep ) {
309 DBM::Deep->_throw_error( "Cannot store something that is tied." );
310 }
311
d6d8e27e 312 unless ( $tmpvar->_engine->storage == $self->storage ) {
75a6a379 313 DBM::Deep->_throw_error( "Cannot store values across DBM::Deep files. Please use export() instead." );
5ef7542f 314 }
5ef7542f 315
d6d8e27e 316 # First, verify if we're storing the same thing to this spot. If we are, then
317 # this should be a no-op. -EJS, 2008-05-19
318 my $loc = $sector->get_data_location_for({
319 key_md5 => $self->_apply_digest( $key ),
320 allow_head => 1,
321 });
322
323 if ( defined($loc) && $loc == $tmpvar->_base_offset ) {
1cff45d7 324 return 1;
325 }
326
edd45134 327 #XXX Can this use $loc?
d6d8e27e 328 my $value_sector = $self->_load_sector( $tmpvar->_base_offset );
329 $sector->write_data({
330 key => $key,
331 key_md5 => $self->_apply_digest( $key ),
332 value => $value_sector,
333 });
334 $value_sector->increment_refcount;
335
336 return 1;
1cff45d7 337 }
edd45134 338
2120a181 339 $class = 'DBM::Deep::Engine::Sector::Reference';
340 $type = substr( $r, 0, 1 );
341 }
342 else {
1cff45d7 343 if ( tied($value) ) {
344 DBM::Deep->_throw_error( "Cannot store something that is tied." );
345 }
2120a181 346 $class = 'DBM::Deep::Engine::Sector::Scalar';
460b1067 347 }
9b2370e0 348
2120a181 349 # Create this after loading the reference sector in case something bad happens.
350 # This way, we won't allocate value sector(s) needlessly.
351 my $value_sector = $class->new({
352 engine => $self,
353 data => $value,
354 type => $type,
355 });
75a6a379 356
2120a181 357 $sector->write_data({
358 key => $key,
359 key_md5 => $self->_apply_digest( $key ),
360 value => $value_sector,
361 });
362
363 # This code is to make sure we write all the values in the $value to the disk
364 # and to make sure all changes to $value after the assignment are reflected
365 # on disk. This may be counter-intuitive at first, but it is correct dwimmery.
366 # NOTE - simply tying $value won't perform a STORE on each value. Hence, the
367 # copy to a temp value.
368 if ( $r eq 'ARRAY' ) {
369 my @temp = @$value;
370 tie @$value, 'DBM::Deep', {
371 base_offset => $value_sector->offset,
372 staleness => $value_sector->staleness,
373 storage => $self->storage,
374 engine => $self,
375 };
376 @$value = @temp;
377 bless $value, 'DBM::Deep::Array' unless Scalar::Util::blessed( $value );
e064ccd1 378 }
2120a181 379 elsif ( $r eq 'HASH' ) {
380 my %temp = %$value;
381 tie %$value, 'DBM::Deep', {
382 base_offset => $value_sector->offset,
383 staleness => $value_sector->staleness,
384 storage => $self->storage,
385 engine => $self,
386 };
e064ccd1 387
2120a181 388 %$value = %temp;
389 bless $value, 'DBM::Deep::Hash' unless Scalar::Util::blessed( $value );
390 }
460b1067 391
2120a181 392 return 1;
e064ccd1 393}
394
2120a181 395# XXX Add staleness here
396sub get_next_key {
460b1067 397 my $self = shift;
2120a181 398 my ($obj, $prev_key) = @_;
70b55428 399
2120a181 400 # XXX Need to add logic about resetting the iterator if any key in the reference has changed
401 unless ( $prev_key ) {
402 $obj->{iterator} = DBM::Deep::Iterator->new({
403 base_offset => $obj->_base_offset,
404 engine => $self,
405 });
406 }
118ba343 407
2120a181 408 return $obj->{iterator}->get_next_key( $obj );
409}
118ba343 410
2120a181 411################################################################################
260a80b4 412
2120a181 413sub setup_fh {
414 my $self = shift;
415 my ($obj) = @_;
359a01ac 416
00d9bd0b 417 return 1 if $obj->_base_offset;
118ba343 418
badf847c 419 my $header = $self->_load_header;
118ba343 420
00d9bd0b 421 # Creating a new file
422 if ( $header->is_new ) {
423 # 1) Create Array/Hash entry
424 my $sector = DBM::Deep::Engine::Sector::Reference->new({
425 engine => $self,
426 type => $obj->_type,
427 });
428 $obj->{base_offset} = $sector->offset;
429 $obj->{staleness} = $sector->staleness;
118ba343 430
00d9bd0b 431 $self->flush;
432 }
433 # Reading from an existing file
434 else {
435 $obj->{base_offset} = $header->size;
436 my $sector = DBM::Deep::Engine::Sector::Reference->new({
437 engine => $self,
438 offset => $obj->_base_offset,
439 });
440 unless ( $sector ) {
441 DBM::Deep->_throw_error("Corrupted file, no master index record");
118ba343 442 }
2120a181 443
00d9bd0b 444 unless ($obj->_type eq $sector->type) {
445 DBM::Deep->_throw_error("File type mismatch");
118ba343 446 }
f1879fdc 447
00d9bd0b 448 $obj->{staleness} = $sector->staleness;
118ba343 449 }
2120a181 450
00d9bd0b 451 $self->storage->set_inode;
452
2120a181 453 return 1;
454}
455
456sub begin_work {
457 my $self = shift;
458 my ($obj) = @_;
459
460 if ( $self->trans_id ) {
461 DBM::Deep->_throw_error( "Cannot begin_work within an active transaction" );
462 }
463
464 my @slots = $self->read_txn_slots;
e9b0b5f0 465 my $found;
466 for my $i ( 0 .. $#slots ) {
2120a181 467 next if $slots[$i];
e9b0b5f0 468
2120a181 469 $slots[$i] = 1;
e9b0b5f0 470 $self->set_trans_id( $i + 1 );
471 $found = 1;
2120a181 472 last;
473 }
e9b0b5f0 474 unless ( $found ) {
475 DBM::Deep->_throw_error( "Cannot allocate transaction ID" );
476 }
2120a181 477 $self->write_txn_slots( @slots );
478
479 if ( !$self->trans_id ) {
480 DBM::Deep->_throw_error( "Cannot begin_work - no available transactions" );
481 }
482
483 return;
484}
485
486sub rollback {
487 my $self = shift;
488 my ($obj) = @_;
489
490 if ( !$self->trans_id ) {
491 DBM::Deep->_throw_error( "Cannot rollback without an active transaction" );
492 }
493
494 # Each entry is the file location for a bucket that has a modification for
495 # this transaction. The entries need to be expunged.
496 foreach my $entry (@{ $self->get_entries } ) {
497 # Remove the entry here
498 my $read_loc = $entry
499 + $self->hash_size
500 + $self->byte_size
e9b0b5f0 501 + $self->byte_size
502 + ($self->trans_id - 1) * ( $self->byte_size + $STALE_SIZE );
2120a181 503
504 my $data_loc = $self->storage->read_at( $read_loc, $self->byte_size );
505 $data_loc = unpack( $StP{$self->byte_size}, $data_loc );
506 $self->storage->print_at( $read_loc, pack( $StP{$self->byte_size}, 0 ) );
507
508 if ( $data_loc > 1 ) {
509 $self->_load_sector( $data_loc )->free;
510 }
260a80b4 511 }
e06824f8 512
2120a181 513 $self->clear_entries;
70b55428 514
2120a181 515 my @slots = $self->read_txn_slots;
e9b0b5f0 516 $slots[$self->trans_id-1] = 0;
2120a181 517 $self->write_txn_slots( @slots );
518 $self->inc_txn_staleness_counter( $self->trans_id );
519 $self->set_trans_id( 0 );
6fde4ed2 520
70b55428 521 return 1;
522}
523
2120a181 524sub commit {
16d1ad9b 525 my $self = shift;
2120a181 526 my ($obj) = @_;
527
528 if ( !$self->trans_id ) {
529 DBM::Deep->_throw_error( "Cannot commit without an active transaction" );
530 }
531
532 foreach my $entry (@{ $self->get_entries } ) {
533 # Overwrite the entry in head with the entry in trans_id
534 my $base = $entry
535 + $self->hash_size
536 + $self->byte_size;
537
538 my $head_loc = $self->storage->read_at( $base, $self->byte_size );
539 $head_loc = unpack( $StP{$self->byte_size}, $head_loc );
e9b0b5f0 540
541 my $spot = $base + $self->byte_size + ($self->trans_id - 1) * ( $self->byte_size + $STALE_SIZE );
2120a181 542 my $trans_loc = $self->storage->read_at(
e9b0b5f0 543 $spot, $self->byte_size,
2120a181 544 );
545
546 $self->storage->print_at( $base, $trans_loc );
547 $self->storage->print_at(
e9b0b5f0 548 $spot,
549 pack( $StP{$self->byte_size} . ' ' . $StP{$STALE_SIZE}, (0) x 2 ),
2120a181 550 );
551
552 if ( $head_loc > 1 ) {
553 $self->_load_sector( $head_loc )->free;
554 }
555 }
556
557 $self->clear_entries;
558
559 my @slots = $self->read_txn_slots;
e9b0b5f0 560 $slots[$self->trans_id-1] = 0;
2120a181 561 $self->write_txn_slots( @slots );
562 $self->inc_txn_staleness_counter( $self->trans_id );
563 $self->set_trans_id( 0 );
564
565 return 1;
16d1ad9b 566}
567
2120a181 568sub read_txn_slots {
d4b1166e 569 my $self = shift;
d7f031fc 570 return $self->_load_header->read_txn_slots(@_);
2120a181 571}
20f7b20c 572
2120a181 573sub write_txn_slots {
574 my $self = shift;
d7f031fc 575 return $self->_load_header->write_txn_slots(@_);
2120a181 576}
577
578sub get_running_txn_ids {
579 my $self = shift;
580 my @transactions = $self->read_txn_slots;
e9b0b5f0 581 my @trans_ids = map { $_+1} grep { $transactions[$_] } 0 .. $#transactions;
2120a181 582}
583
584sub get_txn_staleness_counter {
585 my $self = shift;
d7f031fc 586 return $self->_load_header->get_txn_staleness_counter(@_);
d4b1166e 587}
588
2120a181 589sub inc_txn_staleness_counter {
d4b1166e 590 my $self = shift;
d7f031fc 591 return $self->_load_header->inc_txn_staleness_counter(@_);
2120a181 592}
20f7b20c 593
2120a181 594sub get_entries {
595 my $self = shift;
596 return [ keys %{ $self->{entries}{$self->trans_id} ||= {} } ];
d4b1166e 597}
598
2120a181 599sub add_entry {
ea2f6d67 600 my $self = shift;
2120a181 601 my ($trans_id, $loc) = @_;
ea2f6d67 602
2120a181 603 $self->{entries}{$trans_id} ||= {};
604 $self->{entries}{$trans_id}{$loc} = undef;
605}
ea2f6d67 606
2120a181 607# If the buckets are being relocated because of a reindexing, the entries
608# mechanism needs to be made aware of it.
609sub reindex_entry {
610 my $self = shift;
611 my ($old_loc, $new_loc) = @_;
612
613 TRANS:
614 while ( my ($trans_id, $locs) = each %{ $self->{entries} } ) {
695c88b1 615 if ( exists $locs->{$old_loc} ) {
616 delete $locs->{$old_loc};
617 $locs->{$new_loc} = undef;
618 next TRANS;
2120a181 619 }
ea2f6d67 620 }
ea2f6d67 621}
622
2120a181 623sub clear_entries {
d4b1166e 624 my $self = shift;
2120a181 625 delete $self->{entries}{$self->trans_id};
626}
eea0d863 627
2120a181 628################################################################################
eea0d863 629
2120a181 630sub _apply_digest {
75be6413 631 my $self = shift;
2120a181 632 return $self->{digest}->(@_);
633}
16d1ad9b 634
2120a181 635sub _add_free_blist_sector { shift->_add_free_sector( 0, @_ ) }
badf847c 636sub _add_free_data_sector { shift->_add_free_sector( 1, @_ ) }
2120a181 637sub _add_free_index_sector { shift->_add_free_sector( 2, @_ ) }
badf847c 638sub _add_free_sector { shift->_load_header->add_free_sector( @_ ) }
75be6413 639
badf847c 640sub _request_blist_sector { shift->_request_sector( 0, @_ ) }
641sub _request_data_sector { shift->_request_sector( 1, @_ ) }
642sub _request_index_sector { shift->_request_sector( 2, @_ ) }
643sub _request_sector { shift->_load_header->request_sector( @_ ) }
75be6413 644
badf847c 645################################################################################
75be6413 646
badf847c 647{
648 my %t = (
649 SIG_ARRAY => 'Reference',
650 SIG_HASH => 'Reference',
651 SIG_BLIST => 'BucketList',
652 SIG_INDEX => 'Index',
653 SIG_NULL => 'Null',
654 SIG_DATA => 'Scalar',
655 );
75be6413 656
badf847c 657 my %class_for;
658 while ( my ($k,$v) = each %t ) {
659 $class_for{ DBM::Deep::Engine->$k } = "DBM::Deep::Engine::Sector::$v";
660 }
75be6413 661
badf847c 662 sub load_sector {
663 my $self = shift;
664 my ($offset) = @_;
75be6413 665
badf847c 666 my $data = $self->get_data( $offset )
667 or return;#die "Cannot read from '$offset'\n";
668 my $type = substr( $$data, 0, 1 );
669 my $class = $class_for{ $type };
670 return $class->new({
671 engine => $self,
672 type => $type,
673 offset => $offset,
674 });
675 }
676 *_load_sector = \&load_sector;
75be6413 677
badf847c 678 sub load_header {
679 my $self = shift;
75be6413 680
badf847c 681 #XXX Does this mean we make too many objects? -RobK, 2008-06-23
682 return DBM::Deep::Engine::Sector::FileHeader->new({
683 engine => $self,
684 offset => 0,
685 });
686 }
687 *_load_header = \&load_header;
7b1e1aa1 688
badf847c 689 sub get_data {
690 my $self = shift;
691 my ($offset, $size) = @_;
692 return unless defined $offset;
75be6413 693
badf847c 694 unless ( exists $self->sector_cache->{$offset} ) {
695 # Don't worry about the header sector. It will manage itself.
696 return unless $offset;
2603d86e 697
badf847c 698 if ( !defined $size ) {
699 my $type = $self->storage->read_at( $offset, 1 )
700 or die "($offset): Cannot read from '$offset' to find the type\n";
75be6413 701
badf847c 702 if ( $type eq $self->SIG_FREE ) {
703 return;
704 }
7b1e1aa1 705
badf847c 706 my $class = $class_for{$type}
707 or die "($offset): Cannot find class for '$type'\n";
708 $size = $class->size( $self )
709 or die "($offset): '$class' doesn't return a size\n";
710 $self->sector_cache->{$offset} = $type . $self->storage->read_at( undef, $size - 1 );
711 }
712 else {
713 $self->sector_cache->{$offset} = $self->storage->read_at( $offset, $size )
714 or return;
715 }
716 }
7b1e1aa1 717
badf847c 718 return \$self->sector_cache->{$offset};
7b1e1aa1 719 }
75be6413 720}
721
00d9bd0b 722sub sector_cache {
723 my $self = shift;
724 return $self->{sector_cache} ||= {};
725}
726
727sub clear_sector_cache {
728 my $self = shift;
729 $self->{sector_cache} = {};
730}
731
a8d2331c 732sub dirty_sectors {
733 my $self = shift;
734 return $self->{dirty_sectors} ||= {};
735}
736
00d9bd0b 737sub clear_dirty_sectors {
738 my $self = shift;
739 $self->{dirty_sectors} = {};
740}
741
a8d2331c 742sub add_dirty_sector {
743 my $self = shift;
badf847c 744 my ($offset) = @_;
a8d2331c 745
badf847c 746 $self->dirty_sectors->{ $offset } = undef;
a8d2331c 747}
748
a8d2331c 749sub flush {
750 my $self = shift;
751
00d9bd0b 752 my $sectors = $self->dirty_sectors;
753 for my $offset (sort { $a <=> $b } keys %{ $sectors }) {
badf847c 754 $self->storage->print_at( $offset, $self->sector_cache->{$offset} );
a8d2331c 755 }
756
d451590f 757 # Why do we need to have the storage flush? Shouldn't autoflush take care of things?
758 # -RobK, 2008-06-26
759 $self->storage->flush;
760
a8d2331c 761 $self->clear_dirty_sectors;
00d9bd0b 762
763 $self->clear_sector_cache;
a8d2331c 764}
765
766################################################################################
767
f1879fdc 768sub lock_exclusive {
769 my $self = shift;
770 my ($obj) = @_;
771 return $self->storage->lock_exclusive( $obj );
772}
773
774sub lock_shared {
775 my $self = shift;
776 my ($obj) = @_;
777 return $self->storage->lock_shared( $obj );
778}
779
780sub unlock {
781 my $self = shift;
782 my ($obj) = @_;
a8d2331c 783
784 my $rv = $self->storage->unlock( $obj );
785
786 $self->flush if $rv;
787
788 return $rv;
f1879fdc 789}
790
791################################################################################
792
2120a181 793sub storage { $_[0]{storage} }
794sub byte_size { $_[0]{byte_size} }
795sub hash_size { $_[0]{hash_size} }
796sub hash_chars { $_[0]{hash_chars} }
797sub num_txns { $_[0]{num_txns} }
798sub max_buckets { $_[0]{max_buckets} }
799sub blank_md5 { chr(0) x $_[0]->hash_size }
e9b0b5f0 800sub data_sector_size { $_[0]{data_sector_size} }
801
802# This is a calculated value
803sub txn_bitfield_len {
804 my $self = shift;
805 unless ( exists $self->{txn_bitfield_len} ) {
806 my $temp = ($self->num_txns) / 8;
807 if ( $temp > int( $temp ) ) {
808 $temp = int( $temp ) + 1;
809 }
810 $self->{txn_bitfield_len} = $temp;
811 }
812 return $self->{txn_bitfield_len};
813}
8db25060 814
2120a181 815sub trans_id { $_[0]{trans_id} }
816sub set_trans_id { $_[0]{trans_id} = $_[1] }
8db25060 817
2120a181 818sub trans_loc { $_[0]{trans_loc} }
819sub set_trans_loc { $_[0]{trans_loc} = $_[1] }
820
821sub chains_loc { $_[0]{chains_loc} }
822sub set_chains_loc { $_[0]{chains_loc} = $_[1] }
823
c57b19c6 824sub cache { $_[0]{cache} ||= {} }
825sub clear_cache { %{$_[0]->cache} = () }
826
888453b9 827sub _dump_file {
828 my $self = shift;
00d9bd0b 829 $self->flush;
888453b9 830
831 # Read the header
00d9bd0b 832 my $header_sector = DBM::Deep::Engine::Sector::FileHeader->new({
833 engine => $self,
834 });
888453b9 835
836 my %types = (
837 0 => 'B',
838 1 => 'D',
839 2 => 'I',
840 );
841
842 my %sizes = (
843 'D' => $self->data_sector_size,
844 'B' => DBM::Deep::Engine::Sector::BucketList->new({engine=>$self,offset=>1})->size,
845 'I' => DBM::Deep::Engine::Sector::Index->new({engine=>$self,offset=>1})->size,
846 );
847
848 my $return = "";
c57b19c6 849
a8d2331c 850 # Filesize
851 $return .= "Size: " . (-s $self->storage->{fh}) . $/;
852
c57b19c6 853 # Header values
854 $return .= "NumTxns: " . $self->num_txns . $/;
855
888453b9 856 # Read the free sector chains
857 my %sectors;
858 foreach my $multiple ( 0 .. 2 ) {
859 $return .= "Chains($types{$multiple}):";
860 my $old_loc = $self->chains_loc + $multiple * $self->byte_size;
861 while ( 1 ) {
862 my $loc = unpack(
863 $StP{$self->byte_size},
864 $self->storage->read_at( $old_loc, $self->byte_size ),
865 );
866
867 # We're now out of free sectors of this kind.
868 unless ( $loc ) {
869 last;
870 }
871
872 $sectors{ $types{$multiple} }{ $loc } = undef;
873 $old_loc = $loc + SIG_SIZE + $STALE_SIZE;
874 $return .= " $loc";
875 }
876 $return .= $/;
877 }
878
00d9bd0b 879 my $spot = $header_sector->size;
888453b9 880 SECTOR:
881 while ( $spot < $self->storage->{end} ) {
882 # Read each sector in order.
883 my $sector = $self->_load_sector( $spot );
884 if ( !$sector ) {
885 # Find it in the free-sectors that were found already
886 foreach my $type ( keys %sectors ) {
887 if ( exists $sectors{$type}{$spot} ) {
888 my $size = $sizes{$type};
889 $return .= sprintf "%08d: %s %04d\n", $spot, 'F' . $type, $size;
890 $spot += $size;
891 next SECTOR;
892 }
893 }
894
895 die "********\n$return\nDidn't find free sector for $spot in chains\n********\n";
896 }
897 else {
898 $return .= sprintf "%08d: %s %04d", $spot, $sector->type, $sector->size;
899 if ( $sector->type eq 'D' ) {
900 $return .= ' ' . $sector->data;
901 }
902 elsif ( $sector->type eq 'A' || $sector->type eq 'H' ) {
903 $return .= ' REF: ' . $sector->get_refcount;
904 }
905 elsif ( $sector->type eq 'B' ) {
906 foreach my $bucket ( $sector->chopped_up ) {
907 $return .= "\n ";
908 $return .= sprintf "%08d", unpack($StP{$self->byte_size},
909 substr( $bucket->[-1], $self->hash_size, $self->byte_size),
910 );
c57b19c6 911 my $l = unpack( $StP{$self->byte_size},
912 substr( $bucket->[-1],
913 $self->hash_size + $self->byte_size,
914 $self->byte_size,
915 ),
916 );
917 $return .= sprintf " %08d", $l;
918 foreach my $txn ( 0 .. $self->num_txns - 2 ) {
888453b9 919 my $l = unpack( $StP{$self->byte_size},
920 substr( $bucket->[-1],
c57b19c6 921 $self->hash_size + 2 * $self->byte_size + $txn * ($self->byte_size + $STALE_SIZE),
888453b9 922 $self->byte_size,
923 ),
924 );
925 $return .= sprintf " %08d", $l;
926 }
927 }
928 }
929 $return .= $/;
930
931 $spot += $sector->size;
932 }
933 }
934
935 return $return;
936}
937
a20d9a3f 9381;
939__END__