util/system/file_lock.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
45
46

#include "file_lock.h"
#include "flock.h"

#include <util/generic/yexception.h> 

#include <cerrno>
 
namespace {
    int GetMode(const EFileLockType type) {
        switch (type) {
            case EFileLockType::Exclusive:
                return LOCK_EX;
            case EFileLockType::Shared:
                return LOCK_SH;
            default:
                Y_UNREACHABLE();
        }
        Y_UNREACHABLE();
    }
}

TFileLock::TFileLock(const TString& filename, const EFileLockType type)
    : TFile(filename, OpenAlways | RdOnly)
    , Type(type)
{
}

void TFileLock::Acquire() {
    Flock(GetMode(Type));
}

bool TFileLock::TryAcquire() { 
    try { 
        Flock(GetMode(Type) | LOCK_NB);
        return true; 
    } catch (const TSystemError& e) { 
        if (e.Status() != EWOULDBLOCK) { 
            throw; 
        } 
        return false; 
    } 
} 
 
void TFileLock::Release() {
    Flock(LOCK_UN);
}