--- libaitsched/example/test.c 2012/05/31 22:31:48 1.6 +++ libaitsched/example/test.c 2012/08/02 13:56:19 1.7 @@ -1,4 +1,6 @@ #include +#include +#include #include #include #include @@ -6,9 +8,13 @@ #include #include #include +#include "../inc/config.h" #include intptr_t Kill; +#ifdef AIO_SUPPORT +struct iovec iv[3], wiv[3], riv[3]; +#endif void *event(sched_task_t *arg) { @@ -30,7 +36,7 @@ void *timer(sched_task_t *arg) void *r(sched_task_t *arg) { - printf("read::\n"); + printf("read:: bytes\n"); Kill++; return NULL; } @@ -47,6 +53,68 @@ void *once(sched_task_t *arg) return NULL; } +#ifdef AIO_SUPPORT +void *aioread(sched_task_t *arg); +void *aiowrite(sched_task_t *arg) +{ + char *ole = malloc(BUFSIZ); + + printf("AIO write[%d]:: %d bytes\n%p\n", TASK_FD(arg), (int) TASK_DATLEN(arg), + TASK_DATA(arg)); + free(TASK_DATA(arg)); + + memset(ole, 0, BUFSIZ); + schedAIORead(TASK_ROOT(arg), aioread, NULL, TASK_FD(arg), ole, BUFSIZ - 1, -1); + return NULL; +} + +void *aioread(sched_task_t *arg) +{ + char *ole = malloc(BUFSIZ); + int len; + + printf("AIO read[%d]:: %d bytes\n%s\n-------\n", TASK_FD(arg), (int) TASK_DATLEN(arg), + (char*) TASK_DATA(arg)); + + if (TASK_ARG(arg)) { + len = strlcpy(ole, "++++++BAHURA OR CULTURE .... A CULTURE OR BAHURA :-)\n", BUFSIZ); + printf("sched Write len=%d %p\n", len, ole); + schedAIOWrite(TASK_ROOT(arg), aiowrite, TASK_ARG(arg), TASK_FD(arg), ole, len, -1); + + } + free(TASK_DATA(arg)); + return NULL; +} + +void *aiobulkread(sched_task_t *arg) +{ + struct iovec *iv = TASK_DATA(arg); + register int i; + + printf("aioBULKread::\n"); + for (i = 0; i < 3; i++) { + printf("%d) rlen[%d]=%s\n---\n", i, iv[i].iov_len, (char*) iv[i].iov_base); + free(iv[i].iov_base); + } + + return NULL; +} + +void *aiobulkwrite(sched_task_t *arg) +{ + struct iovec *iv = TASK_DATA(arg); + register int i; + + printf("aioBULKwrite::\n"); + for (i = 0; i < 3; i++) { + printf("%d) wlen=%d\n", i, iv[i].iov_len); + free(iv[i].iov_base); + } + + return NULL; +} +#endif + void sig(int s) { switch (s) { @@ -60,11 +128,15 @@ int main(int argc, char **argv) { sched_root_task_t *root; - int f; + int f, fd; struct sockaddr_in sin; struct timespec ts = { 20, 0 }; // struct timespec p = { 0, 10000000 }; struct sigaction sa; +#ifdef AIO_SUPPORT + char *ole = malloc(BUFSIZ); + register int i; +#endif sa.sa_handler = sig; sigemptyset(&sa.sa_mask); @@ -80,6 +152,11 @@ main(int argc, char **argv) if (bind(f, (struct sockaddr*) &sin, sizeof sin) == -1) return 1; + fd = open("test_aio.dat", O_CREAT | O_RDWR, 0644); + if (fd == -1) + return 1; + printf("fd=%d\n", fd); + root = schedBegin(); if (!root) { printf("Error:: #%d - %s\n", sched_GetErrno(), sched_GetError()); @@ -122,12 +199,54 @@ main(int argc, char **argv) return 6; } +#ifdef AIO_SUPPORT + memset(ole, 0, BUFSIZ); + if (!schedAIORead(root, aioread, (void*) f, fd, ole, BUFSIZ - 1, 0)) + printf("Warning:: #%d - %s\n", sched_GetErrno(), sched_GetError()); + + + iv[0].iov_len = 5; + iv[1].iov_len = 2; + iv[2].iov_len = 50; + for (i = 0; i < 3; i++) + iv[i].iov_base = malloc(iv[i].iov_len); + if (!schedLIORead(root, aiobulkread, NULL, fd, iv, 3, 0)) + printf("Warning:: #%d - %s\n", sched_GetErrno(), sched_GetError()); + fsync(fd); + for (i = 0; i < 3; i++) { + wiv[i].iov_len = 100; + wiv[i].iov_base = malloc(wiv[i].iov_len); + } + strlcpy(wiv[0].iov_base, "12345678900000000000000000\n", wiv[0].iov_len); + wiv[0].iov_len = strlen(wiv[0].iov_base) + 1; + strlcpy(wiv[1].iov_base, "222222222222222222222222\n", wiv[1].iov_len); + wiv[1].iov_len = strlen(wiv[1].iov_base) + 1; + strlcpy(wiv[2].iov_base, "333\n", wiv[2].iov_len); + wiv[2].iov_len = strlen(wiv[2].iov_base) + 1; + if (!schedLIOWrite(root, aiobulkwrite, NULL, fd, wiv, 3, 0)) + printf("Warning:: #%d - %s\n", sched_GetErrno(), sched_GetError()); + + for (i = 0; i < 3; i++) { + riv[i].iov_len = 5; + riv[i].iov_base = malloc(riv[i].iov_len + 1); + memset(riv[i].iov_base, 0, riv[i].iov_len + 1); + } + if (!schedLIORead(root, aiobulkread, NULL, fd, riv, 3, 0)) + printf("Warning:: #%d - %s\n", sched_GetErrno(), sched_GetError()); +#endif + schedCallOnce(root, once, "000000", 42, NULL, 0); // schedPolling(root, &p, NULL); schedRun(root, &Kill); schedEnd(&root); +#ifdef AIO_SUPPORT + for (i = 0; i < 3; i++) + free(iv[i].iov_base); +#endif + + close(fd); close(f); return 0; }