From 3a353aceb280bcf03187255f837d4f294e4e69ac Mon Sep 17 00:00:00 2001 From: Lars Wirzenius Date: Fri, 15 Dec 2017 18:26:43 +0200 Subject: Add: make worker-manager register worker --- worker_manager | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'worker_manager') 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) -- cgit v1.2.1