contrib/clickhouse/src/Access/CachedAccessChecking.cpp


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

#include <Access/CachedAccessChecking.h>
#include <Access/ContextAccess.h>


namespace DB
{
CachedAccessChecking::CachedAccessChecking(const std::shared_ptr<const ContextAccess> & access_, AccessFlags access_flags_)
    : CachedAccessChecking(access_, AccessRightsElement{access_flags_})
{
}

CachedAccessChecking::CachedAccessChecking(const std::shared_ptr<const ContextAccess> & access_, const AccessRightsElement & element_)
    : access(access_), element(element_)
{
}

CachedAccessChecking::~CachedAccessChecking() = default;

bool CachedAccessChecking::checkAccess(bool throw_if_denied)
{
    if (checked)
        return result;
    if (throw_if_denied)
    {
        try
        {
            access->checkAccess(element);
            result = true;
        }
        catch (...)
        {
            result = false;
            throw;
        }
    }
    else
    {
        result = access->isGranted(element);
    }
    checked = true;
    return result;
}

}