Merge pull request #1463 from Bidaya0/feat/add-gevent-option

feat: optimise agent upload.
HXSecurity · May 24, 2023 · a9a3802 · a9a3802
2 parents 78147ad + fdc3b0f
commit a9a3802
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/dongtai_conf/settings.py b/dongtai_conf/settings.py
@@ -10,10 +10,10 @@
 https://docs.djangoproject.com/en/3.0/ref/settings/
 """
 
+import os
 from typing import List
 from ast import literal_eval
 from urllib.parse import urljoin
-import os
 import sys
 from configparser import ConfigParser
 

diff --git a/dongtai_conf/wsgi.py b/dongtai_conf/wsgi.py
@@ -9,6 +9,11 @@
 
 import os
 
+ENABLE_GEVENT = os.environ.get("GEVENT", 'FALSE') == 'TRUE'
+if ENABLE_GEVENT:
+    from gevent import monkey
+    monkey.patch_all()
+
 from django.core.wsgi import get_wsgi_application
 
 os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'dongtai_conf.settings')