forked from RockLib/RockLib.Messaging
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathProgram.cs
More file actions
168 lines (148 loc) · 6.09 KB
/
Copy pathProgram.cs
File metadata and controls
168 lines (148 loc) · 6.09 KB
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
using Newtonsoft.Json.Linq;
using RockLib.Messaging;
using System;
using System.Collections.Generic;
using System.Threading;
namespace Example.Messaging.Http.DotNetCore20
{
class Program
{
static void Main(string[] args)
{
Console.WriteLine("Make a selection:");
Console.WriteLine("1) Create an ISender and prompt for messages.");
Console.WriteLine("2) Create an IReceiver and listen for messages.");
Console.WriteLine("3) Create an ISender and an IReceiver and send one message from one to the other.");
Console.WriteLine("q) Quit");
Console.WriteLine("Tip: Start multiple instances of this app and have them send and receive to each other.");
Console.Write("selection>");
while (true)
{
var c = Console.ReadKey(true).KeyChar;
switch (c)
{
case '1':
Console.WriteLine(c);
RunSender();
return;
case '2':
Console.WriteLine(c);
RunReceiver();
return;
case '3':
Console.WriteLine(c);
SendAndReceiveOneMessage();
return;
case 'q':
Console.WriteLine(c);
return;
}
}
}
private static void RunSender()
{
using (var sender = MessagingScenarioFactory.CreateSender("Sender1"))
{
Console.WriteLine($"Enter a message for sender '{sender.Name}'. Add headers as a trailing JSON object. Leave blank to quit.");
string message;
while (true)
{
Console.Write("message>");
if ((message = Console.ReadLine()) == "")
return;
if (TryExtractHeaders(ref message, out var headers))
sender.Send(new SenderMessage(message) { Headers = headers });
else
sender.Send(message);
}
}
}
private static void RunReceiver()
{
using (var receiver = MessagingScenarioFactory.CreateReceiver("Receiver1"))
{
receiver.Start(m =>
{
foreach (var header in m.Headers)
Console.WriteLine($"{header.Key}: {header.Value}");
Console.WriteLine(m.StringPayload);
// Always acknowledge, reject, or rollback a received http request.
m.Acknowledge();
});
Console.WriteLine($"Receiving messages from receiver '{receiver.Name}'. Press <enter> to quit.");
while (Console.ReadKey(true).Key != ConsoleKey.Enter) { }
}
}
private static void SendAndReceiveOneMessage()
{
// Use a wait handle to pause the main thread while waiting for the message to be received.
var waitHandle = new AutoResetEvent(false);
var sender = MessagingScenarioFactory.CreateSender("Sender1");
var receiver = MessagingScenarioFactory.CreateReceiver("Receiver1");
receiver.Start(m =>
{
var message = m.StringPayload;
Console.WriteLine($"Message received: {message}");
// Always acknowledge, reject, or rollback a received http request.
m.Acknowledge();
waitHandle.Set();
});
sender.Send($"Http test message from {typeof(Program).FullName}");
waitHandle.WaitOne();
receiver.Dispose();
sender.Dispose();
waitHandle.Dispose();
Console.Write("Press any key to exit...");
Console.ReadKey(true);
}
private static bool TryExtractHeaders(ref string message, out IDictionary<string, object> headers)
{
headers = new Dictionary<string, object>();
var trim = message.TrimEnd();
if (trim.Length > 1 && trim[trim.Length - 1] == '}' && trim[trim.Length - 2] != '\\')
{
var stack = new Stack<char>();
stack.Push('}');
for (int i = trim.Length - 2; i >= 0; i--)
{
if (trim[i] == '{')
{
stack.Pop();
if (stack.Count == 0)
{
JObject json;
try
{
json = JObject.Parse(message.Substring(i));
message = message.Substring(0, i);
}
catch (Exception)
{
return false;
}
string GetString(object value)
{
switch (value)
{
case string stringValue:
return stringValue;
case bool boolValue:
return boolValue ? "true" : "false";
default:
return value.ToString();
}
}
foreach (var field in json)
if (field.Value is JValue value)
headers.Add(field.Key, GetString(value.Value));
return true;
}
}
else if (trim[i] == '}')
stack.Push('}');
}
}
return false;
}
}
}