|
| 1 | +@file:Suppress("UNCHECKED_CAST") |
| 2 | + |
1 | 3 | package io.ably.lib.objects |
2 | 4 |
|
3 | | -import com.google.gson.Gson |
4 | | -import com.google.gson.GsonBuilder |
5 | | -import com.google.gson.JsonArray |
| 5 | +import com.fasterxml.jackson.databind.ObjectMapper |
| 6 | +import com.google.gson.* |
| 7 | +import org.msgpack.core.MessagePack |
6 | 8 | import org.msgpack.core.MessagePacker |
7 | 9 | import org.msgpack.core.MessageUnpacker |
| 10 | +import org.msgpack.jackson.dataformat.MessagePackFactory |
| 11 | + |
| 12 | +// Gson instance for JSON serialization/deserialization |
| 13 | +internal val gson: Gson = GsonBuilder().create() |
8 | 14 |
|
9 | | -internal val gson: Gson = createGsonSerializer() |
| 15 | +// Jackson ObjectMapper for MessagePack serialization (respects @JsonProperty annotations) |
| 16 | +// Caches type metadata and serializers for ObjectMessage class after first use, so super fast!! |
| 17 | +private val msgpackMapper = ObjectMapper(MessagePackFactory()) |
| 18 | + |
| 19 | +internal fun ObjectMessage.toJsonObject(): JsonObject { |
| 20 | + return gson.toJsonTree(this).asJsonObject |
| 21 | +} |
10 | 22 |
|
11 | | -private fun createGsonSerializer(): Gson { |
12 | | - return GsonBuilder().create() // Do not call serializeNulls() to omit null values |
| 23 | +internal fun JsonObject.toObjectMessage(): ObjectMessage { |
| 24 | + return gson.fromJson(this, ObjectMessage::class.java) |
13 | 25 | } |
14 | 26 |
|
| 27 | +internal fun ObjectMessage.writeTo(packer: MessagePacker) { |
| 28 | + // Jackson automatically creates the correct msgpack map structure |
| 29 | + val msgpackBytes = msgpackMapper.writeValueAsBytes(this) |
| 30 | + |
| 31 | + // Parse the msgpack bytes to get the structured value |
| 32 | + val tempUnpacker = MessagePack.newDefaultUnpacker(msgpackBytes) |
| 33 | + val msgpackValue = tempUnpacker.unpackValue() |
| 34 | + tempUnpacker.close() |
| 35 | + |
| 36 | + // Pack the structured value using the provided packer |
| 37 | + packer.packValue(msgpackValue) |
| 38 | +} |
| 39 | + |
| 40 | +internal fun MessageUnpacker.readObjectMessage(): ObjectMessage { |
| 41 | + // Read the msgpack value from the unpacker |
| 42 | + val msgpackValue = this.unpackValue() |
| 43 | + |
| 44 | + // Convert the msgpack value back to bytes |
| 45 | + val tempPacker = MessagePack.newDefaultBufferPacker() |
| 46 | + tempPacker.packValue(msgpackValue) |
| 47 | + val msgpackBytes = tempPacker.toByteArray() |
| 48 | + tempPacker.close() |
| 49 | + |
| 50 | + // Let Jackson deserialize the msgpack bytes back to ObjectMessage |
| 51 | + return msgpackMapper.readValue(msgpackBytes, ObjectMessage::class.java) |
| 52 | +} |
| 53 | + |
| 54 | +/** |
| 55 | + * Default implementation of {@link LiveObjectSerializer} that handles serialization/deserialization |
| 56 | + * of ObjectMessage arrays for both JSON and MessagePack formats using Jackson and Gson. |
| 57 | + * Dynamically loaded by LiveObjectsHelper#getLiveObjectSerializer() to avoid hard dependencies. |
| 58 | + */ |
| 59 | +@Suppress("unused") // Used via reflection in LiveObjectsHelper |
15 | 60 | internal class DefaultLiveObjectSerializer : LiveObjectSerializer { |
| 61 | + |
16 | 62 | override fun readMsgpackArray(unpacker: MessageUnpacker): Array<Any> { |
17 | | - TODO("Not yet implemented") |
| 63 | + val objectMessagesCount = unpacker.unpackArrayHeader() |
| 64 | + return Array(objectMessagesCount) { unpacker.readObjectMessage() } |
18 | 65 | } |
19 | 66 |
|
20 | 67 | override fun writeMsgpackArray(objects: Array<out Any>?, packer: MessagePacker) { |
21 | | - TODO("Not yet implemented") |
| 68 | + val objectMessages: Array<ObjectMessage> = objects as Array<ObjectMessage> |
| 69 | + packer.packArrayHeader(objectMessages.size) |
| 70 | + objectMessages.forEach { it.writeTo(packer) } |
22 | 71 | } |
23 | 72 |
|
24 | 73 | override fun readFromJsonArray(json: JsonArray): Array<Any> { |
25 | | - TODO("Not yet implemented") |
| 74 | + return json.map { element -> |
| 75 | + if (element.isJsonObject) element.asJsonObject.toObjectMessage() |
| 76 | + else throw JsonParseException("Expected JsonObject, but found: $element") |
| 77 | + }.toTypedArray() |
26 | 78 | } |
27 | 79 |
|
28 | 80 | override fun asJsonArray(objects: Array<out Any>?): JsonArray { |
29 | | - TODO("Not yet implemented") |
| 81 | + val objectMessages: Array<ObjectMessage> = objects as Array<ObjectMessage> |
| 82 | + val jsonArray = JsonArray() |
| 83 | + for (objectMessage in objectMessages) { |
| 84 | + jsonArray.add(objectMessage.toJsonObject()) |
| 85 | + } |
| 86 | + return jsonArray |
30 | 87 | } |
31 | 88 | } |
0 commit comments