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
|
#include "StarLockFile.hpp"
#include "StarTime.hpp"
#include "StarThread.hpp"
#include <sys/file.h>
#include <fcntl.h>
#include <errno.h>
#include <unistd.h>
namespace Star {
int64_t const LockFile::MaximumSleepMillis;
Maybe<LockFile> LockFile::acquireLock(String const& filename, int64_t lockTimeout) {
LockFile lock(std::move(filename));
if (lock.lock(lockTimeout))
return lock;
return {};
}
LockFile::LockFile(String const& filename) : m_filename(std::move(filename)) {}
LockFile::LockFile(LockFile&& lockFile) {
operator=(std::move(lockFile));
}
LockFile::~LockFile() {
unlock();
}
LockFile& LockFile::operator=(LockFile&& lockFile) {
m_filename = std::move(lockFile.m_filename);
m_handle = std::move(lockFile.m_handle);
return *this;
}
bool LockFile::lock(int64_t timeout) {
auto doFLock = [](String const& filename, bool block) -> shared_ptr<int> {
int fd = open(filename.utf8Ptr(), O_RDONLY | O_CREAT, 0644);
if (fd < 0)
throw StarException(strf("Could not open lock file {}, {}\n", filename, strerror(errno)));
int ret;
if (block)
ret = flock(fd, LOCK_EX);
else
ret = flock(fd, LOCK_EX | LOCK_NB);
if (ret != 0) {
close(fd);
if (errno != EWOULDBLOCK)
throw StarException(strf("Could not lock file {}, {}\n", filename, strerror(errno)));
return {};
}
return make_shared<int>(fd);
};
if (timeout < 0) {
m_handle = doFLock(m_filename, true);
return true;
} else if (timeout == 0) {
m_handle = doFLock(m_filename, false);
return (bool)m_handle;
} else {
int64_t startTime = Time::monotonicMilliseconds();
while (true) {
m_handle = doFLock(m_filename, false);
if (m_handle)
return true;
if (Time::monotonicMilliseconds() - startTime > timeout)
return false;
Thread::sleep(min(timeout / 4, MaximumSleepMillis));
}
}
}
void LockFile::unlock() {
if (m_handle) {
int fd = *(int*)m_handle.get();
unlink(m_filename.utf8Ptr());
close(fd);
m_handle.reset();
}
}
bool LockFile::isLocked() const {
return (bool)m_handle;
}
}
|