ylib/include/net/http_router.h

#pragma once
#include "http_define.h"
#if USE_NET_HTTP_WEBSITE
#include "http_interface.h"
#include <functional>
#include <map>
#include <regex>
#include "util/array.hpp"
#include "util/queue.hpp"
#include "util/thread.h"
class IHPThreadPool;
namespace ylib
{
    namespace network
    {
        namespace http
        {
            class reqpack;
            class http_server_lst;
            class request;
            class response;
            class interceptor;
            class subscribe;
            /*************************************************************************
             * class：路由中专服务
             *************************************************************************/
            class router :public ylib::error_base, public network::http::interface_, private ylib::ithread
            {
            public:
                /*线程参数信息*/
                struct thread_param_info
                {
                    void clear() {}
                    network::http::router* router;
                    network::http::reqpack* reqpack;
                };
            public:
                router();
                ~router();
                /******************************************************************
                 * function：启动
                 * param
                 *      config                  ：           配置项
                 * return：
                 *      失败可通过 last_error() 返回错误信息。
                 ******************************************************************/
                bool start(const router_config& config);
                /******************************************************************
                 * function：关闭
                 ******************************************************************/
                void close();
                /******************************************************************
                 * function：拦截器
                 ******************************************************************/
                network::http::interceptor* interceptor();
                /******************************************************************
                * function：订阅器
                ******************************************************************/
                network::http::subscribe* subscribe();
                /******************************************************************
                 * function：其它
                 * desc：未订阅请求触发该回调
                 * param
                 *      callback                ：           触发回调
                 ******************************************************************/
                void other(std::function<void(network::http::request*, network::http::response*,network::http::websocket_message*)> callback);
                /******************************************************************
                 * function：数据接收后回调
                 ******************************************************************/
                void on_recved(std::function<void(const ylib::buffer& begin, ylib::buffer* end)> callback);

                /******************************************************************
                 * function：关闭回调
                 ******************************************************************/
                void on_close(std::function<void(uint64 connid,network::http::websocket_message* ws_msg)> callback);
                /******************************************************************
                 * function：数据发送前回调
                 * desc：不支持大文件断点传输方式
                 ******************************************************************/
                void on_sendbefore(std::function<void(const ylib::buffer& begin, ylib::buffer* end)> callback);
                /******************************************************************
                * function：线程回调[禁止调用]
                * desc：put提交后投递到线程池，线程池开始执行调用该回调。
                * param
                *      param                    ：           线程参数
                ******************************************************************/
                void __thread_callback(reqpack* rq);
                friend class http_server_lst;
                friend class response;
                /******************************************************************
                 * function：队列数
                 ******************************************************************/
                size_t queue_size();
            private:
                // 添加任务
                void push(reqpack* rp);
                // 添加连接断开任务
                void push_close(uint64 connid,std::shared_ptr<network::http::websocket_message> ws_msg);
                // 是否为代理任务
                bool is_proxy(reqpack* rp);
#if HTTP_LUA_ENGINE == 1
                // LUA执行
                void lua_engine(reqpack *rp,const network::http::subscribe_info& info);
#endif
            private:
                // 线程池
                IHPThreadPool* m_threadpool;
                // [回调] 未订阅请求
                std::function<void(network::http::request*, network::http::response*,network::http::websocket_message*)> m_callback_other;
                // [回调] 接收后
                std::function<void(const ylib::buffer& begin, ylib::buffer* end)> m_callback_recved;
                // [回调] 发送前
                std::function<void(const ylib::buffer& begin, ylib::buffer* end)> m_callback_sendbefore;
                // [回调] 关闭
                std::function<void(uint64 connid,network::http::websocket_message* ws_msg)> m_callback_close;
                // 拦截器
                std::unique_ptr<network::http::interceptor> m_interceptor;
                // 订阅器
                std::unique_ptr<network::http::subscribe> m_subscribe;
                
            private:
                ylib::queue<network::http::reqpack*> m_handle_queue;
                router_config m_config;
                // 通过 ithread 继承
                virtual bool run() override;
            };
        }
    }
}
#endif