blob: 2ce2eee97db63a5182cfad77c2b2a8a75ffd7893 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once
#include <vespa/vespalib/util/count_down_latch.h>
#include <atomic>
#include <vector>
namespace search { class IFlushToken; }
namespace vespalib { class Executor; }
namespace search::diskindex {
class FieldMerger;
class FusionOutputIndex;
/*
* This class has ownership of active field mergers until they are
* done or failed.
*/
class FieldMergersState {
const FusionOutputIndex& _fusion_out_index;
vespalib::Executor& _executor;
std::shared_ptr<IFlushToken> _flush_token;
vespalib::CountDownLatch _done;
std::atomic<uint32_t> _failed;
std::vector<std::unique_ptr<FieldMerger>> _field_mergers;
void destroy_field_merger(FieldMerger& field_merger);
public:
FieldMergersState(const FusionOutputIndex& fusion_out_index, vespalib::Executor& executor, std::shared_ptr<IFlushToken> flush_token);
~FieldMergersState();
FieldMerger& alloc_field_merger(uint32_t id);
void field_merger_done(FieldMerger& field_merger, bool failed);
void wait_field_mergers_done();
void schedule_task(FieldMerger& field_merger);
uint32_t get_failed() const noexcept { return _failed; }
};
}
|
