From 03b956c45781465fa46a6b4513c6283f788e8e49 Mon Sep 17 00:00:00 2001
From: Patrick Jentsch <p.jentsch@uni-bielefeld.de>
Date: Thu, 7 Nov 2019 15:28:07 +0100
Subject: [PATCH] Remove workaround to wait for container

---
 app/corpora/events.py | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/app/corpora/events.py b/app/corpora/events.py
index 8ad98fd3..28d3bad7 100644
--- a/app/corpora/events.py
+++ b/app/corpora/events.py
@@ -5,7 +5,6 @@ from flask import current_app, request
 from flask_login import login_required
 from .CQiClient.CQiClient import CQiClient
 import logging
-import time
 
 
 analysis_clients = {}
@@ -15,7 +14,6 @@ analysis_clients = {}
 @login_required
 def init_corpus_analysis(corpus_id):
     ''' TODO: Check if current_user is allowed to subscribe to this '''
-    time.sleep(5)  # wait for IP of container only for dev
     socketio.start_background_task(observe_corpus_analysis_connection,
                                    current_app._get_current_object(),
                                    corpus_id,
@@ -33,13 +31,17 @@ def observe_corpus_analysis_connection(app, corpus_id, session_id):
     logger = logging.getLogger(__name__)
     with app.app_context():
         corpus = Corpus.query.filter_by(id=corpus_id).first()
-        logger.warning('MÖÖÖÖP')
-        analysis_client = CQiClient(host=corpus.analysis_container_ip)
-        analysis_client.ctrl_connect('opaque', 'opaque')
+        while corpus.status != 'analysing':
+            db.session.refresh(corpus)
+            socketio.sleep(3)
+        analysis_server = '{}_analysis_container{}'.format(corpus.creator.username, corpus.id)
+        analysis_client = CQiClient(host=analysis_server)
         analysis_clients[session_id] = analysis_client
+        analysis_client.ctrl_connect('opaque', 'opaque')
         while session_id in connected_sessions:
             logger.warning(analysis_client.ctrl_ping())
             socketio.sleep(3)
         analysis_client.ctrl_bye()
+        analysis_clients.pop(session_id, None)
         corpus.status = 'stop analysis'
         db.session.commit()
-- 
GitLab