-
Notifications
You must be signed in to change notification settings - Fork 2k
/
EventDataExtensions.cs
68 lines (63 loc) · 2.58 KB
/
EventDataExtensions.cs
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
using System;
using System.Collections.Generic;
using System.Linq;
using Azure.Messaging.EventHubs;
using Orleans.Serialization;
namespace Orleans.ServiceBus.Providers
{
/// <summary>
/// Extends EventData to support streaming
/// </summary>
public static class EventDataExtensions
{
private const string EventDataPropertyStreamNamespaceKey = "StreamNamespace";
/// <summary>
/// Adds stream namespace to the EventData
/// </summary>
/// <param name="eventData"></param>
/// <param name="streamNamespace"></param>
public static void SetStreamNamespaceProperty(this EventData eventData, string streamNamespace)
{
eventData.Properties[EventDataPropertyStreamNamespaceKey] = streamNamespace;
}
/// <summary>
/// Gets stream namespace from the EventData
/// </summary>
/// <param name="eventData"></param>
/// <returns></returns>
public static string GetStreamNamespaceProperty(this EventData eventData)
{
object namespaceObj;
if (eventData.Properties.TryGetValue(EventDataPropertyStreamNamespaceKey, out namespaceObj))
{
return (string)namespaceObj;
}
return null;
}
/// <summary>
/// Serializes event data properties
/// </summary>
/// <param name="eventData"></param>
/// <param name="serializationManager"></param>
/// <returns></returns>
public static byte[] SerializeProperties(this EventData eventData, SerializationManager serializationManager)
{
var writeStream = new BinaryTokenStreamWriter();
serializationManager.Serialize(eventData.Properties.Where(kvp => !string.Equals(kvp.Key, EventDataPropertyStreamNamespaceKey, StringComparison.Ordinal)).ToList(), writeStream);
var result = writeStream.ToByteArray();
writeStream.ReleaseBuffers();
return result;
}
/// <summary>
/// Deserializes event data properties
/// </summary>
/// <param name="bytes"></param>
/// <param name="serializationManager"></param>
/// <returns></returns>
public static IDictionary<string, object> DeserializeProperties(this ArraySegment<byte> bytes, SerializationManager serializationManager)
{
var stream = new BinaryTokenStreamReader(bytes);
return serializationManager.Deserialize<List<KeyValuePair<string, object>>>(stream).ToDictionary(kvp => kvp.Key, kvp => kvp.Value);
}
}
}