Add a setting for HTTP Upgrade.

huntlabs · Mar 3, 2022 · f06465d · f06465d
1 parent 2ca36c2
commit f06465d
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/source/hunt/framework/config/ApplicationConfig.d b/source/hunt/framework/config/ApplicationConfig.d
@@ -105,6 +105,7 @@ class ApplicationConfig {
         size_t keepAliveTimeOut = 30;
         size_t maxHeaderSize = 64 * 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
@@ -60,6 +60,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) {