初认识非阻塞IO

以下内容均为本人学习笔记，若有不当，欢迎指出
上一篇为初认识五种IO模型：初识五种IO模型

非阻塞IO的相关用法

一个文件描述符默认都是阻塞的
需要调用 fcntl() 函数设置文件状态

#include <fcntl.h>
#include <unistd.h>
extern int fcntl (int __fd, int __cmd, ...);

cmd参数设置：(基于cmd设置不同，第三个参数也不同)

• 获取文件状态标记 cmd = F_GETFL
• 设置文件状态标记 cmd = F_SETFL

例一 : 轮询方式读取标准输入

#include <stdio.h>
#include <fcntl.h>
#include <errno.h>
#include <string.h>
#include <unistd.h>
#include <iostream>
using namespace std;

//int fcntl (int __fd, int __cmd, ...);

int SetNoBlock(int fd)
{
  //使用F_GETFL将文件描述符状态返回
  int f1 = fcntl( fd,F_GETFL);
  if(f1< 0)
  {
    printf("get failed\n");
    return -1;
  }
  //使用F_SETFL将文件描述符状态设置为非阻塞
  fcntl(fd,F_SETFL,fd | O_NONBLOCK);
  return 0;
}

int main()
{

  //实现非阻塞式从标准输入读取数据
  if(SetNoBlock(0) == -1)
  {
    return 1;
  }
  char buf[1024] = {0};
 //循环的从标准输入读取数据
  while(1)
  {
    ssize_t read_size = read(0,buf,sizeof(buf)-1);
    if(read_size < 0)
    {
      sleep(1);
      //if(error == EAGAIN)
      //{
      //    printf("data is not ready!\n");
      //}
      printf("%s\n",strerror(error));
      printf("read_size :%ld ,errno %d\n",read_size,errno);
      continue;
      //因为是采用非阻塞式IO,若读取失败，函数返回值为-1，返回出错但不会阻塞
      //错误信息为 Resource temporarily unavailable（资源暂时没有准备好）
      //在非阻塞式read()返回出错时，错误码(errno)被设置为EAGAIN(值为11),描述为 error again，也是为了区分普通读出错errno

    }
    if(read_size == 0)
    {
      printf("read done\n");
      return 0;
    }
    else
    { 
      printf("user say : %s\n",buf);
    }
  }
  return 0;
}

输出信息如下:

Resource temporarily unavailable
read_size :-1 ,errno 11
Resource temporarily unavailable
read_size :-1 ,errno 11
hello world
Resource temporarily unavailable
read_size :-1 ,errno 11
user say : hello world

change world
Resource temporarily unavailable
read_size :-1 ,errno 11
user say : change world

Resource temporarily unavailable
read_size :-1 ,errno 11

我们发现如何没有向标准输入输入数据,就返回错误,打印错误信息,继续循环
若输入数据,就会执行正确逻辑

例二 : 使用select检测标准输入是否就绪

先看接口如何使用

       #include <sys/time.h>
       #include <sys/types.h>
       #include <unistd.h>

       int select(int nfds, fd_set *readfds, fd_set *writefds,
                  fd_set *exceptfds, struct timeval *timeout);

• 第一个参数 nfds 为你所要等待的文件描述符的最大值的值再 +1 ，因为这里的fd_set 文件描述符集其实是一个位图，当每次进行设置修改时都要遍历，这样给定最大范围，可以提高效率
• 第二个参数为要等待的可读的文件描述符集
• 第三个参数为要等待的可写的文件描述符集
• 第四个参数为要等待的异常文件描述符集
• 第五个参数为一个表示时间的结构体，用来设置select()的等待时间

下面的接口用来操作fd_set：

       void FD_CLR(int fd, fd_set *set);
       int  FD_ISSET(int fd, fd_set *set);
       void FD_SET(int fd, fd_set *set);
       void FD_ZERO(fd_set *set);

select的使用案例

#include <stdio.h>
#include <sys/select.h>
#include <fcntl.h>
#include <unistd.h>

//使用select 检测标准输入
int main()
{
  fd_set read_set;
  int fd = 0;//stdin
  FD_ZERO(&read_set);
  FD_SET(fd,&read_set);

  while(1)
  {
    int ret = select(fd + 1,&read_set,NULL,NULL,NULL);
    //此处只检测读就绪，写和异常不用理会，并设置为阻塞式检测
    if(ret < 0)
    {
      perror("select");
      continue;
    }
    printf("ret : %d\n",ret);
    if(FD_ISSET(fd,&read_set) == 0)
    {
      //说明出错
      printf("error!\n");
      continue;
    }
    else
    {
      char buf[1024] = {0};
      if(read(fd,buf,sizeof(buf)-1) > 0)
      {
        printf("%s\n",buf);
      }
    }
    //一次之后，重新调整read_set
    FD_ZERO(&read_set);
    FD_SET(fd,&read_set);
  }
  return 0;
}

select 详细内容查看下一篇文章。
完。