Question

我正在研究重构代码，该代码使用Bootstrap协议来更新机器中多个节点的固件。当前代码看起来像这样（伪代码）：

public void StartUpdate()
{
    Sokcet bootpSocket = new Socket():
    StateObject bootpState = new StateObject(bootpSocket);

    BOOTPReceive(bootpState);
    SendMagicPacket();

    while (!IsError && !IsUpdateComplete)
    {
        //wait for BOOTP/Update to finish before returning to caller
    }
}

private void BOOTPReceive(object state)
{
    bOOTPSocket.BeginReceive(PACKET_DATA, 0, PACKET_DATA.Length, 0, OnBOOTPReceive, state);
}

SendMagicPacket()
{
    //create and send magic packet
    // this will tell the node to respond with a BOOTPPacket
}

private void OnBOOTPReceive(IAsyncResult result)
{
    StateObject state = (StateObject) result.AsyncState;
    Socket handler = state.workSocket;
    int bytesRcvd = handler.EndReceive(result);
    packet = PACKET_DATA;

    if(isValidBOOTP(packet))
    {
        SendBOOTPResponse();    
    }
    else{
        BOOTPReceive(); //keep listening for valid bootp response
    }
}

private void SendBOOTPResponse()
{
    UdpClient udpClient = new UdpClient();
    udpClient.BeginSend(packetData, packetData.Length, BROADCAST_IP, (int)UdpPort.BOOTP_CLIENT_PORT, OnBOOTPSend, udpClient);
}

private void OnBOOTPSend(IAsyncResult result)
{
    UdpClient udpClient = (UdpClient)result.AsyncState;
    int bytesSent = udpClient.EndSend(result);
    udpClient.Close();
}

我想要做的是将其转换为async-await，但仍然要求我不要立即返回给调用者。我该怎么做呢？这可能吗？并且这是正确的事情，因为await-async一直传播到顶部？

我认为这看起来像的伪代码：

public void StartUpdate()
{
    bool result = await SendMagicPacket();
    bool IsError = await BOOTPCommunication(); //Handles all of the BOOTP recieve/sends
    //don't return to caller until BOOTPCommunication is completed. How do i do this?
}

Answer 1

您需要等待以下两项任务：

public async Task StartUpdate()
    {
        var resultTask =  SendMagicPacket();
        var isErrorTask = BOOTPCommunication(); //Handles all of the BOOTP recieve/sends

        await Task.WhenAll(new[]{resultTask, isErrorTask});
        //don't return to caller until BOOTPCommunication is completed. How do i do this?
    }

Answer 2

//等待BOOTP / Update完成后再返回调用者

您根本不需要任何异步IO，因为您希望等到所有操作完成。我假设你复制了一些示例代码。大多数示例代码使用异步套接字API。

将所有内容切换到同步套接字API并完成。

如果由于某种原因想要保持此异步，您确实可以切换到等待并解开此代码。您发布的伪代码看起来是一个很好的目标。但是，它强制周围的方法为async Task。

你可以通过使所有调用者递归异步来处理它。如果您不需要保存线程，则可以阻止该任务并拥有一个主要是同步的调用链。那时你会失去所有的异步利益。

Answer 3

Radin走在正确的轨道上，但我认为你想要的是这样的：

您需要等待以下两项任务：

public async Task StartUpdate()
{
    var resultTask =  SendMagicPacket();
    var isErrorTask = BOOTPCommunication(); //Handles all of the BOOTP recieve/sends

    Task.WhenAll(new[]{resultTask, isErrorTask}).Wait(); //Wait() will block so that the method doesn't return to the caller until both of the asynchronous tasks complete.
}

允许的是SendMagicPacket和BOOTPCommunication同时触发，但等待BOTH完成。使用该模式，您可以同时触发N个事件，同时使用 Wait（）等待所有事件完成，以便方法本身同步返回。

理解.NET中的等待和异步

3 个答案: