+++ /dev/null
-/*
- * Licensed to the Apache Software Foundation (ASF) under one
- * or more contributor license agreements. See the NOTICE file
- * distributed with this work for additional information
- * regarding copyright ownership. The ASF licenses this file
- * to you under the Apache License, Version 2.0 (the
- * "License"); you may not use this file except in compliance
- * with the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing,
- * software distributed under the License is distributed on an
- * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
- * KIND, either express or implied. See the License for the
- * specific language governing permissions and limitations
- * under the License.
- */
-
-#include <thrift/thrift-config.h>
-#include <thrift/concurrency/ThreadManager.h>
-#include <thrift/concurrency/ThreadFactory.h>
-#include <thrift/concurrency/Monitor.h>
-
-#include <assert.h>
-#include <deque>
-#include <set>
-#include <iostream>
-#include <stdint.h>
-
-namespace apache {
-namespace thrift {
-namespace concurrency {
-namespace test {
-
-using namespace apache::thrift::concurrency;
-
-static std::deque<std::shared_ptr<Runnable> > m_expired;
-static void expiredNotifier(std::shared_ptr<Runnable> runnable)
-{
- m_expired.push_back(runnable);
-}
-
-static void sleep_(int64_t millisec) {
- Monitor _sleep;
- Synchronized s(_sleep);
-
- try {
- _sleep.wait(millisec);
- } catch (TimedOutException&) {
- ;
- } catch (...) {
- assert(0);
- }
-}
-
-class ThreadManagerTests {
-
-public:
- class Task : public Runnable {
-
- public:
- Task(Monitor& monitor, size_t& count, int64_t timeout)
- : _monitor(monitor), _count(count), _timeout(timeout), _startTime(0), _endTime(0), _done(false) {}
-
- void run() override {
-
- _startTime = std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::steady_clock::now().time_since_epoch()).count();
-
- sleep_(_timeout);
-
- _endTime = std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::steady_clock::now().time_since_epoch()).count();
-
- _done = true;
-
- {
- Synchronized s(_monitor);
-
- // std::cout << "Thread " << _count << " completed " << std::endl;
-
- _count--;
- if (_count % 10000 == 0) {
- _monitor.notify();
- }
- }
- }
-
- Monitor& _monitor;
- size_t& _count;
- int64_t _timeout;
- int64_t _startTime;
- int64_t _endTime;
- bool _done;
- Monitor _sleep;
- };
-
- /**
- * Dispatch count tasks, each of which blocks for timeout milliseconds then
- * completes. Verify that all tasks completed and that thread manager cleans
- * up properly on delete.
- */
- bool loadTest(size_t count = 100, int64_t timeout = 100LL, size_t workerCount = 4) {
-
- Monitor monitor;
-
- size_t activeCount = count;
-
- shared_ptr<ThreadManager> threadManager = ThreadManager::newSimpleThreadManager(workerCount);
-
- shared_ptr<ThreadFactory> threadFactory
- = shared_ptr<ThreadFactory>(new ThreadFactory(false));
-
- threadManager->threadFactory(threadFactory);
-
- threadManager->start();
-
- std::set<shared_ptr<ThreadManagerTests::Task> > tasks;
-
- for (size_t ix = 0; ix < count; ix++) {
-
- tasks.insert(shared_ptr<ThreadManagerTests::Task>(
- new ThreadManagerTests::Task(monitor, activeCount, timeout)));
- }
-
- int64_t time00 = std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::steady_clock::now().time_since_epoch()).count();
-
- for (auto ix = tasks.begin();
- ix != tasks.end();
- ix++) {
-
- threadManager->add(*ix);
- }
-
- std::cout << "\t\t\t\tloaded " << count << " tasks to execute" << std::endl;
-
- {
- Synchronized s(monitor);
-
- while (activeCount > 0) {
- std::cout << "\t\t\t\tactiveCount = " << activeCount << std::endl;
- monitor.wait();
- }
- }
-
- int64_t time01 = std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::steady_clock::now().time_since_epoch()).count();
-
- int64_t firstTime = 9223372036854775807LL;
- int64_t lastTime = 0;
-
- double averageTime = 0;
- int64_t minTime = 9223372036854775807LL;
- int64_t maxTime = 0;
-
- for (auto ix = tasks.begin();
- ix != tasks.end();
- ix++) {
-
- shared_ptr<ThreadManagerTests::Task> task = *ix;
-
- int64_t delta = task->_endTime - task->_startTime;
-
- assert(delta > 0);
-
- if (task->_startTime < firstTime) {
- firstTime = task->_startTime;
- }
-
- if (task->_endTime > lastTime) {
- lastTime = task->_endTime;
- }
-
- if (delta < minTime) {
- minTime = delta;
- }
-
- if (delta > maxTime) {
- maxTime = delta;
- }
-
- averageTime += delta;
- }
-
- averageTime /= count;
-
- std::cout << "\t\t\tfirst start: " << firstTime << " Last end: " << lastTime
- << " min: " << minTime << "ms max: " << maxTime << "ms average: " << averageTime
- << "ms" << std::endl;
-
- bool success = (time01 - time00) >= ((int64_t)count * timeout) / (int64_t)workerCount;
-
- std::cout << "\t\t\t" << (success ? "Success" : "Failure")
- << "! expected time: " << ((int64_t)count * timeout) / (int64_t)workerCount << "ms elapsed time: " << time01 - time00
- << "ms" << std::endl;
-
- return success;
- }
-
- class BlockTask : public Runnable {
-
- public:
- BlockTask(Monitor& entryMonitor, Monitor& blockMonitor, bool& blocked, Monitor& doneMonitor, size_t& count)
- : _entryMonitor(entryMonitor), _entered(false), _blockMonitor(blockMonitor), _blocked(blocked), _doneMonitor(doneMonitor), _count(count) {}
-
- void run() override {
- {
- Synchronized s(_entryMonitor);
- _entered = true;
- _entryMonitor.notify();
- }
-
- {
- Synchronized s(_blockMonitor);
- while (_blocked) {
- _blockMonitor.wait();
- }
- }
-
- {
- Synchronized s(_doneMonitor);
- if (--_count == 0) {
- _doneMonitor.notify();
- }
- }
- }
-
- Monitor& _entryMonitor;
- bool _entered;
- Monitor& _blockMonitor;
- bool& _blocked;
- Monitor& _doneMonitor;
- size_t& _count;
- };
-
- /**
- * Block test. Create pendingTaskCountMax tasks. Verify that we block adding the
- * pendingTaskCountMax + 1th task. Verify that we unblock when a task completes */
-
- bool blockTest(int64_t timeout = 100LL, size_t workerCount = 2) {
- (void)timeout;
- bool success = false;
-
- try {
-
- Monitor entryMonitor; // not used by this test
- Monitor blockMonitor;
- bool blocked[] = {true, true, true};
- Monitor doneMonitor;
-
- size_t pendingTaskMaxCount = workerCount;
-
- size_t activeCounts[] = {workerCount, pendingTaskMaxCount, 1};
-
- shared_ptr<ThreadManager> threadManager
- = ThreadManager::newSimpleThreadManager(workerCount, pendingTaskMaxCount);
-
- shared_ptr<ThreadFactory> threadFactory
- = shared_ptr<ThreadFactory>(new ThreadFactory());
-
- threadManager->threadFactory(threadFactory);
-
- threadManager->start();
-
- std::vector<shared_ptr<ThreadManagerTests::BlockTask> > tasks;
- tasks.reserve(workerCount + pendingTaskMaxCount);
-
- for (size_t ix = 0; ix < workerCount; ix++) {
-
- tasks.push_back(shared_ptr<ThreadManagerTests::BlockTask>(
- new ThreadManagerTests::BlockTask(entryMonitor, blockMonitor, blocked[0], doneMonitor, activeCounts[0])));
- }
-
- for (size_t ix = 0; ix < pendingTaskMaxCount; ix++) {
-
- tasks.push_back(shared_ptr<ThreadManagerTests::BlockTask>(
- new ThreadManagerTests::BlockTask(entryMonitor, blockMonitor, blocked[1], doneMonitor, activeCounts[1])));
- }
-
- for (auto ix = tasks.begin();
- ix != tasks.end();
- ix++) {
- threadManager->add(*ix);
- }
-
- if (!(success = (threadManager->totalTaskCount() == pendingTaskMaxCount + workerCount))) {
- throw TException("Unexpected pending task count");
- }
-
- shared_ptr<ThreadManagerTests::BlockTask> extraTask(
- new ThreadManagerTests::BlockTask(entryMonitor, blockMonitor, blocked[2], doneMonitor, activeCounts[2]));
-
- try {
- threadManager->add(extraTask, 1);
- throw TException("Unexpected success adding task in excess of pending task count");
- } catch (TooManyPendingTasksException&) {
- throw TException("Should have timed out adding task in excess of pending task count");
- } catch (TimedOutException&) {
- // Expected result
- }
-
- try {
- threadManager->add(extraTask, -1);
- throw TException("Unexpected success adding task in excess of pending task count");
- } catch (TimedOutException&) {
- throw TException("Unexpected timeout adding task in excess of pending task count");
- } catch (TooManyPendingTasksException&) {
- // Expected result
- }
-
- std::cout << "\t\t\t"
- << "Pending tasks " << threadManager->pendingTaskCount() << std::endl;
-
- {
- Synchronized s(blockMonitor);
- blocked[0] = false;
- blockMonitor.notifyAll();
- }
-
- {
- Synchronized s(doneMonitor);
- while (activeCounts[0] != 0) {
- doneMonitor.wait();
- }
- }
-
- std::cout << "\t\t\t"
- << "Pending tasks " << threadManager->pendingTaskCount() << std::endl;
-
- try {
- threadManager->add(extraTask, 1);
- } catch (TimedOutException&) {
- std::cout << "\t\t\t"
- << "add timed out unexpectedly" << std::endl;
- throw TException("Unexpected timeout adding task");
-
- } catch (TooManyPendingTasksException&) {
- std::cout << "\t\t\t"
- << "add encountered too many pending exepctions" << std::endl;
- throw TException("Unexpected timeout adding task");
- }
-
- // Wake up tasks that were pending before and wait for them to complete
-
- {
- Synchronized s(blockMonitor);
- blocked[1] = false;
- blockMonitor.notifyAll();
- }
-
- {
- Synchronized s(doneMonitor);
- while (activeCounts[1] != 0) {
- doneMonitor.wait();
- }
- }
-
- // Wake up the extra task and wait for it to complete
-
- {
- Synchronized s(blockMonitor);
- blocked[2] = false;
- blockMonitor.notifyAll();
- }
-
- {
- Synchronized s(doneMonitor);
- while (activeCounts[2] != 0) {
- doneMonitor.wait();
- }
- }
-
- threadManager->stop();
-
- if (!(success = (threadManager->totalTaskCount() == 0))) {
- throw TException("Unexpected total task count");
- }
-
- } catch (TException& e) {
- std::cout << "ERROR: " << e.what() << std::endl;
- }
-
- std::cout << "\t\t\t" << (success ? "Success" : "Failure") << std::endl;
- return success;
- }
-
-
- bool apiTest() {
-
- // prove currentTime has milliseconds granularity since many other things depend on it
- int64_t a = std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::steady_clock::now().time_since_epoch()).count();
- sleep_(100);
- int64_t b = std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::steady_clock::now().time_since_epoch()).count();
- if (b - a < 50 || b - a > 150) {
- std::cerr << "\t\t\texpected 100ms gap, found " << (b-a) << "ms gap instead." << std::endl;
- return false;
- }
-
- return apiTestWithThreadFactory(shared_ptr<ThreadFactory>(new ThreadFactory()));
-
- }
-
- bool apiTestWithThreadFactory(shared_ptr<ThreadFactory> threadFactory)
- {
- shared_ptr<ThreadManager> threadManager = ThreadManager::newSimpleThreadManager(1);
- threadManager->threadFactory(threadFactory);
-
- std::cout << "\t\t\t\tstarting.. " << std::endl;
-
- threadManager->start();
- threadManager->setExpireCallback(expiredNotifier); // std::bind(&ThreadManagerTests::expiredNotifier, this));
-
-#define EXPECT(FUNC, COUNT) { size_t c = FUNC; if (c != COUNT) { std::cerr << "expected " #FUNC" to be " #COUNT ", but was " << c << std::endl; return false; } }
-
- EXPECT(threadManager->workerCount(), 1);
- EXPECT(threadManager->idleWorkerCount(), 1);
- EXPECT(threadManager->pendingTaskCount(), 0);
-
- std::cout << "\t\t\t\tadd 2nd worker.. " << std::endl;
-
- threadManager->addWorker();
-
- EXPECT(threadManager->workerCount(), 2);
- EXPECT(threadManager->idleWorkerCount(), 2);
- EXPECT(threadManager->pendingTaskCount(), 0);
-
- std::cout << "\t\t\t\tremove 2nd worker.. " << std::endl;
-
- threadManager->removeWorker();
-
- EXPECT(threadManager->workerCount(), 1);
- EXPECT(threadManager->idleWorkerCount(), 1);
- EXPECT(threadManager->pendingTaskCount(), 0);
-
- std::cout << "\t\t\t\tremove 1st worker.. " << std::endl;
-
- threadManager->removeWorker();
-
- EXPECT(threadManager->workerCount(), 0);
- EXPECT(threadManager->idleWorkerCount(), 0);
- EXPECT(threadManager->pendingTaskCount(), 0);
-
- std::cout << "\t\t\t\tadd blocking task.. " << std::endl;
-
- // We're going to throw a blocking task into the mix
- Monitor entryMonitor; // signaled when task is running
- Monitor blockMonitor; // to be signaled to unblock the task
- bool blocked(true); // set to false before notifying
- Monitor doneMonitor; // signaled when count reaches zero
- size_t activeCount = 1;
- shared_ptr<ThreadManagerTests::BlockTask> blockingTask(
- new ThreadManagerTests::BlockTask(entryMonitor, blockMonitor, blocked, doneMonitor, activeCount));
- threadManager->add(blockingTask);
-
- EXPECT(threadManager->workerCount(), 0);
- EXPECT(threadManager->idleWorkerCount(), 0);
- EXPECT(threadManager->pendingTaskCount(), 1);
-
- std::cout << "\t\t\t\tadd other task.. " << std::endl;
-
- shared_ptr<ThreadManagerTests::Task> otherTask(
- new ThreadManagerTests::Task(doneMonitor, activeCount, 0));
-
- threadManager->add(otherTask);
-
- EXPECT(threadManager->workerCount(), 0);
- EXPECT(threadManager->idleWorkerCount(), 0);
- EXPECT(threadManager->pendingTaskCount(), 2);
-
- std::cout << "\t\t\t\tremove blocking task specifically.. " << std::endl;
-
- threadManager->remove(blockingTask);
-
- EXPECT(threadManager->workerCount(), 0);
- EXPECT(threadManager->idleWorkerCount(), 0);
- EXPECT(threadManager->pendingTaskCount(), 1);
-
- std::cout << "\t\t\t\tremove next pending task.." << std::endl;
-
- shared_ptr<Runnable> nextTask = threadManager->removeNextPending();
- if (nextTask != otherTask) {
- std::cerr << "\t\t\t\t\texpected removeNextPending to return otherTask" << std::endl;
- return false;
- }
-
- EXPECT(threadManager->workerCount(), 0);
- EXPECT(threadManager->idleWorkerCount(), 0);
- EXPECT(threadManager->pendingTaskCount(), 0);
-
- std::cout << "\t\t\t\tremove next pending task (none left).." << std::endl;
-
- nextTask = threadManager->removeNextPending();
- if (nextTask) {
- std::cerr << "\t\t\t\t\texpected removeNextPending to return an empty Runnable" << std::endl;
- return false;
- }
-
- std::cout << "\t\t\t\tadd 2 expired tasks and 1 not.." << std::endl;
-
- shared_ptr<ThreadManagerTests::Task> expiredTask(
- new ThreadManagerTests::Task(doneMonitor, activeCount, 0));
-
- threadManager->add(expiredTask, 0, 1);
- threadManager->add(blockingTask); // add one that hasn't expired to make sure it gets skipped
- threadManager->add(expiredTask, 0, 1); // add a second expired to ensure removeExpiredTasks removes both
-
- sleep_(50); // make sure enough time elapses for it to expire - the shortest expiration time is 1 millisecond
-
- EXPECT(threadManager->workerCount(), 0);
- EXPECT(threadManager->idleWorkerCount(), 0);
- EXPECT(threadManager->pendingTaskCount(), 3);
- EXPECT(threadManager->expiredTaskCount(), 0);
-
- std::cout << "\t\t\t\tremove expired tasks.." << std::endl;
-
- if (!m_expired.empty()) {
- std::cerr << "\t\t\t\t\texpected m_expired to be empty" << std::endl;
- return false;
- }
-
- threadManager->removeExpiredTasks();
-
- if (m_expired.size() != 2) {
- std::cerr << "\t\t\t\t\texpected m_expired to be set" << std::endl;
- return false;
- }
-
- if (m_expired.front() != expiredTask) {
- std::cerr << "\t\t\t\t\texpected m_expired[0] to be the expired task" << std::endl;
- return false;
- }
- m_expired.pop_front();
-
- if (m_expired.front() != expiredTask) {
- std::cerr << "\t\t\t\t\texpected m_expired[1] to be the expired task" << std::endl;
- return false;
- }
-
- m_expired.clear();
-
- threadManager->remove(blockingTask);
-
- EXPECT(threadManager->workerCount(), 0);
- EXPECT(threadManager->idleWorkerCount(), 0);
- EXPECT(threadManager->pendingTaskCount(), 0);
- EXPECT(threadManager->expiredTaskCount(), 2);
-
- std::cout << "\t\t\t\tadd expired task (again).." << std::endl;
-
- threadManager->add(expiredTask, 0, 1); // expires in 1ms
- sleep_(50); // make sure enough time elapses for it to expire - the shortest expiration time is 1ms
-
- std::cout << "\t\t\t\tadd worker to consume expired task.." << std::endl;
-
- threadManager->addWorker();
- sleep_(100); // make sure it has time to spin up and expire the task
-
- if (m_expired.empty()) {
- std::cerr << "\t\t\t\t\texpected m_expired to be set" << std::endl;
- return false;
- }
-
- if (m_expired.front() != expiredTask) {
- std::cerr << "\t\t\t\t\texpected m_expired to be the expired task" << std::endl;
- return false;
- }
-
- m_expired.clear();
-
- EXPECT(threadManager->workerCount(), 1);
- EXPECT(threadManager->idleWorkerCount(), 1);
- EXPECT(threadManager->pendingTaskCount(), 0);
- EXPECT(threadManager->expiredTaskCount(), 3);
-
- std::cout << "\t\t\t\ttry to remove too many workers" << std::endl;
- try {
- threadManager->removeWorker(2);
- std::cerr << "\t\t\t\t\texpected InvalidArgumentException" << std::endl;
- return false;
- } catch (const InvalidArgumentException&) {
- /* expected */
- }
-
- std::cout << "\t\t\t\tremove worker.. " << std::endl;
-
- threadManager->removeWorker();
-
- EXPECT(threadManager->workerCount(), 0);
- EXPECT(threadManager->idleWorkerCount(), 0);
- EXPECT(threadManager->pendingTaskCount(), 0);
- EXPECT(threadManager->expiredTaskCount(), 3);
-
- std::cout << "\t\t\t\tadd blocking task.. " << std::endl;
-
- threadManager->add(blockingTask);
-
- EXPECT(threadManager->workerCount(), 0);
- EXPECT(threadManager->idleWorkerCount(), 0);
- EXPECT(threadManager->pendingTaskCount(), 1);
-
- std::cout << "\t\t\t\tadd worker.. " << std::endl;
-
- threadManager->addWorker();
- {
- Synchronized s(entryMonitor);
- while (!blockingTask->_entered) {
- entryMonitor.wait();
- }
- }
-
- EXPECT(threadManager->workerCount(), 1);
- EXPECT(threadManager->idleWorkerCount(), 0);
- EXPECT(threadManager->pendingTaskCount(), 0);
-
- std::cout << "\t\t\t\tunblock task and remove worker.. " << std::endl;
-
- {
- Synchronized s(blockMonitor);
- blocked = false;
- blockMonitor.notifyAll();
- }
- threadManager->removeWorker();
-
- EXPECT(threadManager->workerCount(), 0);
- EXPECT(threadManager->idleWorkerCount(), 0);
- EXPECT(threadManager->pendingTaskCount(), 0);
-
- std::cout << "\t\t\t\tcleanup.. " << std::endl;
-
- blockingTask.reset();
- threadManager.reset();
- return true;
- }
-};
-
-}
-}
-}
-} // apache::thrift::concurrency
-
-using namespace apache::thrift::concurrency::test;