异步IO实例

#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <unistd.h>
#include <fcntl.h>
#include <aio.h>
#include <pthread.h>
#include <arpa/inet.h>

#define BUF_SIZE 1024  

void aio_completion_handler(sigval_t sigval);

void setup_io(int fd, aiocb& my_aiocb)
{
    //初始化AIO请求
    bzero((char *)& my_aiocb, sizeof(struct aiocb));
    my_aiocb.aio_fildes = fd;
    my_aiocb.aio_buf = malloc(BUF_SIZE + );
    my_aiocb.aio_nbytes = BUF_SIZE;
    my_aiocb.aio_offset = ;

    //设置线程回调函数
    my_aiocb.aio_sigevent.sigev_notify = SIGEV_THREAD;
    my_aiocb.aio_sigevent.sigev_notify_function = aio_completion_handler;
    my_aiocb.aio_sigevent.sigev_notify_attributes = NULL;
    my_aiocb.aio_sigevent.sigev_value.sival_ptr = &my_aiocb;
}

//回调函数
void aio_completion_handler(sigval_t sigval)
{
    struct aiocb *req;
    int ret;

    req = (struct aiocb *)sigval.sival_ptr;

    if (aio_error(req) == )
    {
        if ((ret = aio_return(req)) > )
        {
            printf("Thread id %u recv:%s/n", (unsigned int)pthread_self(), (char*)req->aio_buf);
        }
    }

    char* buf = (char*)req->aio_buf;

    if (send(req->aio_fildes, buf, strlen(buf), ) == -)
    {
        perror("send");
        return;
    }

    close(req->aio_fildes);

    return;
}

int main()
{
    int sockfd;
    int sin_size;
    struct sockaddr_in addr, cliaddr;

    //创建socket
    if ((sockfd = socket(AF_INET, SOCK_STREAM, )) < )
    {
        perror("createSocket");
        return -;
    }

    //初始化socket结构
    memset(&addr, , sizeof(addr));
    addr.sin_family = AF_INET;
    addr.sin_port = htons();
    addr.sin_addr.s_addr = htonl(INADDR_ANY);

    //绑定套接口
    if (bind(sockfd, (struct sockaddr *) &addr, sizeof(struct sockaddr)) == -)
    {
        perror("bind");
        return -;
    }

    //创建监听套接口
    if (listen(sockfd, ) == -)
    {
        perror("listen");
        return -;
    }

    printf("server is running! \n");

    //等待连接
    while ()
    {
        struct aiocb my_aiocb;

        struct sockaddr_in client_addr;
        socklen_t cliaddr_len = sizeof(client_addr);
        int NewClient = accept(sockfd, (struct sockaddr*)&client_addr, &cliaddr_len);
        if (NewClient < )
        {
            perror("accept");
            continue;
        }
        char cli_ip[INET_ADDRSTRLEN] = "";
        inet_ntop(AF_INET, &client_addr.sin_addr, cli_ip, INET_ADDRSTRLEN);
        printf("client ip = %s\n", cli_ip);

        setup_io(NewClient, my_aiocb);
        aio_read(&my_aiocb);
    }
    close(sockfd);
}