fix: avoid dispose unmatched handlers (#1614)

* fix: avoid dispose unmatched handlers

* fix: fix for star handlers

---------

Co-authored-by: Dirk Bäumer <[email protected]>

Author: yf-yang

jsonrpc/src/common/connection.ts

@@ -1327,8 +1327,10 @@ export function createMessageConnection(messageReader: MessageReader, messageWri
 			return {
 				dispose: () => {
 					if (method !== undefined) {
-						notificationHandlers.delete(method);
-					} else {
+						if (notificationHandlers.get(method)?.handler === handler) {
+							notificationHandlers.delete(method);
+						}
+					} else if (starNotificationHandler === type) {
 						starNotificationHandler = undefined;
 					}
 				}
@@ -1341,7 +1343,9 @@ export function createMessageConnection(messageReader: MessageReader, messageWri
 			progressHandlers.set(token, handler);
 			return {
 				dispose: () => {
-					progressHandlers.delete(token);
+					if (progressHandlers.get(token) === handler) {
+						progressHandlers.delete(token);
+					}
 				}
 			};
 		},
@@ -1487,8 +1491,10 @@ export function createMessageConnection(messageReader: MessageReader, messageWri
 						return;
 					}
 					if (method !== undefined) {
-						requestHandlers.delete(method);
-					} else {
+						if (requestHandlers.get(method)?.handler === handler) {
+							requestHandlers.delete(method);
+						}
+					} else if (starRequestHandler === type) {
 						starRequestHandler = undefined;
 					}
 				}