#include <iostream>
#include <mutex>

#include <boost/thread/thread.hpp>
#include <boost/lockfree/queue.hpp>
#include <boost/atomic.hpp>

typedef struct Data
{
    uint64_t sec;
    uint64_t nano;
} DataType;

boost::atomic<bool> done (false);
boost::atomic_int producer_count(0);
boost::atomic_int consumer_count(0);
boost::atomic_uint64_t sum(0);

boost::lockfree::queue<DataType ,
    boost::lockfree::fixed_sized<true>,
    boost::lockfree::capacity < 128 >
    > queue;

const int iterations = 1 * 1000 * 1000;
const int producer_thread_count = 2;
const int consumer_thread_count = 1;

std::mutex producerLock;

void producer()
{
    DataType msg;

    struct timespec localTime;
    for (int i = 0; i != iterations; ++i)
    {
        ++producer_count;
        clock_gettime( CLOCK_REALTIME,  &localTime );

        msg.sec = localTime.tv_sec;
        msg.nano = localTime.tv_nsec;

        producerLock.lock();
        while (!queue.push(msg));
        producerLock.unlock();
    }
}




inline void calc(DataType& msg)
{
    struct timespec localTime;
    clock_gettime( CLOCK_REALTIME, &localTime );

    uint64_t sec = (localTime.tv_sec - msg.sec);
    uint64_t nsec = (localTime.tv_nsec - msg.nano) + (1000L * 1000L * 1000L * sec);
    sum += nsec;
    ++consumer_count;
}

void consumer(void)
{
    DataType msg;
    while (!done) {

        while (queue.pop(msg)) {
                calc(msg);
        }
    }

    while (queue.pop(msg))
    {
        calc(msg);
    }
}


void PrintResults()
{
    std::cout << "produced   " << producer_count << " objects." << std::endl;
    std::cout << "consumed   " << consumer_count << " objects." << std::endl;
    std::cout << "Total time " << sum << " ns." << std::endl;
    std::cout << "Messages   " << sum / consumer_count << " ns per msg" << std::endl;
}

int main(int argc, char* argv[])
{
    using namespace std;
    cout << "boost::lockfree::queue is ";
    if (!queue.is_lock_free())
        cout << "not ";
    cout << "lockfree" << endl;

    boost::thread_group producer_threads, consumer_threads;

    for (int i = 0; i != producer_thread_count; ++i)
      producer_threads.create_thread(producer);

    for (int i = 0; i != consumer_thread_count; ++i)
        consumer_threads.create_thread(consumer);

    producer_threads.join_all();
    done = true;
    consumer_threads.join_all();

    PrintResults();

    return 0;
}

boost::lockfree::queue is lockfree
produced   2000000 objects.
consumed   2000000 objects.
Total time 2748765264 ns.
Messages   1374 ns per msg

real    0m1.374s
user    0m2.868s
sys     0m0.880s

boost::lockfree::queue is lockfree
produced   2000000 objects.
consumed   2000000 objects.
Total time 67803937295 ns.
Messages   33901 ns per msg

real    0m0.541s
user    0m1.603s
sys     0m0.002s