From 1b4ed755fe605f04adea60c91366a9bc69bb32f5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Hu=E1=BB=B3nh=20Tr=E1=BA=A7n=20Khanh?=
 <qcdz9r6wpcbh59@gmail.com>
Date: Sat, 9 Mar 2024 15:14:34 +0000
Subject: [PATCH] Fix PairMap

---
 compiler/PairMap.test.js | 5 +++--
 compiler/PairMap.ts      | 8 ++++----
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/compiler/PairMap.test.js b/compiler/PairMap.test.js
index 48084af..35dc90b 100644
--- a/compiler/PairMap.test.js
+++ b/compiler/PairMap.test.js
@@ -25,8 +25,9 @@ describe('PairMap', () => {
   test('set method works for different types of keys', () => {
     pairMap.set([1, 'a'], 'value')
     expect(pairMap.get([1, 'a'])).toBe('value')
-    pairMap.set(['x', { key: 'b' }], 'anotherValue')
-    expect(pairMap.get(['x', { key: 'b' }])).toBe('anotherValue')
+    const key = { key: 'b' }
+    pairMap.set(['x', key], 'anotherValue')
+    expect(pairMap.get(['x', key])).toBe('anotherValue')
   })
 
   test('get method returns undefined for non-existing keys', () => {
diff --git a/compiler/PairMap.ts b/compiler/PairMap.ts
index a797405..8fcefc6 100644
--- a/compiler/PairMap.ts
+++ b/compiler/PairMap.ts
@@ -6,9 +6,9 @@ function createCounter() {
 }
 
 export class PairMap<KeyLeft, KeyRight, Value> {
-  private leftKeyMap = new WeakMap<KeyLeft, number>()
-  private rightKeyMap = new WeakMap<KeyRight, number>()
-  private valueMap = new WeakMap<string, Value>()
+  private leftKeyMap = new Map<KeyLeft, number>()
+  private rightKeyMap = new Map<KeyRight, number>()
+  private valueMap = new Map<string, Value>()
   private getNextCounter: () => number
 
   constructor() {
@@ -16,7 +16,7 @@ export class PairMap<KeyLeft, KeyRight, Value> {
   }
 
   private getOrCreateCounterForKey<T>(
-    map: WeakMap<object, number>,
+    map: Map<T, number>,
     key: T
   ): number {
     if (!map.has(key)) {