Function portforward

misc/python/materialize/cli/orchestratord.py:330–391

(args: argparse.Namespace)

def portforward(args: argparse.Namespace):
    env_kubectl = make_env_kubectl(args)

    resource_id = get_resource_id(args)

    port_forward_targets = {
        "console/http": lambda port: f"http://localhost:{port}/",
        "balancerd/pgwire": lambda port: f"postgres://{os.environ['USER']}@localhost:{port}/materialize",
        "balancerd/http": lambda port: f"http://localhost:{port}/",
        "environmentd/internal-sql": lambda port: f"postgres://mz_system@localhost:{port}/materialize",
        "environmentd/internal-http": lambda port: f"http://localhost:{port}/",
    }

    port_forwards = {}
    sockets = []
    try:
        for port_forward_target in port_forward_targets:
            s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            sockets.append(s)
            s.bind(("", 0))
            s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
            port_forwards[port_forward_target] = s.getsockname()[1]
    finally:
        for s in sockets:
            s.close()

    def port_forward(target: str, port: str):
        target_service, target_port = target.split("/")
        # in theory doing a port forward to the service directly should work,
        # but i can't seem to get it to
        selector = json.loads(
            env_kubectl(
                "get",
                "service",
                f"mz{resource_id}-{target_service}",
                "-o=jsonpath={.spec.selector}",
            )
        )
        pod = env_kubectl(
            "get",
            "pod",
            "-o=name",
            *[f"--selector={k}={v}" for k, v in selector.items()],
        ).splitlines()[0]

        env_kubectl(
            "port-forward",
            pod,
            f"{port}:{target_port}",
        )

    threads = [
        threading.Thread(target=port_forward, args=[target, port])
        for target, port in port_forwards.items()
    ]

    for t in threads:
        t.start()