epoll 封裝

dwsocket.h

#pragma once

#include <string>
#include <sys/epoll.h>

namespace dw {

    using namespace std;

    class Addr
    {
    public:
        int port;
        string ip;
        Addr(string ip, int port);
        Addr();
        ~Addr();
        friend ostream& operator<<(ostream &out, const Addr &addr);
    private:

    };


    /*
     * 服務(wù)端Socket 完全異步
     */
    class ServerSocket
    {
    public:
        Addr *addr = NULL;
        int maxListenNum = 128;//最大監(jiān)聽(tīng)數(shù),libevent也是128
        int epollSize = 65535;//epoll 的最大監(jiān)聽(tīng)數(shù)
        int maxEvents = 1024;
        int maxThread = 1; //最大線程數(shù)(socket開(kāi)啟時(shí),會(huì)會(huì)根據(jù)cpu核心數(shù),重新初始化該值)

        ServerSocket(int port);
        ~ServerSocket();
        int start();

    private:
        int socketfd = -1;
        int epollfd = -1;
        epoll_event* events;

        void initAddr(int port);
        int initSocket();
        int initEpoll();
        void setNonBlock(int fd); //設(shè)置非阻塞socket
        void initMaxThread();
        void loopWait();
        void addSocketFd(int fd);
        int initThreadPool(); //初始化線程池
    };

}

Addr.cpp

#include "dwsocket.h"

#include<iostream>

namespace dw {

    Addr::Addr() {
    }

    Addr::Addr(string ip, int port) {
        this->port = port;
        this->ip = ip;
    }

    Addr::~Addr() {
    }

    ostream& operator<<(ostream &out, const Addr &addr) {
        out << addr.ip << ":" << addr.port;
        return out;
    }


}

ServerSocket.cpp

#include "dwsocket.h"

#include<iostream>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <sys/sysinfo.h>

namespace dw {

    using namespace std;

    ServerSocket::ServerSocket(int port) {
        initAddr(port);
        initMaxThread();
    }

    ServerSocket::~ServerSocket() {
        delete addr;
        shutdown(socketfd, SHUT_RDWR);
    }

    void ServerSocket::initMaxThread() {
        int cpuCoreNum = get_nprocs();
        maxThread = cpuCoreNum * 2;
    }


    void ServerSocket::initAddr(int port) {
        addr = new Addr("127.0.0.1", port);
    }

    int ServerSocket::initSocket() {
        if (maxListenNum <= 0) {
            cout << "Error: maxListenNum 必須大于0 " << endl;
            return -1;
        }

        socketfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
        if (socketfd < 0) {
            cout << "Error: socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)" << endl;
            return -1;
        }

        //設(shè)置socket
        setNonBlock(socketfd);
        int opt = 1;
        setsockopt(socketfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));

        struct sockaddr_in server_addr;
        server_addr.sin_family = AF_INET;
        server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
        server_addr.sin_port = htons(addr->port);

        int err = bind(socketfd, (struct sockaddr*) &server_addr, sizeof(server_addr));
        if (err < 0) {
            cout << "Error: bind(socketfd, (struct sockaddr*) &server_addr, sizeof(server_addr))" << endl;
            return -1;
        }

        err = listen(socketfd, maxListenNum);
        if (err < 0) {
            cout << "Error: listen(socketfd, maxListenNum)" << endl;
            return -1;
        }

        return 0;
    }

    void ServerSocket::setNonBlock(int fd) {
        int fl = fcntl(socketfd, F_GETFL);
        fcntl(socketfd, F_SETFL, fl | O_NONBLOCK);
    }

    int ServerSocket::initEpoll() {

        epollfd = epoll_create(epollSize);
        if (epollfd < 0) {
            cout << "Error: epoll_create(maxEvents)" << endl;
            return -1;
        }

        events = new epoll_event[maxEvents];
        addSocketFd(socketfd);

        return 0;
    }

    void ServerSocket::loopWait() {

        struct sockaddr_in clientaddr;
        socklen_t clilen = sizeof(clientaddr);

        while (1) {
            int num = epoll_wait(epollfd, events, maxEvents, -1); //返回活躍用戶個(gè)數(shù)
            for (int i = 0; i < num; i++) {
                int connfd = -1;
                if (events[i].data.fd == socketfd) {
                    //新連接
                    cout << "新連接" << endl;
                    connfd = accept(socketfd, (struct sockaddr *) &clientaddr, &clilen);
                    if (connfd < 0) {
                        cout << "連接失敗" << endl;
                        continue;
                    }
                    setNonBlock(connfd);
                }
                else if (events[i].events&EPOLLIN) {
                    //以前的連接,有數(shù)據(jù)讀取
                    cout << "以前的連接 讀取" << endl;
                    if ((connfd = events[i].data.fd) < 0) {
                        cout << "連接失敗" << endl;
                        continue;
                    }
                }
                else if (events[i].events&EPOLLOUT) {
                    //以前的連接,有數(shù)據(jù)寫出
                    cout << "以前的連接 寫出" << endl;
                }
                else {
                    cout << "其他" << endl;
                    continue;
                }
                addSocketFd(connfd);
            }
        }
    }

    void ServerSocket::addSocketFd(int fd) {
        struct epoll_event event;
        event.data.fd = fd;
        event.events = EPOLLIN | EPOLLET;
        epoll_ctl(epollfd, EPOLL_CTL_ADD, fd, &event);
    }

    int ServerSocket::start() {
        int err = initSocket();
        if (err < 0) {
            cout << "Error: initSocket()" << endl;
            return -1;
        }

        err = initEpoll();
        if (err < 0) {
            cout << "Error: initEpoll()" << endl;
            return -1;
        }
        loopWait();
        return 0;
    }
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末驳概,一起剝皮案震驚了整個(gè)濱河市授嘀,隨后出現(xiàn)的幾起案子演侯,更是在濱河造成了極大的恐慌第租,老刑警劉巖,帶你破解...
    沈念sama閱讀 218,204評(píng)論 6 506
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件般眉,死亡現(xiàn)場(chǎng)離奇詭異了赵,居然都是意外死亡,警方通過(guò)查閱死者的電腦和手機(jī)甸赃,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,091評(píng)論 3 395
  • 文/潘曉璐 我一進(jìn)店門柿汛,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái),“玉大人埠对,你說(shuō)我怎么就攤上這事络断。” “怎么了项玛?”我有些...
    開(kāi)封第一講書人閱讀 164,548評(píng)論 0 354
  • 文/不壞的土叔 我叫張陵貌笨,是天一觀的道長(zhǎng)。 經(jīng)常有香客問(wèn)我襟沮,道長(zhǎng)锥惋,這世上最難降的妖魔是什么昌腰? 我笑而不...
    開(kāi)封第一講書人閱讀 58,657評(píng)論 1 293
  • 正文 為了忘掉前任,我火速辦了婚禮膀跌,結(jié)果婚禮上遭商,老公的妹妹穿的比我還像新娘。我一直安慰自己捅伤,他們只是感情好株婴,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,689評(píng)論 6 392
  • 文/花漫 我一把揭開(kāi)白布。 她就那樣靜靜地躺著暑认,像睡著了一般。 火紅的嫁衣襯著肌膚如雪大审。 梳的紋絲不亂的頭發(fā)上蘸际,一...
    開(kāi)封第一講書人閱讀 51,554評(píng)論 1 305
  • 那天,我揣著相機(jī)與錄音徒扶,去河邊找鬼粮彤。 笑死,一個(gè)胖子當(dāng)著我的面吹牛姜骡,可吹牛的內(nèi)容都是我干的导坟。 我是一名探鬼主播,決...
    沈念sama閱讀 40,302評(píng)論 3 418
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼圈澈,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼惫周!你這毒婦竟也來(lái)了?” 一聲冷哼從身側(cè)響起康栈,我...
    開(kāi)封第一講書人閱讀 39,216評(píng)論 0 276
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤递递,失蹤者是張志新(化名)和其女友劉穎,沒(méi)想到半個(gè)月后啥么,有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體登舞,經(jīng)...
    沈念sama閱讀 45,661評(píng)論 1 314
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,851評(píng)論 3 336
  • 正文 我和宋清朗相戀三年悬荣,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了菠秒。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 39,977評(píng)論 1 348
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡氯迂,死狀恐怖践叠,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情囚戚,我是刑警寧澤酵熙,帶...
    沈念sama閱讀 35,697評(píng)論 5 347
  • 正文 年R本政府宣布,位于F島的核電站驰坊,受9級(jí)特大地震影響匾二,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,306評(píng)論 3 330
  • 文/蒙蒙 一察藐、第九天 我趴在偏房一處隱蔽的房頂上張望皮璧。 院中可真熱鬧,春花似錦分飞、人聲如沸悴务。這莊子的主人今日做“春日...
    開(kāi)封第一講書人閱讀 31,898評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)讯檐。三九已至,卻和暖如春染服,著一層夾襖步出監(jiān)牢的瞬間别洪,已是汗流浹背。 一陣腳步聲響...
    開(kāi)封第一講書人閱讀 33,019評(píng)論 1 270
  • 我被黑心中介騙來(lái)泰國(guó)打工柳刮, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留挖垛,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 48,138評(píng)論 3 370
  • 正文 我出身青樓秉颗,卻偏偏與公主長(zhǎng)得像痢毒,于是被迫代替她去往敵國(guó)和親。 傳聞我的和親對(duì)象是個(gè)殘疾皇子蚕甥,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,927評(píng)論 2 355

推薦閱讀更多精彩內(nèi)容