Add: make worker-manager register worker

1 files changed, 10 insertions, 0 deletions

diff --git a/worker_manager b/worker_manager
index f517196..3ed7325 100755
--- a/worker_manager
+++ b/worker_manager
@@ -96,6 +96,7 @@ class WorkerManager(cliapp.Application):
         url = self.settings['controller']
         self.show_msg(
             'Worker manager {} starts, controller is {}'.format(name, url))
+        self.register(url, name)
         while True:
             work = self.get_work(url, name)
             if work and self.settings['pretend']:
@@ -105,6 +106,15 @@ class WorkerManager(cliapp.Application):
             else:
                 self.sleep_a_little()
 
+    def register(self, url, name):
+        register_url = '{}/workers'.format(url)
+        headers = self.get_auth_headers()
+        body = {
+            'worker': name,
+        }
+        r = requests.post(register_url, data=body, headers=headers, verify=False)
+        logging.info('Created worker: status code %d', r.status_code)
+
     def sleep_a_little(self):
         secs = self.settings['sleep']
         time.sleep(secs)