首页 > 解决方案 > 如何在后台运行异步任务?

问题描述

我需要转发k8s集群中运行服务的端口。所以我使用了 github repo 中提供的示例来为特定服务进行端口转发。现在我想在我的测试中使用这个例子,但是当我调用端口转发方法时,我总是得到 web socket 异常,但是如果我在控制台应用程序中编写相同的代码,那么我没有得到任何异常,它运行得很好。

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;
        }

Nunit 测试项目:

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

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

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

标签: c#kuberneteswebsocketportforwarding

解决方案


我假设 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
}

推荐阅读