Skip to content
Snippets Groups Projects
Select Git revision
  • f1d30708ce9b3c7a89acf9165ff00bbbd5fdfb22
  • main default protected
2 results

plugins.ts

Blame
  • Code owners
    Assign users and groups as approvers for specific file changes. Learn more.
    threadpooltester.cpp 2.82 KiB
    /*
     *  Copyright (C) 2014-2020 Savoir-faire Linux Inc.
     *
     *  Author: Adrien Béraud <adrien.beraud@savoirfairelinux.com>
     *
     *  This program is free software; you can redistribute it and/or modify
     *  it under the terms of the GNU General Public License as published by
     *  the Free Software Foundation; either version 3 of the License, or
     *  (at your option) any later version.
     *
     *  This program is distributed in the hope that it will be useful,
     *  but WITHOUT ANY WARRANTY; without even the implied warranty of
     *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     *  GNU General Public License for more details.
     *
     *  You should have received a copy of the GNU General Public License
     *  along with this program. If not, see <https://www.gnu.org/licenses/>.
     */
    
    #include "threadpooltester.h"
    
    #include "opendht/thread_pool.h"
    #include <atomic>
    
    namespace test {
    CPPUNIT_TEST_SUITE_REGISTRATION(ThreadPoolTester);
    using clock = std::chrono::steady_clock;
    
    void
    ThreadPoolTester::setUp() {
    
    }
    
    void
    ThreadPoolTester::testThreadPool() {
        dht::ThreadPool pool(16);
    
        constexpr unsigned N = 64 * 1024;
        std::atomic_uint count {0};
        for (unsigned i=0; i<N; i++)
            pool.run([&] {
                count++;
            });
    
        auto start = clock::now();
        while (count.load() != N && clock::now() - start < std::chrono::seconds(10))
            std::this_thread::sleep_for(std::chrono::milliseconds(10));
    
        pool.join();
        CPPUNIT_ASSERT(count.load() == N);
    }
    
    void
    ThreadPoolTester::testExecutor()
    {
        dht::ThreadPool pool(8);
        auto executor1 = std::make_shared<dht::Executor>(pool, 1);
        auto executor4 = std::make_shared<dht::Executor>(pool, 4);
        auto executor8 = std::make_shared<dht::Executor>(pool, 8);
    
        constexpr unsigned N = 64 * 1024;
        unsigned count1 {0};
        std::atomic_uint count4 {0};
        std::atomic_uint count8 {0};
        for (unsigned i=0; i<N; i++) {
            executor1->run([&] { count1++; });
            executor4->run([&] { count4++; });
            executor8->run([&] { count8++; });
        }
    
        auto start = clock::now();
        while ((count1 != N ||
                count4.load() != N ||
                count8.load() != N) && clock::now() - start < std::chrono::seconds(20))
        {
            std::this_thread::sleep_for(std::chrono::milliseconds(10));
        }
        executor1.reset();
        executor4.reset();
        executor8.reset();
        CPPUNIT_ASSERT_EQUAL(N, count1);
        CPPUNIT_ASSERT_EQUAL(N, count4.load());
        CPPUNIT_ASSERT_EQUAL(N, count8.load());
    }
    
    void
    ThreadPoolTester::testContext()
    {
        std::atomic_uint count {0};
        constexpr unsigned N = 64 * 1024;
    
        {
            dht::ExecutionContext ctx(dht::ThreadPool::computation());
            for (unsigned i=0; i<N; i++) {
                ctx.run([&] { count++; });
            }
        }
    
        CPPUNIT_ASSERT_EQUAL(N, count.load());
    
    }
    
    void
    ThreadPoolTester::tearDown() {
    }
    
    }  // namespace test