125 lines
5.7 KiB
C++
125 lines
5.7 KiB
C++
#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"
|
||
//#include "sol/sol.hpp"
|
||
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*)> callback);
|
||
/******************************************************************
|
||
* function：数据接收后回调
|
||
******************************************************************/
|
||
void on_recved(std::function<void(const ylib::buffer& begin, ylib::buffer* end)> 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);
|
||
// 是否为代理任务
|
||
bool is_proxy(reqpack* rp);
|
||
// 是否为CDN服务
|
||
bool is_cdn(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*)> 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::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
|