Directory: | cvmfs/ |
---|---|
File: | cvmfs/ingestion/task_compress.cc |
Date: | 2025-06-08 02:35:55 |
Exec | Total | Coverage | |
---|---|---|---|
Lines: | 42 | 42 | 100.0% |
Branches: | 45 | 72 | 62.5% |
Line | Branch | Exec | Source |
---|---|---|---|
1 | /** | ||
2 | * This file is part of the CernVM File System. | ||
3 | */ | ||
4 | |||
5 | |||
6 | #include "task_compress.h" | ||
7 | |||
8 | #include <cstdlib> | ||
9 | |||
10 | #include "compression/compression.h" | ||
11 | #include "util/logging.h" | ||
12 | #include "util/smalloc.h" | ||
13 | |||
14 | |||
15 | /** | ||
16 | * The data payload of the blocks is replaced by their compressed counterparts. | ||
17 | * The block tags stay the same. | ||
18 | * TODO(jblomer): avoid memory copy with EchoCompressor | ||
19 | */ | ||
20 | 578612 | void TaskCompress::Process(BlockItem *input_block) { | |
21 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 578069 times.
|
578612 | assert(input_block->chunk_item() != NULL); |
22 | |||
23 |
1/2✓ Branch 2 taken 579472 times.
✗ Branch 3 not taken.
|
578069 | zlib::Compressor *compressor = input_block->chunk_item()->GetCompressor(); |
24 | 579472 | const int64_t tag = input_block->tag(); | |
25 | 578985 | const bool flush = input_block->type() == BlockItem::kBlockStop; | |
26 | 579141 | unsigned char *input_data = input_block->data(); | |
27 | 578960 | size_t remaining_in_input = input_block->size(); | |
28 | |||
29 | 579157 | BlockItem *output_block = NULL; | |
30 |
3/4✓ Branch 1 taken 580284 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 249736 times.
✓ Branch 4 taken 330548 times.
|
579157 | if (!tag_map_.Lookup(tag, &output_block)) { |
31 | // So far unseen chunk, start new stream of compressed blocks | ||
32 |
2/4✓ Branch 1 taken 249779 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 249510 times.
✗ Branch 5 not taken.
|
249736 | output_block = new BlockItem(tag, allocator_); |
33 |
1/2✓ Branch 2 taken 249425 times.
✗ Branch 3 not taken.
|
249510 | output_block->SetFileItem(input_block->file_item()); |
34 |
1/2✓ Branch 2 taken 249406 times.
✗ Branch 3 not taken.
|
249425 | output_block->SetChunkItem(input_block->chunk_item()); |
35 |
1/2✓ Branch 1 taken 250380 times.
✗ Branch 2 not taken.
|
249406 | output_block->MakeData(kCompressedBlockSize); |
36 |
1/2✓ Branch 1 taken 250600 times.
✗ Branch 2 not taken.
|
250380 | tag_map_.Insert(tag, output_block); |
37 | } | ||
38 | |||
39 | 581148 | bool done = false; | |
40 | do { | ||
41 | 649314 | unsigned char *output_data = output_block->data() + output_block->size(); | |
42 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 648669 times.
|
648707 | assert(!output_block->IsFull()); |
43 | 648669 | size_t remaining_in_output = output_block->capacity() | |
44 | 648609 | - output_block->size(); | |
45 | |||
46 |
1/2✓ Branch 1 taken 649642 times.
✗ Branch 2 not taken.
|
648678 | done = compressor->Deflate(flush, &input_data, &remaining_in_input, |
47 | &output_data, &remaining_in_output); | ||
48 | // remaining_in_output is now number of consumed bytes | ||
49 | 649642 | output_block->set_size(output_block->size() + remaining_in_output); | |
50 | |||
51 |
2/2✓ Branch 1 taken 135511 times.
✓ Branch 2 taken 513147 times.
|
648928 | if (output_block->IsFull()) { |
52 |
1/2✓ Branch 1 taken 135549 times.
✗ Branch 2 not taken.
|
135511 | tubes_out_->Dispatch(output_block); |
53 |
2/4✓ Branch 1 taken 135551 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 135493 times.
✗ Branch 5 not taken.
|
135549 | output_block = new BlockItem(tag, allocator_); |
54 |
1/2✓ Branch 2 taken 135474 times.
✗ Branch 3 not taken.
|
135493 | output_block->SetFileItem(input_block->file_item()); |
55 |
1/2✓ Branch 2 taken 135464 times.
✗ Branch 3 not taken.
|
135474 | output_block->SetChunkItem(input_block->chunk_item()); |
56 |
1/2✓ Branch 1 taken 135617 times.
✗ Branch 2 not taken.
|
135464 | output_block->MakeData(kCompressedBlockSize); |
57 |
1/2✓ Branch 1 taken 135602 times.
✗ Branch 2 not taken.
|
135617 | tag_map_.Insert(tag, output_block); |
58 | } | ||
59 |
6/6✓ Branch 0 taken 67702 times.
✓ Branch 1 taken 581047 times.
✓ Branch 2 taken 250438 times.
✓ Branch 3 taken 330609 times.
✓ Branch 4 taken 464 times.
✓ Branch 5 taken 249974 times.
|
648749 | } while ((remaining_in_input > 0) || (flush && !done)); |
60 | |||
61 |
2/2✓ Branch 0 taken 249977 times.
✓ Branch 1 taken 330606 times.
|
580583 | if (flush) { |
62 |
1/2✓ Branch 2 taken 249879 times.
✗ Branch 3 not taken.
|
249977 | input_block->chunk_item()->ReleaseCompressor(); |
63 | |||
64 |
2/2✓ Branch 1 taken 249838 times.
✓ Branch 2 taken 2 times.
|
249879 | if (output_block->size() > 0) |
65 |
1/2✓ Branch 1 taken 249904 times.
✗ Branch 2 not taken.
|
249838 | tubes_out_->Dispatch(output_block); |
66 | else | ||
67 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1 times.
|
2 | delete output_block; |
68 |
1/2✓ Branch 1 taken 249858 times.
✗ Branch 2 not taken.
|
249906 | tag_map_.Erase(tag); |
69 | |||
70 |
2/4✓ Branch 1 taken 250211 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 249704 times.
✗ Branch 5 not taken.
|
249858 | BlockItem *stop_block = new BlockItem(tag, allocator_); |
71 |
1/2✓ Branch 1 taken 249535 times.
✗ Branch 2 not taken.
|
249704 | stop_block->MakeStop(); |
72 |
1/2✓ Branch 2 taken 249448 times.
✗ Branch 3 not taken.
|
249535 | stop_block->SetFileItem(input_block->file_item()); |
73 |
1/2✓ Branch 2 taken 249423 times.
✗ Branch 3 not taken.
|
249448 | stop_block->SetChunkItem(input_block->chunk_item()); |
74 |
1/2✓ Branch 1 taken 249606 times.
✗ Branch 2 not taken.
|
249423 | tubes_out_->Dispatch(stop_block); |
75 | } | ||
76 | |||
77 |
2/2✓ Branch 0 taken 579929 times.
✓ Branch 1 taken 283 times.
|
580212 | delete input_block; |
78 | 581129 | } | |
79 |