mirror of https://github.com/yandex/odyssey.git
67 lines
1.1 KiB
C
67 lines
1.1 KiB
C
|
|
#include <machinarium.h>
|
|
#include <odyssey_test.h>
|
|
|
|
static machine_channel_t *channel;
|
|
|
|
static int count_written = 0;
|
|
static int count_read = 0;
|
|
static int pc;
|
|
|
|
static void test_consumer(void *arg)
|
|
{
|
|
(void)arg;
|
|
|
|
while (count_read < count_written) {
|
|
machine_msg_t *msg;
|
|
msg = machine_channel_read(channel, 0);
|
|
if (msg == NULL)
|
|
break;
|
|
machine_msg_free(msg);
|
|
count_read++;
|
|
machine_sleep(0);
|
|
}
|
|
}
|
|
|
|
static void test_pc(void *arg)
|
|
{
|
|
(void)arg;
|
|
|
|
int i = 0;
|
|
for (; i < 1000; i++) {
|
|
machine_msg_t *msg;
|
|
msg = machine_msg_create(0);
|
|
test(msg != NULL);
|
|
machine_msg_set_type(msg, i);
|
|
count_written++;
|
|
machine_channel_write(channel, msg);
|
|
}
|
|
|
|
for (i = 0; i < 100; i++) {
|
|
int rc = machine_coroutine_create(test_consumer, NULL);
|
|
test(rc != -1);
|
|
}
|
|
}
|
|
|
|
void machinarium_test_producer_consumer2(void)
|
|
{
|
|
machinarium_init();
|
|
|
|
channel = machine_channel_create(0);
|
|
test(channel != NULL);
|
|
|
|
pc = machine_create("producer-consumer", test_pc, NULL);
|
|
test(pc != -1);
|
|
|
|
int rc;
|
|
rc = machine_wait(pc);
|
|
test(rc != -1);
|
|
|
|
test(count_read == 1000);
|
|
test(count_written == count_read);
|
|
|
|
machine_channel_free(channel);
|
|
|
|
machinarium_free();
|
|
}
|