connman.py

#!/usr/bin/env python3

__copyright__ = 'Copyright (c) 2021-2022, Utrecht University'
__license__   = 'GPLv3, see LICENSE'

import threading
import time
from typing import Dict, Optional

from irods.session import iRODSSession

TTL = 60 * 30


class Session(object):
    def __init__(self, sid: int, irods: iRODSSession) -> None:
        """Session object storing the iRODS session object.

        :param sid:   Flask session identifier
        :param irods: iRODS session
        """
        self.sid: int            = sid
        self.irods: iRODSSession = irods
        self.time: float         = time.time()
        self.lock                = threading.Lock()

    def __del__(self) -> None:
        self.irods.cleanup()
        print(f"[gc/logout]: Dropped iRODS session of session {self.sid}")


sessions: Dict[int, Session] = dict()  # Custom session dict instead of Flask session (cannot pickle iRODS session)
lock = threading.Lock()


def gc() -> None:
    """Session garbage collection."""
    while True:
        with lock:
            t = time.time()
            global sessions
            sessions = {k: v for k, v in sessions.items() if t - v.time < TTL or v.lock.locked()}

        time.sleep(1)


gc_thread = threading.Thread(target=gc, name='irods-session-gc', daemon=True)
gc_thread.start()


def get(sid: int) -> Optional[iRODSSession]:
    """Retrieve iRODS session object from session."""
    if sid in sessions:
        s: Session = sessions[sid]
        s.lock.acquire()
    else:
        return None

    return s.irods


def add(sid: int, irods: iRODSSession) -> None:
    """Add Flask sid and iRODS session to our custom session dict.

    :param sid:   Flask session identifier
    :param irods: iRODS session
    """
    global sessions
    s: Session = Session(sid, irods)
    sessions[sid] = s
    s.time = time.time()
    s.lock.acquire()
    print(f"[login]: Successfully connected to iRODS for session {sid}'")


def release(sid: int) -> None:
    """Release a session.

    :param sid: Flask session identifier
    """
    global sessions
    if sid in sessions:
        s: Session = sessions[sid]
        s.time = time.time()
        s.lock.release()


def clean(sid: int) -> None:
    """Clean a session.

    :param sid: Flask session identifier
    """
    global sessions
    if sid in sessions:
        del sessions[sid]