Fix IntegrityError during webserver startup (#27297)

apache · Oct 27, 2022 · 8f99c79 · 8f99c79
1 parent 9ab1a6a
commit 8f99c79
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/airflow/cli/commands/webserver_command.py b/airflow/cli/commands/webserver_command.py
@@ -414,6 +414,12 @@ def webserver(args):
 
         run_args += ["airflow.www.app:cached_app()"]
 
+        # To prevent different workers creating the web app and
+        # all writing to the database at the same time, we use the --preload option.
+        # With the preload option, the app is loaded before the workers are forked, and each worker will
+        # then have a copy of the app
+        run_args += ['--preload']
+
         gunicorn_master_proc = None
 
         def kill_proc(signum, _):

diff --git a/tests/cli/commands/test_webserver_command.py b/tests/cli/commands/test_webserver_command.py
@@ -405,6 +405,7 @@ def test_cli_webserver_args(self):
                     "--access-logformat",
                     "custom_log_format",
                     "airflow.www.app:cached_app()",
+                    "--preload",
                 ],
                 close_fds=True,
             )