Add autoreleasePool for request handler

enums · Jan 11, 2021 · 66bfe17 · 66bfe17
1 parent 124de64
commit 66bfe17
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/Sources/Heze/App/HezeApp.swift b/Sources/Heze/App/HezeApp.swift
@@ -148,7 +148,7 @@ public class HezeApp: HezeObject {
             routes += pair.compactMap { (arg) in
                 let (method, handler) = arg
                 handler.bindContext(context)
-                return Route(method: method, uri: path, handler: handler.handleThenComplete)
+                return Route(method: method, uri: path, handler: handler.wrappedhandleThenComplete)
             }
         }
         server.addRoutes(Routes(routes))

diff --git a/Sources/Heze/View/HezeHandler.swift b/Sources/Heze/View/HezeHandler.swift
@@ -6,6 +6,7 @@
 //  Copyright © 2019 Yuu Zheng. All rights reserved.
 //
 
+import Foundation
 import PerfectHTTP
 
 public typealias HezeMetaHandler = HezeHandler
@@ -50,7 +51,7 @@ open class HezeHandler: HezeObject {
         return nil
     }
 
-    internal func handleThenComplete(_ req: HTTPRequest, _ res: HTTPResponse) {
+    open func handleThenComplete(_ req: HTTPRequest, _ res: HTTPResponse) {
         request = req
         defer {
             request = nil
@@ -65,4 +66,14 @@ open class HezeHandler: HezeObject {
         }
     }
 
+    open func wrappedhandleThenComplete(_ req: HTTPRequest, _ res: HTTPResponse) {
+        #if os(macOS)
+        autoreleasepool {
+            handleThenComplete(req, res)
+        }
+        #else
+        handleThenComplete(req, res)
+        #endif
+    }
+
 }