Skip to content

ssh_tunnel_manager.core.monitor

monitor

SSH Tunnel Manager - Monitor Thread

Classes

TunnelMonitorThread

Bases: QThread

Background thread to monitor tunnel status.

Source code in src/ssh_tunnel_manager/core/monitor.py 
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
class TunnelMonitorThread(QThread):
    """Background thread to monitor tunnel status."""

    status_update = Signal(str, bool)  # tunnel_name, is_running
    connection_lost = Signal(str)  # tunnel_name when connection is lost

    def __init__(self, active_tunnels: Dict[str, TunnelProcess]):
        super().__init__()
        self.active_tunnels = active_tunnels
        self.running = True

    def run(self):
        """Main monitoring loop."""
        while self.running:
            for name, tunnel_process in self.active_tunnels.items():
                try:
                    # Check if the process is alive
                    if hasattr(tunnel_process, 'is_alive'):
                        is_alive = tunnel_process.is_alive()
                    else:
                        is_alive = tunnel_process.is_running

                    # If tunnel is in STARTING state, try to transition to RUNNING
                    if (tunnel_process.status == tunnel_process.STATUS_STARTING and 
                        hasattr(tunnel_process, 'transition_to_running_if_healthy')):
                        if tunnel_process.transition_to_running_if_healthy():
                            self.status_update.emit(name, True)
                            continue

                    # Handle connection lost scenarios
                    was_running = tunnel_process.is_running
                    current_running = is_alive and (tunnel_process.status == tunnel_process.STATUS_RUNNING)

                    if was_running and not current_running:
                        # Connection was lost
                        if tunnel_process.connection_lost_count < 10:  # Limit to 10 messages
                            tunnel_process.connection_lost_count += 1
                            self.connection_lost.emit(name)

                    self.status_update.emit(name, current_running)

                except Exception as e:
                    # In case of any error, assume the tunnel is not running
                    self.status_update.emit(name, False)

            time.sleep(MONITOR_INTERVAL)

    def stop(self):
        """Stop the monitoring thread."""
        self.running = False
Functions
run()

Main monitoring loop.

Source code in src/ssh_tunnel_manager/core/monitor.py 
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
def run(self):
    """Main monitoring loop."""
    while self.running:
        for name, tunnel_process in self.active_tunnels.items():
            try:
                # Check if the process is alive
                if hasattr(tunnel_process, 'is_alive'):
                    is_alive = tunnel_process.is_alive()
                else:
                    is_alive = tunnel_process.is_running

                # If tunnel is in STARTING state, try to transition to RUNNING
                if (tunnel_process.status == tunnel_process.STATUS_STARTING and 
                    hasattr(tunnel_process, 'transition_to_running_if_healthy')):
                    if tunnel_process.transition_to_running_if_healthy():
                        self.status_update.emit(name, True)
                        continue

                # Handle connection lost scenarios
                was_running = tunnel_process.is_running
                current_running = is_alive and (tunnel_process.status == tunnel_process.STATUS_RUNNING)

                if was_running and not current_running:
                    # Connection was lost
                    if tunnel_process.connection_lost_count < 10:  # Limit to 10 messages
                        tunnel_process.connection_lost_count += 1
                        self.connection_lost.emit(name)

                self.status_update.emit(name, current_running)

            except Exception as e:
                # In case of any error, assume the tunnel is not running
                self.status_update.emit(name, False)

        time.sleep(MONITOR_INTERVAL)
stop()

Stop the monitoring thread.

Source code in src/ssh_tunnel_manager/core/monitor.py 
62
63
64
def stop(self):
    """Stop the monitoring thread."""
    self.running = False