一种延迟方法调用的实现

需求场景

最近遇到一种场景，需要延迟某个步骤的实现，以便在延迟的过程中能够及早处理，从而取消任务。 比如，现在的服务器CPU过高了，则系统记录下来，然后开始发送邮件通知用户，但是如果在10秒之内CPU恢复了；或者我不希望用户得知这个CPU的信息，因为我就在现场，我可以及时知道怎么处理这种情况，这个时候准备拨打电话的这个方法的调用就不能被执行，需要延迟一段时间，以便能够被取消——以上场景仅供参考。

代码实现

  以下是我的一个实现方式，供大家讨论，或者有更好的方式可以改进。

这种方法旨在能够方便的延迟任意方法的执行：

定义任务接口 

  这是一个基本的线程任务的接口，包括任务名称、线程的运行状态、启停线程。

     interface ITask
2     {
3         string Name { get; }
4         bool IsRunning { get; }
5         void Start();
6         void Stop();
7         void Run();

8     } 

  定义延迟的信息类

  保存当前需要延迟的方法，以及需要延迟的时间等

  1         class LazyItem

 2         {
 3             private readonly int _timeout;
 4             private readonly DateTime _enqueueTime;
 5 
 6             public bool IsTimeout
 7             {
 8                 get { return (DateTime.Now - _enqueueTime).TotalSeconds > _timeout; }
 9             }
10             public string Key { get; private set; }
11             public Action Continue { get; private set; }
12 
13             public LazyItem(string key, int timeout, Action continueAction)
14             {
15                 _enqueueTime = DateTime.Now;
16                 _timeout = timeout;
17                 Key = key;
18                 Continue = continueAction;
19             }
20         }

  延迟的实现

  主要是执行一个定时的任务，去检测被延迟的方法是否达到了超时的时间，如果达到，则执行该方法。另外，提供一个可供取消方法的函数。

  1     class LazyInvoker : ITask

 2     {
 3         public string Name
 4         {
 5             get { return "Lazy Invoke Task"; }
 6         }
 7 
 8         public bool IsRunning { get; private set; }
 9 
10         private readonly ConcurrentDictionary<string, LazyItem> _lazyActions = new ConcurrentDictionary<string, LazyItem>();
11 
12         public void Start()
13         {
14             Task.Factory.StartNew(Run);
15             IsRunning = true;
16         }
17 
18         public void Stop()
19         {
20             IsRunning = false;
21         }
22 
23         /// <summary>
24         /// 检测被延迟的任务，达到超时时间则触发
25         /// </summary>
26         /// Created by:marvin(2014/10/11 12:14)
27         public void Run()
28         {
29             while (IsRunning)
30             {
31                 CheckContinue();
32                 Thread.Sleep(1 * 1000);
33             }
34         }
35 
36         private void CheckContinue()
37         {
38             if (_lazyActions.Count <= 0) return;
39 
40             var removeKeys = (from lazyItem in _lazyActions where lazyItem.Value.IsTimeout select lazyItem.Key).ToList();
41             if (removeKeys.Count <= 0) return;
42 
43             foreach (var key in removeKeys)
44             {
45                 LazyItem tmp;
46                 if (_lazyActions.TryRemove(key, out tmp))
47                 {
48                     tmp.Continue();
49                 }
50             }
51         }
52 
53         /// <summary>
54         /// 延迟工作
55         /// </summary>
56         /// <param name="id">The identifier.</param>
57         /// <param name="lazyTimeout">The lazy timeout.</param>
58         /// <param name="continueAction">The continue action.</param>
59         /// Created by:marvin(2014/10/11 11:48)
60         public void LazyDo(string id, int lazyTimeout, Action continueAction)
61         {
62             if (!_lazyActions.ContainsKey(id))
63             {
64                 if (_lazyActions.TryAdd(id, new LazyItem(id, lazyTimeout, continueAction)))
65                 {
66                     //Console.WriteLine("lazy action : {0} , timeout : {1}", id, lazyTimeout);
67                 }
68             }
69         }
70 
71         /// <summary>
72         /// 取消任务
73         /// </summary>
74         /// <param name="actionKey">The action key.</param>
75         /// Created by:marvin(2014/10/11 12:02)
76         public void Cancel(string actionKey)
77         {
78             if (_lazyActions.ContainsKey(actionKey))
79             {
80                 LazyItem tmp;
81                 if (_lazyActions.TryRemove(actionKey, out tmp))
82                 {
83                     Console.WriteLine("lazy action “{0}” had removed", tmp.Key);
84                 }
85             }
86         }
87     }

 测试

  1     class Program

 2     {
 3         static void Main(string[] args)
 4         {
 5             var lazyInvoker = new LazyInvoker();
 6             lazyInvoker.Start();
 7 
 8             //延迟7秒运行
 9             lazyInvoker.LazyDo(Guid.NewGuid().ToString(), 7, DoSomething);
10             Thread.Sleep(5 * 1000);
11 
12             //延迟3秒运行，但是3秒的时候被取消
13             var id = Guid.NewGuid().ToString();
14             lazyInvoker.LazyDo(id, 5, DoSomething);
15             Thread.Sleep(3 * 1000);
16             lazyInvoker.Cancel(id);
17 
18             Console.ReadKey();
19         }
20 
21         private static void DoSomething()
22         {
23             Console.WriteLine("Now time is :" + DateTime.Now.ToString("HH:mm:ss"));
24         }
25     }

  运行结果如下：