• 仿mudou库one thread one loop式并发服务器


    目录

    1.实现目标

    2.HTTP服务器

    实现高性能服务器-Reactor模型

    模块划分

    SERVER模块:

    HTTP协议模块:

    3.项目中的子功能

    秒级定时任务实现

    时间轮实现

    正则库的简单使用

    通⽤类型any类型的实现

    4.SERVER服务器实现

    日志宏的封装

    缓冲区Buffer类实现

    套接字Socket实现

    事件管理Channel模块实现

    描述符事件监控Poller模块实现

    定时任务管理TimerWheel类实现

    EventLoop线程池类实现

    ​编辑

    LoopThread模块实现

    LoopThreadPool线程池模块实现

    通信连接管理Connection模块实现

    监听描述符管理Acceptor类实现

    TcpServer模块

    HTTP协议模块实现

    Util⼯具类实现

    HttpRequest请求类实现

    HttpResponse响应类实现

    HttpServer服务器模块

    服务器搭建并进行测试

    搭建服务器

    长连接测试

    测试超时连接是否销毁

    错误请求处理

    服务器性能达到瓶颈的处理

    一次发送多个请求测试

    测试大文件传输

    服务器性能测试

    项目源码


    1.实现目标

    通过实现的⾼并发服务器组件,可以简洁快速的完成⼀个⾼性能的服务器搭建。并且,通过组件内提供的不同应⽤层协议⽀持,也可以快速完成⼀个⾼性能应⽤服务器的搭建

    2.HTTP服务器

    HTTP概念:超⽂本传输协议是应⽤层协议,是⼀种简单的请求-响应协议,HTTP服务器本质上就是个TCP服务器,只不过在应⽤层基于HTTP协议格式进⾏数据的组织和解析来明确客⼾端的请求并完成业务处理

    实现高性能服务器-Reactor模型

    概念:
    Reactor模式,是指通过⼀个或多个输⼊同时传递给服务器进⾏请求处理时的事件驱动处理模式。服务端程序处理传⼊多路请求,并将它们同步分派给请求对应的处理线程,Reactor模式也叫Dispatcher模式。使⽤ I/O多路复⽤统⼀监听事件,收到事件后分发给处理进程或线程

    分类:
    1.单Reactor单线程:单I/O多路复⽤+业务处理
    1. 通过IO多路复⽤模型进⾏客⼾端请求监控

    2. 触发事件后,进⾏事件处理
    a. 如果是新建连接请求,则获取新建连接,并添加⾄多路复⽤模型进⾏事件监控。

    b. 如果是数据通信请求,则进⾏对应数据处理(接收数据,处理数据,发送响应)

    优点:所有操作均在同⼀线程中完成,思想流程较为简单,不涉及进程/线程间通信及资源争抢问题

    缺点:⽆法有效利⽤CPU多核资源,很容易达到性能瓶颈。
    适⽤场景:适⽤于客⼾端数量较少,且处理速度较为快速的场景

    2.单Reactor多线程:单I/O多路复⽤+线程池(业务处理)

    1. Reactor线程通过I/O多路复⽤模型进⾏客户端请求监控
    2. 触发事件后,进⾏事件处理
    a. 如果是新建连接请求,则获取新建连接,并添加⾄多路复⽤模型进⾏事件监控。
    b. 如果是数据通信请求,则接收数据后分发给Worker线程池进⾏业务处理。
    c. ⼯作线程处理完毕后,将响应交给Reactor线程进⾏数据响应

    优点:充分利⽤CPU多核资源
    缺点:多线程间的数据共享访问控制较为复杂,单个Reactor承担所有事件的监听和响应,在单线程中运⾏,⾼并发场景下容易成为性能瓶颈

    3.多Reactor多线程:多I/O多路复⽤+线程池(业务处理)

    1. 在主Reactor中处理新连接请求事件,有新连接到来则分发到⼦Reactor中监控
    2. 在⼦Reactor中进⾏客户端通信监控,有事件触发,则接收数据分发给Worker线程池
    3. Worker线程池分配独⽴的线程进⾏具体的业务处理
    a. ⼯作线程处理完毕后,将响应交给⼦Reactor线程进⾏数据响应

    当前项目采用的方式:One Thread One Loop主从Reactor模型⾼并发服务器

    1.主Reactor线程仅仅监控监听描述符,获取新建连接,保证获取新连接的⾼效性,提⾼服务器的并发性能。
    2.主Reactor获取到新连接后分发给从属Reactor进⾏通信事件监控。而从属Reactor线程监控各⾃的描述符的读写事件进⾏数据读写以及业务处理。
    3.One Thread One Loop的思想就是把所有的操作都放到⼀个线程中进⾏⼀个线程对应⼀个事件处理的循环

    简单来说就是:主属Reactor用于对连接的管理,从属Reactor就是把剩下的工作全部做完

    组件使用者需要自行决定是否需要线程池,自己完成实现。

    模块划分

    我们要实现的是⼀个带有协议⽀持的Reactor模型⾼性能服务器,因此我们可以划分成两个模块:

    SERVER模块:实现Reactor模型的TCP服务器

    协议模块:对当前的Reactor模型服务器提供应⽤层协议⽀持

    SERVER模块:

    分成3个方面的管理:

    1.监听连接管理 2.通信连接管理 3.超时连接管理

    根据这3个方面的管理我们可以划分为下面多个子模块:
    Buffer模块:Buffer模块是⼀个缓冲区模块,⽤于实现通信中用户态的接收缓冲区和发送缓冲区功能

    Socket模块:Socket模块是对套接字操作封装的⼀个模块,主要实现的socket的各项操作

    Channel模块:结合Poller模块,对事件进行监控处理,分别监控可读,可写,错误,任意事件的监控,根据不同的事件调用不同的回调函数进行处理

    Poller模块:Poller模块是对epoll进⾏封装的⼀个模块,主要实现epoll的IO事件添加,修改,移除,获取活跃连接功能。

    Connection模块:Connection模块是对Buffer模块,Socket模块,Channel模块的⼀个整体封装,实现了对⼀个通信套接字的整体的管理,每⼀个进⾏数据通信的套接字(也就是accept获取到的新连接)都会使⽤Connection进⾏管理

    Acceptor模块:Acceptor模块是对Socket模块,Channel模块的⼀个整体封装,实现了对⼀个监听套接字的整体的管理

    TimerQueue模块:TimerQueue模块是实现固定时间定时任务的模块,可以理解就是要给定时任务管理器,向定时任务管理器中添加⼀个任务,任务将在固定时间后被执⾏,同时也可以通过刷新定时任务来延迟任务的执⾏。
    这个模块主要是对
    Connection对象的⽣命周期管理对⾮活跃连接进⾏超时后的释放功能

    EventLoop模块:EventLoop模块是对Poller模块,TimerQueue模块,Socket模块的⼀个整体封装,进⾏所有描述符的事件监控,这里为了保证保证整个服务器的线程安全问题,因此要求使⽤者对于Connection的所有操作⼀定要在其对应的EventLoop线程内完成(也就是One Thread One Loop 的思想)

    TcpServer模块:这个模块是⼀个整体Tcp服务器模块的封装,内部封装了Acceptor模块,EventLoopThreadPool模块

    HTTP协议模块:

    Util模块:这个模块是⼀个⼯具模块,主要提供HTTP协议模块所⽤到的⼀些⼯具函数,⽐如url编解码,⽂件读写等

    HttpRequest模块:
    这个模块是HTTP请求数据模块,⽤于保存HTTP请求数据被解析后的各项请求元素信息

    HttpResponse模块:
    这个模块是HTTP响应数据模块,⽤于业务处理后设置并保存HTTP响应数据的的各项元素信息,最终会被按照HTTP协议响应格式组织成为响应信息发送给客户端

    HttpContext模块:
    这个模块是⼀个HTTP请求接收的上下⽂模块,主要是为了防⽌在⼀次接收的数据中,不是⼀个完整的HTTP请求,则解析过程并未完成,⽆法进⾏完整的请求处理,需要在下次接收到新数据后继续根据上下⽂进⾏解析,最终得到⼀个HttpRequest请求信息对象

    HttpServer模块:
    这个模块是最终给组件使⽤者提供的HTTP服务器模块了,⽤于以简单的接⼝实现HTTP服务器的搭建

    3.项目中的子功能

    秒级定时任务实现

    在我们的项目中需要使用到超时销毁的功能, 所以我们需要一个时间轮,能让一个连接在一定的时间进行销毁,这就需要我们写一个简单的秒级定时任务

    在Linux中给我们提供了定时器:
     

    int timerfd_create(int clockid, int flags);

    clockid: CLOCK_REALTIME-系统实时时间,如果修改了系统时间就会出问题; CLOCK_MONOTONIC-从开机到现在的时间是⼀种相对时间(采用这个时间,保证准确性)
    flags: 0-默认阻塞属性

    int timerfd_settime(int fd, int flags, struct itimerspec *new, structitimerspec *old);
    

     fd: timerfd_create返回的⽂件描述符
     flags: 0-相对时间, 1-绝对时间;默认设置为0即可.
     new: ⽤于设置定时器的新超时时间
     old: ⽤于接收原来的超时时间

    1. struct timespec {
    2. time_t tv_sec; /* Seconds */
    3. long tv_nsec; /* Nanoseconds */
    4. };
    5. struct itimerspec {
    6. struct timespec it_interval; /* 第⼀次之后的超时间隔时间 */
    7. struct timespec it_value; /* 第⼀次超时时间 */
    8. };

    定时器会在每次超时⾃动给fd中写⼊8字节的数据,表⽰在上⼀次读取数据到当前读取数据期间超时了多少次。这样我们每次读取定时器的时候都知道他超时了多少次,然后我们就执行多少秒的释放连接即可

    定时器的使用用例:

    1. #include <iostream>
    2. #include <unistd.h>
    3. #include <sys/timerfd.h>
    4. int main()
    5. {
    6. //int timerfd_create(int clockid, int flags);
    7. int timerfd = timerfd_create(CLOCK_MONOTONIC,0);
    8. if(timerfd < 0)
    9. {
    10. perror("timerfd_create fail");
    11. return -1;
    12. }
    13. //int timerfd_settime(int fd, int flags,const struct itimerspec *new_value,struct itimerspec *old_value);
    14. //设置结构体
    15. struct itimerspec itim;
    16. itim.it_value.tv_sec = 1;
    17. itim.it_value.tv_nsec = 0; //设置第一次超时时间
    18. itim.it_interval.tv_sec = 1;
    19. itim.it_interval.tv_nsec = 0; //第一次超时之后每隔1秒超时一次
    20. int ret = timerfd_settime(timerfd,0,&itim,nullptr);
    21. if(ret != 0)
    22. {
    23. perror("timerfd_settime fail");
    24. return -1;
    25. }
    26. while(1){
    27. //因为这个也是一个文件描述符,所以可以使用read进行系统调用,来读取到其中的数据
    28. uint64_t times = 0;
    29. int ret = read(timerfd,&times,8);
    30. if(ret < 0){
    31. perror("read fail");
    32. return -1;
    33. }
    34. std::cout<<times<<std::endl;
    35. }
    36. return 0;
    37. }

    时间轮实现

    如果每次我们都需要遍历一次连接来进行超时销毁,这样效率是非常低的,这里采用时间轮的思想来提高效率

    我们通过Linux提供的定时器,每次将tick指针移动,到指定位置时进行销毁即可

    如果我们需要使用到分级定时器,或者时级定时器,那么为了保证不消耗那么多空间,这里采用像Linux中页面的设计那么,采用多级时间轮。

    这样就可以大大的节省空间了,3个时间轮就可以把一个的每一个时刻都定位到了,如果还想要精确到年,月,日也是可以通过创建更多的时间轮来实现。

    这里主要实现秒级时间轮:

    1. #include <iostream>
    2. #include <functional>
    3. #include <vector>
    4. #include <unistd.h>
    5. #include <memory>
    6. #include <unordered_map>
    7. using OnCloseTime = std::function<void()>; //定时器要执行的任务
    8. using ReleaseTime = std::function<void()>; //删除时间管理对象中weak_ptr的信息
    9. //定时器任务对象
    10. class TimeTask
    11. {
    12. private:
    13. uint32_t _timeout; //超时时间
    14. uint64_t _id; //每个任务的id
    15. bool _cancel; //取消定时任务
    16. OnCloseTime _close_cb; //销毁定时任务的回调
    17. ReleaseTime _release_cb; //因为时间轮中会记录一个weak_ptr对象,所以最后需要销毁
    18. public:
    19. TimeTask(uint32_t timeout,uint64_t id,const OnCloseTime& close_cb):_timeout(timeout),_id(id),_close_cb(close_cb),_cancel(false){}
    20. ~TimeTask(){
    21. if(_cancel== false) _close_cb();
    22. _release_cb();
    23. }
    24. uint64_t id() { return _id; }
    25. void SetRelease(const ReleaseTime& release_cb) { _release_cb = release_cb; }
    26. uint32_t Delay() { return _timeout; }
    27. void Cancel() { _cancel = true; }
    28. };
    29. //时间轮管理对象
    30. class TimeWheel
    31. {
    32. private:
    33. using WeakTask = std::weak_ptr<TimeTask>; //使用weak_ptr防止在shared_ptr直接对对象的操作
    34. using PtrTask = std::shared_ptr<TimeTask>; //使用shared_ptr保证释放时不到0不销毁
    35. int _capacity; //记录时间轮的大小
    36. int _tick; //记录当前指针指向的时间,指到哪里,销毁哪里
    37. std::vector<std::vector<PtrTask>> _wheel; //时间轮
    38. std::unordered_map<uint64_t,WeakTask> _times; //记录id和weak_ptr之间的映射关系
    39. private:
    40. void RemoveTask(uint64_t id)
    41. {
    42. if(_times.find(id) != _times.end())
    43. {
    44. _times.erase(id);
    45. }
    46. }
    47. public:
    48. TimeWheel():_tick(0),_capacity(60),_wheel(_capacity){}
    49. void AddTask(uint64_t id,uint32_t delay,const OnCloseTime close_cb)
    50. {
    51. PtrTask pt(new TimeTask(delay,id,close_cb));
    52. //设置ReleaseTask
    53. pt->SetRelease(std::bind(&TimeWheel::RemoveTask,this,id));
    54. //把任务添加到数组中
    55. int pos = (_tick + delay) %_capacity;
    56. _wheel[pos].push_back(pt);
    57. //将id和weakTask映射关联起来
    58. _times[id] = WeakTask(pt);
    59. }
    60. void CancelTask(uint64_t id)
    61. {
    62. //通过id找到任务,如果没有直接返回,有的话将标志置为true
    63. if(_times.find(id) == _times.end()) return;
    64. PtrTask pt = _times[id].lock(); //获得weak_ptr中的shared_ptr
    65. pt->Cancel();
    66. }
    67. void RefreshTask(uint64_t id)
    68. {
    69. //创建一个新的智能指针对象,然后添加到数组中
    70. //如果在原数组中没有找到,那么直接返回
    71. if(_times.find(id) == _times.end()) return;
    72. std::cout<<"找到了定时任务\n";
    73. PtrTask pt = _times[id].lock(); //获得weak_ptr中的shared_ptr
    74. int delay = pt->Delay();
    75. int pos = (_tick + delay) %_capacity;
    76. _wheel[pos].push_back(pt);
    77. }
    78. void RunTask()
    79. {
    80. _tick = (_tick+1)%_capacity;
    81. _wheel[_tick].clear();
    82. }
    83. };

    后序项目需要使用到的时候直接融合进去即可

    正则库的简单使用

    正则表达式(regular expression)描述了⼀种字符串匹配的模式(pattern),可以⽤来检查⼀个串是否含有某种⼦串、将匹配的⼦串替换或者从某个串中取出符合某个条件的⼦串等。
    正则表达式的使⽤,可以使得HTTP请求的解析更加简单(不需要更多的操作,简化代码编写)注释中有简单的使用方法,这里不一一赘述

    1. #include <iostream>
    2. #include <string>
    3. #include <regex>
    4. int main()
    5. {
    6. // 请求: GET /hello/login?user=xiaoming&passwd=123456 HTTP/1.1\r\n
    7. std::string str = "get /hello/login?user=xiaoming&passwd=123456 HTTP/1.1";
    8. //提取请求方法
    9. std::smatch matches;
    10. std::regex e("(GET|POST|HEAD|DELETE|PUT) ([^?]*)(?:\\?(.*))? (HTTP/1\\.[01])(?:\r|\r\n)?",std::regex::icase); //匹配请求方法的正则表达式
    11. //(GET|POST|HEAD|DELETE|PUT) 匹配其中任意一个请求方法
    12. //([^?]*) 匹配非?字符 *表示0次或者多次
    13. //\\?(.*) 提取?问号之后的字符,直到遇到后面的空格
    14. //[01] 提起0或者1其中任意一个
    15. //(?:\r|\r\n)? ?:表示匹配某个字符串,但是不提取 后面的?表示匹配0次或者1
    16. bool ret = regex_match(str,matches,e);
    17. if(ret == false) return -1;
    18. std::string method = matches[1];
    19. std::transform(method.begin(),method.end(),method.begin(),::toupper); //转换大小写
    20. std::cout<<method<<std::endl;
    21. for(int i = 0;i<matches.size();++i)
    22. {
    23. std::cout<< i <<" : ";
    24. std::cout<<matches[i]<<std::endl;
    25. }
    26. return 0;
    27. }

    通⽤类型any类型的实现

    Connection中需要设置协议处理的上下⽂来控制处理节奏,由于应用层协议非常多,所以我们需要需要使用通用类型来保存不同的数据结构

    实现思想:

    首先Any类一定不是一个模板类,因为模板类的实例化的时候需要传类型,但是我们可以通过在Any类中设计一个父类和一个子类,其中父类不能是模板类,因为Any类访问父类指针的时候就需要父类的类型,所以我们可以把子类设计成模板类,子类继承父类,并通过重写虚函数来实现多态。当需要保存数据时,通过new一个带模板参数的子类对象来保存数据,然后让Any类中的父类指针指向这个子类对象就可以了

    实现如下:

    1. class Any
    2. {
    3. private:
    4. //父类不是模板类,这样可以保证Any类不是模板
    5. class holer
    6. {
    7. public:
    8. virtual ~holer() {}
    9. virtual const std::type_info& type() = 0; //设置成纯虚函数,那么子类想要实例化就必须要重写虚函数
    10. virtual holer* clone() = 0;
    11. };
    12. template <class T>
    13. class placeholer : public holer
    14. {
    15. public:
    16. placeholer(const T& val):_val(val){}
    17. virtual ~placeholer() {}
    18. virtual const std::type_info& type() { return typeid(T);}
    19. virtual placeholer* clone() { return new placeholer(_val);}
    20. public:
    21. T _val;
    22. };
    23. private:
    24. Any& swap(Any& any)
    25. {
    26. std::swap(_holer,any._holer);
    27. return *this;
    28. }
    29. holer* _holer;
    30. public:
    31. Any():_holer(nullptr) {}
    32. ~Any(){ delete _holer; }
    33. Any(const Any& any):_holer(any._holer == nullptr?nullptr:any._holer->clone()){}
    34. template <class T>
    35. Any(const T& val):_holer(new placeholer<T>(val)) {}
    36. template<class T>
    37. T* get()
    38. {
    39. assert(typeid(T) == _holer->type());
    40. return &((placeholer<T>*)_holer)->_val;
    41. }
    42. Any& operator=(Any& any)
    43. {
    44. Any(any).swap(*this);
    45. return *this;
    46. }
    47. template<class T>
    48. Any& operator=(const T& val)
    49. {
    50. Any(val).swap(*this);
    51. return *this;
    52. }
    53. };

    4.SERVER服务器实现

    日志宏的封装

    在项目中为了方便调试,我们可以通过日志打印的方式来高效的确定和自己预期不相同的地方,关于日志宏在上一个五子棋项目中已经有比较详细的解析,主要分成以下几步:

    1.通过time函数获取时间

    2.通过localtime来获取具体的时间,即年月日时分秒

    3.通过strftime将时间以格式化数据存放到缓冲区中

    4.通过打印的方式打印出来,同时还要加上文件以及行号,线程等

    1. #define INF 0
    2. #define DBG 1
    3. #define ERR 2
    4. #define DEFAULT_LOG_LEVEL DBG
    5. #define LOG(level, format, ...) {\
    6. if (level >= DEFAULT_LOG_LEVEL) {\
    7. time_t t = time(NULL);\
    8. struct tm *m = localtime(&t);\
    9. char ts[32] = {0};\
    10. strftime(ts, 31, "%H:%M:%S", m);\
    11. fprintf(stdout, "[%p %s %s:%d] " format "\n", (void*)pthread_self(), ts, __FILE__, __LINE__, ##__VA_ARGS__);\
    12. }\
    13. }
    14. #define ILOG(format, ...) LOG(INF, format, ##__VA_ARGS__);
    15. #define DLOG(format, ...) LOG(DBG, format, ##__VA_ARGS__);
    16. #define ELOG(format, ...) LOG(ERR, format, ##__VA_ARGS__);

    缓冲区Buffer类实现

    Buffer类⽤于实现用户态缓冲区,提供数据缓冲,取出等功能,后序模块中的发送数据其实是像缓冲区中写入,而读取数据则是从缓冲区中取走数据。

    模块模型:

    成员设计

    1. std::vector<char> _buffer; // 缓冲区
    2. uint64_t _read_idx; // 读偏移
    3. uint64_t _write_idx; // 写偏移

    模块主要功能

    1.获取当前写位置地址;

    char *WritePosition() { return Begin() + _write_idx; }

    2.确保可写空间足够;

    uint64_t ReadAbleSize() { return _write_idx - _read_idx; }

    (如果不够就先移动数据,因为前面和后面都是有空余空间的,如果还不够那就直接扩容)

    void EnsureWriteSpace(uint64_t len)

    3.获取末尾空闲空间大小; (总空间-写偏移)

    uint64_t TailIdleSpace() { return _buffer.size() - _write_idx; }

    4.获取前面的空闲空间大小;(读偏移)

    uint64_t HeadIdleSpace() { return _read_idx; }

    5.将写位置向后移动指定长度;(这里就是写入数据之后把可写下标移动)

    void MoveWriteOffSet(uint64_t len)

    6.获取当前读位置地址;

    char *ReadPosition() { return Begin() + _read_idx; }

    7.获取可读数据大小;(可读到可写的区间)

    uint64_t ReadAbleSize() { return _write_idx - _read_idx; }

    8.将读位置向后移动到指定长度;(处理完数据之后移动)

    void MoveReadOffSet(uint64_t len)

    9.清理缓冲区;(把读写偏移放到0即可)

    void Clear()

    由于当前模块实现并不难,这里只有一个函数需要注意一下:

    如果确保可写空间足够:

    扩容策略:

    这样就可以高效的使用内存了,具体代码如下:

    1. // 确保可写空间足够(头部和尾部的空间够则移动数据,不够则扩容)
    2. void EnsureWriteSpace(uint64_t len)
    3. {
    4. if (len <= TailIdleSpace())
    5. return;
    6. else if (len <= TailIdleSpace() + HeadIdleSpace())
    7. {
    8. // 将数据拷贝到最前面
    9. int sz = ReadAbleSize();
    10. std::copy(ReadPosition(), ReadPosition() + sz, Begin());
    11. // 设置读写偏移
    12. _read_idx = 0;
    13. _write_idx = sz;
    14. }
    15. else
    16. {
    17. // 扩容
    18. _buffer.resize(_write_idx + len);
    19. }
    20. }

    其他部分的缓冲区模块实现起来比较简单,后面会有源码

    套接字Socket实现

    在使用网络通信中,我们需要Socket的接口,为了方便使用,所以统一封装了一个套接字类

    成员设计

    int _sockfd; // 套接字

    主要接口

    1.创建套接字;

    1. bool Create()
    2. // int socket(int domain, int type, int protocol);

    2.绑定地址信息;

    1. bool Bind(const std::string &ip, uint16_t port)
    2. // int bind(int sockfd, const struct sockaddr *addr,socklen_t addrlen);

    3.开始监听;

    1. bool Listen(int backlog = MAX_LISTEN)
    2. // int listen(int sockfd, int backlog);

    4.向服务器发起连接;

    1. bool Connect(const std::string &ip, uint16_t port)
    2. // int connect(int sockfd, const struct sockaddr *addr,socklen_t addrlen);

    5.获取新连接;

    1. int Accept()
    2. // int accept(int sockfd, struct sockaddr *addr, socklen_t *addrlen);

    6.接收数据;  

    1. ssize_t Recv(void *buf, size_t len, int flag = 0)
    2. // ssize_t recv(int sockfd, void *buf, size_t len, int flags);

    7.发送数据;

    1. ssize_t Send(const void *buf, size_t len, int flag = 0)
    2. // ssize_t send(int sockfd, const void *buf, size_t len, int flags);

    8.关闭套接字;

    void Close()

    9.创建一个服务端连接;(可以通过该接口快速创建一个服务端套接字)

    1. bool CreateSerber(uint16_t port, const std::string &ip = "0.0.0.0", bool flag = false)
    2. // 1.创建套接字 2.绑定地址 3.监听 4.设置地址重用 5.设置非阻塞

    10.创建一个客户端连接;(可以通过该接口快速创建一个客户端套接字)

    1. bool CreateClient(uint16_t port, const std::string &ip)
    2. // 1.创建套接字 2.连接

    11.开启地址端口重用; (如果是服务器先关闭,为了快速重启服务器可以选择)

    1. void ReuseAddr()
    2. //通过使用该函数来实现:
    3. //int setsockopt(int sockfd, int level, int optname,const void *optval, socklen_t optlen);

    12.设置非阻塞; (给套接字设置,因为我们使用的epoll是需要设置套接字非阻塞的)

    1. void SetNonBlock()
    2. // int fcntl(int fd, int cmd, ... /* arg */ );

    事件管理Channel模块实现

    结合EventLoop模块,通过对描述符的监控,如果某一个描述符事件就绪,那么我们只需要调用对应的回调函数即可

    成员设计

    1.需要对应监控的描述符

    2.EventLoop模块,进行监控

    3.用一个变量来记录监控事件(本质是位图,哪一个事件需要监控哪一个bit就置为1)

    4.用一个变量记录就绪事件

    5.各种回调函数

    1. int _fd; //监控的描述符
    2. EventLoop* _loop;
    3. uint32_t _event; //监控事件
    4. uint32_t _revent; //已经发生的事件
    5. //各种回调函数
    6. using EventCallback = std::function<void()>;
    7. EventCallback _read_callback; //读事件回调
    8. EventCallback _write_callback; //写事件回调
    9. EventCallback _error_callback; //错误事件回调
    10. EventCallback _close_callback; //关闭事件回调
    11. EventCallback _event_callback; //任意事件回调

    主要接口

    1.设置各种回调函数,以及返回描述符,设置事件等

    1. void SetReadCallback(const EventCallback& cb) { _read_callback = cb; }
    2. void SetWriteCallback(const EventCallback& cb) { _write_callback = cb; }
    3. void SetErrorCallback(const EventCallback& cb) { _error_callback = cb; }
    4. void SetCloseCallback(const EventCallback& cb) { _close_callback = cb; }
    5. void SetEventCallback(const EventCallback& cb) { _event_callback = cb; }
    6. int Fd() { return _fd; }
    7. uint32_t Events() { return _event; }
    8. void SetRevent(uint32_t revent) { _revent = revent; }

    2.当前是否监控了可读

    bool ReadAble() { return _event & EPOLLIN;}

    3.当前是否监控了可写

    bool WriteAble() { return _event & EPOLLOUT;}

    4.启动可读事件监控

    void SetRead() { _event |= EPOLLIN; Update();}

    5.启动可写事件监控

    void SetWrite() { _event |= EPOLLOUT; Update();}

    6.关闭可读事件监控

    void CloseRead() { _event &= ~EPOLLIN ; Update();}

    7.关闭可写事件监控

    void CloseWrite() { _event &= ~EPOLLOUT; Update();}

    8.关闭全部事件监控

    void CloseEvent() { _event = 0; Update();}

    9.移除监控(因为这里的移除是通过EventLoop模块进行移除,所以需要在类外实现)

    10.更新事件(和移除监控一样)

    1. void Remove();
    2. void Update();

    11.处理任意事件

    void HandleEvent()

    其他的实现比较简单,这里处理任意事件的逻辑应该如下:

    这里处理任意事件的本质就是刷新活跃度,因为后面实现每一个任务的时候是把任务压入到任务池当中,所以这里可以直接执行任务,而不需要刷新活跃度,等到事件处理完成之后再刷新活跃度,这样就可以防止一个事件可能会处理的事件很长,但是却提前刷新了活跃度导致后序结果有问题。

    具体实现如下:
     

    1. void HandleEvent()
    2. {
    3.         //这里因为是把销毁任务压入到任务池中执行,所以,这里可以直接执行任务而不需要先刷新活跃度
    4.         //满足条件,都会触发的
    5.         if((_revent & EPOLLIN) || (_revent & EPOLLRDHUP) || (_revent & EPOLLPRI))
    6.         {
    7.             if(_read_callback) _read_callback();
    8.         }
    9.         //有可能释放连接的操作,一次只能处理一个
    10.         if(_revent & EPOLLOUT)
    11.         {
    12.             if(_write_callback) _write_callback();
    13.         }
    14.         else if(_revent & EPOLLERR)
    15.         {
    16.             if(_error_callback) _error_callback();
    17.         }
    18.         else if(_revent & EPOLLHUP)
    19.         {
    20.             if(_close_callback) _close_callback();
    21.         }
    22.         //所有事件处理过都需要刷新活跃度
    23.         if(_event_callback) _event_callback();
    24. }

    描述符事件监控Poller模块实现

    通过epoll来对描述符的IO事件监控

    成员设计

    1.需要一个epoll操作句柄

    2.拥有一个struct epoll_event 结构数组,监控保存所有活跃事件

    3.通过hash表管理描述符以及Channel对象的管理

    1. int _epfd;
    2. struct epoll_event _evs[MAX_EVENTSIZE];
    3. std::unordered_map<int,Channel*> _channels; //描述符和channel的映射关系

    接口设计

    1.构造函数创建一个epoll模型

    1. Poller()
    2. {
    3. //创建一个epoll模型
    4. _epfd = epoll_create(MAX_EVENTSIZE);
    5. if(_epfd < 0)
    6. {
    7. ELOG("EPOLL_CREATE FAILED:%s",strerror(errno));
    8. abort(); //退出程序
    9. }
    10. }

    2.添加或者修改监控事件(在channels中找不到就添加,找到就修改)

     void UpdateEvent(Channel* channel)

    3.移除监控事件 (删除hash映射关系,同时移除epoll监控)

    void RemoveEvent(Channel* channel)

    4.开始监控,返回活跃连接   

    1. void Poll(std::vector<Channel*>* active)
    2. //int epoll_wait(int epfd, struct epoll_event *events,int maxevents, int timeout);

    5.对epoll的操作 (上面函数有对epoll的操作统一使用这个接口,设计成私有函数)   

    1. void Update(Channel* channel,int op)
    2. // int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event);

    6.判断一个事件是否被监控 (设计为私有函数)

       

     bool HasChannel(Channel* channel)

    开始监控函数需要我们注意应该有以下流程:

    通过epoll_wait监控得到的事件,首先我们需要通过Channel来设置事件就绪,然后再返回活跃连接给上层进行处理

    1. void Poll(std::vector<Channel*>* active)
    2. {
    3. //int epoll_wait(int epfd, struct epoll_event *events,int maxevents, int timeout);
    4. int nfds = epoll_wait(_epfd,_evs,MAX_EVENTSIZE,-1);
    5. if(nfds < 0)
    6. {
    7. //这里有可能是信号打断
    8. if(errno == EINTR)
    9. {
    10. return;
    11. }
    12. //其他原因
    13. ELOG("EPOLL_WAIT FAILED! %s",strerror(errno));
    14. abort(); //退出程序
    15. }
    16. for(int i = 0;i<nfds;++i)
    17. {
    18. auto it = _channels.find(_evs[i].data.fd);
    19. assert(it != _channels.end());
    20. //向channel设置就绪事件
    21. it->second->SetRevent(_evs[i].events);
    22. //向外输出活跃事件的channel
    23. active->push_back(it->second);
    24. }
    25. }

    定时任务管理TimerWheel类实现

    在前面我们已经实现了定时任务以及时间轮了,这里我们只需要把他们融合到一起即可

    这里主要说明两个回调函数的含义:

    1. using OnCloseTime = std::function<void()>; //定时器要执行的任务
    2. using ReleaseTime = std::function<void()>; //删除时间管理对象中weak_ptr的信息

    其中OnCloseTime的回调就是每一个任务执行的时候销毁定时任务,也就是把智能指针中的引用计数--就可,如果减到0,就销毁这个任务;

    ReleaseTime回调就是当引用计数减到0的时候,需要销毁这个定时任务的时候,同时我们需要把hash表中的id和WeakTask去除关联。

    这里对TimeWheel模块进行解析:

    成员设计

    1.需要一个数组来充当时间轮

    2.我们需要使用到shared_ptr,同时我们每次刷新活跃度的时候都需要创建一个shared_ptr所以我们需要一个weak_ptr,并保存起来

    3.记录id和weak_ptr的映射关闭,这样我们可以随时通过id来创建一个shared_ptr

    4.因为我们添加,取消,刷新任务的时候,我们需要在线程中执行,所以我们需要EventLoop模块指针

    5.我们需要对定时器事件进行管理,需要管理指针

    成员如下:

    1. using WeakTask = std::weak_ptr<TimeTask>; //使用weak_ptr防止在shared_ptr直接对对象的操作
    2. using PtrTask = std::shared_ptr<TimeTask>; //使用shared_ptr保证释放时不到0不销毁
    3. int _capacity; //记录时间轮的大小
    4. int _tick; //记录当前指针指向的时间,指到哪里,销毁哪里
    5. std::vector<std::vector<PtrTask>> _wheel; //时间轮
    6. std::unordered_map<uint64_t,WeakTask> _times; //记录id和weak_ptr之间的映射关系
    7. int _timerfd; //定时器描述符
    8. EventLoop* _loop;
    9. std::unique_ptr<Channel> _timer_channel; //用于定时器事件管理

    功能

    1.添加任务;2.删除任务;3.刷新任务

    因为这些接口需要在各自的线程中执行,所以我们需要添加到线程中,这样就不需要考虑线程安全问题了。

    1. //定时器任务对象
    2. class TimeTask
    3. {
    4. private:
    5. uint32_t _timeout; //超时时间
    6. uint64_t _id; //每个任务的id
    7. bool _cancel; //取消定时任务
    8. OnCloseTime _close_cb; //销毁定时任务的回调
    9. ReleaseTime _release_cb; //因为时间轮中会记录一个weak_ptr对象,所以最后需要销毁
    10. public:
    11. TimeTask(uint32_t timeout,uint64_t id,const OnCloseTime& close_cb):_timeout(timeout),_id(id),_close_cb(close_cb),_cancel(false){}
    12. ~TimeTask(){
    13. if(_cancel== false) _close_cb();
    14. _release_cb();
    15. }
    16. uint64_t id() { return _id; }
    17. void SetRelease(const ReleaseTime& release_cb) { _release_cb = release_cb; }
    18. uint32_t Delay() { return _timeout; }
    19. void Cancel() { _cancel = true; }
    20. };
    21. //时间轮管理对象
    22. class TimeWheel
    23. {
    24. private:
    25. using WeakTask = std::weak_ptr<TimeTask>; //使用weak_ptr防止在shared_ptr直接对对象的操作
    26. using PtrTask = std::shared_ptr<TimeTask>; //使用shared_ptr保证释放时不到0不销毁
    27. int _capacity; //记录时间轮的大小
    28. int _tick; //记录当前指针指向的时间,指到哪里,销毁哪里
    29. std::vector<std::vector<PtrTask>> _wheel; //时间轮
    30. std::unordered_map<uint64_t,WeakTask> _times; //记录id和weak_ptr之间的映射关系
    31. int _timerfd; //定时器描述符
    32. EventLoop* _loop;
    33. std::unique_ptr<Channel> _timer_channel; //用于定时器事件管理
    34. private:
    35. void RemoveTask(uint64_t id)
    36. {
    37. auto it = _times.find(id);
    38. if(it != _times.end())
    39. {
    40. _times.erase(it);
    41. }
    42. }
    43. static int CreateTimerFd()
    44. {
    45. //int timerfd_create(int clockid, int flags);
    46. int timerfd = timerfd_create(CLOCK_MONOTONIC,0);
    47. if(timerfd < 0)
    48. {
    49. ELOG("timerfd_create fail");
    50. abort();
    51. }
    52. //int timerfd_settime(int fd, int flags,const struct itimerspec *new_value,struct itimerspec *old_value);
    53. //设置结构体
    54. struct itimerspec itim;
    55. itim.it_value.tv_sec = 1;
    56. itim.it_value.tv_nsec = 0; //设置第一次超时时间
    57. itim.it_interval.tv_sec = 1;
    58. itim.it_interval.tv_nsec = 0; //第一次超时之后每隔1秒超时一次
    59. timerfd_settime(timerfd,0,&itim,nullptr);
    60. return timerfd;
    61. }
    62. int ReadTimerFd()
    63. {
    64. uint64_t times;
    65. int ret = read(_timerfd,&times,8);
    66. if(ret < 0){
    67. ELOG("READTIMERFD FAILED!");
    68. abort();
    69. }
    70. return times;
    71. }
    72. void RunTask()
    73. {
    74. _tick = (_tick+1)%_capacity;
    75. _wheel[_tick].clear();
    76. }
    77. void Ontime()
    78. {
    79. //读取timerfd中内容,根据实时的超时次数执行任务,这里防止服务器因为处理繁忙而导致这里只进行了一次的的刷新,必须要刷新够次数
    80. int times = ReadTimerFd();
    81. for(int i = 0;i<times;++i)
    82. {
    83. RunTask();
    84. }
    85. }
    86. void AddTaskInLoop(uint64_t id,uint32_t delay,const OnCloseTime close_cb)
    87. {
    88. PtrTask pt(new TimeTask(delay,id,close_cb));
    89. //设置ReleaseTask
    90. pt->SetRelease(std::bind(&TimeWheel::RemoveTask,this,id));
    91. //把任务添加到数组中
    92. int pos = (_tick + delay) %_capacity;
    93. _wheel[pos].push_back(pt);
    94. //将id和weakTask映射关联起来
    95. _times[id] = WeakTask(pt);
    96. }
    97. void CancelTaskInLoop(uint64_t id)
    98. {
    99. //通过id找到任务,如果没有直接返回,有的话将标志置为true
    100. auto it = _times.find(id);
    101. if(it == _times.end()) return;
    102. PtrTask pt = it->second.lock(); //获得weak_ptr中的shared_ptr
    103. if(pt) pt->Cancel();
    104. }
    105. void RefreshTaskInLoop(uint64_t id)
    106. {
    107. //创建一个新的智能指针对象,然后添加到数组中
    108. //如果在原数组中没有找到,那么直接返回
    109. auto it = _times.find(id);
    110. if(it == _times.end()) return;
    111. PtrTask pt = it->second.lock(); //获得weak_ptr中的shared_ptr
    112. int delay = pt->Delay();
    113. int pos = (_tick + delay) %_capacity;
    114. _wheel[pos].push_back(pt);
    115. }
    116. public:
    117. TimeWheel(EventLoop* loop):_tick(0),_capacity(60),_wheel(_capacity),_timerfd(CreateTimerFd())
    118. ,_loop(loop),_timer_channel(new Channel(_loop,_timerfd))
    119. {
    120. //设置读回调,并启动读监控
    121. _timer_channel->SetReadCallback(std::bind(&TimeWheel::Ontime,this));
    122. _timer_channel->SetRead();
    123. }
    124. //因为当前类中有使用到数据结构,为了保证线程安全而又不用加锁的方式来提高效率,那么我们让其在一个线程中执行
    125. void AddTask(uint64_t id,uint32_t delay,const OnCloseTime close_cb);
    126. void CancelTask(uint64_t id);
    127. void RefreshTask(uint64_t id);
    128. //这个接口存在线程安全问题,只能在EventLoop模块中使用
    129. bool HasTimer(uint64_t id)
    130. {
    131. auto it = _times.find(id);
    132. if(it == _times.end())
    133. {
    134. return false;
    135. }
    136. return true;
    137. }
    138. };

    EventLoop线程池类实现

    这个模块设计思想

    1.在线程中对描述符进行事件监控

    2.有描述符就绪则对描述符进行事件处理(要保证操作都在线程中执行,保证线程安全)

    3.所有事件处理完成之后再讲任务队列中的任务一一执行(前面的处理实际上是把任务压入到任务队列,然后在线程中执行)

    通过下图来理解:

    这样就可以保证线程安全了,因为只需要提供一把锁就可以保证,就是对task保护,线程来取出数据过程中是有可能有线程安全问题的。

    类成员设计

    1.要进行事件监控 ->Poller模块

    2.执行任务队列中的任务 -> 一个线程安全的任务队列

    3.添加任务的时候需要定时器,需要一个事件轮

    4.需要有一个事件通知的描述符,来唤醒事件监控的阻塞

    类功能

    1.需要创建一个描述符就行事件通知,来唤醒事件监控的阻塞

    2.该类最主要的函数就是start函数,在这个函数中,通过Poller监控的就绪事件,然后分别调用对应的回调函数对事件进行处理,压入到任务队列中,然后再将任务队列中的任务放到线程中一一执行

    1. class EventLoop
    2. {
    3. private:
    4. using Func = std::function<void()>;
    5. std::thread::id _thread_id; //线程ID
    6. Poller _poller;
    7. int _event_fd;
    8. std::unique_ptr<Channel> _event_channel; //通过channel来管理eventfd
    9. std::vector<Func> _tasks; //任务队列
    10. std::mutex _mutex; //保证任务队列的线程安全
    11. TimeWheel _time_wheel; //时间轮
    12. private:
    13. static int CreateEventfd()
    14. {
    15. int efd = eventfd(0,EFD_CLOEXEC | EFD_NONBLOCK);
    16. if(efd < 0)
    17. {
    18. ELOG("eventfd failed!");
    19. abort(); //退出程序
    20. }
    21. return efd;
    22. }
    23. void ReadEventFd()
    24. {
    25. uint64_t res = 0;
    26. int ret = read(_event_fd,&res,sizeof(res));
    27. if(ret < 0)
    28. {
    29. //信号打断或者读阻塞
    30. if(errno == EINTR || errno == EAGAIN)
    31. {
    32. return;
    33. }
    34. ELOG("READEVENTFD FAILED!");
    35. abort();
    36. }
    37. }
    38. void WeakUpEventFd()
    39. {
    40. uint64_t val = 1;
    41. int ret = write(_event_fd,&val,sizeof(val));
    42. if(ret < 0)
    43. {
    44. if(errno == EINTR)
    45. {
    46. return;
    47. }
    48. ELOG("WEAKUPEVENTFD FAILED!");
    49. abort();
    50. }
    51. }
    52. //执行任务队列中的所有任务
    53. void RunAllTask()
    54. {
    55. std::vector<Func> functor;
    56. {
    57. std::unique_lock<std::mutex> lock(_mutex);
    58. _tasks.swap(functor);
    59. }
    60. //执行任务
    61. for(auto& f:functor) f();
    62. }
    63. public:
    64. EventLoop():_thread_id(std::this_thread::get_id()),_event_fd(CreateEventfd()),_event_channel(new Channel(this,_event_fd))
    65. ,_time_wheel(this)
    66. {
    67. //给eventChannel设置回调函数
    68. _event_channel->SetReadCallback(std::bind(&EventLoop::ReadEventFd,this));
    69. //启动读事件监控
    70. _event_channel->SetRead();
    71. }
    72. //如果要执行的任务在当前线程那就直接执行,不在就压入任务队列
    73. void RunInLoop(const Func& cb)
    74. {
    75. if(IsInloop()) cb();
    76. else QueueInLoop(cb);
    77. }
    78. //断言一个线程是否在当前线程中
    79. void AssertInLoop()
    80. {
    81. assert(_thread_id == std::this_thread::get_id());
    82. }
    83. //压入任务队列
    84. void QueueInLoop(const Func& cb)
    85. {
    86. {
    87. std::unique_lock<std::mutex> lock(_mutex);
    88. _tasks.push_back(cb);
    89. }
    90. //唤醒有可能因为事件还没有就绪的阻塞线程
    91. WeakUpEventFd();
    92. }
    93. //判断当前线程是否是EventLoop线程
    94. bool IsInloop() { return _thread_id == std::this_thread::get_id();}
    95. //添加/更新监控事件
    96. void UpdateEvent(Channel* channel) { return _poller.UpdateEvent(channel);}
    97. //移除监控事件
    98. void RemoveEvent(Channel* channel) { return _poller.RemoveEvent(channel);}
    99. //事件监控 ->就绪事件处理 -> 执行任务
    100. void Start()
    101. {
    102. while(1)
    103. {
    104. std::vector<Channel*> actives;
    105. _poller.Poll(&actives);
    106. //就绪事件处理
    107. for(auto& a:actives)
    108. {
    109. a->HandleEvent();
    110. }
    111. //执行任务
    112. RunAllTask();
    113. }
    114. }
    115. void TimerAdd(uint64_t id,uint32_t delay,const OnCloseTime close_cb) { _time_wheel.AddTask(id,delay,close_cb); }
    116. void TimerRefresh(uint64_t id) { _time_wheel.RefreshTask(id); }
    117. void TimerCancel(uint64_t id) { _time_wheel.CancelTask(id); }
    118. bool HasTimer(uint64_t id) { return _time_wheel.HasTimer(id); }
    119. };

    他们之间的模块关系图:

    LoopThread模块实现

    这个模块的意义:因为EventLoop模块在实例化对象的时候必须是在线程内部,如果我们创建了多个EventLoop对象,又同时创建多个线程,将各个线程id重新给EventLoop进行设置,那么在构造EventLoop对象到设置新的thread_id期间是不可控的。

    我们必须先创建线程,然后在线程的入口函数中进行EventLoop对象的实例化

    类成员设计

    1.创建一个线程

    2.为了保证先实例了loop之后外界才能获取到,所以我们需要使用到条件变量以及互斥锁

    类功能

    1.我们需要创建一个线程,绑定线程入口函数,保证了先创建线程再在线程中实例化EventLoop,这就是One Thread One Loop思想

    2.提供给外部获得到实例化的EventLoop对象(需要使用条件变量控制)

    1. class LoopThread
    2. {
    3. private:
    4. //实现获取loop和构造函数的同步关系,保证先实例化了loop之后才能获取
    5. std::mutex _mutex;
    6. std::condition_variable _cond;
    7. EventLoop* _loop;
    8. std::thread _thread; //一个线程对应一个loop
    9. private:
    10. void ThreadEntry()
    11. {
    12. //1.创建loop 2.通过条件变量来唤醒等待线程 3.运行
    13. EventLoop loop; //这里的临时变量生命周期跟随LoopThread
    14. {
    15. std::unique_lock<std::mutex> lock(_mutex);
    16. _cond.notify_all();
    17. _loop = &loop;
    18. }
    19. _loop->Start();
    20. }
    21. public:
    22. LoopThread():_loop(nullptr),_thread(std::bind(&LoopThread::ThreadEntry,this)){}
    23. EventLoop* GetLoop()
    24. {
    25. EventLoop* loop;
    26. {
    27. std::unique_lock<std::mutex> lock(_mutex);
    28. //通过条件变量保证同步
    29. _cond.wait(lock,[&](){ return _loop != nullptr; });
    30. loop = _loop;
    31. }
    32. return loop;
    33. }
    34. };

    LoopThreadPool线程池模块实现

    对所有的LoopThread进行管理分配

    类成员设计
    1.用户可以根据需求来创建线程的数量

    2.我们这里使用RR轮转的方式进行分配

    3.根据主从Reactor模型,首先我们需要一个主Reactor是一直进行工作的

    4.全部的线程都需要进行管理,使用数组进行管理

    类功能

    1.根据用户需要的线程数量来创建线程

    2.根据RR轮转来给用户提供从属EventLoop

    1. class LoopThreadPool
    2. {
    3. private:
    4. int _thread_count; //创建的LoopThread数量
    5. int _next_idx; //采用RR轮转的方式进行分配
    6. EventLoop* _base_loop; //主EventLoop跟随主线程
    7. std::vector<LoopThread*> _threads; //管理全部的线程
    8. std::vector<EventLoop*> _loops; //管理从属EventLoop
    9. public:
    10. LoopThreadPool(EventLoop* loop):_base_loop(loop),_thread_count(0),_next_idx(0){}
    11. void SetThreadCount(int count) { _thread_count = count; }
    12. //根据数量来创建出对应的LoopThread
    13. void Create()
    14. {
    15. if(_thread_count>0)
    16. {
    17. _threads.resize(_thread_count);
    18. _loops.resize(_thread_count);
    19. for(int i = 0;i<_thread_count;++i)
    20. {
    21. _threads[i] = new LoopThread();
    22. _loops[i] = _threads[i]->GetLoop();
    23. }
    24. }
    25. }
    26. EventLoop* NextLoop()
    27. {
    28. if(_thread_count == 0) return _base_loop;
    29. _next_idx = (_next_idx + 1)% _thread_count;
    30. return _loops[_next_idx];
    31. }
    32. };

    通信连接管理Connection模块实现

    Connection模块是Server模块中最重要的模块,存在的意义:对连接进行管理,所有的操作都是通过这个模块完成的

    类成员设计(管理方式):
    1.对套接字的管理,能进行套接字的各种操作

    2.连接事件的管理,如可读,可写,错误,挂断,任意事件

    3.缓冲区的管理,通过socket对数据进行接收发送

    4.协议上下文的管理,记录数据处理的过程,即使用户没有一次性把数据全部发送过来,但是 也可以保存当前处理的阶段,方便下次处理

    5.回调函数的管理,各种情况应该如何处理交给用户决定,必须有调用业务处理的回调函数

    6.连接状态的管理,不同的连接状态有不同的限制

    7.是否启动非活跃连接超时销毁

    类功能设计

    1.发送数据:提供发送数据接口,但是这个并不是真正发送接口,而是把数据发送到发送缓冲区,然后启动写事件监控,等待写事件就绪进行发送

    2.关闭连接:给用户提供的关闭连接接口,但是并不是实际的关闭,而是先看看输入输出缓冲区中有没有数据等待处理,如果有先处理,然后关闭连接

    3.启动非活跃连接超时销毁 

    4.取消非活跃连接超时销毁

    5.协议切换,一个连接接收数据之后如何进行处理,取决于上下文,以及业务处理回调函数

    6.这里需要注意的是大多说的处理其实都是在线程中执行的,所以我们都需要通过bind,让其在线程中执行。

    1. class Connection;
    2. //DISCONNECTED 断开连接状态
    3. //CONNECTING 连接建立,但是未完成全部工作的过渡态
    4. //CONNECTED 连接建立完成,可以通信状态
    5. //DISCONNECTING 连接待关闭状态,等待处理后序工作之后断开连接
    6. typedef enum {DISCONNECTED,CONNECTING,CONNECTED,DISCONNECTING }ConnectStatus;
    7. using PtrConnection = std::shared_ptr<Connection>;
    8. class Connection : public std::enable_shared_from_this<Connection>
    9. {
    10. private:
    11. int _conn_id; //连接建立的唯一id
    12. //uint64_t _timer_id 这里为了简化使用,直接使用connid来作为timerid
    13. int _socketfd; //连接对应的文件描述符
    14. bool _enable_active_release; //是否启动非活跃连接超时销毁
    15. EventLoop* _loop; //连接所关联的线程
    16. Socket _socket; //套接字管理
    17. Channel _channel; //事件管理
    18. ConnectStatus _status; //连接状态
    19. Buffer _in_buffer; //输入缓冲区,从套接字中读取,然后放入到缓冲区中
    20. Buffer _out_buffer; //输出缓冲区,将待发送数据放到输出缓冲区
    21. Any _context; //连接上下文
    22. private:
    23. //事件处理回调函数
    24. using ConnectCallback = std::function<void(const PtrConnection&)>;
    25. using MessageCallback = std::function<void(const PtrConnection&,Buffer*)>;
    26. using ClosedCallback = std::function<void(const PtrConnection&)>;
    27. using AnyEventCallback = std::function<void(const PtrConnection&)>;
    28. ConnectCallback _connect_callback;
    29. MessageCallback _msg_callback;
    30. ClosedCallback _closed_callback;
    31. AnyEventCallback _event_callback;
    32. //组件内关闭回调函数,组件内使用,因为使用智能指针进行Connect的管理,一旦关闭,就应该在管理的地方进行删除
    33. ClosedCallback _server_event_callback;
    34. //为了保证线程安全,每一个接口函数都应该放入到一个线程中
    35. void SendInLoop(Buffer& buf)
    36. {
    37. //这里并不是实际的发送接口,而是把数据放到发送缓冲区中
    38. if(_status == DISCONNECTED) return;
    39. _out_buffer.WriteBufferAndPush(buf);
    40. //启动写事件监控
    41. if(_channel.WriteAble() == false) _channel.SetWrite();
    42. }
    43. void ShutDownInLoop()
    44. {
    45. //这里也不是实际的关闭操作,而是将待处理的数据处理,待发送的数据发送
    46. _status = DISCONNECTING; //处于待关闭状态
    47. if(_in_buffer.ReadAbleSize() > 0){
    48. if(_msg_callback) _msg_callback(shared_from_this(),&_in_buffer);
    49. }
    50. if(_out_buffer.ReadAbleSize() > 0)
    51. {
    52. //启动写事件监控
    53. if(_channel.WriteAble() == false) _channel.SetWrite();
    54. }
    55. //关闭连接,不管数据有没有处理完,因为这里的数据可能不完整,不需要处理了
    56. if(_out_buffer.ReadAbleSize() == 0) Release();
    57. }
    58. void ReleaseInLoop()
    59. {
    60. //1.修改连接状态
    61. _status = DISCONNECTED;
    62. //2.移除事件监控
    63. _channel.Remove();
    64. //3.关闭描述符
    65. _socket.Close();
    66. //4.取消定时销毁任务
    67. if(_loop->HasTimer(_conn_id)) CancelActiveReleaseInLoop();
    68. //5.调用关闭的回调函数
    69. if(_closed_callback) _closed_callback(shared_from_this());
    70. //组件内调用的关闭函数
    71. if(_server_event_callback) _server_event_callback(shared_from_this());
    72. }
    73. void EstablishedInLoop()
    74. {
    75. //1.修改状态
    76. assert(_status == CONNECTING);
    77. _status = CONNECTED;
    78. //2.启动读事件监控
    79. _channel.SetRead();
    80. //3.调用连接成功回调函数
    81. if(_connect_callback) _connect_callback(shared_from_this());
    82. }
    83. void SetActiveReleaseInloop(int sec)
    84. {
    85. //1.修改判断标志位
    86. _enable_active_release = true;
    87. //2.如果定时任务存在,那就延迟一下
    88. if(_loop->HasTimer(_conn_id)){
    89. return _loop->TimerRefresh(_conn_id);
    90. }
    91. //3.不存在就添加
    92. _loop->TimerAdd(_conn_id,sec,std::bind(&Connection::Release,this));
    93. }
    94. void CancelActiveReleaseInLoop()
    95. {
    96. _enable_active_release = false;
    97. if(_loop->HasTimer(_conn_id)) _loop->TimerCancel(_conn_id);
    98. }
    99. void UpgradeInLoop(const Any& context,const ConnectCallback& conn,const MessageCallback& msg,
    100. const ClosedCallback& closed,const AnyEventCallback& event)
    101. {
    102. //修改各个类成员变量即可
    103. _context = context;
    104. _connect_callback = conn;
    105. _msg_callback = msg;
    106. _closed_callback = closed;
    107. _event_callback = event;
    108. }
    109. //五个Channel事件回调函数
    110. //将socket数据放到接收缓冲区中,调用message_callback进行消息的读取
    111. void HandlerRead()
    112. {
    113. //1.把数据放入到接收缓冲区
    114. char buffer[65536];
    115. ssize_t ret = _socket.NonBlockRecv(buffer,65535);
    116. if(ret < 0)
    117. {
    118. //出错了,不能直接关闭,而是调用ShutDownInLoop()
    119. return ShutDownInLoop();
    120. }
    121. //如果接收到的数据是0就不需要进行消息处理
    122. //2.调用message_callback
    123. _in_buffer.WriteAndPush(buffer,ret);
    124. if(_in_buffer.ReadAbleSize() > 0)
    125. {
    126. _msg_callback(shared_from_this(),&_in_buffer);
    127. }
    128. }
    129. //将发送缓冲区的数据发送
    130. void HandlerWrite()
    131. {
    132. //将发送缓冲区中的数据发送出去
    133. ssize_t ret = _socket.NonBlockSend(_out_buffer.ReadPosition(),_out_buffer.ReadAbleSize());
    134. if(ret < 0)
    135. {
    136. //出错了就看接收缓冲区有没有实际要处理的数据,有的话处理完就实际关闭连接
    137. if(_in_buffer.ReadAbleSize() > 0)
    138. {
    139. _msg_callback(shared_from_this(),&_in_buffer);
    140. }
    141. return Release();
    142. }
    143. //记得把写偏移移动
    144. _out_buffer.MoveReadOffSet(ret);
    145. if(_out_buffer.ReadAbleSize() == 0)
    146. {
    147. _channel.CloseWrite(); //关闭写事件监控
    148. //如果是连接待关闭状态则关闭连接
    149. if(_status == DISCONNECTING)
    150. {
    151. return Release();
    152. }
    153. }
    154. }
    155. //描述符挂断事件处理
    156. void HandlerClose()
    157. {
    158. if(_in_buffer.ReadAbleSize() > 0)
    159. {
    160. _msg_callback(shared_from_this(),&_in_buffer);
    161. }
    162. return Release();
    163. }
    164. //描述符错误事件处理
    165. void HandlerError() { HandlerClose(); }
    166. //描述符触发任意事件
    167. void HandlerEvent() {
    168. //1.判断是否需要刷新活跃度
    169. if(_enable_active_release == true) { _loop->TimerRefresh(_conn_id);}
    170. //2.调用组件使用者的任意事件回调
    171. if(_event_callback) { return _event_callback(shared_from_this()); }
    172. }
    173. public:
    174. Connection(int connid,int socketfd,EventLoop* loop)
    175. :_conn_id(connid),_socketfd(socketfd),_enable_active_release(false),_loop(loop),
    176. _socket(socketfd),_channel(_loop,_socketfd),_status(CONNECTING)
    177. {
    178. //设置channel回调函数
    179. _channel.SetReadCallback(std::bind(&Connection::HandlerRead,this));
    180. _channel.SetWriteCallback(std::bind(&Connection::HandlerWrite,this));
    181. _channel.SetErrorCallback(std::bind(&Connection::HandlerError,this));
    182. _channel.SetCloseCallback(std::bind(&Connection::HandlerClose,this));
    183. _channel.SetEventCallback(std::bind(&Connection::HandlerEvent,this));
    184. }
    185. ~Connection() { DLOG("CONNECTION RELEASE :%p",this); }
    186. //成员变量的接口
    187. int Fd() { return _socketfd; }
    188. //获取连接Id
    189. int Id() { return _conn_id; }
    190. //设置上下文
    191. void SetContext(const Any& context) { _context = context; }
    192. //获取上下文信息
    193. Any* Context() { return &_context; }
    194. //判断当前是否是连接状态
    195. bool IsConnected() { return _status == CONNECTED; }
    196. //设置各种回调函数
    197. void SetConnectCallback(const ConnectCallback& cb) { _connect_callback = cb; }
    198. void SetMessageCallback(const MessageCallback& cb) { _msg_callback = cb; }
    199. void SetClosedCallback(const ClosedCallback& cb) { _closed_callback = cb; }
    200. void SetEventCallback(const AnyEventCallback& cb) { _event_callback = cb; }
    201. void SetSvrCallback(const ClosedCallback& cb) { _server_event_callback = cb; }
    202. //发送数据,将数据放到发送缓冲区,启动写事件监控
    203. void Send(const char* data,size_t len)
    204. {
    205. //这里外面传过来的是一个临时对象,有可能会销毁,所以保存一份变量保证安全性
    206. Buffer buf;
    207. buf.WriteAndPush(data,len);
    208. _loop->RunInLoop(std::bind(&Connection::SendInLoop,this,std::move(buf)));
    209. }
    210. //这是提供给组件使用的关闭操作,但是并不是实际的关闭,需要内部进行处理
    211. void ShutDown()
    212. {
    213. _loop->RunInLoop(std::bind(&Connection::ShutDownInLoop,this));
    214. }
    215. //实际关闭连接的接口,销毁时应该放入任务池中,在执行完任务之后才销毁,否则可能会导致其他任务处理时,连接被释放导致内存访问错误
    216. void Release()
    217. {
    218. _loop->QueueInLoop(std::bind(&Connection::ReleaseInLoop,this));
    219. }
    220. //连接建立成功之后设置channel,启动读事件监控,调用_connect_callback
    221. void Established()
    222. {
    223. _loop->RunInLoop(std::bind(&Connection::EstablishedInLoop,this));
    224. }
    225. //启动非活跃连接销毁
    226. void SetActiveRelease(int sec)
    227. {
    228. _loop->RunInLoop(std::bind(&Connection::SetActiveReleaseInloop,this,sec));
    229. }
    230. //取消非活跃连接销毁
    231. void CancelActiveRelease()
    232. {
    233. _loop->RunInLoop(std::bind(&Connection::CancelActiveReleaseInLoop,this));
    234. }
    235. //协议切换
    236. void Upgrade(const Any& context,const ConnectCallback& conn,const MessageCallback& msg,
    237. const ClosedCallback& closed,const AnyEventCallback& event)
    238. {
    239. //因为协议切换是需要放在线程中,并且应该立即执行,否则用户切换协议之前的数据处理就没有意义了
    240. _loop->AssertInLoop();
    241. _loop->RunInLoop(std::bind(&Connection::UpgradeInLoop,this,context,conn,msg,closed,event));
    242. }
    243. };

    其中模块之间的关系图帮助理解:

    监听描述符管理Acceptor类实现

    Acceptor模块只进行监听连接的管理,有事件新的连接到来就调用对应的回调函数进行处理即可

    类成员设计

    1.创建一个监听套接字

    2.需要一个新连接处理的回调函数

    类功能

    1.创建一个监听套接字

    2.启动读事件监控

    3.事件触发,获取新连接

    4.调用对应的回调函数

    对于新连接如何处理应该是服务器模块来进行管理的

    1. class Acceptor
    2. {
    3. private:
    4. EventLoop* _loop;
    5. Socket _socket;
    6. Channel _channel;
    7. using Accept_callback = std::function<void(int)>;
    8. Accept_callback _accept_cb;
    9. private:
    10. int CreateSocket(int port)
    11. {
    12. bool ret = _socket.CreateSerber(port);
    13. assert(ret);
    14. return _socket.Fd();
    15. }
    16. void HandlerRead()
    17. {
    18. int newfd = _socket.Accept();
    19. if(newfd < 0) return;
    20. if(_accept_cb) _accept_cb(newfd);
    21. }
    22. public:
    23. //构造函数不能立刻启用可读事件监控,否则这里有可能导致回调函数还没有设置,此时如果立刻有连接到来,会导致newfd没有得到处理,最终资源泄露
    24. Acceptor(EventLoop* loop,int port):_loop(loop),_socket(CreateSocket(port)),_channel(_loop,_socket.Fd())
    25. {
    26. _channel.SetReadCallback(std::bind(&Acceptor::HandlerRead,this));
    27. }
    28. void SetAcceptCallback(const Accept_callback& cb) { _accept_cb = cb; }
    29. //开始监听,启动读事件监控
    30. void Listen()
    31. {
    32. _channel.SetRead();
    33. }
    34. };

    模块之前的关系图:

    TcpServer模块

    对所有模块的整合,通过该模块实例化对象,可以非常简单的完成一个服务器搭建

    类成员设计(管理):

    1.Acceptor对象,创建一个监听套接字

    2.EventLoop对象,baseloop对象,实现对监听套接字的事件监控

    3.通过hash表来实现对新连接的管理

    4.LoopThreadPool对象,创建loop线程池,对新连接进行事件监控以及处理

    类功能

    1.设置从属线程池数量

    2.启动服务器

    3.设置各种回调函数(连接建立完成,消息,关闭,任意)

    4.是否启动非活跃连接超时销毁功能

    5.添加定时任务功能

    1. class TcpServer
    2. {
    3. private:
    4. uint64_t _next_id;
    5. int _timeout; //销毁时间
    6. bool _enable_active_release; //是否其实非活跃超时销毁
    7. EventLoop _base_loop; //主线程
    8. Acceptor _acceptor; //监听套接字管理的对象
    9. LoopThreadPool _pool; //从属线程
    10. std::unordered_map<uint64_t,PtrConnection> _conns; //管理连接
    11. using ConnectCallback = std::function<void(const PtrConnection&)>;
    12. using MessageCallback = std::function<void(const PtrConnection&,Buffer*)>;
    13. using ClosedCallback = std::function<void(const PtrConnection&)>;
    14. using AnyEventCallback = std::function<void(const PtrConnection&)>;
    15. using Functor = std::function<void()>;
    16. ConnectCallback _connect_callback;
    17. MessageCallback _msg_callback;
    18. ClosedCallback _closed_callback;
    19. AnyEventCallback _event_callback;
    20. private:
    21. //新连接构造一个connection管理
    22. void NewConnection(int newfd)
    23. {
    24. ++_next_id;
    25. PtrConnection conn(new Connection(_next_id,newfd,_pool.NextLoop()));
    26. conn->SetConnectCallback(_connect_callback);
    27. conn->SetMessageCallback(_msg_callback);
    28. conn->SetClosedCallback(_closed_callback);
    29. conn->SetEventCallback(_event_callback);
    30. conn->SetSvrCallback(std::bind(&TcpServer::RemoveConnection,this,std::placeholders::_1));
    31. if(_enable_active_release) conn->SetActiveRelease(_timeout);
    32. conn->Established();
    33. _conns.insert(std::make_pair(_next_id,conn));
    34. }
    35. //关闭时调用,去除管理Connection
    36. void RemoveConnection(const PtrConnection& conn)
    37. {
    38. _base_loop.RunInLoop(std::bind(&TcpServer::RemoveConnectionInLoop,this,conn));
    39. }
    40. void RemoveConnectionInLoop(const PtrConnection& conn)
    41. {
    42. int id = conn->Id();
    43. _conns.erase(id);
    44. }
    45. void RunAfterInLoop(const Functor& func,int delay)
    46. {
    47. _next_id++;
    48. _base_loop.TimerAdd(_next_id,delay,func);
    49. }
    50. public:
    51. TcpServer(int port):_next_id(0),_enable_active_release(false),_acceptor(&_base_loop,port),_pool(&_base_loop)
    52. {
    53. _acceptor.SetAcceptCallback(std::bind(&TcpServer::NewConnection,this,std::placeholders::_1));
    54. _acceptor.Listen(); //将套接字挂到loop中
    55. }
    56. void SetConnectCallback(const ConnectCallback& cb) { _connect_callback = cb; }
    57. void SetMessageCallback(const MessageCallback& cb) { _msg_callback = cb; }
    58. void SetClosedCallback(const ClosedCallback& cb) { _closed_callback = cb; }
    59. void SetEventCallback(const AnyEventCallback& cb) { _event_callback = cb; }
    60. //设置线程数量
    61. void SetThreadCount(int count) { _pool.SetThreadCount(count); }
    62. //设置非活跃超时销毁
    63. void SetActiveRelease(int timeout) { _timeout = timeout; _enable_active_release = true; }
    64. //添加一个任务
    65. void RunAfter(const Functor& func,int delay)
    66. {
    67. _base_loop.RunInLoop(std::bind(&TcpServer::RunAfterInLoop,this,func,delay));
    68. }
    69. void Start() { _pool.Create(); _base_loop.Start();}
    70. };

    HTTP协议模块实现

    Util⼯具类实现

    工具类中主要包含我们后面可能使用到的函数,所以统一封装成static成员

    成员函数如下

    1.对字符串进行分割 ->对url进行解析的时候需要根据特殊字符进行分割

    2.读取文件中的内容 ->当用户需要获取静态资源页面的时候

    3.向文件中写入数据 ->当用户需要传输大文件的时候,需要把外面的文件写入到服务器中

    4.url编码 -> 在网络传输过程中需要对特殊字符进行编码

    5.url解码 -> 有编码就一定要在特定时候进行解码

    6.通过响应状态码获取响应信息 -> 在进行响应的时候需要响应状态码,已经对应的响应描述

    7.通过文件名获取mime ->在前端展示的时候需要知道这是一个什么类型的文件,响应头部中

    需要进行设置 

    (6.7需要提前准备好hash表把所以的kv都放入到hash中,方便查找,同时应该在全局初始化,这样的话就可以防止每一次都开辟空间,造成资源的浪费)

    8.判断一个文件是否是目录 ->在文件上传的时候需要保证是文件而不是目录

    9.判断一个文件是否是目录 ->通过Linux中提供的stat函数进行判断

    10.判断资源路径的合法性 ->访问的资源如果在web根目录的上级就会有问题,所以必须保证资源合法性 (通过访问的层数进行判断,如果访问的层数小于0,那么就是非法)

    1. //为了提高效率,把映射关系定义成全局
    2. std::unordered_map<int, std::string> _statu_msg = {
    3. {100, "Continue"},
    4. {101, "Switching Protocol"},
    5. {102, "Processing"},
    6. {103, "Early Hints"},
    7. {200, "OK"},
    8. {201, "Created"},
    9. {202, "Accepted"},
    10. {203, "Non-Authoritative Information"},
    11. {204, "No Content"},
    12. {205, "Reset Content"},
    13. {206, "Partial Content"},
    14. {207, "Multi-Status"},
    15. {208, "Already Reported"},
    16. {226, "IM Used"},
    17. {300, "Multiple Choice"},
    18. {301, "Moved Permanently"},
    19. {302, "Found"},
    20. {303, "See Other"},
    21. {304, "Not Modified"},
    22. {305, "Use Proxy"},
    23. {306, "unused"},
    24. {307, "Temporary Redirect"},
    25. {308, "Permanent Redirect"},
    26. {400, "Bad Request"},
    27. {401, "Unauthorized"},
    28. {402, "Payment Required"},
    29. {403, "Forbidden"},
    30. {404, "Not Found"},
    31. {405, "Method Not Allowed"},
    32. {406, "Not Acceptable"},
    33. {407, "Proxy Authentication Required"},
    34. {408, "Request Timeout"},
    35. {409, "Conflict"},
    36. {410, "Gone"},
    37. {411, "Length Required"},
    38. {412, "Precondition Failed"},
    39. {413, "Payload Too Large"},
    40. {414, "URI Too Long"},
    41. {415, "Unsupported Media Type"},
    42. {416, "Range Not Satisfiable"},
    43. {417, "Expectation Failed"},
    44. {418, "I'm a teapot"},
    45. {421, "Misdirected Request"},
    46. {422, "Unprocessable Entity"},
    47. {423, "Locked"},
    48. {424, "Failed Dependency"},
    49. {425, "Too Early"},
    50. {426, "Upgrade Required"},
    51. {428, "Precondition Required"},
    52. {429, "Too Many Requests"},
    53. {431, "Request Header Fields Too Large"},
    54. {451, "Unavailable For Legal Reasons"},
    55. {501, "Not Implemented"},
    56. {502, "Bad Gateway"},
    57. {503, "Service Unavailable"},
    58. {504, "Gateway Timeout"},
    59. {505, "HTTP Version Not Supported"},
    60. {506, "Variant Also Negotiates"},
    61. {507, "Insufficient Storage"},
    62. {508, "Loop Detected"},
    63. {510, "Not Extended"},
    64. {511, "Network Authentication Required"}
    65. };
    66. std::unordered_map<std::string, std::string> mim_msg = {
    67. {".aac", "audio/aac"},
    68. {".abw", "application/x-abiword"},
    69. {".arc", "application/x-freearc"},
    70. {".avi", "video/x-msvideo"},
    71. {".azw", "application/vnd.amazon.ebook"},
    72. {".bin", "application/octet-stream"},
    73. {".bmp", "image/bmp"},
    74. {".bz", "application/x-bzip"},
    75. {".bz2", "application/x-bzip2"},
    76. {".csh", "application/x-csh"},
    77. {".css", "text/css"},
    78. {".csv", "text/csv"},
    79. {".doc", "application/msword"},
    80. {".docx", "application/vnd.openxmlformats-officedocument.wordprocessingml.document"},
    81. {".eot", "application/vnd.ms-fontobject"},
    82. {".epub", "application/epub+zip"},
    83. {".gif", "image/gif"},
    84. {".htm", "text/html"},
    85. {".html", "text/html"},
    86. {".ico", "image/vnd.microsoft.icon"},
    87. {".ics", "text/calendar"},
    88. {".jar", "application/java-archive"},
    89. {".jpeg", "image/jpeg"},
    90. {".jpg", "image/jpeg"},
    91. {".js", "text/javascript"},
    92. {".json", "application/json"},
    93. {".jsonld", "application/ld+json"},
    94. {".mid", "audio/midi"},
    95. {".midi", "audio/x-midi"},
    96. {".mjs", "text/javascript"},
    97. {".mp3", "audio/mpeg"},
    98. {".mpeg", "video/mpeg"},
    99. {".mpkg", "application/vnd.apple.installer+xml"},
    100. {".odp", "application/vnd.oasis.opendocument.presentation"},
    101. {".ods", "application/vnd.oasis.opendocument.spreadsheet"},
    102. {".odt", "application/vnd.oasis.opendocument.text"},
    103. {".oga", "audio/ogg"},
    104. {".ogv", "video/ogg"},
    105. {".ogx", "application/ogg"},
    106. {".otf", "font/otf"},
    107. {".png", "image/png"},
    108. {".pdf", "application/pdf"},
    109. {".ppt", "application/vnd.ms-powerpoint"},
    110. {".pptx", "application/vnd.openxmlformats-officedocument.presentationml.presentation"},
    111. {".rar", "application/x-rar-compressed"},
    112. {".rtf", "application/rtf"},
    113. {".sh", "application/x-sh"},
    114. {".svg", "image/svg+xml"},
    115. {".swf", "application/x-shockwave-flash"},
    116. {".tar", "application/x-tar"},
    117. {".tif", "image/tiff"},
    118. {".tiff", "image/tiff"},
    119. {".ttf", "font/ttf"},
    120. {".txt", "text/plain"},
    121. {".vsd", "application/vnd.visio"},
    122. {".wav", "audio/wav"},
    123. {".weba", "audio/webm"},
    124. {".webm", "video/webm"},
    125. {".webp", "image/webp"},
    126. {".woff", "font/woff"},
    127. {".woff2", "font/woff2"},
    128. {".xhtml", "application/xhtml+xml"},
    129. {".xls", "application/vnd.ms-excel"},
    130. {".xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"},
    131. {".xml", "application/xml"},
    132. {".xul", "application/vnd.mozilla.xul+xml"},
    133. {".zip", "application/zip"},
    134. {".3gp", "video/3gpp"},
    135. {".3g2", "video/3gpp2"},
    136. {".7z", "application/x-7z-compressed"}
    137. };
    138. class Util
    139. {
    140. public:
    141. //分割字符串
    142. static size_t Spilt(const std::string& src,const std::string& sep,std::vector<std::string>* arr)
    143. {
    144. //"abc,sdf,ijk" ","
    145. int offset = 0; //偏移量
    146. while(offset < src.size())
    147. {
    148. int pos = src.find(sep,offset);
    149. if(pos == std::string::npos)
    150. {
    151. //没找到,说明后面的offset后面的字符都是结果
    152. arr->push_back(src.substr(offset,pos-offset));
    153. return arr->size();
    154. }
    155. //这里可能有多个重复sep
    156. if(offset == pos){
    157. offset = pos + sep.size();
    158. continue;
    159. }
    160. //收集结果
    161. arr->push_back(src.substr(offset,pos-offset));
    162. offset = pos + sep.size();
    163. }
    164. return arr->size();
    165. }
    166. //读取文件内容
    167. static bool ReadFile(const std::string& filename,std::string* buf)
    168. {
    169. std::ifstream ifs(filename,std::ios::binary);
    170. if(!ifs.is_open())
    171. {
    172. DLOG("OPEN %s FAILED!",filename);
    173. return false;
    174. }
    175. int fsize = 0;
    176. //先把文件指针偏移到末尾,获取到文件的大小
    177. ifs.seekg(0,ifs.end);
    178. fsize = ifs.tellg();
    179. //回到起始,开始读取文件
    180. ifs.seekg(0,ifs.beg);
    181. buf->resize(fsize);
    182. //读取文件
    183. ifs.read(&(*buf)[0],fsize);
    184. if(!ifs.good())
    185. {
    186. DLOG("READ FILE %s FAILED!",filename);
    187. ifs.close();
    188. return false;
    189. }
    190. //记得关闭文件,防止资源泄露
    191. ifs.close();
    192. return true;
    193. }
    194. //写入文件
    195. static bool WriteFile(const std::string& filename,const std::string& buf)
    196. {
    197. std::ofstream ofs(filename,std::ios::binary | std::ios::trunc);
    198. if(!ofs.is_open())
    199. {
    200. DLOG("OPEN %s FAILED!",filename.c_str());
    201. return false;
    202. }
    203. //向文件写入
    204. ofs.write(buf.c_str(),buf.size());
    205. if(!ofs.good())
    206. {
    207. DLOG("WRITEFILE %s FAILED!",filename.c_str());
    208. ofs.close();
    209. return false;
    210. }
    211. ofs.close();
    212. return true;
    213. }
    214. //url编码
    215. static std::string UrlEnCode(const std::string& url,bool convert_space_to_plus)
    216. {
    217. //". - _ ~"以及数字字母字符采用绝对不编码,convert_space_to_plus -> 是否启用空格转+
    218. std::string ret;
    219. for(auto ch: url)
    220. {
    221. if(ch == '.' || ch == '-' || ch == '_' || ch == '~' || isalnum(ch))
    222. {
    223. ret += ch;
    224. continue;
    225. }
    226. //空格转+
    227. if(convert_space_to_plus && ch == ' ')
    228. {
    229. ret += '+';
    230. continue;
    231. }
    232. //其余都是需要转换的字符
    233. char tmp[4] = {0};
    234. snprintf(tmp,4,"%%%02X",ch);
    235. ret += tmp;
    236. }
    237. return ret;
    238. }
    239. static char HexToI(char ch)
    240. {
    241. if(ch >= '0' && ch <= '9')
    242. {
    243. return ch - '0';
    244. }
    245. else if(ch >= 'a' && ch <= 'z')
    246. {
    247. return ch - 'a' + 10;
    248. }
    249. else if(ch >= 'A' && ch <= 'Z')
    250. {
    251. return ch - 'A' + 10;
    252. }
    253. return -1; //错误
    254. }
    255. //url解码
    256. static std::string UrlDeCode(const std::string& url,bool convert_plus_to_space)
    257. {
    258. std::string ret;
    259. for(int i = 0;i<url.size();++i)
    260. {
    261. //判断convert_plus_to_space条件是否满足
    262. if(convert_plus_to_space && url[i] == '+')
    263. {
    264. ret += ' ';
    265. continue;
    266. }
    267. //遇到%后面的数,就把第一个数转换成16进制第二个数相加
    268. if(url[i] == '%')
    269. {
    270. char v1 = HexToI(url[i+1]);
    271. char v2 = HexToI(url[i+2]);
    272. char res = (v1 << 4) + v2;
    273. ret += res;
    274. i += 2;
    275. continue;
    276. }
    277. //其他情况直接放入结果集即可
    278. ret += url[i];
    279. }
    280. return ret;
    281. }
    282. //通过响应状态码获得响应信息
    283. static std::string StatuDesc(int statu)
    284. {
    285. auto it = _statu_msg.find(statu);
    286. if (it != _statu_msg.end()) {
    287. return it->second;
    288. }
    289. return "Unknow";
    290. }
    291. static std::string GetMime(const std::string& filename)
    292. {
    293. //a.txt 找到文件后缀
    294. size_t pos = filename.find_last_of('.');
    295. if(pos == std::string::npos)
    296. {
    297. //没找到就是二进制流数据
    298. return "application/octet-stream";
    299. }
    300. std::string tmp = filename.substr(pos);
    301. auto it = mim_msg.find(tmp);
    302. if(it == mim_msg.end())
    303. {
    304. return "application/octet-stream";
    305. }
    306. return it->second;
    307. }
    308. //判断一个文件是否是目录
    309. static bool IsDir(const std::string& filename)
    310. {
    311. struct stat st;
    312. int ret = stat(filename.c_str(),&st);
    313. if(ret < 0)
    314. {
    315. return false;
    316. }
    317. return S_ISDIR(st.st_mode);
    318. }
    319. //判断一个文件是否是普通文件
    320. static bool IsRegular(const std::string& filename)
    321. {
    322. struct stat st;
    323. int ret = stat(filename.c_str(),&st);
    324. if(ret < 0)
    325. {
    326. return false;
    327. }
    328. return S_ISREG(st.st_mode);
    329. }
    330. //资源路径的有效性判断
    331. static bool IsValPath(const std::string& path)
    332. {
    333. //根据层数来判断当前路径是否在web根目录下
    334. //"/index.html" "/../index.html"
    335. std::vector<std::string> vs;
    336. Spilt(path,"/",&vs);
    337. int level = 0;
    338. for(auto& str:vs)
    339. {
    340. if(str == "..")
    341. {
    342. --level;
    343. if(level < 0) return false; //任意一层小于0都是有问题的
    344. continue;
    345. }
    346. ++level;
    347. }
    348. return true;
    349. }
    350. };

    HttpRequest请求类实现

    该模块主要存储HTTP请求信息要素,提供简单的功能性接口

    请求要素如下:

    HTTP常见header

    Content-Type:数据类型(text/html等)。
    Content-Length:正文的长度。
    Host:客户端告知服务器,所请求的资源是在哪个主机的哪个端口上。
    User-Agent:声明用户的操作系统和浏览器的版本信息。
    Referer:当前页面是哪个页面跳转过来的。
    Location:搭配3XX状态码使用,告诉客户端接下来要去哪里访问。
    Cookie:用户在客户端存储少量信息,用于实现会话(session)的功能。

    类成员设计

    1.请求方法 

    2.资源路径url

    3.协议版本

    4.请求正文

    5.正则提取字符串 ->我们需要使用使用之前提供过的正则库已经正则表达式来对请求信息进行匹配,从而简化代码的编写

    6.头部字段 ->把请求中的头部字段以kv的方式存储在hash表中,方便查询

    7.查询字符串 ->在url中可能有查询字符串,同样以kv的方式存储在hash中

    类功能性接口

    1.将成员变量设置为公有,这样方便直接访问

    2.提供头部字段以及查询字符串的插入以及查询功能

    3.获取正文长度

    4.判断是否为长连接,如果是短连接就需要处理完之后直接关闭连接(Connection头部字段  Connection: close/keep-alive)

    1. class HttpRequest
    2. {
    3. public:
    4. std::string _method; //请求方法
    5. std::string _path; //资源路径
    6. std::string _version;//协议版本
    7. std::string _body; //请求正文
    8. std::smatch _matches; //正则提取查询字符串
    9. std::unordered_map<std::string,std::string> _headers; //头部字段
    10. std::unordered_map<std::string,std::string> _parame; //查询字符串
    11. public:
    12. HttpRequest():_version("HTTP/1.1"){}
    13. void ReSet()
    14. {
    15. _method.clear();
    16. _path.clear();
    17. _version.clear();
    18. _body.clear();
    19. std::smatch tmp; //用于交换清除
    20. _matches.swap(tmp);
    21. _headers.clear();
    22. _parame.clear();
    23. }
    24. //插入头部字段
    25. void SetHeader(const std::string& key,const std::string& val)
    26. {
    27. _headers.insert(std::make_pair(key,val));
    28. }
    29. //获取指定头部的值
    30. std::string GetHeader(const std::string& key)
    31. {
    32. auto it = _headers.find(key);
    33. if(it != _headers.end())
    34. {
    35. return it->second;
    36. }
    37. return "";
    38. }
    39. //查询指定头部的值是否存在
    40. bool HasHeader(const std::string& key)
    41. {
    42. auto it = _headers.find(key);
    43. if(it != _headers.end())
    44. {
    45. return true;
    46. }
    47. return false;
    48. }
    49. //插入查询字符串
    50. void SetParam(const std::string& key,const std::string& val)
    51. {
    52. _parame.insert(std::make_pair(key,val));
    53. }
    54. //获取指定查询字符串
    55. std::string GetParam(const std::string& key)
    56. {
    57. auto it = _parame.find(key);
    58. if(it != _parame.end())
    59. {
    60. return it->second;
    61. }
    62. return "";
    63. }
    64. //判断是否存在某个查询字符串
    65. bool HasParam(const std::string& key)
    66. {
    67. auto it = _parame.find(key);
    68. if(it != _parame.end())
    69. {
    70. return true;
    71. }
    72. return false;
    73. }
    74. //获取正文长度
    75. size_t Content_Length()
    76. {
    77. auto it = _headers.find("Content-Length");
    78. if(it == _headers.end())
    79. {
    80. return 0;
    81. }
    82. return std::stol(it->second);
    83. }
    84. //判断是否为短连接
    85. bool Close()
    86. {
    87. if(HasHeader("Connection") && GetHeader("Connection") == "keep-alive")
    88. {
    89. return false;
    90. }
    91. return true;
    92. }
    93. };

    HttpResponse响应类实现

    类功能:存储HTTP响应信息要素,提供简单的功能性接口

    响应信息要素

    类成员设计

    1.响应状态 ->我们需要知道当前处理是否有问题,所以需要给用户返回响应状态

    2.是否重定向 -> 如果该url需要重定向,那么我们就需要把重定向的url保存起来

    3.响应正文

    4.头部字段 ->设置头部kv到hash中,方便查询

    类功能性接口

    1.为了方便成员的访问,将成员设置成公有

    2.头部字段的新增,查询,获取

    3.正文的设置

    4.重定向的设置

    5.长短连接的判断

    1. class HttpResponse
    2. {
    3. public:
    4. int _statu; //响应状态
    5. bool _redirect_flag; //是否重定向
    6. std::string _body; //响应正文
    7. std::string _redirect_url; //重定向url
    8. std::unordered_map<std::string,std::string> _headers; //头部字段
    9. public:
    10. HttpResponse():_statu(200),_redirect_flag(false){}
    11. HttpResponse(int statu):_statu(statu),_redirect_flag(false){}
    12. void ReSet()
    13. {
    14. _statu = 200;
    15. _redirect_flag = false;
    16. _body.clear();
    17. _redirect_url.clear();
    18. _headers.clear();
    19. }
    20. //插入头部字段
    21. void SetHeader(const std::string& key,const std::string& val)
    22. {
    23. _headers.insert(std::make_pair(key,val));
    24. }
    25. //获取指定头部的值
    26. std::string GetHeader(const std::string& key)
    27. {
    28. auto it = _headers.find(key);
    29. if(it != _headers.end())
    30. {
    31. return it->second;
    32. }
    33. return "";
    34. }
    35. //查询指定头部的值是否存在
    36. bool HasHeader(const std::string& key)
    37. {
    38. auto it = _headers.find(key);
    39. if(it != _headers.end())
    40. {
    41. return true;
    42. }
    43. return false;
    44. }
    45. //设置响应正文
    46. void SetContent(const std::string& body,const std::string& type)
    47. {
    48. _body = body;
    49. SetHeader("Content-Length",type);
    50. }
    51. //设置重定向
    52. void SetRedirect(const std::string& url,int statu = 302)
    53. {
    54. _statu = statu;
    55. _redirect_flag = true;
    56. _redirect_url = url;
    57. }
    58. //判断是否为短连接
    59. bool Close()
    60. {
    61. if(HasHeader("Connection") && GetHeader("Connection") == "keep-alive")
    62. {
    63. return false;
    64. }
    65. return true;
    66. }
    67. };

    HttpContent上下文类实现

    该类是处理的Http请求响应的上下文模块,当一个请求没有一次性发送过来的时候,那么处理的时候就不能只处理当次的,而是要保证处理一个完整的请求,所以需要上下文模块来记录上一次处理,以便处理一个完整的请求

    类成员设计

    1.我们需要一个响应状态码,因为每一次处理的结果都可能是不一样的响应状态

    2.需要记录当前解析到的状态 ,方便下一次解析

    3.解析得到的信息需要放到HttpRequest中

    类功能

    这里之后一个函数,就是进行解析工作,但是其中分成几个子函数:

    1.接收请求行 (从缓冲区中取出数据,并更新解析状态)

    2.解析请求行  (通过正则表达式进行处理,并把查询字符串放入到Request)

    3.接收头部  (将Buffer中数据进行分割)

    4.解析头部  (以kv的形式存储在Request中)

    5.接收正文   (通过Content-Length获取正文长度,并根据需求来获取正文)

    需要注意的是:
    任意一个状态有问题,都需要立即停止剩下的工作,应该那是没有意义的。而且上一个工作没有完成是不能进行下一步进行处理的。

    1. typedef enum {
    2. RECV_HTTP_ERROR,
    3. RECV_HTTP_LINE,
    4. RECV_HTTP_HEAD,
    5. RECV_HTTP_BODY,
    6. RECV_HTTP_OVER,
    7. }HttpRecvStatu;
    8. #define MAX_LINE 8192
    9. class HttpContent
    10. {
    11. private:
    12. int _resp_statu; //响应状态码
    13. HttpRecvStatu _recv_statu; //当前解析阶段
    14. HttpRequest _request; //已经解析得到的信息
    15. private:
    16. bool ParseLine(const std::string& line)
    17. {
    18. std::smatch matches;
    19. std::regex e("(GET|POST|HEAD|DELETE|PUT) ([^?]*)(?:\\?(.*))? (HTTP/1\\.[01])(?:\r|\r\n)?",std::regex::icase); //匹配请求方法的正则表达式
    20. bool ret = regex_match(line,matches,e);
    21. if(ret == false) return false;
    22. //结果如下:
    23. //0 : GET /hello/login?user=xiaoming&passwd=123456 HTTP/1.1
    24. //1 : GET
    25. //2 : /hello/login
    26. //3 : user=xiaoming&passwd=123456
    27. //4 : HTTP/1.1
    28. _request._method = matches[1];
    29. //这里统一转换为大写,兼容性更强
    30. std::transform(_request._method.begin(),_request._method.end(),_request._method.begin(),::toupper);
    31. //这里资源路径需要解码且不需要+转空格
    32. _request._path = Util::UrlDeCode(matches[2],false);
    33. _request._version = matches[4];
    34. //这个查询字符串要特殊处理一下
    35. std::string param = matches[3];
    36. std::vector<std::string> vs;
    37. Util::Spilt(param,"&",&vs);
    38. //通过等号进行继续分割,得到key val
    39. for(auto& str:vs)
    40. {
    41. size_t pos = str.find('=');
    42. if(pos == std::string::npos)
    43. {
    44. //没找到,认为是有问题的,返回错误
    45. _resp_statu = 400; //BAD REQUEST
    46. _recv_statu = RECV_HTTP_ERROR;
    47. return false;
    48. }
    49. //这里得到的key和val也是需要进行解码的且需要空格转+
    50. std::string key = Util::UrlDeCode(str.substr(0,pos),true);
    51. std::string val = Util::UrlDeCode(str.substr(pos+1),true);
    52. _request.SetParam(key,val);
    53. }
    54. return true;
    55. }
    56. bool RecvLine(Buffer* buf)
    57. {
    58. if(_recv_statu != RECV_HTTP_LINE) return false;
    59. //接收一行
    60. std::string line = buf->GetLineAndPop();
    61. if(line.size() == 0)
    62. {
    63. //没有一行,有可能是一行的数据太大了,也可能是不够一行的数据
    64. if(buf->ReadAbleSize() > MAX_LINE)
    65. {
    66. //{414, "URI Too Long"}
    67. _resp_statu = 414;
    68. _recv_statu = RECV_HTTP_ERROR;
    69. return false;
    70. }
    71. return true;
    72. }
    73. if(line.size() > MAX_LINE)
    74. {
    75. _resp_statu = 414;
    76. _recv_statu = RECV_HTTP_ERROR;
    77. return false;
    78. }
    79. bool ret = ParseLine(line);
    80. if(ret == false)
    81. {
    82. return false;
    83. }
    84. //首行处理完毕,进入头部阶段
    85. _recv_statu = RECV_HTTP_HEAD;
    86. return true;
    87. }
    88. bool RecvHead(Buffer* buf)
    89. {
    90. if(_recv_statu != RECV_HTTP_HEAD) return false;
    91. //这里是有多行数据,而且数据都是以key: val的形式
    92. while(true)
    93. {
    94. std::string line = buf->GetLineAndPop();
    95. if(line.size() == 0)
    96. {
    97. //没有一行,有可能是一行的数据太大了,也可能是不够一行的数据
    98. if(buf->ReadAbleSize() > MAX_LINE)
    99. {
    100. //{414, "URI Too Long"}
    101. _resp_statu = 414;
    102. _recv_statu = RECV_HTTP_ERROR;
    103. return false;
    104. }
    105. return true;
    106. }
    107. if(line.size() > MAX_LINE)
    108. {
    109. _resp_statu = 414;
    110. _recv_statu = RECV_HTTP_ERROR;
    111. return false;
    112. }
    113. if(line == "\n" || line == "\r\n") break; //这里说明头部已经解析完成
    114. bool ret = ParseHead(line);
    115. if(ret == false) return false;
    116. }
    117. //头部解析完成,进行正文解析
    118. _recv_statu = RECV_HTTP_BODY;
    119. return true;
    120. }
    121. bool ParseHead(std::string& line)
    122. {
    123. //去掉换行符
    124. if(line.back() == '\n') line.pop_back();
    125. if(line.back() == '\r') line.pop_back();
    126. //这里都是key: val的形式
    127. size_t pos = line.find(": ");
    128. if(pos == std::string::npos)
    129. {
    130. //解析失败
    131. _resp_statu = 400;
    132. _recv_statu = RECV_HTTP_ERROR;
    133. return false;
    134. }
    135. std::string key = line.substr(0,pos);
    136. std::string val = line.substr(pos+2);
    137. _request.SetHeader(key,val);
    138. return true;
    139. }
    140. bool RecvBody(Buffer* buf)
    141. {
    142. if(_recv_statu != RECV_HTTP_BODY) return false;
    143. //正文数据大小为0,则直接返回即可
    144. size_t content_len = _request.Content_Length();
    145. if(content_len == 0){
    146. //解析完毕,直接返回
    147. _recv_statu = RECV_HTTP_OVER;
    148. return true;
    149. }
    150. //获得剩下应该获取的正文长度
    151. size_t real_len = content_len - _request._body.size();
    152. //缓冲区的数据大于剩余正文长度,取出当前所需
    153. if(buf->ReadAbleSize() >= real_len)
    154. {
    155. _request._body.append(buf->ReadPosition(),real_len);
    156. //这里不要忘了向后移动
    157. buf->MoveReadOffSet(real_len);
    158. _recv_statu = RECV_HTTP_OVER;
    159. return true;
    160. }
    161. //这里的缓冲区数据不够,则把数据放到正文中,然后返回即可
    162. _request._body.append(buf->ReadPosition(),buf->ReadAbleSize());
    163. //这里不要忘了向后移动
    164. buf->MoveReadOffSet(buf->ReadAbleSize());
    165. return true;
    166. }
    167. public:
    168. HttpContent():_resp_statu(200),_recv_statu(RECV_HTTP_LINE){};
    169. void Reset()
    170. {
    171. _resp_statu = 200;
    172. _recv_statu = RECV_HTTP_LINE;
    173. _request.ReSet();
    174. }
    175. int Statu() { return _resp_statu; }
    176. HttpRecvStatu RecvStatu() { return _recv_statu; }
    177. HttpRequest& RecvReq() { return _request; }
    178. //接收并解析请求
    179. void HttpRecvReqest(Buffer* buffer)
    180. {
    181. //根据不同的状态进入不同的函数,这里一个函数处理完成之后会立刻进入下一个函数,完成全过程
    182. switch(_recv_statu){
    183. case RECV_HTTP_LINE: RecvLine(buffer);
    184. case RECV_HTTP_HEAD: RecvHead(buffer);
    185. case RECV_HTTP_BODY: RecvBody(buffer);
    186. }
    187. }
    188. };

    HttpServer服务器模块

    该类主要是实现HTTP服务器的搭建,可以让用户快速搭建使用服务器

    实现思想
    我们可以通过一个请求路由表来做统一的处理,这个表记录了那个请求对应哪一个方法,通过hash的方式,什么方法,怎么处理这个有用户自己决定,服务器只需要执行对应的函数即可。

    优势:用户只需要实现业务处理函数,处理好请求以及处理函数的映射关系,添加到服务器中;而服务器只需要接收数据,解析数据,查找路由表中的映射关系,同时执行对应的函数即可

    类成员设计

    1.各种方法的请求路由表 (GET/POST/PUT/DELETE)

    2.静态资源根目录 ->实现静态资源获取的处理

    3.高性能的TCP服务器 ->进行连接的IO处理

    服务器处理流程
    1.从socket接受数据,放入到接收缓冲区

    2.调用OnMessage回调函数进行处理

    3.对请求进行解析,得到一个HttpRequest,里面填写好所有的请求要素

    4.通过请求路由查找对应的方法进行处理

        4.1如果是静态资源的请求  -> 实体文件资源的请求

            将静态资源文件中的数据读取出来填到HttpResponse中

        4.2如果是功能性请求  -> 查找对应的路由函数,执行对应的方法,并填写HttpResponse

    5.业务处理完成之后,已经得到一个HttpResponse结构,组织http响应并进行发送

    类功能函数设计
    1.添加请求 ->建立函数映射表(GET/POST/PUT/DELETE)

    2.设置静态资源根目录 

    3.设置是否启动超时连接关闭

    4.设置线程池中线程的数量 

    5.启动服务器 (调用TcpServer中的start接口)

    1. class HttpServer
    2. {
    3. private:
    4. using Functor = std::function<void(const HttpRequest&,HttpResponse*)>;
    5. using Handler = std::vector<std::pair<std::regex,Functor>>;
    6. //各种方法的路由表,第一个是正则表达式,第二个是方法
    7. Handler _get_route;
    8. Handler _post_route;
    9. Handler _put_route;
    10. Handler _delete_route;
    11. std::string _base_dir; //静态资源根目录
    12. TcpServer _server;
    13. private:
    14. void ErrorHandler(const HttpRequest& req,HttpResponse* resp)
    15. {
    16. //1.组织一个错误的页面
    17. std::string body;
    18. body += "";
    19. body += "";
    20. body += "";
    21. body += "";
    22. body += "";
    23. body += "

      ";

    24. body += std::to_string(resp->_statu);
    25. body += " ";
    26. body += Util::StatuDesc(resp->_statu);
    27. body += "";
    28. body += "";
    29. body += "";
    30. //2.讲页面数据放入到resp中
    31. resp->SetContent(body,"text/html");
    32. }
    33. //根据http来组织httpResponse各要素
    34. void WriteResponse(const PtrConnection& conn,HttpRequest& req,HttpResponse& resp)
    35. {
    36. //1.完善头部字段
    37. if(req.Close() == true)
    38. {
    39. resp.SetHeader("Connection","Close");
    40. }
    41. else{
    42. resp.SetHeader("Connection","keep-alive");
    43. }
    44. if(!resp._body.empty() && !resp.HasHeader("Content-Length"))
    45. {
    46. resp.SetHeader("Content-Length",std::to_string(resp._body.size()));
    47. }
    48. if(!resp._body.empty() && !resp.HasHeader("Content-Type"))
    49. {
    50. resp.SetHeader("Content-Type","application/octet-stream");
    51. }
    52. //是否设置重定向
    53. if(resp._redirect_flag)
    54. {
    55. resp.SetHeader("Location",resp._redirect_url);
    56. }
    57. //2.填充resp信息 协议版本,响应状态码,响应状态码描述
    58. std::stringstream str;
    59. str << req._version <<" "<< std::to_string(resp._statu) << " " << Util::StatuDesc(resp._statu) << "\r\n";
    60. //响应头部
    61. for(auto& head : resp._headers)
    62. {
    63. str << head.first <<": " << head.second << "\r\n";
    64. }
    65. str << "\r\n";
    66. str << resp._body;
    67. //3.发送数据
    68. conn->Send(str.str().c_str(),str.str().size());
    69. }
    70. //静态资源处理
    71. bool IsFilerHandler(const HttpRequest& req)
    72. {
    73. //1.必须设置静态资源根目录
    74. if(req._path.empty())
    75. {
    76. return false;
    77. }
    78. //2.必须是GET或者是HEAD方法
    79. if(req._method != "GET" && req._method != "HEAD"){
    80. return false;
    81. }
    82. //3.必须是合法的资源路径
    83. if(!Util::IsValPath(req._path))
    84. {
    85. return false;
    86. }
    87. //4.资源必须存在
    88. //先添加上静态资源根目录,防止后面出错用另外一个变量来接收
    89. std::string req_path = _base_dir + req._path;
    90. // "/" 最后一个字符是/,那么添加上"index.html"
    91. if(req_path.back() == '/'){
    92. req_path += "index.html";
    93. }
    94. //判断是否是一个文件
    95. if(!Util::IsRegular(req_path)){
    96. return false;
    97. }
    98. return true;
    99. }
    100. bool FilerHandler(HttpRequest& req,HttpResponse* resp)
    101. {
    102. //先确定路径
    103. std::string req_path = _base_dir + req._path;
    104. if(req_path.back() == '/'){
    105. req_path += "index.html";
    106. }
    107. //把要获取的资源读取出来放到resp中的body中,并设置mime
    108. bool ret = Util::ReadFile(req_path,&(resp->_body));
    109. if(ret == false){
    110. return false;
    111. }
    112. //设置mime
    113. resp->SetHeader("Content-Type",Util::GetMime(req_path));
    114. return true;
    115. }
    116. //功能性资源处理
    117. void Dispatcher(HttpRequest& req,HttpResponse* resp,Handler& handlers)
    118. {
    119. //通过查看不同的方法的路由表,如果找到方法就执行对应的方法,如果没有找到就返回404
    120. for(auto& handler:handlers)
    121. {
    122. const std::regex& re = handler.first;
    123. const Functor& func = handler.second;
    124. bool ret = std::regex_match(req._path,req._matches,re);
    125. if(ret == false)
    126. {
    127. continue; //没有找到,继续查找
    128. }
    129. //找到就执行对应的函数
    130. return func(req,resp);
    131. }
    132. //返回404
    133. resp->_statu = 404;
    134. }
    135. void Route(HttpRequest& req,HttpResponse* resp)
    136. {
    137. //分成两种 -> 如果是静态资源请求则调用静态资源请求处理
    138. //如果是功能性资源请求则调用对应的函数处理
    139. if(IsFilerHandler(req))
    140. {
    141. FilerHandler(req,resp);
    142. }
    143. //功能性请求
    144. if(req._method == "GET" || req._method == "HEAD"){
    145. return Dispatcher(req,resp,_get_route);
    146. }
    147. else if(req._method == "POST"){
    148. return Dispatcher(req,resp,_post_route);
    149. }else if(req._method == "PUT"){
    150. return Dispatcher(req,resp,_put_route);
    151. }
    152. else if(req._method == "DELETE"){
    153. return Dispatcher(req,resp,_delete_route);
    154. }
    155. //不是上述方法,则返回405
    156. resp->_statu = 405; //Method Not Allowed
    157. }
    158. //设置上下文
    159. void Onconnect(const PtrConnection& conn)
    160. {
    161. conn->SetContext(HttpContent());
    162. DLOG("New Connection %p",conn.get());
    163. }
    164. //解析+处理
    165. void OnMessage(const PtrConnection& conn,Buffer* buf)
    166. {
    167. while(buf->ReadAbleSize() > 0)
    168. {
    169. //1.获取上下文
    170. HttpContent* content = conn->Context()->get<HttpContent>();
    171. //2.通过解析上下文得到httprequest
    172. content->HttpRecvReqest(buf);
    173. HttpRequest& req = content->RecvReq();
    174. HttpResponse resp(content->Statu());
    175. //这里数据解析可能出错
    176. if(content->Statu() >= 400)
    177. {
    178. //进行错误响应,关闭连接
    179. ErrorHandler(req,&resp); //填充一个错误的页面返回
    180. WriteResponse(conn,req,resp); //组织+响应
    181. //关闭连接之前应该重置一下上下文,防止状态一直处理错误的状态
    182. content->Reset();
    183. //这里出错了,为了提高服务器的效率,直接把缓冲区中的数据清空
    184. buf->MoveReadOffSet(buf->ReadAbleSize());
    185. conn->ShutDown(); //关闭连接
    186. return;
    187. }
    188. //这里有可能解析还没有完成,重新开始
    189. if(content->RecvStatu() != RECV_HTTP_OVER){
    190. return;
    191. }
    192. //3.请求路由
    193. Route(req,&resp);
    194. //4.组织httpresponse并发送
    195. WriteResponse(conn,req,resp);
    196. //5.重置上下文
    197. content->Reset();
    198. //6.是否是短连接,关闭
    199. if(resp.Close()) conn->ShutDown();
    200. }
    201. }
    202. public:
    203. HttpServer(int port,int timeout = DEFAULT_TIMEOUT):_server(port)
    204. {
    205. _server.SetActiveRelease(timeout);
    206. _server.SetConnectCallback(std::bind(&HttpServer::Onconnect,this,std::placeholders::_1));
    207. _server.SetMessageCallback(std::bind(&HttpServer::OnMessage,this,std::placeholders::_1,std::placeholders::_2));
    208. }
    209. void SetBaseDir(const std::string& path){
    210. assert(Util::IsDir(path) == true);
    211. _base_dir = path;
    212. }
    213. //建立正则表达式和对应处理函数的映射关系
    214. void Get(const std::string& pattern,const Functor& func){
    215. _get_route.push_back(std::make_pair(std::regex(pattern),func));
    216. }
    217. void Post(const std::string& pattern,const Functor& func){
    218. _post_route.push_back(std::make_pair(std::regex(pattern),func));
    219. }
    220. void Put(const std::string& pattern,const Functor& func){
    221. _put_route.push_back(std::make_pair(std::regex(pattern),func));
    222. }
    223. void Delete(const std::string& pattern,const Functor& func){
    224. _delete_route.push_back(std::make_pair(std::regex(pattern),func));
    225. }
    226. void SetThreadCount(int count)
    227. {
    228. _server.SetThreadCount(count);
    229. }
    230. void Listen(){
    231. _server.Start();
    232. }
    233. };

    服务器搭建并进行测试

    搭建服务器

    我们需要编写简单的业务处理函数来填写路由表对应的处理方法,然后再调用接口来启动服务器即可

    1. #include "http.hpp"
    2. #define WWWROOT "./wwwroot/"
    3. std::string RequestStr(const HttpRequest& req)
    4. {
    5. std::stringstream ss;
    6. //请求方法 资源路径 协议版本
    7. ss << req._method <<" " << req._path <<" " << req._version <<"\r\n";
    8. //查询字符串
    9. for(auto& it : req._parame)
    10. {
    11. ss << it.first <<": " <<it.second << "\r\n";
    12. }
    13. //请求头部
    14. for(auto& it : req._headers)
    15. {
    16. ss << it.first <<": " <<it.second << "\r\n";
    17. }
    18. ss << "\r\n";
    19. //请求正文
    20. ss << req._body;
    21. return ss.str();
    22. }
    23. void GetFile(const HttpRequest& req,HttpResponse* resp)
    24. {
    25. resp->SetContent(RequestStr(req),"text.plain");
    26. //sleep(15); //这里模拟服务器处理时间很长,可能会导致的其他连接超时销毁
    27. }
    28. void Login(const HttpRequest& req,HttpResponse* resp)
    29. {
    30. resp->SetContent(RequestStr(req),"text.plain");
    31. }
    32. void PutFile(const HttpRequest& req,HttpResponse* resp)
    33. {
    34. std::string req_path = WWWROOT + req._path;
    35. Util::WriteFile(req_path,req._body);
    36. }
    37. void DeleteFile(const HttpRequest& req,HttpResponse* resp)
    38. {
    39. resp->SetContent(RequestStr(req),"text.plain");
    40. }
    41. int main()
    42. {
    43. HttpServer server(8080);
    44. server.SetBaseDir(WWWROOT);
    45. server.SetThreadCount(3);
    46. server.Get("/hello",GetFile);
    47. server.Post("/login",Login);
    48. server.Put("/submit.txt",PutFile);
    49. server.Delete("/submit.txt",DeleteFile);
    50. server.Listen();
    51. return 0;
    52. }

    长连接测试

    1. #include "../source/server.hpp"
    2. int main()
    3. {
    4. Socket cli_sock;
    5. cli_sock.CreateClient(8080, "127.0.0.1");
    6. std::string req = "GET /hello HTTP/1.1\r\nConnection: keep-alive\r\nContent-Length: 0\r\n\r\n";
    7. while(1)
    8. {
    9. assert(cli_sock.Send(req.c_str(),req.size()) != -1);
    10. char buffer[1024];
    11. assert(cli_sock.Recv(buffer,1023) != -1);
    12. DLOG("[%s]",buffer);
    13. sleep(3);
    14. }
    15. cli_sock.Close();
    16. return 0;
    17. }

    测试结果:

    这里服务器设置了30秒的超时连接销毁,通过一定时间的观察我们可以得出长连接测试是没问题的。

    测试超时连接是否销毁

    1. #include "../source/server.hpp"
    2. int main()
    3. {
    4. Socket cli_sock;
    5. cli_sock.CreateClient(8080, "127.0.0.1");
    6. std::string req = "GET /hello HTTP/1.1\r\nConnection: keep-alive\r\nContent-Length: 0\r\n\r\n";
    7. while(1)
    8. {
    9. sleep(40);
    10. }
    11. cli_sock.Close();
    12. return 0;
    13. }

    测试结果:

    我们可以看到服务器这里经过了30秒之后就自动关闭了,而客户端那里由于设置了死循环,所以没有任何反应,但是这也足够测试出来结果是没有问题的

    错误请求处理

    1. //只发送一次小的请求,服务器得不到完整的数据,就不会进行业务处理
    2. //给服务器发送多条小的请求,服务器会把后面的请求当正文处理,但是后面的请求会失败
    3. #include "../source/server.hpp"
    4. int main()
    5. {
    6. Socket cli_sock;
    7. cli_sock.CreateClient(8080, "127.0.0.1");
    8. std::string req = "GET /hello HTTP/1.1\r\nConnection: keep-alive\r\nContent-Length: 100\r\n\r\nhow are you?";
    9. while(1)
    10. {
    11. assert(cli_sock.Send(req.c_str(),req.size()) != -1);
    12. assert(cli_sock.Send(req.c_str(),req.size()) != -1);
    13. assert(cli_sock.Send(req.c_str(),req.size()) != -1);
    14. char buffer[1024];
    15. assert(cli_sock.Recv(buffer,1023) != -1);
    16. DLOG("[%s]",buffer);
    17. sleep(3);
    18. }
    19. cli_sock.Close();
    20. return 0;
    21. }

    测试结果:

    我们可以看到确实这里只处理了一次,把后面的请求也放入到了第一次请求的请求正文中了,同时服务器也没有关闭,所以测试还是没有问题的

    服务器性能达到瓶颈的处理

    理论上,如果服务器性能达到瓶颈,那么怎么处理应该都不为过,这里我们采用这种方式来模拟服务器性能达到瓶颈:通过请求GET方法,但是让其睡眠来模拟

    测试代码:

    1. //测试在服务器达到瓶颈的时候对连接的处理,其他的连接可能因为这个连接处理而导致超时销毁
    2. #include "../source/server.hpp"
    3. int main()
    4. {
    5. signal(SIGCHLD,SIG_IGN);
    6. for(int i = 0;i<10;++i)
    7. {
    8. pid_t pid = fork();
    9. if(pid < 0)
    10. {
    11. DLOG("FORK ERROR");
    12. return -1;
    13. }
    14. else if(pid == 0)
    15. {
    16. //子进程
    17. Socket cli_sock;
    18. cli_sock.CreateClient(8080, "127.0.0.1");
    19. std::string req = "GET /hello HTTP/1.1\r\nConnection: keep-alive\r\nContent-Length: 0\r\n\r\n";
    20. while(1)
    21. {
    22. assert(cli_sock.Send(req.c_str(),req.size()) != -1);
    23. char buffer[1024];
    24. assert(cli_sock.Recv(buffer,1023) != -1);
    25. DLOG("[%s]",buffer);
    26. sleep(3);
    27. }
    28. cli_sock.Close();
    29. exit(0);
    30. }
    31. }
    32. while(1) sleep(1);
    33. return 0;
    34. }

    测试结果:

    client:

    server:

    我们可以看到性能瓶颈并没有导致服务器进行关闭,这些连接都得到了合理得处理结果

    一次发送多个请求测试

    1. //一次性向服务器发送多个请求,查看服务器的响应情况
    2. //每一条请求都应该得到正确的处理
    3. #include "../source/server.hpp"
    4. int main()
    5. {
    6. Socket cli_sock;
    7. cli_sock.CreateClient(8080, "127.0.0.1");
    8. std::string req = "GET /hello HTTP/1.1\r\nConnection: keep-alive\r\nContent-Length: 0\r\n\r\n";
    9. req += "GET /hello HTTP/1.1\r\nConnection: keep-alive\r\nContent-Length: 0\r\n\r\n";
    10. req += "GET /hello HTTP/1.1\r\nConnection: keep-alive\r\nContent-Length: 0\r\n\r\n";
    11. while(1)
    12. {
    13. assert(cli_sock.Send(req.c_str(),req.size()) != -1);
    14. char buffer[1024];
    15. assert(cli_sock.Recv(buffer,1023) != -1);
    16. DLOG("[%s]",buffer);
    17. sleep(3);
    18. }
    19. cli_sock.Close();
    20. return 0;
    21. }

    测试结果:

    它们都得到了预期的处理结果,服务器也一直在处理中,知道用户关闭客户端

    测试大文件传输

    1. #include "../source/http/http.hpp"
    2. int main()
    3. {
    4. Socket cli_sock;
    5. cli_sock.CreateClient(8080, "127.0.0.1");
    6. std::string req = "PUT /submit.txt HTTP/1.1\r\nConnection: keep-alive\r\n";
    7. std::string body;
    8. bool ret = Util::ReadFile("./hello.txt",&body);
    9. req += "Content-Length: " + std::to_string(body.size()) + "\r\n\r\n";
    10. assert(cli_sock.Send(req.c_str(),req.size()) != -1);
    11. //传输正文
    12. assert(cli_sock.Send(body.c_str(),body.size()) != -1);
    13. char buffer[1024];
    14. assert(cli_sock.Recv(buffer,1023) != -1);
    15. DLOG("[%s]",buffer);
    16. sleep(3);
    17. cli_sock.Close();
    18. return 0;
    19. }

    这里传送300M的文件给服务器,因为这里使用的是云服务器,资源没有那么多,就使用300M来测试

    通过上面的指令我们就可以申请到一个300M的文件大写,这里为了让文件内容不全为0,这里追加了字符到文件中

    测试结果:

    我们在服务器这里也得到了一个文件:

    这是我们只需要通过md5sum来比较以下这两个文件是否相同即可:

    这两个文件内容也是一样的,说明我们的测试是没有问题的

    服务器性能测试

    通过webbench来对服务器进行测试:

    测试环境:服务器是两核2G带宽4M的云服务器

    5000个客户端的情况下:

    使用webbench以5000并发量对服务器发送请求,进行1分钟测试,得出的QPS为:2080

    即每秒处理的包的数量

    项目源码

    项目源码

  • 相关阅读:
    字符串思维题练习 DAY5(CF1137 B , CF 733D , CF 1360 F)
    磁盘管理:磁盘结构
    Springboot集成redis--不同环境切换
    Django反向解析函数reverse与resolve
    customRef
    Jetbrains学生邮箱教育认证,操作避坑指南!!!!!!
    智能音箱,扫地机器人,传感器,窗帘电机等产品有何优点?(下)
    xray长亭是自动化Web漏洞扫描神器
    39个你需要知道的Git命令
    NEUQ2022第一次双周赛题解
  • 原文地址:https://blog.csdn.net/weixin_72068014/article/details/134129998