acl/lib_fiber/samples/thread_mutex/main.cpp

189 lines
3.4 KiB
C++
Raw Normal View History

#include "stdafx.h"
#include <stdio.h>
#include <stdlib.h>
#include <vector>
static bool __show = false;
static int __nthreads = 2;
static int __nfibers = 2;
static int __nloop = 2;
static int __delay = 0;
static acl::atomic_long __counter = 0;
class myfiber : public acl::fiber
{
public:
myfiber(ACL_FIBER_EVENT* event, int& nfibers)
: event_(event)
, nfibers_(nfibers)
{
}
protected:
// @override
void run(void)
{
for (int i = 0; i < __nloop; i++)
{
if (__show)
printf("thread-%lu, fiber-%u begin lock\r\n",
acl::thread::thread_self(),
acl::fiber::self());
acl_fiber_event_wait(event_);
if (__show)
printf("thread-%lu, fiber-%u lock ok\r\n",
acl::thread::thread_self(),
acl::fiber::self());
if (__delay > 0)
acl_doze(__delay);
if (__show)
printf("thread-%lu, fiber-%u begin unlock\r\n",
acl::thread::thread_self(),
acl::fiber::self());
2017-11-24 11:40:01 +08:00
acl_fiber_event_notify(event_);
__counter++;
if (__show)
printf("thread-%lu, fiber-%u unlock ok, counter-%lld\r\n",
acl::thread::thread_self(),
acl::fiber::self(),
__counter.value());
}
--nfibers_;
if (nfibers_ == 0)
{
printf("--nfibers_=%d\r\n", nfibers_);
acl::fiber::schedule_stop();
}
delete this;
}
private:
ACL_FIBER_EVENT* event_;
int& nfibers_;
~myfiber(void) {}
};
class fiber_sleep : public acl::fiber
{
public:
fiber_sleep(void) {}
~fiber_sleep(void) {}
private:
void run(void)
{
sleep(1);
}
};
class mythread : public acl::thread
{
public:
mythread(ACL_FIBER_EVENT* event) : event_(event)
{
this->set_detachable(false);
}
~mythread(void) {}
private:
ACL_FIBER_EVENT* event_;
// @override
void* run(void)
{
int nfibers = __nfibers, n = nfibers;
for (int i = 0; i < n; i++)
{
acl::fiber* fb = new myfiber(event_, nfibers);
fb->start();
}
acl::fiber::schedule();
return NULL;
}
};
//////////////////////////////////////////////////////////////////////////////
static void usage(const char* procname)
{
printf("usage: %s -h [help]\r\n"
" -t nthreads[default: 2]\r\n"
" -c nfibers[default: 2]\r\n"
" -n nloop[default: 2]\r\n"
" -d delay[default: 100 ms]\r\n"
" -S [show info]\r\n"
" -m [use thread_mutex]\r\n",
procname);
}
int main(int argc, char *argv[])
{
int ch;
acl::acl_cpp_init();
acl::log::stdout_open(true);
while ((ch = getopt(argc, argv, "hc:t:n:d:s:S")) > 0)
{
switch (ch)
{
case 'h':
usage(argv[0]);
return 0;
case 'c':
__nfibers = atoi(optarg);
break;
case 't':
__nthreads = atoi(optarg);
break;
case 'n':
__nloop = atoi(optarg);
break;
case 'd':
__delay = atoi(optarg);
break;
case 'S':
__show = true;
break;
default:
break;
}
}
ACL_FIBER_EVENT* event = acl_fiber_event_create();
std::vector<acl::thread*> threads;
for (int i = 0; i < __nthreads; i++)
{
acl::thread* thr = new mythread(event);
threads.push_back(thr);
thr->start();
}
for (std::vector<acl::thread*>::iterator it = threads.begin();
it != threads.end(); ++it)
{
(*it)->wait();
delete *it;
}
printf("all over, thread=%d, fibers=%d, nloop=%d, counter=%lld\r\n",
__nthreads, __nfibers, __nloop, __counter.value());
acl_fiber_event_free(event);
return 0;
}