add worker flags for streamer

1 files changed, 5 insertions, 1 deletions

diff --git a/src/flufigut.py b/src/flufigut.py
index 5bdff0a..6be8cb2 100755
--- a/src/flufigut.py
+++ b/src/flufigut.py
@@ -493,6 +493,8 @@ class Planet:
 
         comp_name = '%s-%s%i-stream-%s-%s-%s' % (stream['type'], stream_name, idx + 1, mux, format, profile)
         mount_point = '/%s-%s-%s.%s' % (mux, format, profile, self._desc.globals['formats'][format]['muxer'])
+        worker = self.__get_worker(comp_name)
+
         hostname = stream['hostname']
         hostname_next = None
         if cnt > 1:
@@ -506,7 +508,7 @@ class Planet:
             'type': "%s-stream" % stream['type'],
             'desc': "%s streamer for %s %s-%s (part %i of %i in %s cluster)" % (
                 stream['type'], mux, format, profile, idx + 1, cnt, stream_name),
-            'worker': str(self.__get_worker(comp_name)),
+            'worker': str(worker),
             'feeder': feeder,
             'properties': {
                 'description': self._desc.globals['description'],
@@ -521,6 +523,8 @@ class Planet:
         }
         self.__set_stream_mux_instance_props(comp_name, stream, porter.port, mount_point, hostname_next)
         self.__add_stream_mux_instance_plugs(comp_name, stream_name, idx, mux, format, profile)
+        self.__add_worker_flag(worker, "stream", stream_name)
+        self.__add_worker_flag(worker, "stream-hostname-%s" % stream_name, hostname)
 
     def __generate_stream_instance(self, stream_name, stream, idx, cnt):
         port = stream['port']