483-手写C++muduo库(socket封装fd,Acceptor)

骑猪看日落 2022-09-08 14:48 120阅读 0赞

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16]  
**我们在之前已经完成了对前面5个模块的剖析和重写。  
现在我们看Acceptor**  
Acceptor就是处理accept，监听新用户的连接，拿到跟客户端通信的clientfd，打包成channel，根据muduo的轮询算法找一个subloop，把subloop唤醒，把接收的channel给subloop。  
![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 1]  
**我们打开TCPServer，处理一下Acceptor**  
![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 2]  
![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 3]  
**Acceptor运行在我们的baseloop（mainreactor）里面**  
![在这里插入图片描述][eb24823a21b346ddbc69c41c1d6884b1.png]  
**封装了socket**

## socket封装fd ##

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 4]

## Socket.h ##

#pragma once
    
    #include "noncopyable.h"
    
    class InetAddress;
    
    //封装socket fd
    class Socket : noncopyable
    {
        
    public:
        explicit Socket(int sockfd)
            : sockfd_(sockfd)
        {
        }
    
        ~Socket();
    
        int fd() const {
         return sockfd_; }
        void bindAddress(const InetAddress &localaddr);
        void listen();
        int accept(InetAddress *peeraddr);
    
        void shutdownWrite();
    
        void setTcpNoDelay(bool on);//直接发送，数据不进行TCP缓存 
        void setReuseAddr(bool on);
        void setReusePort(bool on);
        void setKeepAlive(bool on);
    private:
        const int sockfd_;
    };

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 5]  
![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 6]

## Socket.cc ##

#include "Socket.h"
    #include "Logger.h"
    #include "InetAddress.h"
    
    #include <unistd.h>
    #include <sys/types.h>         
    #include <sys/socket.h>
    #include <strings.h>
    #include <netinet/tcp.h>
    #include <sys/socket.h>
    
    Socket::~Socket()
    {
        
        close(sockfd_);//调用系统的close 
    }
    
    void Socket::bindAddress(const InetAddress &localaddr)
    {
        
        if (0 != ::bind(sockfd_, (sockaddr*)localaddr.getSockAddr(), sizeof(sockaddr_in)))
        {
        
            LOG_FATAL("bind sockfd:%d fail \n", sockfd_);
        }
    }
    
    void Socket::listen()
    {
        
        if (0 != ::listen(sockfd_, 1024))
        {
        
            LOG_FATAL("listen sockfd:%d fail \n", sockfd_);
        }
    }
    
    int Socket::accept(InetAddress *peeraddr)
    {
        
        /**
         * 1. accept函数的参数不合法
         * 2. 对返回的connfd没有设置非阻塞
         * Reactor模型 one loop per thread
         * poller + non-blocking IO
         */ 
        sockaddr_in addr;
        socklen_t len = sizeof addr;
        bzero(&addr, sizeof addr);
        int connfd = ::accept4(sockfd_, (sockaddr*)&addr, &len, SOCK_NONBLOCK | SOCK_CLOEXEC);
        if (connfd >= 0)
        {
        
            peeraddr->setSockAddr(addr);
        }
        return connfd;
    }
    
    void Socket::shutdownWrite()
    {
        
        if (::shutdown(sockfd_, SHUT_WR) < 0)
        {
        
            LOG_ERROR("shutdownWrite error");
        }
    }
    
    void Socket::setTcpNoDelay(bool on)
    {
        
        int optval = on ? 1 : 0;
        ::setsockopt(sockfd_, IPPROTO_TCP, TCP_NODELAY, &optval, sizeof optval);
    }
    
    void Socket::setReuseAddr(bool on)
    {
        
        int optval = on ? 1 : 0;
        ::setsockopt(sockfd_, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof optval);
    }
    
    void Socket::setReusePort(bool on)
    {
        
        int optval = on ? 1 : 0;
        ::setsockopt(sockfd_, SOL_SOCKET, SO_REUSEPORT, &optval, sizeof optval);
    }
    
    void Socket::setKeepAlive(bool on)
    {
        
        int optval = on ? 1 : 0;
        ::setsockopt(sockfd_, SOL_SOCKET, SO_KEEPALIVE, &optval, sizeof optval);
    }

## Acceptor ##

![,][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 7]  
![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 8]  
**Acceptor底层是listenfd，运行在baseloop（mainreactor），**

## Acceptor.h ##

#pragma once
    #include "noncopyable.h"
    #include "Socket.h"
    #include "Channel.h"
    
    #include <functional>
    
    class EventLoop;
    class InetAddress;
    
    class Acceptor : noncopyable
    {
        
    public:
        using NewConnectionCallback = std::function<void(int sockfd, const InetAddress&)>;
        Acceptor(EventLoop *loop, const InetAddress &listenAddr, bool reuseport);
        ~Acceptor();
    
        void setNewConnectionCallback(const NewConnectionCallback &cb) 
        {
        
            newConnectionCallback_ = cb;
        }
    
        bool listenning() const {
         return listenning_; }
        void listen();
    private:
        void handleRead();
        
        EventLoop *loop_;//Acceptor用的就是用户定义的那个baseLoop，也称作mainLoop
        Socket acceptSocket_;
        Channel acceptChannel_;
        NewConnectionCallback newConnectionCallback_;
        bool listenning_;
    };

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 9]

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 10]

## Acceptor.cc ##

#include "Acceptor.h"
    #include "Logger.h"
    #include "InetAddress.h"
    
    #include <sys/types.h>    
    #include <sys/socket.h>
    #include <errno.h>
    #include <unistd.h>
    
    
    static int createNonblocking()//创建非阻塞的I/O 
    {
        
        int sockfd = ::socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK | SOCK_CLOEXEC, 0);
        if (sockfd < 0) 
        {
        
            LOG_FATAL("%s:%s:%d listen socket create err:%d \n", __FILE__, __FUNCTION__, __LINE__, errno);
        }
    }
    
    Acceptor::Acceptor(EventLoop *loop, const InetAddress &listenAddr, bool reuseport)//构造函数 
        : loop_(loop)
        , acceptSocket_(createNonblocking())//创建socket套接字 
        , acceptChannel_(loop, acceptSocket_.fd())//channel和poller都是通过请求本线程的loop和poller通信 
        , listenning_(false)
    {
        
        acceptSocket_.setReuseAddr(true);
        acceptSocket_.setReusePort(true);
        acceptSocket_.bindAddress(listenAddr);//bind绑定套接字 
        //TcpServer::start() Acceptor.listen  如果有新用户的连接，就要执行一个回调（connfd=》打包成channel=》唤醒subloop）
        //baseLoop => acceptChannel_(listenfd)有事件发生 => 底层反应堆调用回调 
        acceptChannel_.setReadCallback(std::bind(&Acceptor::handleRead, this));//绑定回调 
    }
    
    Acceptor::~Acceptor()
    {
        
        acceptChannel_.disableAll();
        acceptChannel_.remove();
    }
    
    void Acceptor::listen()
    {
        
        listenning_ = true;
        acceptSocket_.listen();//listen
        acceptChannel_.enableReading();//acceptChannel_ => Poller
    }
    
    //listenfd有事件发生了，就是有新用户连接了
    void Acceptor::handleRead()
    {
        
        InetAddress peerAddr;
        int connfd = acceptSocket_.accept(&peerAddr);
        if (connfd >= 0)
        {
        
            if (newConnectionCallback_)
            {
        
                newConnectionCallback_(connfd, peerAddr);//轮询找到subLoop，唤醒，分发当前的新客户端的Channel
            }
            else//客户端没有办法去服务 
            {
        
                ::close(connfd);
            }
        }
        else
        {
        
            LOG_ERROR("%s:%s:%d accept err:%d \n", __FILE__, __FUNCTION__, __LINE__, errno);
            if (errno == EMFILE)
            {
        
                LOG_ERROR("%s:%s:%d sockfd reached limit! \n", __FILE__, __FUNCTION__, __LINE__);
            }
        }
    }

[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16]: /images/20220829/4c9f7c48ef0945788d36c17f651ba556.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 1]: /images/20220829/04833e4fbed347509c294291f38dac37.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 2]: /images/20220829/7ba6f9f1651c4377b55ac017e3758b6e.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 3]: /images/20220829/af6e3fa4bc2b4d42b8cae3b1db0ef742.png
[eb24823a21b346ddbc69c41c1d6884b1.png]: /images/20220829/f394419add7145aa93283b5596b6a3d5.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 4]: /images/20220829/7d1dd6bfac1a4f22a42406cfe1633fe1.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 5]: /images/20220829/cb2f772543904420ab2795741060f6b8.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 6]: /images/20220829/cf926b246a5a42409315de8f1ec9f9e1.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 7]: /images/20220829/e252d764c2684e63aec201fb2d78bc36.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 8]: /images/20220829/848d75c9ebe8484b85fc3f0373c300aa.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 9]: /images/20220829/f9920d2241f1465fa8103934546aed68.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBALeael-azveWuhw_size_20_color_FFFFFF_t_70_g_se_x_16 10]: /images/20220829/e4d0949708e34ed29ac7680b50e5f796.png