blob: 4d65e012716f769cc8b30377989b0f166db87a10 (
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
#pragma once
#include "fwd.h"
#include <util/generic/ptr.h>
namespace NYql::NPureCalc::NPrivate {
template <typename TNew, typename TOld, typename TFunctor>
class TMappingStream final: public IStream<TNew> {
private:
THolder<IStream<TOld>> Old_;
TFunctor Functor_;
public:
TMappingStream(THolder<IStream<TOld>> old, TFunctor functor)
: Old_(std::move(old))
, Functor_(std::move(functor))
{
}
public:
TNew Fetch() override {
return Functor_(Old_->Fetch());
}
};
template <typename TNew, typename TOld, typename TFunctor>
class TMappingConsumer final: public IConsumer<TNew> {
private:
THolder<IConsumer<TOld>> Old_;
TFunctor Functor_;
public:
TMappingConsumer(THolder<IConsumer<TOld>> old, TFunctor functor)
: Old_(std::move(old))
, Functor_(std::move(functor))
{
}
public:
void OnObject(TNew object) override {
Old_->OnObject(Functor_(object));
}
void OnFinish() override {
Old_->OnFinish();
}
};
template <typename T, typename C>
class TNonOwningConsumer final: public IConsumer<T> {
private:
C Consumer;
public:
explicit TNonOwningConsumer(const C& consumer)
: Consumer(consumer)
{
}
public:
void OnObject(T t) override {
Consumer->OnObject(t);
}
void OnFinish() override {
Consumer->OnFinish();
}
};
}
|
