aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/yt/error/error_attributes.h
blob: 88771f836019ceb1bc5f5198967aa2cfdfd7f936 (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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
#pragma once

#include "mergeable_dictionary.h"

#include <library/cpp/yt/misc/optional.h>

namespace NYT {

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

// For now this is just an opaque handle to error attributes
// used to remove dependency on IAttributeDictionary in public API.
// Eventually it would be a simple hash map.
// NB(arkady-e1ppa): For now most methods are defined in yt/yt/core/misc/stripped_error.cpp
// eventually they will be moved here.
class TErrorAttributes
{
public:
    using TKey = TString;
    using TValue = NYson::TYsonString;
    using TKeyValuePair = std::pair<TKey, TValue>;

    //! Returns the list of all keys in the dictionary.
    std::vector<TString> ListKeys() const;

    //! Returns the list of all key-value pairs in the dictionary.
    std::vector<TKeyValuePair> ListPairs() const;

    //! Returns the value of the attribute (null indicates that the attribute is not found).
    NYson::TYsonString FindYson(TStringBuf key) const;

    //! Sets the value of the attribute.
    void SetYson(const TString& key, const NYson::TYsonString& value);

    //! Removes the attribute.
    //! Returns |true| if the attribute was removed or |false| if there is no attribute with this key.
    bool Remove(const TString& key);

    //! Removes all attributes.
    void Clear();

    //! Returns the value of the attribute (throws an exception if the attribute is not found).
    NYson::TYsonString GetYson(TStringBuf key) const;

    //! Same as #GetYson but removes the value.
    NYson::TYsonString GetYsonAndRemove(const TString& key);

    //! Returns |true| iff the given key is present.
    bool Contains(TStringBuf key) const;

    // TODO(arkady-e1ppa): By default deserialization is located at yt/core
    // consider using deserialization of some default types (guid, string, int, double)
    // to be supported and everything else not supported without inclusion of yt/core.
    //! Finds the attribute and deserializes its value.
    //! Throws if no such value is found.
    template <class T>
    T Get(TStringBuf key) const;

    //! Same as #Get but removes the value.
    template <class T>
    T GetAndRemove(const TString& key);

    //! Finds the attribute and deserializes its value.
    //! Uses default value if no such attribute is found.
    template <class T>
    T Get(TStringBuf key, const T& defaultValue) const;

    //! Same as #Get but removes the value if it exists.
    template <class T>
    T GetAndRemove(const TString& key, const T& defaultValue);

    //! Finds the attribute and deserializes its value.
    //! Returns null if no such attribute is found.
    template <class T>
    typename TOptionalTraits<T>::TOptional Find(TStringBuf key) const;

    //! Same as #Find but removes the value if it exists.
    template <class T>
    typename TOptionalTraits<T>::TOptional FindAndRemove(const TString& key);

    template <CMergeableDictionary TDictionary>
    void MergeFrom(const TDictionary& dict);

private:
    void* Attributes_; // IAttributesDictionary*

    friend class TErrorOr<void>;
    explicit TErrorAttributes(void* attributes);

    TErrorAttributes(const TErrorAttributes& other) = default;
    TErrorAttributes& operator= (const TErrorAttributes& other) = default;

    TErrorAttributes(TErrorAttributes&& other) = default;
    TErrorAttributes& operator= (TErrorAttributes&& other) = default;
};

bool operator == (const TErrorAttributes& lhs, const TErrorAttributes& rhs);

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

} // namespace NYT

#define ERROR_ATTRIBUTES_INL_H_
#include "error_attributes-inl.h"
#undef ERROR_ATTRIBUTES_INL_H_