forked from envoyproxy/nighthawk
-
Notifications
You must be signed in to change notification settings - Fork 0
/
sink_impl.h
35 lines (28 loc) · 965 Bytes
/
sink_impl.h
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
#pragma once
#include "nighthawk/sink/sink.h"
#include "absl/container/flat_hash_map.h"
namespace Nighthawk {
/**
* Filesystem based implementation of Sink. Uses /tmp/nh/{execution_id}/ to store and load
* data.
*/
class FileSinkImpl : public Sink {
public:
absl::Status
StoreExecutionResultPiece(const nighthawk::client::ExecutionResponse& response) override;
absl::StatusOr<std::vector<nighthawk::client::ExecutionResponse>>
LoadExecutionResult(absl::string_view id) const override;
};
/**
* Memory based implementation of Sink.
*/
class InMemorySinkImpl : public Sink {
public:
absl::Status
StoreExecutionResultPiece(const nighthawk::client::ExecutionResponse& response) override;
absl::StatusOr<std::vector<nighthawk::client::ExecutionResponse>>
LoadExecutionResult(absl::string_view id) const override;
private:
absl::flat_hash_map<std::string, std::vector<nighthawk::client::ExecutionResponse>> map_;
};
} // namespace Nighthawk