diff options
author | Eric Wong <e@80x24.org> | 2020-11-13 11:11:41 +0000 |
---|---|---|
committer | Eric Wong <e@80x24.org> | 2020-11-15 02:52:24 +0000 |
commit | 58a964c3c8a2f1699065358e4041b529d3ee531c (patch) | |
tree | 9183d171c3f0053fcdd8193a58b692e837148bef /lib/PublicInbox/IdxStack.pm | |
parent | 0246654927fd9ddc9718ca0d7d5c7e29ce004d5c (diff) | |
download | public-inbox-58a964c3c8a2f1699065358e4041b529d3ee531c.tar.gz |
This will set us up for supporting graceful shutdown on -index without repeating any work.
Diffstat (limited to 'lib/PublicInbox/IdxStack.pm')
-rw-r--r-- | lib/PublicInbox/IdxStack.pm | 16 |
1 files changed, 11 insertions, 5 deletions
diff --git a/lib/PublicInbox/IdxStack.pm b/lib/PublicInbox/IdxStack.pm index ce75b46a..e7e10de9 100644 --- a/lib/PublicInbox/IdxStack.pm +++ b/lib/PublicInbox/IdxStack.pm @@ -6,7 +6,7 @@ package PublicInbox::IdxStack; use v5.10.1; use strict; use Fcntl qw(:seek); -use constant FMT => eval { pack('Q', 1) } ? 'A1QQH*' : 'A1IIH*'; +use constant PACK_FMT => eval { pack('Q', 1) } ? 'A1QQH*H*' : 'A1IIH*H*'; # start off in write-only mode sub new { @@ -16,9 +16,15 @@ sub new { # file_char = [d|m] sub push_rec { - my ($self, $file_char, $at, $ct, $blob_oid) = @_; - my $rec = pack(FMT, $file_char, $at, $ct, $blob_oid); - $self->{rec_size} //= length($rec); + my ($self, $file_char, $at, $ct, $blob_oid, $cmt_oid) = @_; + my $rec = pack(PACK_FMT, $file_char, $at, $ct, $blob_oid, $cmt_oid); + $self->{unpack_fmt} //= do { + my $len = length($cmt_oid); + my $fmt = PACK_FMT; + $fmt =~ s/H\*/H$len/g; + $self->{rec_size} = length($rec); + $fmt; + }; print { $self->{wr} } $rec or die "print: $!"; $self->{tot_size} += length($rec); } @@ -46,7 +52,7 @@ sub pop_rec { my $r = read($io, my $buf, $sz); defined($r) or die "read: $!"; $r == $sz or die "read($r != $sz)"; - unpack(FMT, $buf); + unpack($self->{unpack_fmt}, $buf); } 1; |