mirror of
https://github.com/tearshark/librf.git
synced 2024-10-01 15:57:07 +08:00
108 lines
2.5 KiB
C++
108 lines
2.5 KiB
C++
//验证channel是否线程安全
|
||
|
||
#include <chrono>
|
||
#include <iostream>
|
||
#include <string>
|
||
#include <conio.h>
|
||
#include <thread>
|
||
#include <deque>
|
||
#include <mutex>
|
||
|
||
#include "librf.h"
|
||
|
||
using namespace resumef;
|
||
|
||
using namespace std::chrono;
|
||
static std::mutex cout_mutex;
|
||
std::atomic<intptr_t> gcounter = 0;
|
||
|
||
#define OUTPUT_DEBUG 1
|
||
|
||
future_t<> test_channel_consumer(const channel_t<std::string> & c, size_t cnt)
|
||
{
|
||
for (size_t i = 0; i < cnt; ++i)
|
||
{
|
||
try
|
||
{
|
||
auto val = co_await c.read();
|
||
++gcounter;
|
||
#if OUTPUT_DEBUG
|
||
{
|
||
scoped_lock<std::mutex> __lock(cout_mutex);
|
||
std::cout << "R " << val << "@" << std::this_thread::get_id() << std::endl;
|
||
}
|
||
#endif
|
||
}
|
||
catch (channel_exception e)
|
||
{
|
||
//MAX_CHANNEL_QUEUE=0,并且先读后写，会触发read_before_write异常
|
||
scoped_lock<std::mutex> __lock(cout_mutex);
|
||
std::cout << e.what() << std::endl;
|
||
}
|
||
|
||
#if OUTPUT_DEBUG
|
||
co_await sleep_for(50ms);
|
||
#endif
|
||
}
|
||
}
|
||
|
||
future_t<> test_channel_producer(const channel_t<std::string> & c, size_t cnt)
|
||
{
|
||
for (size_t i = 0; i < cnt; ++i)
|
||
{
|
||
co_await c.write(std::to_string(i));
|
||
#if OUTPUT_DEBUG
|
||
{
|
||
scoped_lock<std::mutex> __lock(cout_mutex);
|
||
std::cout << "W " << i << "@" << std::this_thread::get_id() << std::endl;
|
||
}
|
||
#endif
|
||
}
|
||
}
|
||
|
||
const size_t N = 8;
|
||
const size_t BATCH = 1000000;
|
||
const size_t MAX_CHANNEL_QUEUE = N + 1; //0, 1, 5, 10, -1
|
||
|
||
void resumable_main_channel_mult_thread()
|
||
{
|
||
channel_t<std::string> c(MAX_CHANNEL_QUEUE);
|
||
|
||
std::thread write_th([&]
|
||
{
|
||
local_scheduler my_scheduler; //2017/12/14日，仍然存在BUG。真多线程下调度，存在有协程无法被调度完成的BUG
|
||
go test_channel_producer(c, BATCH * N);
|
||
#if RESUMEF_ENABLE_MULT_SCHEDULER
|
||
this_scheduler()->run_until_notask();
|
||
#endif
|
||
std::cout << "Write OK\r\n";
|
||
});
|
||
|
||
//std::this_thread::sleep_for(100ms);
|
||
|
||
std::thread read_th[N];
|
||
for (size_t i = 0; i < N; ++i)
|
||
{
|
||
read_th[i] = std::thread([&]
|
||
{
|
||
local_scheduler my_scheduler; //2017/12/14日，仍然存在BUG。真多线程下调度，存在有协程无法被调度完成的BUG
|
||
go test_channel_consumer(c, BATCH);
|
||
#if RESUMEF_ENABLE_MULT_SCHEDULER
|
||
this_scheduler()->run_until_notask();
|
||
#endif
|
||
std::cout << "Read OK\r\n";
|
||
});
|
||
}
|
||
|
||
#if !RESUMEF_ENABLE_MULT_SCHEDULER
|
||
std::this_thread::sleep_for(100ms);
|
||
scheduler_t::g_scheduler.run_until_notask();
|
||
#endif
|
||
for(auto & th : read_th)
|
||
th.join();
|
||
write_th.join();
|
||
|
||
std::cout << "OK: counter = " << gcounter.load() << std::endl;
|
||
(void)_getch();
|
||
}
|