diff options
| -rwxr-xr-x | worker_manager | 10 |
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) |