refactor: optimize process for finding the fastest endpoint

packages/extension-polkagate/src/util/misc.ts

@@ -167,21 +167,8 @@ export function extractBaseUrl (url: string | undefined) {
 export async function fastestConnection (endpoints: DropdownOption[]): Promise<FastestConnectionType> {
   try {
     const urls = endpoints.map(({ value }) => ({ value: value as string }));
-    const { api, connections } = await fastestEndpoint(urls);
 
-    // @ts-ignore
-    // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
-    const selectedEndpoint = api.registry.knownTypes.provider.endpoint as string;
-    const connectionsToDisconnect = connections.filter(({ wsProvider }) => wsProvider.endpoint !== selectedEndpoint);
-
-    connectionsToDisconnect.forEach(({ wsProvider }) => {
-      wsProvider.disconnect().catch(console.error);
-    });
-
-    return {
-      api,
-      selectedEndpoint
-    };
+    return await fastestEndpoint(urls);
   } catch (error) {
     console.error('Unable to make an API connection!', error);
 
@@ -219,4 +206,4 @@ export const removeZeroBalanceRecords = (toBeSavedAssets: SavedAssets): SavedAss
   });
 
   return _toBeSavedAssets;
-};
+};

packages/extension-polkagate/src/util/workers/shared-helpers/getAssetOnAssetHub.js

@@ -2,7 +2,7 @@
 // SPDX-License-Identifier: Apache-2.0
 
 import { FETCHING_ASSETS_FUNCTION_NAMES } from '../../constants';
-import { closeWebsockets, fastestEndpoint, getChainEndpoints, metadataFromApi, toGetNativeToken } from '../utils';
+import { fastestEndpoint, getChainEndpoints, metadataFromApi, toGetNativeToken } from '../utils';
 import { getAssets } from './getAssets.js';
 
 /**
@@ -15,7 +15,7 @@ import { getAssets } from './getAssets.js';
  */
 export async function getAssetOnAssetHub (addresses, assetsToBeFetched, chainName, userAddedEndpoints, port) {
   const endpoints = getChainEndpoints(chainName, userAddedEndpoints);
-  const { api, connections } = await fastestEndpoint(endpoints);
+  const { api } = await fastestEndpoint(endpoints);
 
   const { metadata } = metadataFromApi(api);
 
@@ -42,5 +42,6 @@ export async function getAssetOnAssetHub (addresses, assetsToBeFetched, chainNam
 
   console.info(chainName, ': account assets fetched.');
   port.postMessage(JSON.stringify({ functionName: FETCHING_ASSETS_FUNCTION_NAMES.ASSET_HUB, results }));
-  closeWebsockets(connections);
+
+  api.disconnect().catch(console.error);
 }

packages/extension-polkagate/src/util/workers/shared-helpers/getAssetOnMultiAssetChain.js

@@ -8,7 +8,7 @@ import { getSubstrateAddress } from '../../address';
 import { FETCHING_ASSETS_FUNCTION_NAMES } from '../../constants';
 import { toTitleCase } from '../../string';
 // eslint-disable-next-line import/extensions
-import { balancifyAsset, closeWebsockets, fastestEndpoint, getChainEndpoints, metadataFromApi, toGetNativeToken } from '../utils';
+import { balancifyAsset, fastestEndpoint, getChainEndpoints, metadataFromApi, toGetNativeToken } from '../utils';
 
 /**
  *
@@ -20,7 +20,7 @@ import { balancifyAsset, closeWebsockets, fastestEndpoint, getChainEndpoints, me
  */
 export async function getAssetOnMultiAssetChain (assetsToBeFetched, addresses, chainName, userAddedEndpoints, port) {
   const endpoints = getChainEndpoints(chainName, userAddedEndpoints);
-  const { api, connections } = await fastestEndpoint(endpoints);
+  const { api } = await fastestEndpoint(endpoints);
 
   const { metadata } = metadataFromApi(api);
 
@@ -37,7 +37,6 @@ export async function getAssetOnMultiAssetChain (assetsToBeFetched, addresses, c
       return;
     }
 
-    // @ts-ignore
     const [formatted, assetIdRaw] = entry[0].toHuman() ?? [];
 
     let assetId;
@@ -51,7 +50,6 @@ export async function getAssetOnMultiAssetChain (assetsToBeFetched, addresses, c
 
     const storageKey = entry[0].toString();
 
-    // @ts-ignore
     let maybeAssetInfo = assetsToBeFetched.find((_asset) => {
       const currencyId = _asset?.extras?.['currencyIdScale'].replace('0x', '');
 
@@ -79,7 +77,6 @@ export async function getAssetOnMultiAssetChain (assetsToBeFetched, addresses, c
     }
 
     if (maybeAssetInfo) {
-      // @ts-ignore
       const totalBalance = balance.free.add(balance.reserved);
 
       const asset = {
@@ -97,7 +94,6 @@ export async function getAssetOnMultiAssetChain (assetsToBeFetched, addresses, c
 
       const address = getSubstrateAddress(formatted);
 
-      // @ts-ignore
       results[address]?.push(asset) ?? (results[address] = [asset]);
     } else {
       console.info(`NOTE: There is an asset on ${chainName} for ${formatted} which is not whitelisted. assetInfo`, storageKey, balance?.toHuman());
@@ -106,5 +102,6 @@ export async function getAssetOnMultiAssetChain (assetsToBeFetched, addresses, c
 
   console.info(chainName, ': account assets fetched.');
   port.postMessage(JSON.stringify({ functionName: FETCHING_ASSETS_FUNCTION_NAMES.MULTI_ASSET, results }));
-  closeWebsockets(connections);
+
+  api.disconnect().catch(console.error);
 }

packages/extension-polkagate/src/util/workers/shared-helpers/getAssetOnRelayChain.js

@@ -3,7 +3,7 @@
 
 import { FETCHING_ASSETS_FUNCTION_NAMES, NATIVE_TOKEN_ASSET_ID, TEST_NETS } from '../../constants';
 import { getPriceIdByChainName } from '../../misc';
-import { balancify, closeWebsockets } from '../utils';
+import { balancify } from '../utils';
 import { getBalances } from './getBalances.js';
 
 /**
@@ -16,9 +16,9 @@ export async function getAssetOnRelayChain (addresses, chainName, userAddedEndpo
   const results = {};
 
   try {
-    const { api, balanceInfo, connectionsToBeClosed } = await getBalances(chainName, addresses, userAddedEndpoints, port) ?? {};
+    const { api, balanceInfo } = await getBalances(chainName, addresses, userAddedEndpoints, port) ?? {};
 
-    if (!api || !balanceInfo || !connectionsToBeClosed) {
+    if (!api || !balanceInfo) {
       return;
     }
 
@@ -46,7 +46,7 @@ export async function getAssetOnRelayChain (addresses, chainName, userAddedEndpo
       }];
     });
 
-    closeWebsockets(connectionsToBeClosed);
+    api.disconnect().catch(console.error);
   } catch (error) {
     console.error(`getAssetOnRelayChain: Error fetching balances for ${chainName}:`, error);
   } finally {

packages/extension-polkagate/src/util/workers/shared-helpers/getBalances.js

@@ -17,7 +17,7 @@ import { getStakingBalances } from './getStakingBalances';
  */
 export async function getBalances (chainName, addresses, userAddedEndpoints, port) {
   const chainEndpoints = getChainEndpoints(chainName, userAddedEndpoints);
-  const { api, connections } = await fastestEndpoint(chainEndpoints);
+  const { api } = await fastestEndpoint(chainEndpoints);
 
   if (api.isConnected && api.derive.balances) {
     const { metadata } = metadataFromApi(api);
@@ -49,7 +49,7 @@ export async function getBalances (chainName, addresses, userAddedEndpoints, por
       };
     });
 
-    return { api, balanceInfo: await Promise.all(requests), connectionsToBeClosed: connections };
+    return { api, balanceInfo: await Promise.all(requests) };
   }
 
   return undefined;

packages/extension-polkagate/src/util/workers/shared-helpers/getNFTs.js

@@ -5,7 +5,7 @@
 
 import { SUPPORTED_NFT_CHAINS } from '../../../fullscreen/nft/utils/constants';
 import { getFormattedAddress } from '../../address';
-import { closeWebsockets, fastestEndpoint, getChainEndpoints } from '../utils';
+import { fastestEndpoint, getChainEndpoints } from '../utils';
 
 const NFT_FUNCTION_NAME = 'getNFTs';
 
@@ -191,9 +191,9 @@ async function getNFTs (addresses) {
     const formattedAddresses = addresses.map((address) => getFormattedAddress(address, undefined, prefix));
     const endpoints = getChainEndpoints(name, undefined);
 
-    const { api, connections } = await fastestEndpoint(endpoints);
+    const { api } = await fastestEndpoint(endpoints);
 
-    return ({ api, chainName, connections, formattedAddresses, originalAddresses: addresses });
+    return ({ api, chainName, formattedAddresses, originalAddresses: addresses });
   });
 
   const apis = await Promise.all(apiPromises);
@@ -224,7 +224,9 @@ async function getNFTs (addresses) {
     return itemsByAddress;
   } finally {
     // Ensure all websocket connections are closed
-    apis.forEach(({ connections }) => closeWebsockets(connections));
+    apis.forEach(({ api }) => {
+      api.disconnect().catch(console.error);
+    });
   }
 }
 

packages/extension-polkagate/src/util/workers/shared-helpers/getPool.js

@@ -5,7 +5,7 @@ import { BN, BN_ZERO, bnMax, hexToString } from '@polkadot/util';
 
 import getChainName from '../../getChainName';
 import getPoolAccounts from '../../getPoolAccounts';
-import { closeWebsockets, fastestEndpoint, getChainEndpoints } from '../utils';
+import { fastestEndpoint, getChainEndpoints } from '../utils';
 
 /**
  * Get all information regarding a pool
@@ -31,7 +31,7 @@ export async function getPool (genesisHash, stakerAddress, id, port) {
   const chainName = getChainName(genesisHash);
   const endpoints = getChainEndpoints(chainName ?? '');
 
-  const { api, connections } = await fastestEndpoint(endpoints);
+  const { api } = await fastestEndpoint(endpoints);
 
   console.log(`getPool is called for ${stakerAddress} on chain ${chainName}`);
   id && console.log('getPool is called to fetch the pool with poolId:', id);
@@ -133,5 +133,5 @@ export async function getPool (genesisHash, stakerAddress, id, port) {
 
   port.postMessage(JSON.stringify({ functionName: 'getPool', results: JSON.stringify(poolInfo) }));
 
-  closeWebsockets(connections);
+  api.disconnect().catch(console.error);
 }

packages/extension-polkagate/src/util/workers/shared-helpers/getValidatorsInformation.js

@@ -6,7 +6,7 @@ import { hexToString } from '@polkadot/util';
 
 import { KUSAMA_GENESIS_HASH, POLKADOT_GENESIS_HASH } from '../../constants';
 import getChainName from '../../getChainName';
-import { closeWebsockets, fastestEndpoint, getChainEndpoints } from '../utils';
+import { fastestEndpoint, getChainEndpoints } from '../utils';
 
 const BATCH_SIZE = 50;
 
@@ -72,7 +72,7 @@ export default async function getValidatorsInformation (genesisHash, port) {
   const endpoints = getChainEndpoints(chainName);
 
   try {
-    const { api, connections } = await fastestEndpoint(endpoints);
+    const { api } = await fastestEndpoint(endpoints);
 
     console.log('getting validators information on ' + chainName);
 
@@ -84,14 +84,14 @@ export default async function getValidatorsInformation (genesisHash, port) {
 
     console.log('electedInfo, waitingInfo, currentEra fetched successfully');
 
-    // Close the initial connections to the relay chain
-    closeWebsockets(connections);
+    // Close the initial connection to the relay chain
+    api.disconnect().catch(console.error);
 
     // Start connect to the People chain endpoints in order to fetch identities
     console.log('Connecting to People chain endpoints...');
     const peopleChainName = getPeopleChainName(genesisHash);
     const peopleEndpoints = getChainEndpoints(peopleChainName);
-    const { api: peopleApi, connections: peopleConnections } = await fastestEndpoint(peopleEndpoints);
+    const { api: peopleApi } = await fastestEndpoint(peopleEndpoints);
 
     // Keep elected and waiting validators separate
     const electedValidatorsInfo = electedInfo.info;
@@ -136,7 +136,7 @@ export default async function getValidatorsInformation (genesisHash, port) {
     await processSubIdentities(peopleApi, waitingMayHaveSubId, waitingValidatorsInformation, waitingAccountSubInfo);
     await processParentIdentities(peopleApi, waitingAccountSubInfo, waitingValidatorsInformation);
 
-    closeWebsockets(peopleConnections);
+    peopleApi.disconnect().catch(console.error);
 
     const results = {
       eraIndex: Number(currentEra?.toString() || '0'),