увеличить asio семафор-подобное решение - PullRequest
Новая
       27

увеличить asio семафор-подобное решение

1 голос
/ 17 октября 2011

Этот вопрос является продолжением этого вопроса. Во всяком случае, это мотивация - немного к / п из оригинального комментария вопроса:

Я хотел бы иметь возможность публиковать одну группу заданий в нескольких потоках ( CalcFib функций), а затем, когда задания заканчиваются, другую группу заданий ( CalcFib2 ) функции), также на несколько потоков. Этот цикл повторяется много раз (здесь два), поэтому я подумал, что лучше всего создать boost::asio::io_service и создать потоки в начале цикла, поэтому мне не нужно создавать и уничтожать потоки каждый раз, когда цикл начинается / заканчивается.

Я создал две int переменные с неправильным именем семафора _ ** и уменьшил их в вышеупомянутых функциях. Код, ожидающий завершения группы заданий, прост: while, как показано ниже. Решение работает, по крайней мере, как я вижу.

Действительно ли ожидание, использующее while, действительно является подходом? Что мне не хватает? Есть ли лучший способ сделать это?

Мой код выглядит так: 

#include <boost/asio.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/thread.hpp>
#include <boost/thread/mutex.hpp>
#include <boost/bind.hpp>
#include <iostream>

boost::mutex global_stream_lock;
int semaphore_fib = 0;
int semaphore_fib2 = 0;

void WorkerThread( boost::shared_ptr< boost::asio::io_service > io_service)
{
    global_stream_lock.lock();
    std::cout << "[" << boost::this_thread::get_id()
        << "] Thread Start" << std::endl;
    global_stream_lock.unlock();

    io_service->run();

    global_stream_lock.lock();
    std::cout << "[" << boost::this_thread::get_id()
        << "] Thread Finish" << std::endl;
    global_stream_lock.unlock();
}

size_t fib( size_t n )
{
    if ( n <= 1 )
    {
        return n;
    }
    boost::this_thread::sleep( boost::posix_time::milliseconds( 1000 ) );
    return fib( n - 1 ) + fib( n - 2);
}

void CalcFib( size_t n )
{
    global_stream_lock.lock();
    std::cout << "[" << boost::this_thread::get_id()
        << "] Now calculating fib( " << n << " ) " << std::endl;
    global_stream_lock.unlock();

    size_t f = fib( n );

    global_stream_lock.lock();
    std::cout << "[" << boost::this_thread::get_id()
        << "] fib( " << n << " ) = " << f << std::endl;
    semaphore_fib = semaphore_fib-1;
    global_stream_lock.unlock();
}

void CalcFib2( size_t n )
{
    global_stream_lock.lock();
    std::cout << "\t\t[" << boost::this_thread::get_id()
        << "] Now calculating fib2( " << n << " ) " << std::endl;
    global_stream_lock.unlock();

    size_t f = fib( n );

    global_stream_lock.lock();
    std::cout << "\t\t[" << boost::this_thread::get_id()
        << "] fib2( " << n << " ) = " << f << std::endl;
    semaphore_fib2=semaphore_fib2-1;
    global_stream_lock.unlock();
}

int main( int argc, char * argv[] )
{
    boost::shared_ptr< boost::asio::io_service > io_service(
        new boost::asio::io_service
        );
    boost::shared_ptr< boost::asio::io_service::work > work(
        new boost::asio::io_service::work( *io_service )
        );
    boost::asio::io_service::strand strand( *io_service );

    global_stream_lock.lock();
    std::cout << "[" << boost::this_thread::get_id()
        << "] The program will exit when all work has finished."
        << std::endl;
    global_stream_lock.unlock();

    boost::thread_group worker_threads;
    for( int x = 0; x < 2; ++x )
    {
        worker_threads.create_thread( 
            boost::bind( &WorkerThread, io_service)
            );
    }
    for(int loop_no=0; loop_no<2; ++loop_no)
    {
        semaphore_fib=3;
        io_service->post( boost::bind( CalcFib, 5 ) );
        io_service->post( boost::bind( CalcFib, 4 ) );
        io_service->post( boost::bind( CalcFib, 3 ) );
        while(semaphore_fib>0)
        {
            // waiting
        }

        global_stream_lock.lock();
        std::cout << "[" << boost::this_thread::get_id()
            << "] ******* CalcFib group finished ********" << std::endl;
        global_stream_lock.unlock();

        semaphore_fib2=3;
        io_service->post( boost::bind( CalcFib2, 2 ) );
        io_service->post( boost::bind( CalcFib2, 1 ) );
        io_service->post( boost::bind( CalcFib2, 1 ) );
        while(semaphore_fib2>0)
        {
            // waiting
        }
        global_stream_lock.lock();
        std::cout << "[" << boost::this_thread::get_id()
            << "] ******* CalcFib2 group finished  ********" << std::endl;
        global_stream_lock.unlock();
    }
    work.reset();
    worker_threads.join_all();

    return 0;
}

1 Ответ

0 голосов
/ 18 октября 2011

Вот как я решил слишком много if операторов в цикле while в main().

  • Добавил boost::condition_variable cv; и boost::mutex mx; в качестве глобальных переменных
  • Добавлено boost::lock_guard<boost::mutex> lk(mx);cv.notify_all(); в конце функций CalcFib и CalcFib2
  • Добавлено boost::unique_lock<boost::mutex> lk(mx); cv.wait(lk); в каждый while цикл

Таким образом, я достиг того, что *Оператор 1021 * в цикле while выполняется только при выполнении cv.notify_all().

...