aboutsummaryrefslogtreecommitdiffstats
path: root/yt/cpp/mapreduce/interface/temp.h
blob: 08a87eeafb44ee91134762ff8e1ae40c5a802bc7 (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
71
72
73
74
75
76
#pragma once

#include <yt/cpp/mapreduce/interface/client.h>

namespace NYT {

////////////////////////////////////////////////////////////////////////////////

///
/// @brief RAII class for working with temporary tables. Constructor of this class creates temporary table and destructor removes it.
///
/// CAVEAT: when using with transactions destructor of TTempTable should be called before commiting transaction.
class TTempTable
{
public:
    ///
    /// @brief Constructor creates temporary table
    ///
    /// @param client -- YT client or transaction object.
    /// @param prefix -- table name prefix
    /// @param directory -- path to directory where temporary table will be created.
    /// @param options -- options to be passed for table creation (might be useful if we want to set table attributes)
    explicit TTempTable(
        IClientBasePtr client,
        const TString& prefix = {},
        const TYPath& directory = {},
        const TCreateOptions& options = {});

    TTempTable(const TTempTable&) = delete;
    TTempTable& operator=(const TTempTable&) = delete;

    TTempTable(TTempTable&&);
    TTempTable& operator=(TTempTable&&);

    ~TTempTable();

    ///
    /// @brief Create table with given path that will be autoremoved in RAII manner.
    ///
    /// In contrast to TTempTable constructor this function uses provided path to table and don't modify it in any manner.
    /// It's user responsibility to provide unique path.
    static TTempTable CreateAutoremovingTable(IClientBasePtr client, TYPath path, const TCreateOptions& options);

    /// Return full path to the table.
    TString Name() const &;
    TString Name() && = delete;

    /// Release table and return its path. Table will not be deleted by TTempTable destructor after this call.
    TString Release();

private:
    struct TPrivateConstuctorTag
    { };
    TTempTable(TPrivateConstuctorTag, IClientBasePtr client, TYPath path, const TCreateOptions& options);

private:
    IClientBasePtr Client_;
    TYPath Name_;
    bool Owns_ = true;

private:
    void RemoveTable();
};

///
/// @brief Create empty table with unique name in given directory and return its path.
///
/// @param client -- YT client or transaction object.
/// @param prefix -- table name prefix
/// @param directory -- path to directory where temporary table will be created.
/// @param options -- options to be passed for table creation (might be useful if we want to set table attributes)
TYPath CreateTempTable(const IClientBasePtr& client, const TString& prefix, const TYPath& directory, const TCreateOptions& options);

////////////////////////////////////////////////////////////////////////////////

} // namespace NYT