Merge pull request #11733 from overleaf/msm-close-site-sp-shutdwn

[CE/SP] close site on shutdown before flushing

GitOrigin-RevId: e36c80f2f4c6f10eda3a536319a2fcc5dfda4fa4

server-ce/Dockerfile

@@ -70,6 +70,11 @@ COPY server-ce/config/production.json /overleaf/services/history-v1/config/produ
 ADD server-ce/bin/grunt /usr/local/bin/grunt
 RUN chmod +x /usr/local/bin/grunt
 
+# File that controls open|closed status of the site
+# -------------------------------------------------
+ENV SITE_MAINTENANCE_FILE "/etc/sharelatex/site_status"
+RUN touch $SITE_MAINTENANCE_FILE
+
 # Set Environment Variables
 # --------------------------------
 ENV SHARELATEX_CONFIG /etc/sharelatex/settings.js

server-ce/init_preshutdown_scripts/00_close_site

@@ -0,0 +1,23 @@
+#!/bin/sh
+
+SITE_MAINTENANCE_FILE_BAK="$SITE_MAINTENANCE_FILE.bak.shutdown
+
+mv "${SITE_MAINTENANCE_FILE}" "${SITE_MAINTENANCE_FILE_BAK}"
+echo "closed" > "${SITE_MAINTENANCE_FILE}"
+
+# status file is polled every 5 seconds
+sleep 5
+
+# giving a grace period of 5 seconds for users before disconnecting them and start shutting down
+cd /overleaf/services/web && node scripts/disconnect_all_users.js 5 >> /var/log/sharelatex/web.log 2>&1
+
+EXIT_CODE="$?"
+if [ $EXIT_CODE -ne 0 ]
+then
+  echo "scripts/disconnect_all_users.js failed with exit code $EXIT_CODE"
+fi
+
+# wait for disconnection
+sleep 5
+
+exit 0

server-ce/init_scripts/00_restore_site_status

@@ -0,0 +1,12 @@
+#!/bin/bash
+
+set -e
+
+# pre-shutdown scripts close the site by overriding the content of SITE_MAINTENANCE_FILE,
+# this script restores the original value on container restart
+SITE_MAINTENANCE_FILE_BAK="$SITE_MAINTENANCE_FILE.bak.shutdown"
+
+if [ -f "${SITE_MAINTENANCE_FILE_BAK}" ]; then
+    mv -f "${SITE_MAINTENANCE_FILE_BAK}" "${SITE_MAINTENANCE_FILE}"
+    rm -f "${SITE_MAINTENANCE_FILE_BAK}"
+fi