From dd7807206945171e0f018a335c810decf8c8b20d Mon Sep 17 00:00:00 2001
From: Najib Boutaib <najib.boutaib@datadoghq.com>
Date: Wed, 18 Sep 2024 15:16:38 +0200
Subject: [PATCH] Init mapping ids to serialised nodes

---
 .../src/domain/record/serialization/serializationUtils.ts   | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/packages/rum/src/domain/record/serialization/serializationUtils.ts b/packages/rum/src/domain/record/serialization/serializationUtils.ts
index 57f6ac6dba..46da358f0f 100644
--- a/packages/rum/src/domain/record/serialization/serializationUtils.ts
+++ b/packages/rum/src/domain/record/serialization/serializationUtils.ts
@@ -4,6 +4,7 @@ import type { NodePrivacyLevel } from '@datadog/browser-rum-core'
 import type { NodeWithSerializedNode } from './serialization.types'
 
 const serializedNodeIds = new WeakMap<Node, number>()
+const serializedNodesByIds = new Map<number, Node>()
 
 export function hasSerializedNode(node: Node): node is NodeWithSerializedNode {
   return serializedNodeIds.has(node)
@@ -26,8 +27,13 @@ export function getSerializedNodeId(node: Node) {
   return serializedNodeIds.get(node)
 }
 
+export function getSerializedNodeById(serializeNodeId: number): Node | undefined {
+  return serializedNodesByIds.get(serializeNodeId)
+}
+
 export function setSerializedNodeId(node: Node, serializeNodeId: number) {
   serializedNodeIds.set(node, serializeNodeId)
+  serializedNodesByIds.set(serializeNodeId, node)
 }
 
 /**