openreplay/api/chalicelib/core/assist.py
Kraiem Taha Yassine c818e25ff0
Assist (#51)
* feat(api): assist peerJS server

* feat(api): install assist_server dependencies and start it with the API

* feat(api): assist: list live sessions

* feat(nginx): expose assist_server and block peers listing

* feat(api): merged sourcemaps reader and assist-server
feat(api): change image definition
feat(api): changed service start command
feat(utilities): created full server & image definition
feat(nginx): reset chalice configuration

* feat(utilities): utilities.yaml

* feat(nginx): utilities URL

* feat(utilities): utilities template

* feat(ci): Adding utilities GH action.

Signed-off-by: Rajesh Rajendran <rjshrjndrn@gmail.com>

* feat(utilities): build script

Co-authored-by: Rajesh Rajendran <rjshrjndrn@gmail.com>
2021-06-23 21:17:31 +02:00

42 lines
1.7 KiB
Python

from chalicelib.utils import pg_client, helper
from chalicelib.core import projects
import requests
SESSION_PROJECTION_COLS = """s.project_id,
s.session_id::text AS session_id,
s.user_uuid,
s.user_id,
s.user_agent,
s.user_os,
s.user_browser,
s.user_device,
s.user_device_type,
s.user_country,
s.start_ts,
s.user_anonymous_id,
s.platform
"""
def get_live_sessions(project_id):
project_key = projects.get_project_key(project_id)
connected_peers = requests.get(f"http://127.0.0.1:9000/peers/{project_key}")
if connected_peers.status_code != 200:
print("!! issue with the peer-server")
print(connected_peers.text)
return []
connected_peers = connected_peers.json().get("data", [])
if len(connected_peers) == 0:
return []
connected_peers = tuple(connected_peers)
with pg_client.PostgresClient() as cur:
query = cur.mogrify(f"""\
SELECT {SESSION_PROJECTION_COLS}
FROM public.sessions AS s
WHERE s.project_id = %(project_id)s
AND session_id IN %(connected_peers)s
AND duration IS NULL;""",
{"project_id": project_id, "connected_peers": connected_peers})
cur.execute(query)
results = cur.fetchall()
return helper.list_to_camel_case(results)