消息队列现今的应用场景越来越大,常用的有RabbmitMQ和KafKa。
我们用BlockingCollection来实现简单的消息队列。
BlockingCollection实现了生产者/消费者模式,是对IProducerConsumerCollection<T>接口的实现。与其他Concurrent集合一样,每次Add或Take元素,都会导致对集合的lock。只有当确定需要在内存中创建一个生产者,消费者模式时,再考虑这个类。
MSDN中的示例用法:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
using (BlockingCollection< int > bc = new BlockingCollection< int >()) { Task.Factory.StartNew(() => { for ( int i = 0; i < 1000; i++) { bc.Add(i); Thread.Sleep(50); } // Need to do this to keep foreach below from hanging bc.CompleteAdding(); }); // Now consume the blocking collection with foreach. // Use bc.GetConsumingEnumerable() instead of just bc because the // former will block waiting for completion and the latter will // simply take a snapshot of the current state of the underlying collection. foreach (var item in bc.GetConsumingEnumerable()) { Console.WriteLine(item); } } |
实现消息队列
用Vs2017创建一个控制台应用程序。创建DemoQueueBlock类,封装一些常用判断。
- HasEle,判断是否有元素
- Add向队列中添加元素
- Take从队列中取出元素
为了不把BlockingCollection直接暴漏给使用者,我们封装一个DemoQueueBlock类
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
/// <summary> /// BlockingCollection演示消息队列 /// </summary> /// <typeparam name="T"></typeparam> public class DemoQueueBlock<T> where T : class { private static BlockingCollection<T> Colls; public DemoQueueBlock() { } public static bool IsComleted() { if (Colls != null && Colls.IsCompleted) { return true ; } return false ; } public static bool HasEle() { if (Colls != null && Colls.Count>0) { return true ; } return false ; } public static bool Add(T msg) { if (Colls == null ) { Colls = new BlockingCollection<T>(); } Colls.Add(msg); return true ; } public static T Take() { if (Colls == null ) { Colls = new BlockingCollection<T>(); } return Colls.Take(); } } /// <summary> /// 消息体 /// </summary> public class DemoMessage { public string BusinessType { get ; set ; } public string BusinessId { get ; set ; } public string Body { get ; set ; } } |
添加元素进队列
通过控制台,添加元素
1
2
3
4
5
6
7
8
9
10
11
12
|
//添加元素 while ( true ) { Console.WriteLine( "请输入队列" ); var read = Console.ReadLine(); if (read == "exit" ) { return ; } DemoQueueBlock<DemoMessage>.Add( new DemoMessage() { BusinessId = read }); } |
消费队列
通过判断IsComleted,来确定是否获取队列
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
Task.Factory.StartNew(() => { //从队列中取元素。 while (!DemoQueueBlock<DemoMessage>.IsComleted()) { try { var m = DemoQueueBlock<DemoMessage>.Take(); Console.WriteLine( "已消费:" + m.BusinessId); } catch (Exception ex) { Console.WriteLine(ex.Message); } } }); |
查看运行结果
运行结果
这样我们就实现了简易的消息队列。
示例源码:简易队列
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。
原文链接:https://www.jianshu.com/p/5c0cd3424574