-
Notifications
You must be signed in to change notification settings - Fork 1k
/
PersistenceMessageSerializer.cs
246 lines (210 loc) · 10.3 KB
/
PersistenceMessageSerializer.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
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
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
//-----------------------------------------------------------------------
// <copyright file="PersistenceMessageSerializer.cs" company="Akka.NET Project">
// Copyright (C) 2009-2023 Lightbend Inc. <http://www.lightbend.com>
// Copyright (C) 2013-2023 .NET Foundation <https://github.com/akkadotnet/akka.net>
// </copyright>
//-----------------------------------------------------------------------
using System;
using System.Collections.Generic;
using System.Collections.Immutable;
using System.Reflection;
using System.Runtime.Serialization;
using Akka.Actor;
using Akka.Persistence.Fsm;
using Akka.Persistence.Serialization.Proto.Msg;
using Akka.Serialization;
using Akka.Util;
using Google.Protobuf;
namespace Akka.Persistence.Serialization
{
public sealed class PersistenceMessageSerializer : Serializer
{
public PersistenceMessageSerializer(ExtendedActorSystem system) : base(system)
{
}
public override bool IncludeManifest { get; } = true;
public override byte[] ToBinary(object obj)
{
if (obj is IPersistentRepresentation repr) return GetPersistentMessage(repr).ToByteArray();
if (obj is AtomicWrite aw) return GetAtomicWrite(aw).ToByteArray();
if (obj is AtLeastOnceDeliverySnapshot snap) return GetAtLeastOnceDeliverySnapshot(snap).ToByteArray();
if (obj is PersistentFSM.StateChangeEvent stateEvent) return GetStateChangeEvent(stateEvent).ToByteArray();
if (obj.GetType().IsGenericType
&& obj.GetType().GetGenericTypeDefinition() == typeof(PersistentFSM.PersistentFSMSnapshot<>)) return GetPersistentFSMSnapshot(obj).ToByteArray();
throw new ArgumentException($"Can't serialize object of type [{obj.GetType()}] in [{GetType()}]");
}
private PersistentMessage GetPersistentMessage(IPersistentRepresentation persistent)
{
var message = new PersistentMessage();
if (persistent.PersistenceId != null) message.PersistenceId = persistent.PersistenceId;
if (persistent.Manifest != null) message.Manifest = persistent.Manifest;
if (persistent.WriterGuid != null) message.WriterGuid = persistent.WriterGuid;
if (persistent.Sender != null) message.Sender = Akka.Serialization.Serialization.SerializedActorPath(persistent.Sender);
if (persistent.Timestamp > 0L) message.Timestamp = persistent.Timestamp;
message.Payload = GetPersistentPayload(persistent.Payload);
message.SequenceNr = persistent.SequenceNr;
message.Deleted = persistent.IsDeleted;
return message;
}
private PersistentPayload GetPersistentPayload(object obj)
{
PersistentPayload BuildPayload()
{
var serializer = system.Serialization.FindSerializerFor(obj);
var payload = new PersistentPayload();
var manifest = Akka.Serialization.Serialization.ManifestFor(serializer, obj);
if (!string.IsNullOrEmpty(manifest))
{
payload.PayloadManifest = ByteString.CopyFromUtf8(manifest);
}
payload.Payload = ByteString.CopyFrom(serializer.ToBinary(obj));
payload.SerializerId = serializer.Identifier;
return payload;
}
var oldInfo = Akka.Serialization.Serialization.CurrentTransportInformation;
try
{
if (oldInfo == null)
Akka.Serialization.Serialization.CurrentTransportInformation =
system.Provider.SerializationInformation;
return BuildPayload();
}
finally
{
Akka.Serialization.Serialization.CurrentTransportInformation = oldInfo;
}
}
private Proto.Msg.AtomicWrite GetAtomicWrite(AtomicWrite write)
{
var message = new Proto.Msg.AtomicWrite();
foreach (var pr in (IImmutableList<IPersistentRepresentation>)write.Payload)
{
message.Payload.Add(GetPersistentMessage(pr));
}
return message;
}
private Proto.Msg.AtLeastOnceDeliverySnapshot GetAtLeastOnceDeliverySnapshot(AtLeastOnceDeliverySnapshot snapshot)
{
var message = new Proto.Msg.AtLeastOnceDeliverySnapshot
{
CurrentDeliveryId = snapshot.CurrentDeliveryId
};
foreach (var unconfirmed in snapshot.UnconfirmedDeliveries)
{
message.UnconfirmedDeliveries.Add(new Proto.Msg.UnconfirmedDelivery
{
DeliveryId = unconfirmed.DeliveryId,
Destination = unconfirmed.Destination.ToString(),
Payload = GetPersistentPayload(unconfirmed.Message)
});
}
return message;
}
private PersistentStateChangeEvent GetStateChangeEvent(PersistentFSM.StateChangeEvent changeEvent)
{
var message = new PersistentStateChangeEvent
{
StateIdentifier = changeEvent.StateIdentifier
};
if (changeEvent.Timeout.HasValue)
{
message.TimeoutMillis = (long)changeEvent.Timeout.Value.TotalMilliseconds;
}
return message;
}
private PersistentFSMSnapshot GetPersistentFSMSnapshot(object obj)
{
Type type = obj.GetType();
var message = new PersistentFSMSnapshot
{
StateIdentifier = (string)type.GetProperty("StateIdentifier")?.GetValue(obj),
Data = GetPersistentPayload(type.GetProperty("Data")?.GetValue(obj))
};
TimeSpan? timeout = (TimeSpan?)type.GetProperty("Timeout")?.GetValue(obj);
if (timeout.HasValue)
{
message.TimeoutMillis = (long)timeout.Value.TotalMilliseconds;
}
return message;
}
public override object FromBinary(byte[] bytes, Type type)
{
if(type == null) return GetPersistentRepresentation(PersistentMessage.Parser.ParseFrom(bytes));
if (type == typeof(Persistent)) return GetPersistentRepresentation(PersistentMessage.Parser.ParseFrom(bytes));
if (type == typeof(IPersistentRepresentation)) return GetPersistentRepresentation(PersistentMessage.Parser.ParseFrom(bytes));
if (type == typeof(AtomicWrite)) return GetAtomicWrite(bytes);
if (type == typeof(AtLeastOnceDeliverySnapshot)) return GetAtLeastOnceDeliverySnapshot(bytes);
if (type == typeof(PersistentFSM.StateChangeEvent)) return GetStateChangeEvent(bytes);
if (type.IsGenericType
&& type.GetGenericTypeDefinition() == typeof(PersistentFSM.PersistentFSMSnapshot<>)) return GetPersistentFSMSnapshot(type, bytes);
throw new SerializationException($"Unimplemented deserialization of message with type [{type}] in [{GetType()}]");
}
private IPersistentRepresentation GetPersistentRepresentation(PersistentMessage message)
{
IActorRef sender = ActorRefs.NoSender;
if (message.Sender != null)
{
sender = system.Provider.ResolveActorRef(message.Sender);
}
var repr = new Persistent(
GetPayload(message.Payload),
message.SequenceNr,
message.PersistenceId,
message.Manifest,
message.Deleted,
sender,
message.WriterGuid);
return message.Timestamp > 0L ? repr.WithTimestamp(message.Timestamp) : repr;
}
private object GetPayload(PersistentPayload payload)
{
string manifest = "";
if (payload.PayloadManifest != null) manifest = payload.PayloadManifest.ToStringUtf8();
return system.Serialization.Deserialize(payload.Payload.ToByteArray(), payload.SerializerId, manifest);
}
private AtomicWrite GetAtomicWrite(byte[] bytes)
{
var message = Proto.Msg.AtomicWrite.Parser.ParseFrom(bytes);
var payloads = new List<IPersistentRepresentation>();
foreach (var payload in message.Payload)
{
payloads.Add(GetPersistentRepresentation(payload));
}
return new AtomicWrite(payloads.ToImmutableList());
}
private AtLeastOnceDeliverySnapshot GetAtLeastOnceDeliverySnapshot(byte[] bytes)
{
var message = Proto.Msg.AtLeastOnceDeliverySnapshot.Parser.ParseFrom(bytes);
var unconfirmedDeliveries = new List<UnconfirmedDelivery>();
foreach (var unconfirmed in message.UnconfirmedDeliveries)
{
ActorPath.TryParse(unconfirmed.Destination, out var actorPath);
unconfirmedDeliveries.Add(new UnconfirmedDelivery(unconfirmed.DeliveryId, actorPath, GetPayload(unconfirmed.Payload)));
}
return new AtLeastOnceDeliverySnapshot(message.CurrentDeliveryId, unconfirmedDeliveries.ToArray());
}
private PersistentFSM.StateChangeEvent GetStateChangeEvent(byte[] bytes)
{
var message = PersistentStateChangeEvent.Parser.ParseFrom(bytes);
TimeSpan? timeout = null;
if (message.TimeoutMillis > 0)
{
timeout = TimeSpan.FromMilliseconds(message.TimeoutMillis);
}
return new PersistentFSM.StateChangeEvent(message.StateIdentifier, timeout);
}
private object GetPersistentFSMSnapshot(Type type, byte[] bytes)
{
var message = PersistentFSMSnapshot.Parser.ParseFrom(bytes);
TimeSpan? timeout = null;
if (message.TimeoutMillis > 0)
{
timeout = TimeSpan.FromMilliseconds(message.TimeoutMillis);
}
// use reflection to create the generic type of PersistentFSM.PersistentFSMSnapshot
Type[] types = { typeof(string), type.GenericTypeArguments[0], typeof(TimeSpan?) };
object[] arguments = { message.StateIdentifier, GetPayload(message.Data), timeout };
return type.GetConstructor(types).Invoke(arguments);
}
}
}