read_txn_slots has been moved into the FileHeader sector.
[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;
40963fba 570 return $self->_load_header->read_txn_slots;
2120a181 571}
20f7b20c 572
2120a181 573sub write_txn_slots {
574 my $self = shift;
e9b0b5f0 575 my $num_bits = $self->txn_bitfield_len * 8;
2120a181 576 $self->storage->print_at( $self->trans_loc,
e9b0b5f0 577 pack( 'b'.$num_bits, join('', @_) ),
7dcefff3 578 );
2120a181 579}
580
581sub get_running_txn_ids {
582 my $self = shift;
583 my @transactions = $self->read_txn_slots;
e9b0b5f0 584 my @trans_ids = map { $_+1} grep { $transactions[$_] } 0 .. $#transactions;
2120a181 585}
586
587sub get_txn_staleness_counter {
588 my $self = shift;
589 my ($trans_id) = @_;
20f7b20c 590
2120a181 591 # Hardcode staleness of 0 for the HEAD
592 return 0 unless $trans_id;
f37c15ab 593
e9b0b5f0 594 return unpack( $StP{$STALE_SIZE},
2120a181 595 $self->storage->read_at(
888453b9 596 $self->trans_loc + $self->txn_bitfield_len + $STALE_SIZE * ($trans_id - 1),
597 $STALE_SIZE,
2120a181 598 )
599 );
d4b1166e 600}
601
2120a181 602sub inc_txn_staleness_counter {
d4b1166e 603 my $self = shift;
2120a181 604 my ($trans_id) = @_;
20f7b20c 605
2120a181 606 # Hardcode staleness of 0 for the HEAD
888453b9 607 return 0 unless $trans_id;
20f7b20c 608
2120a181 609 $self->storage->print_at(
888453b9 610 $self->trans_loc + $self->txn_bitfield_len + $STALE_SIZE * ($trans_id - 1),
e9b0b5f0 611 pack( $StP{$STALE_SIZE}, $self->get_txn_staleness_counter( $trans_id ) + 1 ),
86867f3a 612 );
2120a181 613}
20f7b20c 614
2120a181 615sub get_entries {
616 my $self = shift;
617 return [ keys %{ $self->{entries}{$self->trans_id} ||= {} } ];
d4b1166e 618}
619
2120a181 620sub add_entry {
ea2f6d67 621 my $self = shift;
2120a181 622 my ($trans_id, $loc) = @_;
ea2f6d67 623
2120a181 624 $self->{entries}{$trans_id} ||= {};
625 $self->{entries}{$trans_id}{$loc} = undef;
626}
ea2f6d67 627
2120a181 628# If the buckets are being relocated because of a reindexing, the entries
629# mechanism needs to be made aware of it.
630sub reindex_entry {
631 my $self = shift;
632 my ($old_loc, $new_loc) = @_;
633
634 TRANS:
635 while ( my ($trans_id, $locs) = each %{ $self->{entries} } ) {
695c88b1 636 if ( exists $locs->{$old_loc} ) {
637 delete $locs->{$old_loc};
638 $locs->{$new_loc} = undef;
639 next TRANS;
2120a181 640 }
ea2f6d67 641 }
ea2f6d67 642}
643
2120a181 644sub clear_entries {
d4b1166e 645 my $self = shift;
2120a181 646 delete $self->{entries}{$self->trans_id};
647}
eea0d863 648
2120a181 649################################################################################
eea0d863 650
2120a181 651sub _apply_digest {
75be6413 652 my $self = shift;
2120a181 653 return $self->{digest}->(@_);
654}
16d1ad9b 655
2120a181 656sub _add_free_blist_sector { shift->_add_free_sector( 0, @_ ) }
badf847c 657sub _add_free_data_sector { shift->_add_free_sector( 1, @_ ) }
2120a181 658sub _add_free_index_sector { shift->_add_free_sector( 2, @_ ) }
badf847c 659sub _add_free_sector { shift->_load_header->add_free_sector( @_ ) }
75be6413 660
badf847c 661sub _request_blist_sector { shift->_request_sector( 0, @_ ) }
662sub _request_data_sector { shift->_request_sector( 1, @_ ) }
663sub _request_index_sector { shift->_request_sector( 2, @_ ) }
664sub _request_sector { shift->_load_header->request_sector( @_ ) }
75be6413 665
badf847c 666################################################################################
75be6413 667
badf847c 668{
669 my %t = (
670 SIG_ARRAY => 'Reference',
671 SIG_HASH => 'Reference',
672 SIG_BLIST => 'BucketList',
673 SIG_INDEX => 'Index',
674 SIG_NULL => 'Null',
675 SIG_DATA => 'Scalar',
676 );
75be6413 677
badf847c 678 my %class_for;
679 while ( my ($k,$v) = each %t ) {
680 $class_for{ DBM::Deep::Engine->$k } = "DBM::Deep::Engine::Sector::$v";
681 }
75be6413 682
badf847c 683 sub load_sector {
684 my $self = shift;
685 my ($offset) = @_;
75be6413 686
badf847c 687 my $data = $self->get_data( $offset )
688 or return;#die "Cannot read from '$offset'\n";
689 my $type = substr( $$data, 0, 1 );
690 my $class = $class_for{ $type };
691 return $class->new({
692 engine => $self,
693 type => $type,
694 offset => $offset,
695 });
696 }
697 *_load_sector = \&load_sector;
75be6413 698
badf847c 699 sub load_header {
700 my $self = shift;
75be6413 701
badf847c 702 #XXX Does this mean we make too many objects? -RobK, 2008-06-23
703 return DBM::Deep::Engine::Sector::FileHeader->new({
704 engine => $self,
705 offset => 0,
706 });
707 }
708 *_load_header = \&load_header;
7b1e1aa1 709
badf847c 710 sub get_data {
711 my $self = shift;
712 my ($offset, $size) = @_;
713 return unless defined $offset;
75be6413 714
badf847c 715 unless ( exists $self->sector_cache->{$offset} ) {
716 # Don't worry about the header sector. It will manage itself.
717 return unless $offset;
2603d86e 718
badf847c 719 if ( !defined $size ) {
720 my $type = $self->storage->read_at( $offset, 1 )
721 or die "($offset): Cannot read from '$offset' to find the type\n";
75be6413 722
badf847c 723 if ( $type eq $self->SIG_FREE ) {
724 return;
725 }
7b1e1aa1 726
badf847c 727 my $class = $class_for{$type}
728 or die "($offset): Cannot find class for '$type'\n";
729 $size = $class->size( $self )
730 or die "($offset): '$class' doesn't return a size\n";
731 $self->sector_cache->{$offset} = $type . $self->storage->read_at( undef, $size - 1 );
732 }
733 else {
734 $self->sector_cache->{$offset} = $self->storage->read_at( $offset, $size )
735 or return;
736 }
737 }
7b1e1aa1 738
badf847c 739 return \$self->sector_cache->{$offset};
7b1e1aa1 740 }
75be6413 741}
742
00d9bd0b 743sub sector_cache {
744 my $self = shift;
745 return $self->{sector_cache} ||= {};
746}
747
748sub clear_sector_cache {
749 my $self = shift;
750 $self->{sector_cache} = {};
751}
752
a8d2331c 753sub dirty_sectors {
754 my $self = shift;
755 return $self->{dirty_sectors} ||= {};
756}
757
00d9bd0b 758sub clear_dirty_sectors {
759 my $self = shift;
760 $self->{dirty_sectors} = {};
761}
762
a8d2331c 763sub add_dirty_sector {
764 my $self = shift;
badf847c 765 my ($offset) = @_;
a8d2331c 766
badf847c 767 $self->dirty_sectors->{ $offset } = undef;
a8d2331c 768}
769
a8d2331c 770sub flush {
771 my $self = shift;
772
00d9bd0b 773 my $sectors = $self->dirty_sectors;
774 for my $offset (sort { $a <=> $b } keys %{ $sectors }) {
badf847c 775 $self->storage->print_at( $offset, $self->sector_cache->{$offset} );
a8d2331c 776 }
777
778 $self->clear_dirty_sectors;
00d9bd0b 779
780 $self->clear_sector_cache;
a8d2331c 781}
782
783################################################################################
784
f1879fdc 785sub lock_exclusive {
786 my $self = shift;
787 my ($obj) = @_;
788 return $self->storage->lock_exclusive( $obj );
789}
790
791sub lock_shared {
792 my $self = shift;
793 my ($obj) = @_;
794 return $self->storage->lock_shared( $obj );
795}
796
797sub unlock {
798 my $self = shift;
799 my ($obj) = @_;
a8d2331c 800
801 my $rv = $self->storage->unlock( $obj );
802
803 $self->flush if $rv;
804
805 return $rv;
f1879fdc 806}
807
808################################################################################
809
2120a181 810sub storage { $_[0]{storage} }
811sub byte_size { $_[0]{byte_size} }
812sub hash_size { $_[0]{hash_size} }
813sub hash_chars { $_[0]{hash_chars} }
814sub num_txns { $_[0]{num_txns} }
815sub max_buckets { $_[0]{max_buckets} }
816sub blank_md5 { chr(0) x $_[0]->hash_size }
e9b0b5f0 817sub data_sector_size { $_[0]{data_sector_size} }
818
819# This is a calculated value
820sub txn_bitfield_len {
821 my $self = shift;
822 unless ( exists $self->{txn_bitfield_len} ) {
823 my $temp = ($self->num_txns) / 8;
824 if ( $temp > int( $temp ) ) {
825 $temp = int( $temp ) + 1;
826 }
827 $self->{txn_bitfield_len} = $temp;
828 }
829 return $self->{txn_bitfield_len};
830}
8db25060 831
2120a181 832sub trans_id { $_[0]{trans_id} }
833sub set_trans_id { $_[0]{trans_id} = $_[1] }
8db25060 834
2120a181 835sub trans_loc { $_[0]{trans_loc} }
836sub set_trans_loc { $_[0]{trans_loc} = $_[1] }
837
838sub chains_loc { $_[0]{chains_loc} }
839sub set_chains_loc { $_[0]{chains_loc} = $_[1] }
840
c57b19c6 841sub cache { $_[0]{cache} ||= {} }
842sub clear_cache { %{$_[0]->cache} = () }
843
888453b9 844sub _dump_file {
845 my $self = shift;
00d9bd0b 846 $self->flush;
888453b9 847
848 # Read the header
00d9bd0b 849 my $header_sector = DBM::Deep::Engine::Sector::FileHeader->new({
850 engine => $self,
851 });
888453b9 852
853 my %types = (
854 0 => 'B',
855 1 => 'D',
856 2 => 'I',
857 );
858
859 my %sizes = (
860 'D' => $self->data_sector_size,
861 'B' => DBM::Deep::Engine::Sector::BucketList->new({engine=>$self,offset=>1})->size,
862 'I' => DBM::Deep::Engine::Sector::Index->new({engine=>$self,offset=>1})->size,
863 );
864
865 my $return = "";
c57b19c6 866
a8d2331c 867 # Filesize
868 $return .= "Size: " . (-s $self->storage->{fh}) . $/;
869
c57b19c6 870 # Header values
871 $return .= "NumTxns: " . $self->num_txns . $/;
872
888453b9 873 # Read the free sector chains
874 my %sectors;
875 foreach my $multiple ( 0 .. 2 ) {
876 $return .= "Chains($types{$multiple}):";
877 my $old_loc = $self->chains_loc + $multiple * $self->byte_size;
878 while ( 1 ) {
879 my $loc = unpack(
880 $StP{$self->byte_size},
881 $self->storage->read_at( $old_loc, $self->byte_size ),
882 );
883
884 # We're now out of free sectors of this kind.
885 unless ( $loc ) {
886 last;
887 }
888
889 $sectors{ $types{$multiple} }{ $loc } = undef;
890 $old_loc = $loc + SIG_SIZE + $STALE_SIZE;
891 $return .= " $loc";
892 }
893 $return .= $/;
894 }
895
00d9bd0b 896 my $spot = $header_sector->size;
888453b9 897 SECTOR:
898 while ( $spot < $self->storage->{end} ) {
899 # Read each sector in order.
900 my $sector = $self->_load_sector( $spot );
901 if ( !$sector ) {
902 # Find it in the free-sectors that were found already
903 foreach my $type ( keys %sectors ) {
904 if ( exists $sectors{$type}{$spot} ) {
905 my $size = $sizes{$type};
906 $return .= sprintf "%08d: %s %04d\n", $spot, 'F' . $type, $size;
907 $spot += $size;
908 next SECTOR;
909 }
910 }
911
912 die "********\n$return\nDidn't find free sector for $spot in chains\n********\n";
913 }
914 else {
915 $return .= sprintf "%08d: %s %04d", $spot, $sector->type, $sector->size;
916 if ( $sector->type eq 'D' ) {
917 $return .= ' ' . $sector->data;
918 }
919 elsif ( $sector->type eq 'A' || $sector->type eq 'H' ) {
920 $return .= ' REF: ' . $sector->get_refcount;
921 }
922 elsif ( $sector->type eq 'B' ) {
923 foreach my $bucket ( $sector->chopped_up ) {
924 $return .= "\n ";
925 $return .= sprintf "%08d", unpack($StP{$self->byte_size},
926 substr( $bucket->[-1], $self->hash_size, $self->byte_size),
927 );
c57b19c6 928 my $l = unpack( $StP{$self->byte_size},
929 substr( $bucket->[-1],
930 $self->hash_size + $self->byte_size,
931 $self->byte_size,
932 ),
933 );
934 $return .= sprintf " %08d", $l;
935 foreach my $txn ( 0 .. $self->num_txns - 2 ) {
888453b9 936 my $l = unpack( $StP{$self->byte_size},
937 substr( $bucket->[-1],
c57b19c6 938 $self->hash_size + 2 * $self->byte_size + $txn * ($self->byte_size + $STALE_SIZE),
888453b9 939 $self->byte_size,
940 ),
941 );
942 $return .= sprintf " %08d", $l;
943 }
944 }
945 }
946 $return .= $/;
947
948 $spot += $sector->size;
949 }
950 }
951
952 return $return;
953}
954
a20d9a3f 9551;
956__END__