/**
 * File threadpool.h
 * 1、用于创建线程池
 * 2、每个线程中等待事件队列中添加新事件（EventBase 指针指向的派生类）
 * 3、有新事件时，分配给一个线程处理，线程中调用事件的process()方法处理该事件
 */
static int tnum = 0;
class ThreadPool
{
public:
    // 初始化线程池、互斥访问事件队列的互斥量、表示队列中事件的信号量
    ThreadPool(int threadNum);
    ~ThreadPool();

public:
    // 向事件队列中添加一个待处理的事件，线程池中的线程会循环处理其中的事件
    int appendEvent(EventBase *event, const std::string eventType);

private:
    // 创建线程时指定的运行函数，参数传递this指针，实现在子线程中可以访问到该对象的成员
    static void *worker(void *arg);

    // 在线程中执行该函数等待处理事件队列中的事件
    void run();

private:
    int m_threadNum;      // 线程池中的线程个数
    pthread_t *m_threads; // 保存线程池中的所有线程

    std::queue<EventBase *> m_workQueue; // 事件队列 保存所有待处理的事件
    pthread_mutex_t queueLocker;         // 用于互斥访问事件队列的锁
    sem_t queueEventNum;                 // 表示队列中事件个数变化的信号量
};
#endif

#include "./fileserver/fileserver.h"

int main(void)
{
    WebServer server;

    // 创建线程池
    int ret = server.createThreadPool(4); // 四核线程池
    if (ret != 0)
    {
        std::cout << outHead("error") << "创建线程池失败" << std::endl;
        return -1;
    }

    // 初始化用于监听的套接字
    int port = 8888;
    ret = server.createListenFd(port);
    if (ret != 0)
    {
        std::cout << outHead("error") << "创建并初始化监听套接字失败" << std::endl;
        return -2;
    }

    // 初始化监听的epoll例程
    ret = server.createEpoll();
    if (ret != 0)
    {
        std::cout << outHead("error") << "初始化监听的epoll例程失败" << std::endl;
        return -3;
    }

    // 向epoll中添加监听套接字
    ret = server.epollAddListenFd();
    if (ret != 0)
    {
        std::cout << outHead("error") << "epoll添加监听套接字失败" << std::endl;
        return -4;
    }

    // 开启监听并处理请求
    ret = server.waitEpoll();
    if (ret != 0)
    {
        std::cout << outHead("error") << "epoll例程监听失败" << std::endl;
        return -5;
    }
    return 0;
}

#ifndef FILESERVER_H
#define FILESERVER_H
#include <signal.h>
#include <sys/types.h>
#include <stdexcept>
#include <errno.h>

#include "../threadpool/threadpool.h"

#define MAX_RESEVENT_SIZE 1024 // 事件的最大个数

class WebServer
{
public:
    WebServer();
    ~WebServer();

    // 创建套接字等待客户端连接，并开启监听
    int createListenFd(int port, const char *ip = nullptr);

    // 创建epoll例程用于监听套接字
    int createEpoll();

    // 向epoll中添加监控Listen 套接字
    int epollAddListenFd();

    // 设置监听事件处理的管道
    int epollAddEventPipe();

    // 设置term和alarm信号的处理
    int addHandleSig(int signo = -1);

    // 信号处理函数
    static void setSigHandler(int signo);

    // 主线程中负责监听所有事件
    int waitEpoll();

    // 创建线程池
    int createThreadPool(int threadNum = 8); // 默认8线程

private:
    int m_listenfd;           // 服务端的套接字
    sockaddr_in m_serverAddr; // 服务端套接字绑定的地址信息
    static int m_epollfd;     // I/O复用的epoll例程未见描述符
    static bool isStop;       // 是否暂停服务器

    static int eventHandlerPipe[2]; // 用于统一事件源传递信号的管道

    epoll_event resEvents[MAX_RESEVENT_SIZE]; // 保存 epoll_wait 结果的数组

    ThreadPool *threadPool;
};
#endif