master_daemon & other daemons updates

1 files changed, 6 insertions, 0 deletions

diff --git a/scripts/automation/trex_control_plane/server/trex_server.py b/scripts/automation/trex_control_plane/server/trex_server.py
index c1760236..45ef9ac1 100755
--- a/scripts/automation/trex_control_plane/server/trex_server.py
+++ b/scripts/automation/trex_control_plane/server/trex_server.py
@@ -29,6 +29,11 @@ import re
 import shlex
 import tempfile
 
+try:
+    from .singleton_daemon import register_socket
+except:
+    from singleton_daemon import register_socket
+
 
 # setup the logger
 CCustomLogger.setup_custom_logger('TRexServer')
@@ -102,6 +107,7 @@ class CTRexServer(object):
     def start(self):
         """This method fires up the daemon server based on initialized parameters of the class"""
         # initialize the server instance with given resources
+        register_socket('trex_daemon_server')
         try:
             print "Firing up TRex REST daemon @ port {trex_port} ...\n".format( trex_port = self.trex_daemon_port )
             logger.info("Firing up TRex REST daemon @ port {trex_port} ...".format( trex_port = self.trex_daemon_port ))