329818b1b0
Summary: The dumper is now a function and doesn't have to worry about any state. The function streams the Cypher queries directly to the client. This diff also makes the dumper work with storage v2. Reviewers: teon.banek Reviewed By: teon.banek Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D2545
57 lines
1.7 KiB
C++
57 lines
1.7 KiB
C++
#pragma once
|
|
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#include "query/typed_value.hpp"
|
|
#include "utils/memory.hpp"
|
|
|
|
namespace query {
|
|
|
|
/**
|
|
* `AnyStream` can wrap *any* type implementing the `Stream` concept into a
|
|
* single type.
|
|
*
|
|
* The type erasure technique is used. The original type which an `AnyStream`
|
|
* was constructed from is "erased", as `AnyStream` is not a class template and
|
|
* doesn't use the type in any way. Client code can then program just for
|
|
* `AnyStream`, rather than using static polymorphism to handle any type
|
|
* implementing the `Stream` concept.
|
|
*/
|
|
class AnyStream final {
|
|
public:
|
|
template <class TStream>
|
|
AnyStream(TStream *stream, utils::MemoryResource *memory_resource)
|
|
: content_{utils::Allocator<GenericWrapper<TStream>>{memory_resource}
|
|
.template new_object<GenericWrapper<TStream>>(stream),
|
|
[memory_resource](Wrapper *ptr) {
|
|
utils::Allocator<GenericWrapper<TStream>>{memory_resource}
|
|
.template delete_object<GenericWrapper<TStream>>(
|
|
static_cast<GenericWrapper<TStream> *>(ptr));
|
|
}} {}
|
|
|
|
void Result(const std::vector<TypedValue> &values) {
|
|
content_->Result(values);
|
|
}
|
|
|
|
private:
|
|
struct Wrapper {
|
|
virtual void Result(const std::vector<TypedValue> &values) = 0;
|
|
};
|
|
|
|
template <class TStream>
|
|
struct GenericWrapper final : public Wrapper {
|
|
explicit GenericWrapper(TStream *stream) : stream_{stream} {}
|
|
|
|
void Result(const std::vector<TypedValue> &values) override {
|
|
stream_->Result(values);
|
|
}
|
|
|
|
TStream *stream_;
|
|
};
|
|
|
|
std::unique_ptr<Wrapper, std::function<void(Wrapper *)>> content_;
|
|
};
|
|
|
|
} // namespace query
|