main_no_debug.cpp

Go to the documentation of this file.

#include <iostream>             // std::cout
#include <thread>               // std::thread
#include <mutex>                // std::mutex, std::adopt_lock, std::unique_lock
#include <chrono>               // std::chrono
#include <condition_variable>   // std::condition_variable
#include <random>               // std::default_random_generator

#include "LockGuard.h"          // chal::LockGuard

#define NUM_THDS 3              // number of threads to execute


// Globals
std::mutex mtx;                 // shared mutex
std::condition_variable cond;   // shared condition variable



void thd_printer(int id, std::string msg) {

    std::cout << "thread" << id+1 << ": " << msg << std::endl;
}


void thd_worker (const int id, int &next_thd, std::default_random_engine &rand_e) {

    int wait_tm; 

    thd_printer(id, "starting, waiting.");

    // repeat forever
    while(1) {
    
        // lock mutex
        mtx.lock();
        std::unique_lock<std::mutex> locker (mtx, std::adopt_lock);
        
        // wait for condition signal
        // Upon condition signal, check if current thread is next
        // if yes continue, if not keep waiting
        // lambda function creates condition predicate 
        cond.wait(locker, [&]() { return id == next_thd; });

        thd_printer(id, "signal received, doing work ....");

        // generate pseudo-random value between 1 and 5 seconds
        wait_tm = 1 + rand_e() % 5; 
         
        // sleep to simulate work
        std::this_thread::sleep_for(std::chrono::seconds(wait_tm));

        thd_printer(id, "done with work, signal next thread");

        // if topmost thread, reset next_thd
        if(next_thd == NUM_THDS-1) 
            next_thd = 0;
        else 
            ++next_thd; 

        cond.notify_all();  
    }
}



int main () {

    std::thread threads[NUM_THDS]; 
    int next_thd = -1; // prevent a spurious wake from prematurely activating thread 0

    std::default_random_engine rand_e; 
   
    std::cout << "main: starting all threads" << std::endl;

    // spawn NUM_THDS threads:
    for ( int i=0; i<NUM_THDS; ++i ) {

        // populate the array of thread objects
        // pass in: * their unique ID by value
        //          * an integer to keep track of thread order by reference
        //          * a shared psuedo-random number generator by reference
        threads[i] = std::thread(thd_worker, i, std::ref(next_thd), std::ref(rand_e));
    }

    std::this_thread::sleep_for(std::chrono::seconds(3));

    cond.notify_all();  // start sequence
    next_thd = 0;       // allow thread 0 to be activated

    // clean up
    for ( auto& th : threads ) {
        th.join();
    }

    return 0;
}