1#pragma once
2
3#include <map>
4#include <shared_mutex>
5
6#include <ext/shared_ptr_helper.h>
7
8#include <Poco/File.h>
9
10#include <Storages/IStorage.h>
11#include <Common/FileChecker.h>
12#include <Common/escapeForFileName.h>
13#include <Core/Defines.h>
14
15
16namespace DB
17{
18
19/** Implements a table engine that is suitable for small chunks of the log.
20 * In doing so, stores all the columns in a single Native file, with a nearby index.
21 */
22class StorageStripeLog : public ext::shared_ptr_helper<StorageStripeLog>, public IStorage
23{
24friend class StripeLogBlockInputStream;
25friend class StripeLogBlockOutputStream;
26friend struct ext::shared_ptr_helper<StorageStripeLog>;
27
28public:
29 std::string getName() const override { return "StripeLog"; }
30 std::string getTableName() const override { return table_name; }
31 std::string getDatabaseName() const override { return database_name; }
32
33 BlockInputStreams read(
34 const Names & column_names,
35 const SelectQueryInfo & query_info,
36 const Context & context,
37 QueryProcessingStage::Enum processed_stage,
38 size_t max_block_size,
39 unsigned num_streams) override;
40
41 BlockOutputStreamPtr write(const ASTPtr & query, const Context & context) override;
42
43 void rename(const String & new_path_to_table_data, const String & new_database_name, const String & new_table_name, TableStructureWriteLockHolder &) override;
44
45 CheckResults checkData(const ASTPtr & /* query */, const Context & /* context */) override;
46
47 /// Data of the file.
48 struct ColumnData
49 {
50 Poco::File data_file;
51 };
52 using Files_t = std::map<String, ColumnData>;
53
54 std::string fullPath() const { return path; }
55
56 Strings getDataPaths() const override { return {fullPath()}; }
57
58 void truncate(const ASTPtr &, const Context &, TableStructureWriteLockHolder &) override;
59
60private:
61 String base_path;
62 String path;
63 String table_name;
64 String database_name;
65
66 size_t max_compress_block_size;
67
68 FileChecker file_checker;
69 mutable std::shared_mutex rwlock;
70
71 Logger * log;
72
73protected:
74 StorageStripeLog(
75 const std::string & relative_path_,
76 const std::string & database_name_,
77 const std::string & table_name_,
78 const ColumnsDescription & columns_,
79 const ConstraintsDescription & constraints_,
80 bool attach,
81 size_t max_compress_block_size_,
82 const Context & context_);
83};
84
85}
86