mirror of
https://github.com/matrix-construct/construct
synced 2024-12-24 14:34:00 +01:00
336 lines
5.5 KiB
C++
336 lines
5.5 KiB
C++
// Matrix Construct
|
|
//
|
|
// Copyright (C) Matrix Construct Developers, Authors & Contributors
|
|
// Copyright (C) 2016-2018 Jason Volk <jason@zemos.net>
|
|
//
|
|
// Permission to use, copy, modify, and/or distribute this software for any
|
|
// purpose with or without fee is hereby granted, provided that the above
|
|
// copyright notice and this permission notice is present in all copies. The
|
|
// full license for this software is available in the LICENSE file.
|
|
|
|
#include "db.h"
|
|
|
|
//
|
|
// Mutex
|
|
//
|
|
|
|
static_assert
|
|
(
|
|
sizeof(rocksdb::port::Mutex) <= sizeof(pthread_mutex_t) + 1,
|
|
"link-time punning of our structure won't work if the structure is larger "
|
|
"than the one rocksdb has assumed space for."
|
|
);
|
|
|
|
rocksdb::port::Mutex::Mutex()
|
|
noexcept
|
|
{
|
|
memset(this, 0x0, sizeof(pthread_mutex_t));
|
|
|
|
#ifdef RB_DEBUG_DB_PORT_
|
|
if(unlikely(!ctx::current))
|
|
return;
|
|
|
|
log::debug
|
|
{
|
|
db::log, "mutex %lu %p CTOR", ctx::id(), this
|
|
};
|
|
#endif
|
|
}
|
|
|
|
rocksdb::port::Mutex::Mutex(bool adaptive)
|
|
noexcept
|
|
:Mutex{}
|
|
{
|
|
}
|
|
|
|
rocksdb::port::Mutex::~Mutex()
|
|
noexcept
|
|
{
|
|
#ifdef RB_DEBUG_DB_PORT_
|
|
if(unlikely(!ctx::current))
|
|
return;
|
|
|
|
log::debug
|
|
{
|
|
db::log, "mutex %lu %p DTOR", ctx::id(), this
|
|
};
|
|
#endif
|
|
}
|
|
|
|
void
|
|
rocksdb::port::Mutex::Lock()
|
|
noexcept
|
|
{
|
|
if(unlikely(!ctx::current))
|
|
return;
|
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
log::debug
|
|
{
|
|
db::log, "mutex %lu %p LOCK", ctx::id(), this
|
|
};
|
|
#endif
|
|
|
|
assert_main_thread();
|
|
const ctx::uninterruptible::nothrow ui;
|
|
mu.lock();
|
|
}
|
|
|
|
void
|
|
rocksdb::port::Mutex::Unlock()
|
|
noexcept
|
|
{
|
|
if(unlikely(!ctx::current))
|
|
return;
|
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
log::debug
|
|
{
|
|
db::log, "mutex %lu %p UNLOCK", ctx::id(), this
|
|
};
|
|
#endif
|
|
|
|
assert_main_thread();
|
|
assert(mu.locked());
|
|
const ctx::uninterruptible::nothrow ui;
|
|
mu.unlock();
|
|
}
|
|
|
|
void
|
|
rocksdb::port::Mutex::AssertHeld()
|
|
noexcept
|
|
{
|
|
assert(!ctx::current || mu.locked());
|
|
}
|
|
|
|
//
|
|
// RWMutex
|
|
//
|
|
|
|
static_assert
|
|
(
|
|
sizeof(rocksdb::port::RWMutex) <= sizeof(pthread_rwlock_t),
|
|
"link-time punning of our structure won't work if the structure is larger "
|
|
"than the one rocksdb has assumed space for."
|
|
);
|
|
|
|
rocksdb::port::RWMutex::RWMutex()
|
|
noexcept
|
|
{
|
|
memset(this, 0x0, sizeof(pthread_rwlock_t));
|
|
|
|
#ifdef RB_DEBUG_DB_PORT_
|
|
if(unlikely(!ctx::current))
|
|
return;
|
|
|
|
log::debug
|
|
{
|
|
db::log, "shared_mutex %lu %p CTOR", ctx::id(), this
|
|
};
|
|
#endif
|
|
}
|
|
|
|
rocksdb::port::RWMutex::~RWMutex()
|
|
noexcept
|
|
{
|
|
#ifdef RB_DEBUG_DB_PORT_
|
|
if(unlikely(!ctx::current))
|
|
return;
|
|
|
|
log::debug
|
|
{
|
|
db::log, "shared_mutex %lu %p DTOR", ctx::id(), this
|
|
};
|
|
#endif
|
|
}
|
|
|
|
void
|
|
rocksdb::port::RWMutex::ReadLock()
|
|
noexcept
|
|
{
|
|
if(unlikely(!ctx::current))
|
|
return;
|
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
log::debug
|
|
{
|
|
db::log, "shared_mutex %lu %p LOCK SHARED", ctx::id(), this
|
|
};
|
|
#endif
|
|
|
|
assert_main_thread();
|
|
const ctx::uninterruptible::nothrow ui;
|
|
mu.lock_shared();
|
|
}
|
|
|
|
void
|
|
rocksdb::port::RWMutex::WriteLock()
|
|
noexcept
|
|
{
|
|
if(unlikely(!ctx::current))
|
|
return;
|
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
log::debug
|
|
{
|
|
db::log, "shared_mutex %lu %p LOCK", ctx::id(), this
|
|
};
|
|
#endif
|
|
|
|
assert_main_thread();
|
|
const ctx::uninterruptible::nothrow ui;
|
|
mu.lock();
|
|
}
|
|
|
|
void
|
|
rocksdb::port::RWMutex::ReadUnlock()
|
|
noexcept
|
|
{
|
|
if(unlikely(!ctx::current))
|
|
return;
|
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
log::debug
|
|
{
|
|
db::log, "shared_mutex %lu %p UNLOCK SHARED", ctx::id(), this
|
|
};
|
|
#endif
|
|
|
|
assert_main_thread();
|
|
const ctx::uninterruptible::nothrow ui;
|
|
mu.unlock_shared();
|
|
}
|
|
|
|
void
|
|
rocksdb::port::RWMutex::WriteUnlock()
|
|
noexcept
|
|
{
|
|
if(unlikely(!ctx::current))
|
|
return;
|
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
log::debug
|
|
{
|
|
db::log, "shared_mutex %lu %p UNLOCK", ctx::id(), this
|
|
};
|
|
#endif
|
|
|
|
assert_main_thread();
|
|
const ctx::uninterruptible::nothrow ui;
|
|
mu.unlock();
|
|
}
|
|
|
|
//
|
|
// CondVar
|
|
//
|
|
|
|
static_assert
|
|
(
|
|
sizeof(rocksdb::port::CondVar) <= sizeof(pthread_cond_t) + sizeof(void *),
|
|
"link-time punning of our structure won't work if the structure is larger "
|
|
"than the one rocksdb has assumed space for."
|
|
);
|
|
|
|
rocksdb::port::CondVar::CondVar(Mutex *mu)
|
|
noexcept
|
|
{
|
|
memset(this, 0x0, sizeof(pthread_cond_t) + sizeof(Mutex *));
|
|
this->mu = mu;
|
|
|
|
#ifdef RB_DEBUG_DB_PORT_
|
|
if(unlikely(!ctx::current))
|
|
return;
|
|
|
|
log::debug
|
|
{
|
|
db::log, "cond %lu %p %p CTOR", ctx::id(), this, mu
|
|
};
|
|
#endif
|
|
}
|
|
|
|
rocksdb::port::CondVar::~CondVar()
|
|
noexcept
|
|
{
|
|
#ifdef RB_DEBUG_DB_PORT_
|
|
if(unlikely(!ctx::current))
|
|
return;
|
|
|
|
log::debug
|
|
{
|
|
db::log, "cond %lu %p %p DTOR", ctx::id(), this, mu
|
|
};
|
|
#endif
|
|
}
|
|
|
|
void
|
|
rocksdb::port::CondVar::Wait()
|
|
noexcept
|
|
{
|
|
assert(ctx::current);
|
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
log::debug
|
|
{
|
|
db::log, "cond %lu %p %p WAIT", ctx::id(), this, mu
|
|
};
|
|
#endif
|
|
|
|
assert(mu);
|
|
assert_main_thread();
|
|
mu->AssertHeld();
|
|
const ctx::uninterruptible::nothrow ui;
|
|
cv.wait(mu->mu);
|
|
}
|
|
|
|
// Returns true if timeout occurred
|
|
bool
|
|
rocksdb::port::CondVar::TimedWait(uint64_t abs_time_us)
|
|
noexcept
|
|
{
|
|
assert(ctx::current);
|
|
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
log::debug
|
|
{
|
|
db::log, "cond %lu %p %p WAIT_UNTIL %lu", ctx::id(), this, mu, abs_time_us
|
|
};
|
|
#endif
|
|
|
|
assert(mu);
|
|
assert_main_thread();
|
|
mu->AssertHeld();
|
|
const std::chrono::microseconds us(abs_time_us);
|
|
const std::chrono::system_clock::time_point tp(us);
|
|
const ctx::uninterruptible::nothrow ui;
|
|
return cv.wait_until(mu->mu, tp) == std::cv_status::timeout;
|
|
}
|
|
|
|
void
|
|
rocksdb::port::CondVar::Signal()
|
|
noexcept
|
|
{
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
log::debug
|
|
{
|
|
db::log, "cond %lu %p %p NOTIFY", ctx::id(), this, mu
|
|
};
|
|
#endif
|
|
|
|
assert_main_thread();
|
|
cv.notify_one();
|
|
}
|
|
|
|
void
|
|
rocksdb::port::CondVar::SignalAll()
|
|
noexcept
|
|
{
|
|
#ifdef RB_DEBUG_DB_PORT
|
|
log::debug
|
|
{
|
|
db::log, "cond %lu %p %p BROADCAST", ctx::id(), this, mu
|
|
};
|
|
#endif
|
|
|
|
assert_main_thread();
|
|
cv.notify_all();
|
|
}
|