Merge pull request #239 from vedgy/concurrent-queue-fixes

Test, fix, improve, optimize ConcurrentQueue
This commit is contained in:
Luis Ángel San Martín 2021-12-29 10:53:17 +01:00 committed by GitHub
commit 3a60d89f06
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
16 changed files with 879 additions and 109 deletions

View File

@ -1,3 +1,4 @@
TEMPLATE = subdirs
SUBDIRS = YACReader YACReaderLibrary YACReaderLibraryServer
YACReaderLibrary.depends = YACReader
!CONFIG(no_tests): SUBDIRS += tests

View File

@ -6,7 +6,7 @@ QMAKE_TARGET_BUNDLE_PREFIX = "com.yacreader"
DEPENDPATH += . \
release
DEFINES += NOMINMAX YACREADER
DEFINES += YACREADER
#load default build flags
include (../config.pri)

View File

@ -12,7 +12,7 @@ INCLUDEPATH += . \
./comic_vine \
./comic_vine/model
DEFINES += SERVER_RELEASE NOMINMAX YACREADER_LIBRARY
DEFINES += SERVER_RELEASE YACREADER_LIBRARY
# load default build flags
include (../config.pri)
@ -160,6 +160,7 @@ HEADERS += comic_flow.h \
}
SOURCES += comic_flow.cpp \
../common/concurrent_queue.cpp \
create_library_dialog.cpp \
db/comic_query_result_processor.cpp \
db/folder_query_result_processor.cpp \

View File

@ -16,7 +16,7 @@ YACReader::ComicQueryResultProcessor::ComicQueryResultProcessor()
void YACReader::ComicQueryResultProcessor::createModelData(const YACReader::SearchModifiers modifier, const QString &filter, const QString &databasePath)
{
querySearchQueue.cancellPending();
querySearchQueue.cancelPending();
querySearchQueue.enqueue([=] {
QString connectionName = "";

View File

@ -22,7 +22,7 @@ YACReader::FolderQueryResultProcessor::FolderQueryResultProcessor(FolderModel *m
void YACReader::FolderQueryResultProcessor::createModelData(const YACReader::SearchModifiers modifier, const QString &filter, bool includeComics)
{
querySearchQueue.cancellPending();
querySearchQueue.cancelPending();
querySearchQueue.enqueue([=] {
QString connectionName = "";

View File

@ -10,7 +10,7 @@ INCLUDEPATH += ../YACReaderLibrary \
../YACReaderLibrary/server \
../YACReaderLibrary/db
DEFINES += SERVER_RELEASE NOMINMAX YACREADER_LIBRARY
DEFINES += SERVER_RELEASE YACREADER_LIBRARY
# load default build flags
# do a basic dependency check
include(headless_config.pri)

View File

@ -33,6 +33,11 @@ jobs:
qmake CONFIG+="7zip" %DEFINES_VAR%
nmake
displayName: 'Build'
- script: |
call "C:\Program Files (x86)\Microsoft Visual Studio\2019\Enterprise\VC\Auxiliary\Build\${{ parameters.vc_vars }}"
set PATH=C:\Qt\${{ parameters.qt_version }}\${{ parameters.qt_spec }}\bin;%PATH%
nmake check TESTARGS="-maxwarnings 100000"
displayName: 'Run tests'
# - script: |
# set PATH=C:\Qt\${{ parameters.qt_version }}\${{ parameters.qt_spec }}\bin;%PATH%
# cd $(Build.SourcesDirectory)\ci\win

View File

@ -33,6 +33,11 @@ jobs:
qmake CONFIG+="7zip" %DEFINES_VAR%
nmake
displayName: 'Build'
- script: |
call "C:\Program Files (x86)\Microsoft Visual Studio\2019\Enterprise\VC\Auxiliary\Build\${{ parameters.vc_vars }}"
set PATH=C:\Qt\${{ parameters.qt_version }}\${{ parameters.qt_spec }}\bin;%PATH%
nmake check TESTARGS="-maxwarnings 100000"
displayName: 'Run tests'
- script: |
set PATH=C:\Qt\${{ parameters.qt_version }}\${{ parameters.qt_spec }}\bin;%PATH%
cd $(Build.SourcesDirectory)\ci\win

View File

@ -72,6 +72,9 @@ jobs:
qmake CONFIG+="unarr" $DEFINES_VAR
make
displayName: 'Build'
- script: |
make check TESTARGS="-maxwarnings 100000"
displayName: 'Run tests'
- task: CopyFiles@2
inputs:
sourceFolder: $(Build.SourcesDirectory)/tarball
@ -118,6 +121,11 @@ jobs:
SKIP_CODESIGN="$(tr [A-Z] [a-z] <<< "$IS_FORK")"
./compileOSX.sh $VERSION $(Build.BuildNumber) $SKIP_CODESIGN
displayName: 'Build'
- script: |
cd $(Build.SourcesDirectory)/tests
qmake
make check TESTARGS="-maxwarnings 100000"
displayName: 'Build and run tests'
- task: CopyFiles@2
inputs:
contents: '*.dmg'

107
common/concurrent_queue.cpp Normal file
View File

@ -0,0 +1,107 @@
#include "concurrent_queue.h"
#include <cassert>
#include <cstddef>
#include <mutex>
#include <utility>
using namespace YACReader;
ConcurrentQueue::ConcurrentQueue(std::size_t threadCount)
{
threads.reserve(threadCount);
for (; threadCount != 0; --threadCount)
threads.emplace_back(&ConcurrentQueue::nextJob, this);
}
ConcurrentQueue::~ConcurrentQueue()
{
{
std::lock_guard<std::mutex> lock(queueMutex);
assert(!bailout);
bailout = true;
}
jobAvailableVar.notify_all();
for (auto &x : threads)
x.join();
assert(jobsLeft == _queue.size() && "Only not yet started jobs are left.");
}
void ConcurrentQueue::enqueue(Job job)
{
{
std::lock_guard<std::mutex> lock(jobsLeftMutex);
++jobsLeft;
}
{
std::lock_guard<std::mutex> lock(queueMutex);
_queue.emplace(std::move(job));
}
jobAvailableVar.notify_one();
}
std::size_t ConcurrentQueue::cancelPending()
{
decltype(_queue) oldQueue;
{
const std::lock_guard<std::mutex> lock(queueMutex);
// The mutex locking time is lower with swap() compared to assigning a
// temporary (which destroys _queue's elements and deallocates memory).
_queue.swap(oldQueue);
}
const auto size = oldQueue.size();
if (size != 0)
finalizeJobs(size);
return size;
}
void ConcurrentQueue::waitAll() const
{
std::unique_lock<std::mutex> lock(jobsLeftMutex);
_waitVar.wait(lock, [this] { return jobsLeft == 0; });
}
void ConcurrentQueue::nextJob()
{
while (true) {
Job job;
{
std::unique_lock<std::mutex> lock(queueMutex);
jobAvailableVar.wait(lock, [this] {
return _queue.size() > 0 || bailout;
});
if (bailout) {
return;
}
job = std::move(_queue.front());
_queue.pop();
}
job();
finalizeJobs(1);
}
}
void ConcurrentQueue::finalizeJobs(std::size_t count)
{
assert(count > 0);
std::size_t remainingJobs;
{
std::lock_guard<std::mutex> lock(jobsLeftMutex);
assert(jobsLeft >= count);
jobsLeft -= count;
remainingJobs = jobsLeft;
}
if (remainingJobs == 0)
_waitVar.notify_all();
}

View File

@ -6,126 +6,48 @@
#include <functional>
#include <condition_variable>
#include <queue>
#include <vector>
namespace YACReader {
//! All functions in this class are thread-safe in the Qt documentation sense.
class ConcurrentQueue
{
public:
explicit ConcurrentQueue(int threadCount)
: jobsLeft(0),
bailout(false)
{
threads = std::vector<std::thread>(threadCount);
for (int index = 0; index < threadCount; ++index) {
threads[index] = std::thread([this] {
this->nextJob();
});
}
}
//! @brief Creates and starts executing @p threadCount worker threads.
//! @note ConcurrentQueue is unable to execute jobs if @p threadCount == 0.
explicit ConcurrentQueue(std::size_t threadCount);
~ConcurrentQueue()
{
joinAll();
}
//! Cancels all jobs that have not been picked up by worker threads yet,
//! waits for all worker threads to complete their jobs and joins them.
~ConcurrentQueue();
void enqueue(std::function<void(void)> job)
{
{
std::lock_guard<std::mutex> lock(queueMutex);
_queue.emplace(job);
}
using Job = std::function<void()>;
{
std::lock_guard<std::mutex> lock(jobsLeftMutex);
++jobsLeft;
}
//! @brief Adds @p job to the queue.
//! @note A worker thread may start executing @p job immediately if it is idle.
//! Worker threads start executing jobs in the same order as they are enqueued.
void enqueue(Job job);
jobAvailableVar.notify_one();
}
//! @brief Cancels all jobs that have not been picked up by worker threads yet.
//! @return The number of jobs that were canceled.
std::size_t cancelPending();
void cancellPending()
{
std::unique_lock<std::mutex> lockQueue(queueMutex);
std::unique_lock<std::mutex> lockJobsLeft(jobsLeftMutex);
_queue = std::queue<std::function<void(void)>>();
jobsLeft = 0;
}
void waitAll()
{
std::unique_lock<std::mutex> lock(jobsLeftMutex);
if (jobsLeft > 0) {
_waitVar.wait(lock, [this] {
return jobsLeft == 0;
});
}
}
//! @brief Blocks the current thread until all enqueued jobs are completed.
void waitAll() const;
private:
//! @invariant all worker threads are joinable until the destructor is called.
std::vector<std::thread> threads;
std::queue<std::function<void(void)>> _queue;
int jobsLeft;
bool bailout;
std::queue<Job> _queue;
std::size_t jobsLeft = 0; //!< @invariant jobsLeft >= _queue.size()
bool bailout = false; //!< @invariant is false until the destructor is called.
std::condition_variable jobAvailableVar;
std::condition_variable _waitVar;
std::mutex jobsLeftMutex;
mutable std::condition_variable _waitVar;
mutable std::mutex jobsLeftMutex;
std::mutex queueMutex;
void nextJob()
{
while (true) {
std::function<void(void)> job;
{
std::unique_lock<std::mutex> lock(queueMutex);
if (bailout) {
return;
}
jobAvailableVar.wait(lock, [this] {
return _queue.size() > 0 || bailout;
});
if (bailout) {
return;
}
job = _queue.front();
_queue.pop();
}
job();
{
std::lock_guard<std::mutex> lock(jobsLeftMutex);
--jobsLeft;
}
_waitVar.notify_one();
}
}
void joinAll()
{
{
std::lock_guard<std::mutex> lock(queueMutex);
if (bailout) {
return;
}
bailout = true;
}
jobAvailableVar.notify_all();
for (auto &x : threads) {
if (x.joinable()) {
x.join();
}
}
}
void nextJob();
void finalizeJobs(std::size_t count);
};
}

View File

@ -5,6 +5,8 @@
CONFIG += c++17
win32:QMAKE_CXXFLAGS += /std:c++17 #enable c++17 explicitly in msvc
DEFINES += NOMINMAX
if(unix|mingw):QMAKE_CXXFLAGS_RELEASE += -DNDEBUG
win32:msvc:QMAKE_CXXFLAGS_RELEASE += /DNDEBUG

View File

@ -0,0 +1,699 @@
#include "concurrent_queue.h"
#include <QDebug>
#include <QDebugStateSaver>
#include <QMetaType>
#include <QObject>
#include <QString>
#include <QTest>
#include <QTime>
#include <QVector>
#include <array>
#include <atomic>
#include <chrono>
#include <cstdint>
#include <memory>
#include <numeric>
#include <random>
#include <sstream>
#include <thread>
#include <vector>
namespace chrono = std::chrono;
using Clock = chrono::steady_clock;
using YACReader::ConcurrentQueue;
namespace {
double toMilliseconds(Clock::duration duration)
{
return chrono::duration_cast<chrono::microseconds>(duration).count() / 1000.0;
}
QString currentThreadInfo()
{
std::ostringstream os;
os << std::this_thread::get_id();
return QString::fromStdString(os.str());
}
//! This test prints thousands of lines of detailed output. The output allows to analyze
//! how ConcurrentQueue is being tested, how it works and why the test fails or crashes
//! (normally it passes). The default maximum number of warnings in Qt Test is 2000,
//! which is too low for this test. Therefore, the following warning is printed before
//! the log output is suppressed: "Maximum amount of warnings exceeded. Use -maxwarnings
//! to override.". Passing `-maxwarnings 100000` command line option to the test lets it
//! print everything. Passing -silent command line option to the test suppresses all its
//! output except for RandomEngineProvider's root seeds, which are necessary to reproduce
//! interesting test results.
QDebug log()
{
return qInfo().noquote() << currentThreadInfo() << '|'
<< QTime::currentTime().toString(Qt::ISODateWithMs) << '|';
}
using Total = std::atomic<int>;
struct JobData {
int summand;
Clock::duration sleepingTime;
};
using JobDataSet = QVector<JobData>;
int expectedTotal(JobDataSet::const_iterator first, JobDataSet::const_iterator last)
{
return std::accumulate(first, last, 0,
[](int total, JobData job) {
return total + job.summand;
});
}
int expectedTotal(const JobDataSet &jobs)
{
return expectedTotal(jobs.cbegin(), jobs.cend());
}
int expectedTotal(const JobDataSet &jobs, std::size_t canceledCount)
{
const auto count = jobs.size() - static_cast<int>(canceledCount);
if (count < 0)
qFatal("Canceled more than the total number of jobs somehow!");
return expectedTotal(jobs.cbegin(), jobs.cbegin() + count);
}
int expectedTotal(const QVector<JobDataSet> &jobs)
{
return std::accumulate(jobs.cbegin(), jobs.cend(), 0,
[](int total, const JobDataSet &dataSet) {
return total + expectedTotal(dataSet);
});
}
class Id
{
public:
explicit Id(int threadId, int jobId)
: threadId { threadId }, jobId { jobId } { }
QString toString() const { return QStringLiteral("[%1.%2]").arg(threadId).arg(jobId); }
private:
const int threadId;
const int jobId;
};
QDebug operator<<(QDebug debug, Id id)
{
QDebugStateSaver saver(debug);
debug.noquote() << id.toString();
return debug;
}
class Job
{
public:
explicit Job(Total &total, JobData data, Id id)
: total { total }, data { data }, id { id } { }
void operator()()
{
log().nospace() << id << " sleep " << toMilliseconds(data.sleepingTime) << " ms...";
std::this_thread::sleep_for(data.sleepingTime);
const auto updatedTotal = (total += data.summand);
log().nospace() << id << " +" << data.summand << " => " << updatedTotal;
}
private:
Total &total;
const JobData data;
const Id id;
};
class Enqueuer
{
public:
explicit Enqueuer(ConcurrentQueue &queue, Total &total, const JobDataSet &jobs, int threadId)
: queue { queue }, total { total }, jobs { jobs }, threadId { threadId } { }
void operator()()
{
const char *const jobStr = jobs.size() == 1 ? "job" : "jobs";
log() << QStringLiteral("#%1 enqueuing %2 %3...").arg(threadId).arg(jobs.size()).arg(jobStr);
for (int i = 0; i < jobs.size(); ++i)
queue.enqueue(Job(total, jobs.at(i), Id(threadId, i + 1)));
log() << QStringLiteral("#%1 enqueuing complete.").arg(threadId);
}
private:
ConcurrentQueue &queue;
Total &total;
const JobDataSet jobs;
const int threadId;
};
class QueueControlMessagePrinter
{
public:
explicit QueueControlMessagePrinter(const Total &total, int threadId, int threadCount)
: total { total }, threadId { threadId }, threadCount { threadCount } { }
void printStartedMessage() const
{
log() << threadMessageFormatString().arg("started");
}
void printCanceledMessage(std::size_t canceledCount) const
{
const char *const jobStr = canceledCount == 1 ? "job" : "jobs";
const auto format = messageFormatString().arg("%1 %2 %3");
log() << format.arg("canceled").arg(canceledCount).arg(jobStr);
}
void printBeginWaitingMessage() const
{
log() << threadMessageFormatString().arg("begin waiting for");
}
void printEndWaitingMessage() const
{
log() << threadMessageFormatString().arg("end waiting for");
}
private:
QString messageFormatString() const
{
return QStringLiteral("#%1 %3 => %2").arg(threadId).arg(total.load());
}
QString threadMessageFormatString() const
{
const char *const threadStr = threadCount == 1 ? "thread" : "threads";
const auto format = messageFormatString().arg("%3 %1 %2");
return format.arg(threadCount).arg(threadStr);
}
const Total &total;
const int threadId;
const int threadCount;
};
std::size_t cancelAndPrint(ConcurrentQueue &queue, const QueueControlMessagePrinter &printer)
{
const auto canceledCount = queue.cancelPending();
printer.printCanceledMessage(canceledCount);
return canceledCount;
}
void waitAndPrint(const ConcurrentQueue &queue, const QueueControlMessagePrinter &printer)
{
printer.printBeginWaitingMessage();
queue.waitAll();
printer.printEndWaitingMessage();
}
template<typename T, std::size_t size>
QDebug operator<<(QDebug debug, const std::array<T, size> &array)
{
QDebugStateSaver saver(debug);
debug.nospace();
debug << '(';
if (size != 0) {
debug << array.front();
for (std::size_t i = 1; i != size; ++i)
debug << ", " << array[i];
}
debug << ')';
return debug;
}
using RandomEngine = std::mt19937_64;
class RandomEngineProvider
{
public:
RandomEngineProvider()
{
std::random_device rd;
const auto randomValues = generate<rootSeedCount>(rd);
// Qt Test does not suppress output from the constructor of a test class
// even when -silent command line option is passed. This is fortunate
// because the root seeds can be used to reproduce a test failure.
log() << "RandomEngineProvider's root seeds:" << randomValues;
std::seed_seq seedSeq(randomValues.begin(), randomValues.end());
rootEngine.reset(new std::mt19937(seedSeq));
}
void resetEngines(std::size_t engineCount)
{
engines.clear();
engines.reserve(engineCount);
for (; engineCount != 0; --engineCount) {
const auto randomValues = generate<seedCount>(*rootEngine);
std::seed_seq seedSeq(randomValues.begin(), randomValues.end());
engines.emplace_back(seedSeq);
}
}
RandomEngine &engine(std::size_t index)
{
return engines.at(index);
}
private:
// In this test we don't really care about uniformly choosing an initial state
// from the entire state-space of the engine. It is possible to generate more
// random numbers at the cost of performance and system entropy pool exhaustion.
static constexpr std::size_t rootSeedCount { 8 };
static constexpr std::size_t seedCount { 32 };
template<std::size_t size, typename Generator>
static std::array<std::uint32_t, size> generate(Generator &generator)
{
std::array<std::uint32_t, size> result;
for (auto &value : result)
value = generator();
return result;
}
std::unique_ptr<std::mt19937> rootEngine;
std::vector<RandomEngine> engines;
};
//! Calls random member functions of ConcurrentQueue for a limited time.
//! Ensures that total equals 0 when all jobs are complete/canceled by:
//! * setting each job's summand to 1;
//! * subtracting a job set's size from total before enqueuing jobs in the set;
//! * adding canceled job count to total after cancelation.
class RandomCaller
{
public:
explicit RandomCaller(ConcurrentQueue &queue, Total &total, int threadId,
int queueThreadCount, RandomEngine &engine,
bool boostEnqueueOperationWeight)
: queue(queue),
total(total),
threadId { threadId },
boostEnqueueOperationWeight { boostEnqueueOperationWeight },
printer(total, threadId, queueThreadCount),
engine(engine)
{
}
void operator()()
{
constexpr auto testDuration = chrono::milliseconds(10);
const auto testStartTime = Clock::now();
auto operation = operationDistribution();
do {
switch (operation(engine)) {
case 0:
enqueue();
break;
case 1:
cancel();
break;
case 2:
waitAndPrint(queue, printer);
break;
default:
qFatal("Unsupported operation.");
}
} while (Clock::now() - testStartTime < testDuration);
}
private:
int randomInt(int a, int b)
{
return uniformInt(engine, decltype(uniformInt)::param_type(a, b));
}
using OperationDistribution = std::discrete_distribution<int>;
void printProbabilities(const OperationDistribution &distribution) const
{
auto p = distribution.probabilities();
constexpr std::size_t expectedProbabilityCount { 3 };
if (p.size() != expectedProbabilityCount)
qFatal("Wrong number of operation probabilities: %zu != %zu", p.size(), expectedProbabilityCount);
for (auto &x : p)
x *= 100; // Convert to percentages.
log() << QStringLiteral("#%1 operation probabilities: e=%2%, c=%3%, w=%4%.").arg(threadId).arg(p[0]).arg(p[1]).arg(p[2]);
}
OperationDistribution operationDistribution()
{
auto distribution = boostEnqueueOperationWeight ? boostedEnqueueOperationDistribution()
: almostUniformOperationDistribution();
printProbabilities(distribution);
return distribution;
}
OperationDistribution almostUniformOperationDistribution()
{
constexpr int sumOfProbabilities { 100 };
const auto enqueueProbability = randomInt(0, sumOfProbabilities);
const auto cancelProbability = randomInt(0, sumOfProbabilities - enqueueProbability);
const auto waitProbability = sumOfProbabilities - enqueueProbability - cancelProbability;
if (enqueueProbability + cancelProbability + waitProbability != sumOfProbabilities)
qFatal("The sum of probabilities is not 100%%.");
const auto real = [](int x) { return static_cast<double>(x); };
return { real(enqueueProbability), real(cancelProbability), real(waitProbability) };
}
OperationDistribution boostedEnqueueOperationDistribution()
{
// Make enqueue the most frequent operation to stress-test executing
// jobs rather than canceling them almost immediately.
const auto enqueueWeight = std::lognormal_distribution<double>(2, 0.5)(engine);
const auto cancelWeight = 1.0;
// Waiting is uninteresting as it doesn't even modify the queue => make it rare.
const auto waitWeight = std::uniform_real_distribution<double>(0, 0.2)(engine);
return { enqueueWeight, cancelWeight, waitWeight };
}
JobDataSet createJobs()
{
constexpr int minJobCount { 1 }, maxJobCount { 5 };
JobDataSet jobs(randomInt(minJobCount, maxJobCount));
for (auto &job : jobs) {
constexpr int minSleepingTime { 0 }, maxSleepingTime { 5 };
const auto sleepingTime = randomInt(minSleepingTime, maxSleepingTime);
job = { 1, sleepingTime * chrono::microseconds(1) };
}
return jobs;
}
void enqueue()
{
const auto jobs = createJobs();
total -= jobs.size();
Enqueuer(queue, total, jobs, threadId)();
}
void cancel()
{
const auto canceledCount = cancelAndPrint(queue, printer);
total += canceledCount;
}
ConcurrentQueue &queue;
Total &total;
const int threadId;
const bool boostEnqueueOperationWeight;
const QueueControlMessagePrinter printer;
RandomEngine &engine;
std::uniform_int_distribution<int> uniformInt;
};
}
Q_DECLARE_METATYPE(Clock::duration)
Q_DECLARE_METATYPE(JobData)
class ConcurrentQueueTest : public QObject
{
Q_OBJECT
private slots:
void init();
void singleUserThread_data();
void singleUserThread();
void multipleUserThreads_data();
void multipleUserThreads();
void cancelPending1UserThread_data();
void cancelPending1UserThread();
void waitAllFromMultipleThreads_data();
void waitAllFromMultipleThreads();
void randomCalls_data();
void randomCalls();
private:
static constexpr int primaryThreadId { 0 };
QueueControlMessagePrinter makeMessagePrinter(int threadCount) const
{
return QueueControlMessagePrinter(total, primaryThreadId, threadCount);
}
Total total { 0 };
RandomEngineProvider randomEngineProvider;
};
void ConcurrentQueueTest::init()
{
total = 0;
}
void ConcurrentQueueTest::singleUserThread_data()
{
QTest::addColumn<int>("threadCount");
QTest::addColumn<JobDataSet>("jobs");
using ms = chrono::milliseconds;
QTest::newRow("-") << 0 << JobDataSet {};
QTest::newRow("0") << 7 << JobDataSet {};
QTest::newRow("A") << 1 << JobDataSet { { 5, ms(0) } };
QTest::newRow("B") << 5 << JobDataSet { { 12, ms(1) } };
QTest::newRow("C") << 1 << JobDataSet { { 1, ms(0) }, { 5, ms(2) }, { 3, ms(1) } };
QTest::newRow("D") << 4 << JobDataSet { { 20, ms(1) }, { 8, ms(5) }, { 5, ms(2) } };
QTest::newRow("E") << 2 << JobDataSet { { 1, ms(2) }, { 2, ms(1) } };
QTest::newRow("F") << 3 << JobDataSet { { 8, ms(3) }, { 5, ms(4) }, { 2, ms(1) }, { 11, ms(1) }, { 100, ms(3) } };
}
void ConcurrentQueueTest::singleUserThread()
{
QFETCH(const int, threadCount);
QFETCH(const JobDataSet, jobs);
const auto printer = makeMessagePrinter(threadCount);
ConcurrentQueue queue(threadCount);
printer.printStartedMessage();
Enqueuer(queue, total, jobs, primaryThreadId)();
waitAndPrint(queue, printer);
QCOMPARE(total.load(), expectedTotal(jobs));
}
void ConcurrentQueueTest::multipleUserThreads_data()
{
QTest::addColumn<int>("threadCount");
QTest::addColumn<QVector<JobDataSet>>("jobs");
using ms = chrono::milliseconds;
JobDataSet jobs1 { { 1, ms(1) } };
JobDataSet jobs2 { { 2, ms(4) } };
QVector<JobDataSet> allJobs { jobs1, jobs2 };
QTest::newRow("A1") << 1 << allJobs;
QTest::newRow("A2") << 2 << allJobs;
jobs1.push_back({ 5, ms(3) });
jobs2.push_back({ 10, ms(1) });
allJobs = { jobs1, jobs2 };
QTest::newRow("B1") << 2 << allJobs;
QTest::newRow("B2") << 3 << allJobs;
QTest::newRow("B3") << 8 << allJobs;
jobs1.push_back({ 20, ms(0) });
jobs2.push_back({ 40, ms(2) });
allJobs = { jobs1, jobs2 };
QTest::newRow("C") << 4 << allJobs;
JobDataSet jobs3 { { 80, ms(0) }, { 160, ms(2) }, { 320, ms(1) }, { 640, ms(0) }, { 2000, ms(3) } };
allJobs.push_back(jobs3);
QTest::newRow("D1") << 3 << allJobs;
QTest::newRow("D2") << 5 << allJobs;
JobDataSet jobs4 { { 4000, ms(1) }, { 8000, ms(3) } };
allJobs.push_back(jobs4);
QTest::newRow("E1") << 4 << allJobs;
QTest::newRow("E2") << 6 << allJobs;
}
void ConcurrentQueueTest::multipleUserThreads()
{
QFETCH(const int, threadCount);
QFETCH(const QVector<JobDataSet>, jobs);
const auto printer = makeMessagePrinter(threadCount);
ConcurrentQueue queue(threadCount);
printer.printStartedMessage();
if (!jobs.empty()) {
std::vector<std::thread> enqueuerThreads;
enqueuerThreads.reserve(jobs.size() - 1);
for (int i = 1; i < jobs.size(); ++i)
enqueuerThreads.emplace_back(Enqueuer(queue, total, jobs.at(i), i));
Enqueuer(queue, total, jobs.constFirst(), primaryThreadId)();
for (auto &t : enqueuerThreads)
t.join();
}
waitAndPrint(queue, printer);
QCOMPARE(total.load(), expectedTotal(jobs));
}
void ConcurrentQueueTest::cancelPending1UserThread_data()
{
QTest::addColumn<int>("threadCount");
QTest::addColumn<JobDataSet>("jobs");
QTest::addColumn<Clock::duration>("cancelDelay");
const auto ms = [](int count) -> Clock::duration { return chrono::milliseconds(count); };
const auto us = [](int count) -> Clock::duration { return chrono::microseconds(count); };
QTest::newRow("-") << 0 << JobDataSet {} << ms(0);
QTest::newRow("01") << 2 << JobDataSet {} << ms(0);
QTest::newRow("02") << 3 << JobDataSet {} << ms(1);
QTest::newRow("A") << 1 << JobDataSet { { 5, ms(3) } } << ms(1);
QTest::newRow("B") << 5 << JobDataSet { { 12, ms(1) } } << ms(1);
JobDataSet dataSet { { 1, ms(3) }, { 5, ms(2) }, { 3, ms(1) } };
QTest::newRow("C1") << 1 << dataSet << ms(1);
QTest::newRow("C2") << 1 << dataSet << ms(4);
QTest::newRow("C3") << 2 << dataSet << ms(1);
QTest::newRow("C4") << 3 << dataSet << ms(1);
QTest::newRow("C5") << 1 << dataSet << ms(7);
dataSet.push_back({ 10, ms(5) });
dataSet.push_back({ 20, ms(8) });
dataSet.push_back({ 40, ms(20) });
dataSet.push_back({ 80, ms(2) });
QTest::newRow("D1") << 1 << dataSet << ms(1);
QTest::newRow("D2") << 1 << dataSet << ms(15);
QTest::newRow("D3") << 1 << dataSet << ms(50);
QTest::newRow("D4") << 2 << dataSet << ms(4);
QTest::newRow("D5") << 3 << dataSet << ms(4);
QTest::newRow("D6") << 4 << dataSet << ms(4);
QTest::newRow("D7") << 2 << dataSet << us(300);
QTest::newRow("D8") << 3 << dataSet << us(500);
QTest::newRow("D9") << 4 << dataSet << us(700);
QTest::newRow("E") << 4 << JobDataSet { { 20, ms(1) }, { 8, ms(5) }, { 5, ms(2) } } << ms(1);
}
void ConcurrentQueueTest::cancelPending1UserThread()
{
QFETCH(const int, threadCount);
QFETCH(const JobDataSet, jobs);
QFETCH(const Clock::duration, cancelDelay);
const auto printer = makeMessagePrinter(threadCount);
ConcurrentQueue queue(threadCount);
printer.printStartedMessage();
Enqueuer(queue, total, jobs, primaryThreadId)();
std::this_thread::sleep_for(cancelDelay);
const auto canceledCount = cancelAndPrint(queue, printer);
QVERIFY(canceledCount <= static_cast<std::size_t>(jobs.size()));
waitAndPrint(queue, printer);
QCOMPARE(total.load(), expectedTotal(jobs, canceledCount));
}
void ConcurrentQueueTest::waitAllFromMultipleThreads_data()
{
QTest::addColumn<int>("waitingThreadCount");
for (int i : { 1, 2, 4, 7, 19 })
QTest::addRow("%d", i) << i;
}
void ConcurrentQueueTest::waitAllFromMultipleThreads()
{
QFETCH(const int, waitingThreadCount);
QVERIFY(waitingThreadCount > 0);
constexpr auto queueThreadCount = 2;
const auto printer = makeMessagePrinter(queueThreadCount);
ConcurrentQueue queue(queueThreadCount);
printer.printStartedMessage();
using ms = chrono::milliseconds;
const JobDataSet jobs { { 5, ms(1) }, { 7, ms(2) } };
Enqueuer(queue, total, jobs, primaryThreadId)();
std::vector<std::thread> waitingThreads;
waitingThreads.reserve(waitingThreadCount - 1);
for (int id = 1; id < waitingThreadCount; ++id) {
waitingThreads.emplace_back([=, &queue] {
waitAndPrint(queue, QueueControlMessagePrinter(total, id, queueThreadCount));
});
}
waitAndPrint(queue, printer);
for (auto &t : waitingThreads)
t.join();
QCOMPARE(total.load(), expectedTotal(jobs));
}
void ConcurrentQueueTest::randomCalls_data()
{
QTest::addColumn<int>("queueThreadCount");
QTest::addColumn<int>("userThreadCount");
QTest::addColumn<bool>("boostEnqueueOperationWeight");
const auto suffix = [](bool boost) { return boost ? " +enqueue" : ""; };
for (bool boost : { false, true })
for (int q : { 1, 2, 4, 9, 12, 20 })
for (int u : { 1, 2, 4, 7, 11, 18 })
QTest::addRow("queue{%d}; %d user thread(s)%s", q, u, suffix(boost)) << q << u << boost;
}
void ConcurrentQueueTest::randomCalls()
{
QFETCH(const int, queueThreadCount);
QFETCH(const int, userThreadCount);
QVERIFY(userThreadCount > 0);
QFETCH(const bool, boostEnqueueOperationWeight);
const auto printer = makeMessagePrinter(queueThreadCount);
ConcurrentQueue queue(queueThreadCount);
printer.printStartedMessage();
randomEngineProvider.resetEngines(userThreadCount);
std::vector<std::thread> userThreads;
userThreads.reserve(userThreadCount - 1);
for (int id = 1; id < userThreadCount; ++id) {
userThreads.emplace_back(RandomCaller(queue, total, id, queueThreadCount,
randomEngineProvider.engine(id),
boostEnqueueOperationWeight));
}
RandomCaller(queue, total, primaryThreadId, queueThreadCount,
randomEngineProvider.engine(primaryThreadId),
boostEnqueueOperationWeight)();
for (auto &t : userThreads)
t.join();
waitAndPrint(queue, printer);
QCOMPARE(total.load(), 0);
}
QTEST_APPLESS_MAIN(ConcurrentQueueTest)
#include "concurrent_queue_test.moc"

View File

@ -0,0 +1,9 @@
include(../qt_test.pri)
PATH_TO_common = ../../common
INCLUDEPATH += $$PATH_TO_common
HEADERS += $${PATH_TO_common}/concurrent_queue.h
SOURCES += \
$${PATH_TO_common}/concurrent_queue.cpp \
concurrent_queue_test.cpp

9
tests/qt_test.pri Normal file
View File

@ -0,0 +1,9 @@
QT += testlib
QT -= gui
CONFIG += qt console warn_on testcase no_testcase_installs
CONFIG -= app_bundle
TEMPLATE = app
include(../config.pri)

2
tests/tests.pro Normal file
View File

@ -0,0 +1,2 @@
TEMPLATE = subdirs
SUBDIRS += concurrent_queue_test