【Linux高级IO（一）】明白五种IO模型 - IT评测·应用市场-qidao123.com技术社区

void SetNoBlock(int fd) {
int fl = fcntl(fd, F_GETFL); //获得指定文件描述符的标记位
if (fl < 0) //小于0，获取失败
{
perror("fcntl");
return;
}
fcntl(fd, F_SETFL, fl | O_NONBLOCK);
//为指定的文件描述符设置标记位在老的标记位f1的基础上添加 O_NONBLOCK（非阻塞）
}

复制代码

else if(errno == EWOULDBLOCK)
{
cout << "0 fd data not ready, try again!" << endl;
}

复制代码

#include <fcntl.h>
#include <unistd.h>
#include<errno.h>
#include <cstdlib>
#include <iostream>
using namespace std;
//对指定的fd设置非阻塞
void SetNonBlock(int fd) {
int fl = fcntl(fd, F_GETFL);
if (fl < 0) {
cerr << "fcntl error" << endl;
exit(1);
}
fcntl(fd, F_SETFL, fl | O_NONBLOCK);
}
int main() {
SetNonBlock(0);
while (1) {
char buffer[1024];
ssize_t s = read(0, buffer, sizeof(buffer) - 1);
if (s > 0) {
buffer[s] = 0;
cout << buffer << endl;
} else if (s == 0) {
cout << "读到文件结尾了" << endl;
break;
}
else
{
//1. 数据没用准备好 2. 真的出错了. 都以-1的返回值返回
// 数据没有准备好,不算出错. 需要区分这两种情况
if(errno == EWOULDBLOCK || errno == EAGAIN)
{
cout<<"os底层数据还没就绪"<<endl;
cout<<errno<<endl;
}
//被信号中断, 也不算read出错
else if(errno == EINTR)
{
cout<<"IO interrupted by signal"<<endl;
}
else
{
cout<<"read error"<<endl;
break;
}
}
sleep(1);
}
}

复制代码

#pragma once
#include <iostream>
#include <sys/select.h>
#include <sys/time.h>
#include "Socket.hpp"
using namespace std;
static const uint16_t defaultport = 8080;
static const int fd_num_max = (sizeof(fd_set) * 8);
int defaultfd = -1;
class SelectServer
{
public:
SelectServer(uint16_t port = defaultport) : _port(port)
{
for(int i = 0; i < fd_num_max; i++)
{
fd_array[i] = defaultfd;
std::cout <<"fd_array[" << i << "]" << " : " << fd_array[i] << std::endl;
}
}
bool Init()
{
_listensock.Socket();
_listensock.Bind(_port);
_listensock.Listen();
return true;
}
void Accepter()
{
// 连接事件就绪了
std::string clientip;
uint16_t clientport = 0;
int sock = _listensock.Accept(&clientip, &clientport); // 获取连接不会阻塞在这里
if (sock < 0)
{
return;
}
lg(Info, "accept success, %s:%d", clientip.c_str(), clientport);
// 以前走到这里可以直接读取数据，可是，现在这里不可以
// 以前是多线程、多进程，文件描述符其实是托管给执行流的他阻塞是不影响的
// 我们现在是单进程，不能建立完连接，立马就进行读如果不发，当前的进程就阻塞了
// select里面只有一个listen套接字要将sock设置进select里这样读文件描述符集里的文件描述符就会变得越来越多
// accept获取新连接，不能直接读，因为不清楚是否就绪，selcet清楚是否就绪，所以要将新获取的连接添加到辅助数组里
// select把数据处理完之后，下次循环时会重新再进行把文件描述符添加到rfds里，再交给select由他来监听
int pos = 1;
for (; pos < fd_num_max; pos++)
{
if (fd_array[pos] != default)
continue; // 被占用
else
break;
}
if (pos == fd_num_max)
{
// 全部被占用
lg(Warning, "server is full, close %d now!", sock);
close(sock);
}
else // 提前break说明由-1位置（即没有被占用）
{
fd_array[pos] = sock; // 新获取的连接往数组中添加
}
}
void Recver(int fd, int pos)
{
char buffer[1024];
ssize_t n = read(fd, buffer, sizeof(buffer) - 1);
if (n > 0)
{
buffer[n] = 0; // 把它当字符串
cout << "get a message: " << buffer << endl;
}
else if (n == 0) // 读失败
{
lg(Info, " client quit,me too, close fd is : ", fd);
close(fd);
fd_array[pos] = defaultfd; // 这里本质是从select中移除
}
}
void Dispatchar(fd_set &rfds) // 读事件就绪 //事件派发器
{
for (int i = 0; i < fd_num_max; i++)
{
int fd = fd_arry[i];
if (fd == dafaultfd)
conitnue;
if (FD_ISSET(fd, &rfds)) // 判断listen套接字是否在集合里，即是否就绪
{
if(fd == _listensock.Fd())
{
Accepter(); //连接管理器
}
else //不是listenfd
{
Recver(fd, i);
}
}
}
}
void Start()
{
int listensock = _listensock.Fd();
fd_array[0] = listensock;
for (;;)
{
Fd_set rfds;
Fd_ZERO(rfds);
int maxfd = fd_array[0];
for(int i = 0; i < fd_num_max; i++)
{
if(fd_arry[i] == default)
continue; //没有被设置过的
Fd_SET(fd_array[i], &rfds); // 将文件描述符添加到集合里
if(maxfd < fd_array[i])
{
maxfd = fd_array[i];
}
}
// accept 的本质是检测listensocket上面有没有连接事件即底层有三次握手他一次只能等一个文件描述符所以要交给select
// 新连接到来，对于select来讲就是读事件就绪
// 读文件描述符集，他是一个位图
struct timeval timeout = {5, 0}; //需要被重复设置因为如果不重复设置就会剩余的时间替换
//输入输出参数，每次从内核返回的时候值可能就被改过了，所以需要重复设置
int n = select(maxfd + 1, &rfds, nullptr, nullptr, &timeout); // 告诉OS关心这个文件描述符上的读事件
//select 是如果事件就绪，上层不处理，select会一直通知你
//select告诉你就绪了，接下来的一次读取，fd不会阻塞
switch(n)
{
case 0:
//等待超时，在等待期间任何事情都没有发生
break;
case -1:
//异常
break;
default:
//有事件就绪
HandlerEvent(rfds);//就绪的事件在rfds里
break;
}
}
}
~SelectServer()
{
_listensock.Close();
}
private:
Sock _listensock;
uint16_t _port;
int fd_array[fd_num_max]{defaultfd};//设置辅助数组
};

复制代码