Add a setting for HTTP Upgrade.

huntlabs · Mar 3, 2022 · 46a624b · 46a624b
1 parent 92717b4
commit 46a624b
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/source/hunt/framework/config/ApplicationConfig.d b/source/hunt/framework/config/ApplicationConfig.d
@@ -22,7 +22,7 @@ import std.socket : Address, parseAddress;
 import std.string;
 
 import hunt.http.MultipartOptions;
-import hunt.logging.ConsoleLogger;
+import hunt.logging.Logger;
 import hunt.util.Configuration;
 
 
@@ -106,6 +106,7 @@ class ApplicationConfig {
         size_t keepAliveTimeOut = 30;
         size_t maxHeaderSize = 60 * 1024;
         int cacheControl;
+        bool canUpgrade = true;
         bool enableCors = false; // CORS support
         string allowOrigin = "*";
         string allowMethods = "*";

diff --git a/source/hunt/framework/controller/Controller.d b/source/hunt/framework/controller/Controller.d
@@ -112,6 +112,8 @@ abstract class Controller
             }
 
             HttpConnection httpConnection = context.httpConnection();
+            if(httpConnection is null) 
+                throw new Error("The http connection is null!");
 
             _request = new Request(context.getRequest(), 
                                     httpConnection.getRemoteAddress(),

diff --git a/source/hunt/framework/provider/HttpServiceProvider.d b/source/hunt/framework/provider/HttpServiceProvider.d
@@ -16,7 +16,7 @@ import hunt.http.server.WebSocketHandler;
 import hunt.http.WebSocketPolicy;
 import hunt.http.WebSocketCommon;
 
-import hunt.logging.ConsoleLogger;
+import hunt.logging.Logger;
 
 import poodinis;
 
@@ -62,6 +62,7 @@ class HttpServiceProvider : ServiceProvider {
             .maxRequestSize(cast(int)appConfig.http.maxHeaderSize)
             .maxFileSize(cast(int)appConfig.upload.maxSize)
             .ioThreadSize(appConfig.http.ioThreads)
+            .canUpgrade(appConfig.http.canUpgrade)
             .resourceCacheTime(staticFilesConfig.cacheTime.seconds);
 
         version(WITH_HUNT_TRACE) {