c++进程池实现回射服务器

小咪咪 2023-08-17 16:58 113阅读 0赞

进程池相对于线程池就没那么麻烦，没有条件变量，没有互斥锁，这些东西！比较简单！参考网上资料，实现一个回射服务器，尽管看着人家实现思路写的，还是出现很多问题，太菜了！确实太菜了！就当学习了gdb多进程调试，哈哈！有关进程调试的文章这里有个学姐写的可以参考哦！[gdb多进程调试][gdb]

`process.h`

#pragma once
    #include <iostream>
    #include <assert.h>
    #include <string.h>
    #include <sys/epoll.h>
    #include <arpa/inet.h>
    #include <sys/socket.h>
    #include <sys/types.h>
    #include <unistd.h>
    #include <map>
    #include <fcntl.h>
    #include <signal.h>
    #include <sys/wait.h>
    #include <vector>
    #include <memory> 
    const int EPOLLNUM = 1024 ;
    const int BUFFERSIZE = 1024 ;
    using namespace std ;
    
    class process ;
    class tool ;
    
    class process {
        
    public:
        process() : pid(-1){
        }
        ~process() {
        }
    public :
        int createSocketPair() ;
        int getReadFd() {
         return pipe[0] ; }
        int getWriteFd() {
         return pipe[1] ; }
    public :
        pid_t pid ;
    private :
        int pipe[2] ;
    };
    
    //进程池使用单实例模式
    template<typename T>  
    class processPool {
        
    public :
        processPool(int listenFd, int processNum = 8) ;
        ~processPool() {
        }
    public :
        //保证时钟只产生一个进程池
        static shared_ptr<processPool<T>> create(int listenFd, 
                                                 int processNum = 8) ;
    public :
        void run() ;
    private :
        //设置进程对应的管道
        void setupSigPipe() ;
        //父进程运行
        void runParent() ;
        //子进程运行
        void runChild() ;
    private :
        static const int maxProcessNum = 16 ; //进程池允许的最大子进程数量
        static const int userPreProcess = 65535 ;//每个进程最多能处理的客户数量
        static const int maxEventNum = 10000 ; //epoll最多处理的事件数量
        static shared_ptr<processPool<T>> object ; //进程池的静态实例 
    
        int curProcessNum ; //进程池目前能处理的进城数量
        int index ; //子进程在进程中的序号
        int epollFd ;  //epoll句柄
        int listenFd ; //监听的文件描述符
        bool stop ; //进程终止标记
        vector<shared_ptr<process>> proDescInfo ; // 保存进程的描述信息
    };
    
    class tool {
        
    public :
        static int setNoBlock(int fd) ;
        static int addFd(int epollFd, int fd) ;
        static int removeFd(int epollFd, int fd) ;
        static void addSig(int sig, void(handle)(int), bool restart=true) ;
    };
    
    template<typename T> 
    shared_ptr<processPool<T>> processPool<T> :: object = nullptr ;
    
    //创建进程池
    template<typename T>
    processPool<T> :: processPool(int listenFd, int pNum)
        : listenFd(listenFd), curProcessNum(pNum), index(-1), stop(false) {
        
        assert(pNum > 0 && pNum <= maxProcessNum) ;
        proDescInfo.reserve(pNum)  ;
        for(int i=0; i<pNum; i++) {
        
            proDescInfo[i] = make_shared<process>() ;
        }
        //创建子进程
        for(int i=0; i<pNum; i++) {
        
            //创建unix双向匿名管道
            int ret = proDescInfo[i]->createSocketPair() ;
            assert(ret > 0) ;
            proDescInfo[i]->pid = fork() ;
            assert(proDescInfo[i]->pid >= 0) ;
            //父进程
            if(proDescInfo[i]->pid > 0) {
        
                //关掉读端
                close(proDescInfo[i]->getReadFd()) ;
                continue ;
            }
            //子进程
            else {
        
                close(proDescInfo[i]->getWriteFd()) ;
                index = i ; //自己在进程池中的下标
                break ; //这里一定得跳出，否则子进程会继续创建进程
            }
        }
    }
    
    template <typename T>
    shared_ptr<processPool<T>> processPool<T> :: create(int listenFd, 
                                                        int processNum) {
        
        if(object == nullptr) {
        
            object = make_shared<processPool<T>>(listenFd, processNum) ;
        }
        return object ;
    }
    
    //父进程在进程中的下标-1
    template<typename T>
    void processPool<T>::run() {
        
        if(index != -1) runChild() ;
        else runParent() ;
    }
    
    //统一事件
    template<typename T> 
    void processPool<T> :: setupSigPipe() {
        
        epollFd = epoll_create(EPOLLNUM) ;
        assert(epollFd > 0) ;
        tool :: addSig(SIGPIPE, SIG_IGN) ;
    }
    
    template<typename T> 
    void processPool<T> :: runParent() {
        
        cout << "在父进程"  << "　　　" << "序号:" << index << endl ;
        //设置信号
        setupSigPipe() ;
        tool :: addFd(epollFd, listenFd)  ; 
        epoll_event ev[maxEventNum] ;
        int count = 0 ;
        char newConn = 'c' ;
        while(!stop) {
        
            int number = epoll_wait(epollFd, ev, maxEventNum, -1) ;
            if(number < 0) {
        
                cout << __LINE__ << "      " << __FILE__ << endl ;
                return  ;
            }
            for(int i=0; i<number; i++) {
        
                int sockFd = ev[i].data.fd ;
                if(sockFd == listenFd) {
        
                    int index = count ;
                    do{
        
                        if(proDescInfo[index]->pid != -1) break ;
                        index = (index+1)%curProcessNum ;
                    }while(index != count) ;
    
                    if(proDescInfo[index]->pid == -1) {
         //没有子进程
                        stop = true ;
                        break ;
                    }
                    count = (index+1)%curProcessNum ;
                    int ret = send(proDescInfo[index]->getWriteFd(), &newConn, sizeof(newConn), 0) ;
                    if(ret < 0) {
        
                        cout << __LINE__ << "         " << __FILE__ << endl ;
                        return ;
                    }
                    cout << "给进程" << index << "发送消息"<< endl ;
                }
            }   
        }
        close(epollFd) ;
    }
    
    //子进程的执行函数
    template<typename T>
    void processPool<T> :: runChild() {
        
        cout << "在子进程中" << "     " << "序号:" << index << endl ;
        setupSigPipe() ;
        int pipeFd = proDescInfo[index]->getReadFd() ;
        //将管道描述符加入到epoll中
        tool::addFd(epollFd, pipeFd) ;
        epoll_event ev[maxEventNum]  ;
        map<int, shared_ptr<T>> user ; 
        //为客户端设置处理的对象的池子
        while(!stop) {
        
            int num = epoll_wait(epollFd, ev, maxEventNum, -1) ;
            if(num < 0&&errno != EINTR) {
        
                cout << __LINE__ << "        " << __FILE__ << endl ;
                return ;
            } 
            for(int i=0; i<num; i++) {
        
                int fd =ev[i].data.fd ;
                if(fd == pipeFd&&(ev[i].events&EPOLLIN)) {
        
                    cout << "管道信号！" << endl ;
                    char client ;
                    int ret = recv(fd, &client, sizeof(client), 0) ;
                    if(ret < 0) {
        
                        cout << __LINE__ << "      " << __FILE__ << endl ;
                        return ;
                    }
                    cout << "接收到信号！" << endl ;
                    int connFd = accept(listenFd, NULL, NULL) ;
                    if(connFd < 0) {
        
                        cout << __LINE__ << "       " << __FILE__ << endl ;
                        return ;
                    }
                    cout << "接收到新连接！" << endl ;
                    ret = tool::addFd(epollFd, connFd) ;
                    if(ret < 0) {
        
                        cout << __LINE__ <<  "         " << __FILE__ << endl ;
                        return ;
                    }
                    cout << "创建用户对象" << endl ;
                    //创建用户对象
                    user[connFd] = make_shared<T>(epollFd, connFd) ;
                } 
                //处理事件
                else if(ev[i].events&EPOLLIN) {
        
                    user[fd]->process() ;
                }
            }
        }
        close(pipeFd) ;
        close(epollFd) ;
    }
    
    class echo {
        
    public :
        echo(int epollFd, int fd) {
        
            echo::epollFd = epollFd ;
            sockFd = fd ;
            memset(buf, '\0', BUFFERSIZE) ;
            endIndex = 0 ;
        }
        ~echo() {
        close(sockFd) ;}
        void process() {
        
            while(true) {
        
                int ret = recv(sockFd, buf, BUFFERSIZE-1, 0) ;
                if(ret < 0) {
        
                    if(errno != EAGAIN) {
        
                        tool :: removeFd(epollFd, sockFd) ;
                        break ;
                    }
                    cout << __LINE__ << "        " << __FILE__ << endl ;
                    return ;
                }
                else if(ret == 0) {
        
                    int res = tool :: removeFd(epollFd, sockFd) ;
                    if(res < 0) {
        
                        cout << "移除失败！" << endl ;
                        return ;
                    }
                    break ;
                }
                else {
        
                    cout << "客户端数据：" << buf << endl ;
                    ret  = send(sockFd, buf, sizeof(buf), 0) ;
                    if(ret < 0) {
        
                        cout << __LINE__ << "         " << __FILE__ << endl ;
                        break ;
                    }
                    break ;
                }
            }
        }
    
    public :
        static int epollFd ;
        int sockFd ;
        char buf[BUFFERSIZE] ;
        int endIndex ;
    } ;
    
    
    int echo :: epollFd ;
    
    int tool :: setNoBlock(int fd) {
        
        int old = fcntl(fd, F_GETFL) ;
        int ret = fcntl(fd, F_SETFL, old|O_NONBLOCK) ;
        if(ret < 0) {
        
            cout << __FILE__ << "       " << __LINE__ << endl ;
            return -1 ;
        }
        return old ;
    }
    
    int tool :: addFd(int epollFd, int fd) {
        
        epoll_event ev ;
        ev.data.fd = fd ;
        ev.events = EPOLLIN|EPOLLET ;
        int ret = epoll_ctl(epollFd, EPOLL_CTL_ADD, fd, &ev) ;
        if(ret < 0) {
        
            cout << __FILE__ << "      " << __LINE__ << endl ;
        }
        ret = setNoBlock(fd) ;
        if(ret < 0) {
        
            cout << __LINE__ << "      " << __FILE__ << endl ;
            return -1 ;
        }
        return 1 ;
    }
    
    int tool :: removeFd(int epollFd, int fd) {
        
       int ret = epoll_ctl(epollFd, EPOLL_CTL_DEL, fd, 0) ;
       if(ret < 0) {
        
            cout << __LINE__ << "     " << __FILE__ << endl ;
            return -1 ;
       }
       return 1 ;
    }
    
    void tool :: addSig(int sig, void(handle)(int), bool restart) {
        
        struct sigaction sa ;
        memset(&sa, '\0', sizeof(sa)) ;
        sa.sa_handler = handle ;
        if(restart) 
            int res = sa.sa_flags|SA_RESTART ;
        sigfillset(&sa.sa_mask) ;
    }
    
    int process :: createSocketPair() {
        
        int ret = socketpair(AF_UNIX, SOCK_STREAM, 0, pipe) ;
        if(ret < 0) {
        
            cout << __LINE__ << "      " << __FILE__ << endl ;
            return -1 ; 
        }
        return 1 ;
    }

`main.cpp`

#include "process.h"
    void handle(int sig) {
        
        if(sig == SIGINT) {
        
            exit(0) ;
        }
    }
    
    int main(int argc, char** argv) {
        
        if(argc != 3) {
        
            cout << "error usage" << endl ;
            return 1 ;
        }
        struct sockaddr_in addr ;
        bzero(&addr, sizeof(addr)) ;
        addr.sin_family = AF_INET ;
        int ret = inet_pton(AF_INET, argv[1], &addr.sin_addr) ;
        if(ret < 0) {
        
            cout << __LINE__ << "        " << __FILE__ << endl ;
            return 0 ;
        }
        addr.sin_port = htons(atoi(argv[2])) ;
        int listenFd = socket(AF_INET, SOCK_STREAM, 0) ; 
        if(listenFd < 0) {
        
            cout << __LINE__ << "       " << __FILE__ << endl ;
            return 1;
        }
        int use = 1 ;
        ret = setsockopt(listenFd, SOL_SOCKET, SO_REUSEADDR, &use, sizeof(use)) ;
        if(ret < 0) {
        
            cout << __LINE__ << "       " << __FILE__ << endl ;
            return 0;
        }
        ret = bind(listenFd, (const sockaddr*)&addr, sizeof(addr)) ;
        if(ret < 0) {
        
            cout << __FILE__ << "       " << __LINE__ << endl ;
            return 0 ;
        }
        ret = listen(listenFd, 100) ;
        if(ret < 0) {
        
            cout << "监听失败！" << endl ;
            return -1 ;
        }
        //中断回收进程后再退出
        signal(SIGINT, handle) ;
        //创建
        shared_ptr<processPool<echo>> pool = processPool<echo> :: create(listenFd, 2);
        if(pool)
        pool->run() ;
        close(listenFd) ;
    }

实现思路相对较清晰，也比较简单！所以想和大家分享一下！

[gdb]: https://blog.csdn.net/kkkkde/article/details/88679706