2020-03-26 17:26:39 +08:00
|
|
|
|
//验证channel是否线程安全
|
2017-11-27 18:37:18 +08:00
|
|
|
|
|
|
|
|
|
#include <chrono>
|
|
|
|
|
#include <iostream>
|
|
|
|
|
#include <string>
|
|
|
|
|
#include <thread>
|
|
|
|
|
#include <deque>
|
|
|
|
|
#include <mutex>
|
|
|
|
|
|
|
|
|
|
#include "librf.h"
|
|
|
|
|
|
|
|
|
|
using namespace resumef;
|
|
|
|
|
|
|
|
|
|
using namespace std::chrono;
|
|
|
|
|
static std::mutex cout_mutex;
|
2017-12-14 10:50:45 +08:00
|
|
|
|
std::atomic<intptr_t> gcounter = 0;
|
2017-11-27 18:37:18 +08:00
|
|
|
|
|
2020-02-21 09:36:59 +08:00
|
|
|
|
#define OUTPUT_DEBUG 0
|
2017-11-27 18:37:18 +08:00
|
|
|
|
|
2020-03-09 17:59:07 +08:00
|
|
|
|
future_t<> test_channel_consumer(channel_t<std::string> c, size_t cnt)
|
2017-11-27 18:37:18 +08:00
|
|
|
|
{
|
|
|
|
|
for (size_t i = 0; i < cnt; ++i)
|
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
auto val = co_await c.read();
|
2017-12-14 10:50:45 +08:00
|
|
|
|
++gcounter;
|
2017-11-27 18:37:18 +08:00
|
|
|
|
#if OUTPUT_DEBUG
|
|
|
|
|
{
|
|
|
|
|
scoped_lock<std::mutex> __lock(cout_mutex);
|
|
|
|
|
std::cout << "R " << val << "@" << std::this_thread::get_id() << std::endl;
|
|
|
|
|
}
|
|
|
|
|
#endif
|
|
|
|
|
}
|
2020-02-29 01:11:21 +08:00
|
|
|
|
catch (channel_exception& e)
|
2017-11-27 18:37:18 +08:00
|
|
|
|
{
|
|
|
|
|
//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
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2020-03-09 17:59:07 +08:00
|
|
|
|
future_t<> test_channel_producer(channel_t<std::string> c, size_t cnt)
|
2017-11-27 18:37:18 +08:00
|
|
|
|
{
|
|
|
|
|
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
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2020-03-09 14:23:35 +08:00
|
|
|
|
const size_t WRITE_THREAD = 6;
|
|
|
|
|
const size_t READ_THREAD = 6;
|
|
|
|
|
const size_t READ_BATCH = 1000000;
|
|
|
|
|
const size_t MAX_CHANNEL_QUEUE = 5; //0, 1, 5, 10, -1
|
2017-11-27 18:37:18 +08:00
|
|
|
|
|
|
|
|
|
void resumable_main_channel_mult_thread()
|
|
|
|
|
{
|
|
|
|
|
channel_t<std::string> c(MAX_CHANNEL_QUEUE);
|
|
|
|
|
|
2020-03-09 14:23:35 +08:00
|
|
|
|
std::thread write_th[WRITE_THREAD];
|
|
|
|
|
for (size_t i = 0; i < WRITE_THREAD; ++i)
|
2017-11-27 18:37:18 +08:00
|
|
|
|
{
|
2020-03-09 14:23:35 +08:00
|
|
|
|
write_th[i] = std::thread([&]
|
|
|
|
|
{
|
2020-03-31 15:30:45 +08:00
|
|
|
|
local_scheduler_t my_scheduler;
|
2020-03-09 14:23:35 +08:00
|
|
|
|
go test_channel_producer(c, READ_BATCH * READ_THREAD / WRITE_THREAD);
|
|
|
|
|
this_scheduler()->run_until_notask();
|
2020-03-29 23:10:38 +08:00
|
|
|
|
|
2020-03-09 14:23:35 +08:00
|
|
|
|
{
|
|
|
|
|
scoped_lock<std::mutex> __lock(cout_mutex);
|
|
|
|
|
std::cout << "Write OK\r\n";
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
}
|
2017-11-27 18:37:18 +08:00
|
|
|
|
|
2020-02-21 09:36:59 +08:00
|
|
|
|
std::this_thread::sleep_for(100ms);
|
2017-11-27 18:37:18 +08:00
|
|
|
|
|
2020-03-09 14:23:35 +08:00
|
|
|
|
std::thread read_th[READ_THREAD];
|
|
|
|
|
for (size_t i = 0; i < READ_THREAD; ++i)
|
2017-11-27 18:37:18 +08:00
|
|
|
|
{
|
2017-12-13 17:49:38 +08:00
|
|
|
|
read_th[i] = std::thread([&]
|
2017-11-27 18:37:18 +08:00
|
|
|
|
{
|
2020-03-31 15:30:45 +08:00
|
|
|
|
local_scheduler_t my_scheduler;
|
2020-03-09 14:23:35 +08:00
|
|
|
|
go test_channel_consumer(c, READ_BATCH);
|
2017-11-27 18:37:18 +08:00
|
|
|
|
this_scheduler()->run_until_notask();
|
2020-03-29 23:10:38 +08:00
|
|
|
|
|
2020-02-21 09:36:59 +08:00
|
|
|
|
{
|
|
|
|
|
scoped_lock<std::mutex> __lock(cout_mutex);
|
|
|
|
|
std::cout << "Read OK\r\n";
|
|
|
|
|
}
|
2017-11-27 18:37:18 +08:00
|
|
|
|
});
|
|
|
|
|
}
|
2017-12-14 10:50:45 +08:00
|
|
|
|
|
|
|
|
|
std::this_thread::sleep_for(100ms);
|
2020-02-15 15:20:16 +08:00
|
|
|
|
scheduler_t::g_scheduler.run_until_notask();
|
2020-03-09 14:23:35 +08:00
|
|
|
|
|
2017-12-13 17:49:38 +08:00
|
|
|
|
for(auto & th : read_th)
|
2017-11-27 18:37:18 +08:00
|
|
|
|
th.join();
|
2020-03-09 14:23:35 +08:00
|
|
|
|
for (auto& th : write_th)
|
|
|
|
|
th.join();
|
2017-11-27 18:37:18 +08:00
|
|
|
|
|
2017-12-14 10:50:45 +08:00
|
|
|
|
std::cout << "OK: counter = " << gcounter.load() << std::endl;
|
2017-11-27 18:37:18 +08:00
|
|
|
|
}
|