blob: d9a26542e599e80db2b4154bf9e1a8db544c44a2 (
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
41
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once
#include "bucketexecutor.h"
namespace storage::spi {
/**
* Simple Bucket task that wraps a lambda that does the job.
*/
template<class FunctionType, class FailedFunction>
class LambdaBucketTask : public BucketTask {
public:
explicit LambdaBucketTask(FunctionType &&func, FailedFunction &&failed)
: _func(std::move(func)),
_failed(std::move(failed))
{}
~LambdaBucketTask() override = default;
void run(const Bucket & bucket, std::shared_ptr<vespalib::IDestructorCallback> onComplete) override {
_func(bucket, std::move(onComplete));
}
void fail(const Bucket & bucket) override {
_failed(bucket);
}
private:
FunctionType _func;
FailedFunction _failed;
};
template<class FunctionType, class FailedFunction>
std::unique_ptr<BucketTask>
makeBucketTask(FunctionType &&function, FailedFunction && failed) {
return std::make_unique<LambdaBucketTask<std::decay_t<FunctionType>, std::decay_t<FailedFunction>>>
(std::forward<FunctionType>(function), std::forward<FailedFunction>(failed));
}
}
|
