mariadb-columnstore-engine/utils/threadpool/tdriver.cpp

/* Copyright (C) 2014 InfiniDB, Inc.

   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; version 2 of
   the License.

   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, write to the Free Software
   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
   MA 02110-1301, USA. */

// $Id: tdriver.cpp 3495 2013-01-21 14:09:51Z rdempsey $
#include <string>
#include <stdexcept>
#include <iostream>
#include <fstream>
using namespace std;
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <boost/thread.hpp>
#include <boost/scoped_ptr.hpp>
#include <boost/scoped_array.hpp>
#include <boost/shared_ptr.hpp>

#include <cppunit/extensions/HelperMacros.h>

#include "threadpool.h"

int thecount = 0;
boost::mutex mutex;



class ThreadPoolTestSuite : public CppUnit::TestFixture
{

    CPPUNIT_TEST_SUITE( ThreadPoolTestSuite );

    CPPUNIT_TEST( test_1 );

    CPPUNIT_TEST_SUITE_END();

private:

    // Functor class
    struct foo
    {
        void operator ()()
        {
            for (int i = 0; i < 500; i++)
            {
                // simulate some work
                fData++;
            }

            boost::mutex::scoped_lock lock(mutex);

            std::cout << "count = " << ++thecount << ' ' << fData << std::endl;
        }

        foo(int i):
                fData(i)
    {}

        foo(const foo& copy)
                : fData(copy.fData)
        {}

        int fData;

    };

public:
    void setUp()
    {}

    void tearDown()
    {}

    void test_1()
    {


        threadpool::ThreadPool pool( 5, 10 );

        for (int y = 0; y < 10; y++)
        {
            foo bar(y);

            for (int i = 0; i < 10; ++i)
            {
                pool.invoke(bar);
            }

            // Wait until all of the queued up and in-progress work has finished
            pool.wait();
            pool.dump();
        }


    }

};


CPPUNIT_TEST_SUITE_REGISTRATION( ThreadPoolTestSuite );

#include <cppunit/extensions/TestFactoryRegistry.h>
#include <cppunit/ui/text/TestRunner.h>


int main( int argc, char **argv)
{
    CppUnit::TextUi::TestRunner runner;
    CppUnit::TestFactoryRegistry &registry = CppUnit::TestFactoryRegistry::getRegistry();
    runner.addTest( registry.makeTest() );
    bool wasSuccessful = runner.run( "", false );
    return (wasSuccessful ? 0 : 1);
}