From 21014e20bedb21fd4862120d0e338098284783f7 Mon Sep 17 00:00:00 2001
From: Daniil Baturin <daniil@baturin.org>
Date: Wed, 21 Aug 2019 09:08:27 +0200
Subject: T1598: clean up vyos-hostsd state dump on clean shutdown.

---
 src/services/vyos-hostsd | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'src/services')

diff --git a/src/services/vyos-hostsd b/src/services/vyos-hostsd
index aec06c72c..1bcb2e1f5 100755
--- a/src/services/vyos-hostsd
+++ b/src/services/vyos-hostsd
@@ -20,6 +20,7 @@ import os
 import sys
 import time
 import json
+import signal
 import traceback
 
 import zmq
@@ -218,8 +219,16 @@ def handle_message(msg_json):
     with open(STATE_FILE, 'w') as f:
         json.dump(STATE, f)
 
+def exit_handler(sig, frame):
+    """ Clean up the state when shutdown correctly """
+    print("Cleaning up state")
+    os.unlink(STATE_FILE)
+    sys.exit(0)
+
 
 if __name__ == '__main__':
+    signal.signal(signal.SIGTERM, exit_handler)
+
     # Create a directory for state checkpoints
     os.makedirs(DATA_DIR, exist_ok=True)
     if os.path.exists(STATE_FILE):
-- 
cgit v1.2.3