SIM-PIPE/forwarding.py at main · DataCloud-project/SIM-PIPE · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
import argparse
import random
import signal
import subprocess
import sys
import threading
import time

services = [
    {
        "name": "cadvisor",
        "ports": [8081, 8080],
        "enabled": False
    },
    {
        "name": "grafana",
        "ports": [8082, 80],
        "enabled": False
    },
    {
        "name": "sftpgo",
        "ports": [8083, 80],
        "enabled": False
    },
    {
        "name": "argo",
        "fullname": "argo-workflows-server",
        "ports": [8084, 2746],
        "enabled": True
    },
    {
        "name": "minio-backend",
        "fullname": "minio",
        "ports": [8085, 9000],
        "enabled": True
    },
    {
        "name": "prometheus",
        "ports": [8086, 9090],
        "prefix": "",
        "fullname": "prometheus-operated",
    },
    {
        "name": "controller",
        "ports": [8087, 9000]
    },
    {
        "name": "frontend",
        "ports": [8088, 3000]
    },
    {
        "name": "minio-frontend",
        "fullname": "minio",
        "ports": [8089, 9001],
        "enabled": True
    },
    {
        "name": "carbontracker",
        "ports": [8000, 8000],
        "enabled": True
    },
]


def port_forward(service, prefix, delay):
    prefix = service["prefix"] if "prefix" in service else prefix
    name = service["fullname"] if "fullname" in service else service["name"]

    while True:
        try:
            subprocess.run(
                [
                    "kubectl",
                    "port-forward",
                    f"service/{prefix}{name}",
                    f"{service['ports'][0]}:{service['ports'][1]}",
                    # Strong hate towards IPv6
                    # "--address=127.0.0.1",
                ],
                check=True,
                stdout=subprocess.DEVNULL,
                # stderr=subprocess.DEVNULL,
            )
        except subprocess.CalledProcessError as e:
            print(f"Error port forwarding {service['name']}: {e}", file=sys.stderr)
            time.sleep(delay)
            delay = min(delay + random.uniform(0, 5), 10)
        else:
            break


def handle_exit(signal, frame):
    sys.exit(0)


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Start kubernetes port forwardings")
    parser.add_argument(
        "--disable",
        type=str,
        nargs="*",
        help="list of enabled by default services to disable",
    )
    parser.add_argument(
        "--enable",
        type=str,
        nargs="*",
        help="list of disabled by default services to enable",
    )
    parser.add_argument(
        "--prefix", type=str, default="simpipe-", help="prefix for service name"
    )
    args = parser.parse_args()

    disabled = args.disable if args.disable is not None else []
    enabled = args.enable if args.enable is not None else []
    prefix = args.prefix

    signal.signal(signal.SIGINT, handle_exit)
    signal.signal(signal.SIGTERM, handle_exit)

    threads = []
    for service in services:
        if service["name"] not in disabled and (
            service.get("enabled", True) or service["name"] in enabled
        ):
            print(f"{service['name']}: http://localhost:{service['ports'][0]}")
            thread = threading.Thread(target=port_forward, args=(service, prefix, 3))
            thread.start()
            threads.append(thread)

    for thread in threads:
        thread.join()