mirror of
https://gitee.com/acl-dev/acl.git
synced 2024-12-15 01:10:52 +08:00
113 lines
2.4 KiB
C
113 lines
2.4 KiB
C
#include "lib_acl.h"
|
|
#include "../stamp.h"
|
|
#include "taskq.h"
|
|
|
|
static ACL_ATOMIC *__counter;
|
|
static int __max = 10000;
|
|
|
|
static void consumer(void *ctx acl_unused)
|
|
{
|
|
//int *o = (int*) ctx;
|
|
long long n = acl_atomic_int64_add_fetch(__counter, 1);
|
|
if (n <= 10) {
|
|
printf("thread: %ld, n=%lld\r\n", (long)acl_pthread_self(), n);
|
|
}
|
|
//free(o);
|
|
}
|
|
|
|
static void *producer(void *ctx)
|
|
{
|
|
TASKQ *taskq = (TASKQ*) ctx;
|
|
int i;
|
|
|
|
for (i = 0; i <__max; i++) {
|
|
//int *o = (int*) malloc(sizeof(int));
|
|
//*o = i;
|
|
taskq_push(taskq, consumer, NULL);
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
static void usage(const char *procname)
|
|
{
|
|
printf("usage: %s -h [help]\r\n"
|
|
" -q qsize[default: 10]\r\n"
|
|
" -c nconsumers[default: 20]\r\n"
|
|
" -p nproducers[default: 5]\r\n"
|
|
" -n max_loop_per_thread[default: 10000]\r\n"
|
|
, procname);
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
unsigned qsize = 10, nconsumers = 20, nproducers = 5, i;
|
|
TASKQ* taskq;
|
|
acl_pthread_t *producers;
|
|
acl_pthread_attr_t attr;
|
|
struct timeval begin, end;
|
|
double spent, speed;
|
|
static long long value = 0, counter;
|
|
int ch;
|
|
|
|
while ((ch = getopt(argc, argv, "hq:c:p:n:")) > 0) {
|
|
switch (ch) {
|
|
case 'h':
|
|
usage(argv[0]);
|
|
return 0;
|
|
case 'q':
|
|
qsize = (unsigned) atoi(optarg);
|
|
break;
|
|
case 'c':
|
|
nconsumers = (unsigned) atoi(optarg);
|
|
break;
|
|
case 'p':
|
|
nproducers = (unsigned) atoi(optarg);
|
|
break;
|
|
case 'n':
|
|
__max = (unsigned) atoi(optarg);
|
|
break;
|
|
default:
|
|
usage(argv[0]);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
taskq = taskq_create(qsize, nconsumers);
|
|
producers = (acl_pthread_t*) acl_mycalloc(nproducers, sizeof(acl_pthread_t));
|
|
__counter = acl_atomic_new();
|
|
|
|
acl_atomic_set(__counter, &value);
|
|
acl_atomic_int64_set(__counter, 0);
|
|
|
|
acl_pthread_attr_init(&attr);
|
|
|
|
gettimeofday(&begin, NULL);
|
|
|
|
for (i = 0; i < nproducers; i++) {
|
|
int ret = acl_pthread_create(&producers[i], &attr, producer, taskq);
|
|
assert(ret == 0);
|
|
}
|
|
|
|
for (i = 0; i < nproducers; i++) {
|
|
acl_pthread_join(producers[i], NULL);
|
|
}
|
|
|
|
counter = nproducers * __max;
|
|
while (value < counter) {
|
|
usleep(100);
|
|
}
|
|
|
|
gettimeofday(&end, NULL);
|
|
spent = stamp_sub(&end, &begin);
|
|
speed = (counter * 1000) / (spent >= 1.0 ? spent : 1);
|
|
printf("counter=%lld, loop=%lld, spent %.2f ms ,speed=%.2f\r\n",
|
|
value, counter, spent, speed);
|
|
|
|
acl_atomic_free(__counter);
|
|
taskq_destroy(taskq);
|
|
acl_myfree(producers);
|
|
|
|
return 0;
|
|
}
|