Question

我需要转发k8s集群中正在运行的服务的端口。因此，我已经使用github repo中提供的示例为特定服务进行端口转发。现在，我想在测试中使用该示例，但是当我调用端口转发方法时，我总是会收到Web套接字异常，但是如果我在控制台应用程序中编写了相同的代码，那么我就不会得到任何异常，它可以很好地运行。

private async Task<bool> ForwardPortOperation(string podName, int[] originalPort, int forwardPort)
        {
            var pod = GetPodObject(podName);

            var webSocket = await _config.GetK8SClient().WebSocketNamespacedPodPortForwardAsync(pod.Metadata.Name, pod.Metadata.NamespaceProperty, originalPort, WebSocketProtocol.V4BinaryWebsocketProtocol);

            var demux = new StreamDemuxer(webSocket);
            demux.Start();

            var stream = demux.GetStream((byte?)0, (byte?)0);

            IPAddress ipAddress = IPAddress.Loopback;
            IPEndPoint localEndPoint = new IPEndPoint(ipAddress, forwardPort);
            Socket listener = new Socket(ipAddress.AddressFamily, SocketType.Stream, ProtocolType.Tcp);
            listener.Bind(localEndPoint);
            listener.Listen(100);

            Socket handler = null;

            var accept = Task.Run(() =>
            {
                while (true)
                {
                    handler = listener.Accept();

                    var bytes = new byte[4096];
                    while (true)
                    {
                        int bytesRec = handler.Receive(bytes);
                        stream.Write(bytes, 0, bytesRec);
                        if (bytesRec == 0 || Encoding.Default.GetString(bytes, 0, bytesRec).IndexOf("<EOF>", StringComparison.InvariantCultureIgnoreCase) > -1)
                        {
                            break;
                        }
                        listener.Close();
                    }
                }
            });

            var copy = Task.Run(() =>
            {
                var buff = new byte[4096];
                while (true)
                {
                    var read = stream.Read(buff, 0, 4096);
                    handler.Send(buff, read, 0);
                }
            });

            await accept;
            await copy;

            return true;
        }

端口转发方法：

public bool PortForward(string podName, int[] originalPort, int forwardPort)
        {
            return Task.Run(() => ForwardPortOperation(podName, originalPort, forwardPort)).Result;
        }

单元测试项目：

[Test]
 public void VerifyPortForward()
{
    new Pod(_config).PortForward("web_service_application", new int[3100], 1615);
}

如何在后台运行这样的异步方法？这样，一旦端口转发成功，我就可以继续进行另一项测试。

例外： https://gist.github.com/gittadesushil/bc3bf008b7a4fc62c33e97ab1fbf687f

Answer 1

我认为方法PortForward应该通过以下方式重写为异步方式：

public async Task<bool> PortForward(string podName, int[] originalPort, int forwardPort)
{
    return await Task.Run(() => ForwardPortOperation(podName, originalPort, forwardPort));
}

之后，您可以编写一个NUnit测试：

[Test]
public async Task VerifyPortForward()
{
    var pod = new Pod(_config);
    var result = await pod.PortForward("web_service_application", new int[3100], 1615);
    // TODO: assert the result
}

如何在后台运行异步任务？

1 个答案: