Feature add support for gunicorn worker class (#21)

* Add support for other gunicorn worker types * Remove gthread worker_class
jortencio · Jul 27, 2023 · 1fcd4ac · 1fcd4ac
1 parent e093ecd
commit 1fcd4ac
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/manifests/init.pp b/manifests/init.pp
@@ -151,7 +151,7 @@
   String                                                $admin_lastname = 'admin',
   String                                                $admin_email = '[email protected]',
   String                                                $gunicorn_install_dir = '/home/superset',
-  String                                                $gunicorn_worker_class = 'gevent',
+  Enum['sync','eventlet','gevent','tornado']            $gunicorn_worker_class = 'gevent',
   Integer                                               $gunicorn_workers = 10,
   Integer                                               $gunicorn_timeout = 120,
   String                                                $gunicorn_bind = '0.0.0.0:8088',

diff --git a/manifests/install.pp b/manifests/install.pp
@@ -92,7 +92,11 @@
 
   # Install python libraries for managing the webserver
   if $superset::manage_webserver {
-    $webserver_venv_pip_pkg = ['gunicorn', 'gevent']
+    if $superset::gunicorn_worker_class == 'sync' {
+      $webserver_venv_pip_pkg = ['gunicorn']
+    } else {
+      $webserver_venv_pip_pkg = ['gunicorn', $superset::gunicorn_worker_class]
+    }
     $webserver_venv_pip_pkg.each | String $pkgname | {
       python::pip { $pkgname:
         ensure     => 'present',

diff --git a/templates/superset.gunicorn.epp b/templates/superset.gunicorn.epp
@@ -1,7 +1,7 @@
 <%- |
   String  $install_dir,
   Integer $workers,
-  Enum['sync','eventlet','gevent','tornado','gthread'] $worker_class,
+  Enum['sync','eventlet','gevent','tornado'] $worker_class,
   String  $bind,
   Integer $timeout,
   Integer $limit_request_line,