Bound CLS cleanup time and support stopped-instance retries
This commit is contained in:
1 parent
d9af783ff2
commit
e2157443b7
3 files changed
+31
-12
No files matched your search
+13
-11
@@ -270,18 +270,20 @@ def remove_cls(build, execute=False):
|
||||
if execute:
|
||||
current = json.loads((INSTANCE / 'deployment.json').read_text())
|
||||
old_command = compose(INSTANCE, current)
|
||||
for _ in range(60):
|
||||
running = any(container['State']['Running'] for container in inspect_services(old_command))
|
||||
if running:
|
||||
for _ in range(60):
|
||||
idle = json.loads(run('docker', 'exec', '-i', 'AlphaPulse-dev-api', 'python', '-', input=IDLE))
|
||||
if not idle['busy']:
|
||||
break
|
||||
time.sleep(5)
|
||||
else:
|
||||
raise RuntimeError('Chat or trading analysis remained busy')
|
||||
run('docker', 'pause', 'AlphaPulse-dev-trader')
|
||||
idle = json.loads(run('docker', 'exec', '-i', 'AlphaPulse-dev-api', 'python', '-', input=IDLE))
|
||||
if not idle['busy']:
|
||||
break
|
||||
time.sleep(5)
|
||||
else:
|
||||
raise RuntimeError('Chat or trading analysis remained busy')
|
||||
run('docker', 'pause', 'AlphaPulse-dev-trader')
|
||||
idle = json.loads(run('docker', 'exec', '-i', 'AlphaPulse-dev-api', 'python', '-', input=IDLE))
|
||||
if idle['busy']:
|
||||
run('docker', 'unpause', 'AlphaPulse-dev-trader')
|
||||
raise RuntimeError('Analysis became active; retry after it completes')
|
||||
if idle['busy']:
|
||||
run('docker', 'unpause', 'AlphaPulse-dev-trader')
|
||||
raise RuntimeError('Analysis became active; retry after it completes')
|
||||
run(*old_command, 'stop')
|
||||
print(run(*command, 'python', '-m', 'alphapulse.crawler.remove_cls', *(['--execute'] if execute else [])), flush=True)
|
||||
if execute:
|
||||
|
||||
@@ -9,7 +9,7 @@ import deploy
|
||||
|
||||
|
||||
class RemoveClsTests(unittest.TestCase):
|
||||
def exercise(self, execute, failure=False):
|
||||
def exercise(self, execute, failure=False, running=True):
|
||||
with tempfile.TemporaryDirectory() as temporary:
|
||||
directory = Path(temporary)
|
||||
build = {'commit': 'a'*40, 'version': '20261104-aaaaaaaa'}
|
||||
@@ -26,6 +26,7 @@ class RemoveClsTests(unittest.TestCase):
|
||||
|
||||
with patch.object(deploy, 'INSTANCE', directory), patch.object(deploy, 'validate_images'), \
|
||||
patch.object(deploy, 'branch_head', return_value=build['commit']), \
|
||||
patch.object(deploy, 'inspect_services', return_value=[{'State': {'Running': running}}]), \
|
||||
patch.object(deploy, 'run', side_effect=command), patch.object(deploy, 'deploy') as update:
|
||||
if failure:
|
||||
with self.assertRaisesRegex(RuntimeError, 'cleanup failed'):
|
||||
@@ -54,3 +55,8 @@ class RemoveClsTests(unittest.TestCase):
|
||||
calls = self.exercise(True, True)
|
||||
self.assertTrue(any('stop' in args for args in calls))
|
||||
self.assertFalse(any('up' in args or 'alembic' in args for args in calls))
|
||||
|
||||
def test_execute_resumes_with_all_writers_stopped(self):
|
||||
calls = self.exercise(True, running=False)
|
||||
self.assertFalse(any(args[:2] == ('docker', 'pause') for args in calls))
|
||||
self.assertTrue(any('--execute' in args for args in calls))
|
||||
Reference in new issue
Block a user