acl/lib_fiber/samples/thread_event/main.cpp

132 lines
2.3 KiB
C++

#include "stdafx.h"
#include <stdio.h>
#include <stdlib.h>
#include <vector>
static int __nthreads = 2;
static int __nloop = 2;
static int __delay = 0;
static acl::atomic_long __counter = 0;
static long long __counter2 = 0;
class mythread : public acl::thread
{
public:
mythread(acl::fiber_event& event, std::map<int, int>* table)
: event_(event)
, table_(table)
{
this->set_detachable(false);
}
private:
~mythread(void) {}
private:
acl::fiber_event& event_;
std::map<int, int>* table_;
// @override
void* run(void)
{
for (int i = 0; i < __nloop; i++)
{
assert(event_.wait());
if (__delay > 0)
acl_doze(__delay);
__counter2++;
(*table_)[i] = i;
if (!event_.notify()) {
printf("tid=%ld, notify error\r\n", self());
exit (1);
}
__counter++;
}
return NULL;
}
};
//////////////////////////////////////////////////////////////////////////////
static void usage(const char* procname)
{
printf("usage: %s -h [help]\r\n"
" -t nthreads[default: 2]\r\n"
" -n nloop[default: 2]\r\n"
" -d delay[default: 0 ms]\r\n"
" -m [use thread mutex, default: false]\r\n",
procname);
}
#include <sys/eventfd.h>
static void test(void)
{
int fd = eventfd(0, 0);
long long n = 100;
write(fd, &n, sizeof(n));
read(fd, &n, sizeof(n));
printf("n=%lld\r\n", n);
}
int main(int argc, char *argv[])
{
int ch;
bool use_mutex = false;
acl::acl_cpp_init();
acl::log::stdout_open(true);
while ((ch = getopt(argc, argv, "ht:n:d:m")) > 0)
{
switch (ch)
{
case 'h':
usage(argv[0]);
return 0;
case 't':
__nthreads = atoi(optarg);
break;
case 'n':
__nloop = atoi(optarg);
break;
case 'd':
__delay = atoi(optarg);
break;
case 'm':
use_mutex = true;
break;
default:
break;
}
}
test();
std::map<int, int> table;
acl::fiber_event event(use_mutex);
std::vector<acl::thread*> threads;
for (int i = 0; i < __nthreads; i++)
{
acl::thread* thr = new mythread(event, &table);
threads.push_back(thr);
thr->set_detachable(false);
thr->start();
}
for (std::vector<acl::thread*>::iterator it = threads.begin();
it != threads.end(); ++it)
{
(*it)->wait();
delete *it;
}
printf("all over, thread=%d, nloop=%d, counter=%lld, %lld\r\n",
__nthreads, __nloop, __counter.value(), __counter2);
return 0;
}