Init mapping ids to serialised nodes

DataDog · Sep 18, 2024 · dd78072 · dd78072
1 parent 642be7a
commit dd78072
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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)
 }
 
 /**