支持 http 服务绑定 onAccept 和 onClose 回调

http/server/HttpHandler.cpp

@@ -42,6 +42,7 @@ HttpHandler::HttpHandler(hio_t* io) :
     tid(0),
     // for http
     io(io),
+    server(NULL),
     service(NULL),
     api_handler(NULL),
     // for websocket

http/server/HttpHandler.h

@@ -51,6 +51,7 @@ class HttpHandler {
 
     // for http
     hio_t                   *io;
+    void                    *server;
     HttpService             *service;
     HttpRequestPtr          req;
     HttpResponsePtr         resp;

http/server/HttpServer.cpp

@@ -37,14 +37,17 @@ static void on_recv(hio_t* io, void* buf, int readbytes) {
 static void on_close(hio_t* io) {
     HttpHandler* handler = (HttpHandler*)hevent_userdata(io);
     if (handler == NULL) return;
-
-    hevent_set_userdata(io, NULL);
-    delete handler;
+    http_server_t* server = (http_server_t*)handler->server;
 
     EventLoop* loop = currentThreadEventLoop;
     if (loop) {
+        if (server->onClose) {
+            server->onClose(io);
+        }
         --loop->connectionNum;
     }
+    hevent_set_userdata(io, NULL);
+    delete handler;
 }
 
 static void on_accept(hio_t* io) {
@@ -65,6 +68,12 @@ static void on_accept(hio_t* io) {
         hio_close(io);
         return;
     }
+    if (server->onAccept) {
+        if (!server->onAccept(io)) {
+            hio_close(io);
+            return;
+        }
+    }
     ++loop->connectionNum;
 
     hio_setcb_close(io, on_close);
@@ -82,6 +91,8 @@ static void on_accept(hio_t* io) {
     sockaddr_u* peeraddr = (sockaddr_u*)hio_peeraddr(io);
     sockaddr_ip(peeraddr, handler->ip, sizeof(handler->ip));
     handler->port = sockaddr_port(peeraddr);
+    // http server
+    handler->server = server;
     // http service
     handler->service = service;
     // websocket service

http/server/HttpServer.h

@@ -29,6 +29,8 @@ typedef struct http_server_s {
     // hooks
     std::function<void()> onWorkerStart;
     std::function<void()> onWorkerStop;
+    std::function<bool(hio_t* io)> onAccept;
+    std::function<void(hio_t* io)> onClose;
     // SSL/TLS
     hssl_ctx_t  ssl_ctx;
     unsigned    alloced_ssl_ctx: 1;