.Net 異步方法加上“timeout”


在本羊讀大學的時候,Thread讓本羊雲里霧里,代碼寫的痛不欲生,真的是讓本羊腦袋里很多“線”纏繞在一起。

之后,Task讓本羊代碼寫的飛起,甚至有時候根本不需要Task的時候還是要寫上,那樣顯得檔次較高:多線程!充分利用CPU!niubility!

再之后,async/await語法糖橫空出世,更讓本羊欲罷不能!

 

然而,

好東西吃多了總會膩的——吃貨

常在河邊走,哪有不濕鞋?——文藝青年

出來混總是要還的——“四道杠”騷年

飆車一時爽,“全家”火葬場——老司機

 

扯遠了,反正大概意思就是,async/await 也會有不爽的時候。

比如:UdpClient.ReceiveAsync()

 

某領導:只等待5秒,過時不候。這樣比較節約性能。

——哎喲,很有道理的樣子,那就改改?

 

於是,本羊很快就找到了一個屬性:UdpClient.Client.ReceiveTimeout

 

UdpClient.Client.ReceiveTimeout=5000;//領導說了只等5秒 <--注釋在這里

var r=UdpClient.ReceiveAsync();

……(省略其他代碼)

 

看到沒,本羊還很善意的給代碼加上了“注釋”,方便其他小伙伴理解本羊的優雅的高深的簡潔的代碼。

然后拿起水杯,小小的抿了一口,伸出右手食指,輕輕的按下“F5”,左手在桌子上很有節奏的敲了5下,哎喲,啥情況?

再敲5下,再敲5下,再敲5下……

好吧,本羊輸了。

是時候打開MSDN了,原來,ReceiveTimeout這玩意兒只對同步方法有效,ReceiveAsync根本不管這Y的。

 

再研究研究?

 

經過一番刻苦鑽研,新的代碼來了:

var t=new CancellationTokenSource();//這玩意兒就是用來配合Task,做取消功能的

Task.Delay(5000, t.Token).ContinueWith(task =>
{
     if (!task.IsCanceled && task.IsCompleted)//不是被取消而且已經完成
         {
             client.Close();//釋放UdpClient
          }
});//咱走着瞧(不會阻塞當前線程),5秒之后再來

try

{

  var r=await client.ReceiveAsync();

  return r;//返回結果

}

catch

{

  throw new TimeoutException("過時不候!");//優雅的拋出錯誤提示

}

 

以上代碼使用了很長一段時間,直到今天本羊又看到一些關於Task的文章,原來還有更優雅的實現方式!

不多說,上代碼:

var t=new CancellationTokenSource();//又是這玩意兒

var r=await Task.WhenAny(client.ReceiveAsync(), Task.Delay(5000, t.Token)) as Task<UdpReceiveResult>;//還是5秒,過時不候

if(r!=null)//如果是Delay先返回,是不能 as Task<UdpReceiveResult>的,r=null。
{

  t.Cancel();//取消那個Delay,其實也可以不用處理,反正5秒后那家伙就自己去西天了

  return r.Result;

}

else

{

  client.Close();//釋放UdpClient,不然還是在ReceiveAsync

  throw new TimeoutException("過時不候!");

}

 

再來一個:

var tasks=new Task[]{client.ReceiveAsync()};

var index=awati Task.Run(()=>

{

  return Task.WaitAny(tasks, 5000);//返回完成的Task在集合中的序號

});//Wait[xxx]會阻塞線程,所以用一個Run包裹住

if(index==0)

{

  return (tasks[0] as Task<UdpReceiveResult>).Result;

}

else

{

  client.Close();

  throw new TimeoutException("過時不候!");

}

又或者(重磅推薦):

var t = client.ReceiveAsync();
if (await Task.Run(() => { return t.Wait(5000); }))
{
  return t.Result;
}
else
{
  client.Close();
  throw new TimeoutException("過時不候!");
}

 

CancellationTokenSource、Task.Delay(delay)都用不到了,爽歪歪~~

 

Wait[xxx]有多個重載,可以設置timeout、CancellationToken,不知道為啥When[xxx]不能設置?

 

最后放上Task的拓展方法:

 

 1 public static class TaskExtension
 2     {
 3         /// <summary>
 4         /// 異步等待
 5         /// 任務完成立即返回,否則在指定時間后拋出TimeoutException
 6         /// </summary>
 7         /// <param name="task"></param>
 8         /// <param name="timeout">等待時間(毫秒)</param>
 9         /// <returns></returns>
10         public static async Task WaitAsync(this Task task, int timeout)
11         {
12             if (!await Task.Run(() => { return task.Wait(timeout); }))
13                 throw new TimeoutException();
14         }
15 
16         /// <summary>
17         /// 異步等待
18         /// 任務完成立即返回,否則在指定時間后拋出TimeoutException
19         /// </summary>
20         /// <param name="task"></param>
21         /// <param name="timeout">等待時間(毫秒)</param>
22         /// <returns></returns>
23         public static async Task<T> WaitAsync<T>(this Task<T> task, int timeout)
24         {
25             if (await Task.Run(() => { return task.Wait(timeout); }))
26                 return task.Result;
27             throw new TimeoutException();
28         }
29 
30         /// <summary>
31         /// 給“無法取消”的任務傳入CancellationToken
32         /// </summary>
33         /// <param name="task">無法傳入CancellationToken的任務</param>
34         /// <param name="token">CancellationToken</param>
35         /// <returns></returns>
36         public static async Task WaitAsync(this Task task, CancellationToken token)
37         {
38             if (!await Task.Run(() => { try { return task.Wait(-1, token); } catch { return false; } }))
39                 throw new TaskCanceledException();
40         }
41 
42         /// <summary>
43         /// 給“無法取消”的任務傳入CancellationToken
44         /// </summary>
45         /// <typeparam name="T">返回類型</typeparam>
46         /// <param name="task">無法傳入CancellationToken的任務</param>
47         /// <param name="token">CancellationToken</param>
48         /// <returns></returns>
49         public static async Task<T> WaitAsync<T>(this Task<T> task, CancellationToken token)
50         {
51             if (await Task.Run(() => { try { return task.Wait(-1, token); } catch { return false; } }))
52                 return task.Result;
53             throw new TaskCanceledException();
54         }
55 
56         /// <summary>
57         /// 封裝“無法取消”的任務
58         /// </summary>
59         /// <typeparam name="T">返回類型</typeparam>
60         /// <param name="task">無法傳入CancellationToken的任務</param>
61         /// <param name="watcher">可取消的無限等待任務[Task.Delay(-1,CancellationToken)]</param>
62         /// <returns></returns>
63         public static async Task<T> WaitAsync<T>(this Task<T> task, Task watcher)
64         {
65             var r = await Task.WhenAny(task, watcher);
66             if (r == task)
67             {
68                 return task.Result;
69             }
70             throw new TaskCanceledException();
71         }
72 
73 
74         public static CancellationTokenRegistration RegisterWidth<T>(this CancellationToken token, T state, Action<T> action)
75         {
76             return token.Register((p) => { action((T)p); }, state);
77         }
78     }
View Code

 

 

 

 

PS:其實本羊的工作一直是單槍匹馬,意思就是公司里只有本羊一個程序猿,所以,某領導是沒有的,其他小伙伴也是沒有的,一切的一切都是本羊虛構的,包括你們。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM