@metamask-previews/assets-controller 3.1.1-preview-17c76be24 → 3.1.1-preview-dbd93752f

package/CHANGELOG.md

@@ -7,6 +7,18 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+### Added
+
+- Add Sentry traces for Assets Health dashboard ([#8310](https://github.com/MetaMask/core/pull/8310))
+  - `Assets Data Source Timing` — per-source latency for each middleware in the fetch pipeline
+  - `Assets Data Source Error` — tracks middleware failures with source names and error counts
+  - `Assets Full Fetch` — end-to-end fetch timing with asset/price/chain/account counts
+  - `Assets Update Pipeline` — enrichment pipeline timing for pushed data source updates
+  - `Assets Subscription Error` — subscription failure tracking per data source
+  - `Assets State Size` — entry counts for balances, metadata, prices, and custom assets (once on app start)
+  - `Aggregated Balance Selector` — balance selector computation time with asset/network/account counts
+  - Add optional `trace` parameter to `getAggregatedBalanceForAccount` selector
+
 ### Changed
 
 - Bump `@metamask/transaction-controller` from `^63.1.0` to `^63.2.0` ([#8301](https://github.com/MetaMask/core/pull/8301))

package/dist/AssetsController.cjs

@@ -13,7 +13,7 @@ var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
-var _AssetsController_instances, _AssetsController_isEnabled, _AssetsController_isBasicFunctionality, _AssetsController_defaultUpdateInterval, _AssetsController_trace, _AssetsController_firstInitFetchReported, _AssetsController_uiOpen, _AssetsController_keyringUnlocked, _AssetsController_controllerMutex, _AssetsController_activeSubscriptions, _AssetsController_enabledChains, _AssetsController_selectedAccounts_get, _AssetsController_backendWebsocketDataSource, _AssetsController_accountsApiDataSource, _AssetsController_snapDataSource, _AssetsController_rpcDataSource, _AssetsController_stakedBalanceDataSource, _AssetsController_allBalanceDataSources_get, _AssetsController_priceDataSource, _AssetsController_detectionMiddleware, _AssetsController_tokenDataSource, _AssetsController_unsubscribeBasicFunctionality, _AssetsController_onActiveChainsUpdated, _AssetsController_initializeState, _AssetsController_extractEnabledChains, _AssetsController_normalizeChainReference, _AssetsController_subscribeToEvents, _AssetsController_updateActive, _AssetsController_registerActionHandlers, _AssetsController_handleActiveChainsUpdate, _AssetsController_executeMiddlewares, _AssetsController_resolveNativeAssetIds, _AssetsController_getNativeAssetIdsForEnabledChains, _AssetsController_getNativeAssetIdsForAccount, _AssetsController_ensureNativeBalancesDefaultZero, _AssetsController_updateState, _AssetsController_getAssetsFromState, _AssetsController_tokenStandardToAssetType, _AssetsController_start, _AssetsController_stop, _AssetsController_subscribeAssets, _AssetsController_subscribeAssetsBalance, _AssetsController_subscribeStakedBalance, _AssetsController_buildChainToAccountsMap, _AssetsController_subscribeDataSource, _AssetsController_unsubscribeDataSource, _AssetsController_buildDataRequest, _AssetsController_getEnabledChainsForAccount, _AssetsController_handleAccountGroupChanged, _AssetsController_handleEnabledNetworksChanged;
+var _AssetsController_instances, _AssetsController_isEnabled, _AssetsController_isBasicFunctionality, _AssetsController_defaultUpdateInterval, _AssetsController_trace, _AssetsController_firstInitFetchReported, _AssetsController_stateSizeReported, _AssetsController_emitTrace, _AssetsController_emitStateSizeTrace, _AssetsController_uiOpen, _AssetsController_keyringUnlocked, _AssetsController_controllerMutex, _AssetsController_activeSubscriptions, _AssetsController_enabledChains, _AssetsController_selectedAccounts_get, _AssetsController_backendWebsocketDataSource, _AssetsController_accountsApiDataSource, _AssetsController_snapDataSource, _AssetsController_rpcDataSource, _AssetsController_stakedBalanceDataSource, _AssetsController_allBalanceDataSources_get, _AssetsController_priceDataSource, _AssetsController_detectionMiddleware, _AssetsController_tokenDataSource, _AssetsController_unsubscribeBasicFunctionality, _AssetsController_onActiveChainsUpdated, _AssetsController_initializeState, _AssetsController_extractEnabledChains, _AssetsController_normalizeChainReference, _AssetsController_subscribeToEvents, _AssetsController_updateActive, _AssetsController_registerActionHandlers, _AssetsController_handleActiveChainsUpdate, _AssetsController_executeMiddlewares, _AssetsController_resolveNativeAssetIds, _AssetsController_getNativeAssetIdsForEnabledChains, _AssetsController_getNativeAssetIdsForAccount, _AssetsController_ensureNativeBalancesDefaultZero, _AssetsController_updateState, _AssetsController_getAssetsFromState, _AssetsController_tokenStandardToAssetType, _AssetsController_start, _AssetsController_stop, _AssetsController_subscribeAssets, _AssetsController_subscribeAssetsBalance, _AssetsController_subscribeStakedBalance, _AssetsController_buildChainToAccountsMap, _AssetsController_subscribeDataSource, _AssetsController_unsubscribeDataSource, _AssetsController_buildDataRequest, _AssetsController_getEnabledChainsForAccount, _AssetsController_handleAccountGroupChanged, _AssetsController_handleEnabledNetworksChanged;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AssetsController = exports.getDefaultAssetsControllerState = void 0;
 const base_controller_1 = require("@metamask/base-controller");
@@ -220,6 +220,8 @@ class AssetsController extends base_controller_1.BaseController {
         _AssetsController_trace.set(this, void 0);
         /** Whether we have already reported first init fetch for this session (reset on #stop). */
         _AssetsController_firstInitFetchReported.set(this, false);
+        /** Whether we have already reported state size for this session (reset on #stop). */
+        _AssetsController_stateSizeReported.set(this, false);
         /** Whether the client (UI) is open. Combined with #keyringUnlocked for #updateActive. */
         _AssetsController_uiOpen.set(this, false);
         /** Whether the keyring is unlocked. Combined with #uiOpen for #updateActive. */
@@ -369,23 +371,28 @@ class AssetsController extends base_controller_1.BaseController {
                 options.chainIds.length < __classPrivateFieldGet(this, _AssetsController_enabledChains, "f").size;
             const updateMode = options?.updateMode ?? (isPartialChainFetch ? 'merge' : 'full');
             await __classPrivateFieldGet(this, _AssetsController_instances, "m", _AssetsController_updateState).call(this, { ...response, updateMode });
-            if (__classPrivateFieldGet(this, _AssetsController_trace, "f") && !__classPrivateFieldGet(this, _AssetsController_firstInitFetchReported, "f")) {
+            const durationMs = Date.now() - startTime;
+            // Emit trace for every full fetch (Assets Health dashboard)
+            __classPrivateFieldGet(this, _AssetsController_instances, "m", _AssetsController_emitTrace).call(this, 'Assets Full Fetch', {
+                duration_ms: durationMs,
+                chain_count: chainIds.length,
+                account_count: accounts.length,
+                basic_functionality: __classPrivateFieldGet(this, _AssetsController_isBasicFunctionality, "f").call(this),
+                asset_count: response.assetsBalance
+                    ? Object.values(response.assetsBalance).reduce((sum, acct) => sum + Object.keys(acct).length, 0)
+                    : 0,
+                price_count: response.assetsPrice
+                    ? Object.keys(response.assetsPrice).length
+                    : 0,
+                ...durationByDataSource,
+            });
+            if (!__classPrivateFieldGet(this, _AssetsController_firstInitFetchReported, "f")) {
                 __classPrivateFieldSet(this, _AssetsController_firstInitFetchReported, true, "f");
-                const durationMs = Date.now() - startTime;
-                try {
-                    await __classPrivateFieldGet(this, _AssetsController_trace, "f").call(this, {
-                        name: 'AssetsController First Init Fetch',
-                        data: {
-                            duration_ms: durationMs,
-                            chain_ids: JSON.stringify(chainIds),
-                            ...durationByDataSource,
-                        },
-                        tags: { controller: 'AssetsController' },
-                    }, () => undefined);
-                }
-                catch {
-                    // Telemetry failure must not break.
-                }
+                __classPrivateFieldGet(this, _AssetsController_instances, "m", _AssetsController_emitTrace).call(this, 'AssetsController First Init Fetch', {
+                    duration_ms: durationMs,
+                    chain_ids: JSON.stringify(chainIds),
+                    ...durationByDataSource,
+                });
             }
         }
         const result = __classPrivateFieldGet(this, _AssetsController_instances, "m", _AssetsController_getAssetsFromState).call(this, accounts, chainIds, assetTypes);
@@ -711,6 +718,7 @@ class AssetsController extends base_controller_1.BaseController {
      * @param request - Optional original request for context when enriching
      */
     async handleAssetsUpdate(response, sourceId, request) {
+        const updateStart = Date.now();
         log('Assets updated from data source', {
             sourceId,
             hasBalance: Boolean(response.assetsBalance),
@@ -730,6 +738,16 @@ class AssetsController extends base_controller_1.BaseController {
             dataTypes: ['balance', 'metadata', 'price'],
         }, response);
         await __classPrivateFieldGet(this, _AssetsController_instances, "m", _AssetsController_updateState).call(this, enrichedResponse);
+        __classPrivateFieldGet(this, _AssetsController_instances, "m", _AssetsController_emitTrace).call(this, 'Assets Update Pipeline', {
+            source: sourceId,
+            duration_ms: Date.now() - updateStart,
+            has_balance: Boolean(response.assetsBalance),
+            has_price: Boolean(response.assetsPrice),
+            has_metadata: Boolean(enrichedResponse.assetsInfo),
+            balance_account_count: response.assetsBalance
+                ? Object.keys(response.assetsBalance).length
+                : 0,
+        });
     }
     // ============================================================================
     // CLEANUP
@@ -766,7 +784,44 @@ class AssetsController extends base_controller_1.BaseController {
     }
 }
 exports.AssetsController = AssetsController;
-_AssetsController_isEnabled = new WeakMap(), _AssetsController_isBasicFunctionality = new WeakMap(), _AssetsController_defaultUpdateInterval = new WeakMap(), _AssetsController_trace = new WeakMap(), _AssetsController_firstInitFetchReported = new WeakMap(), _AssetsController_uiOpen = new WeakMap(), _AssetsController_keyringUnlocked = new WeakMap(), _AssetsController_controllerMutex = new WeakMap(), _AssetsController_activeSubscriptions = new WeakMap(), _AssetsController_enabledChains = new WeakMap(), _AssetsController_backendWebsocketDataSource = new WeakMap(), _AssetsController_accountsApiDataSource = new WeakMap(), _AssetsController_snapDataSource = new WeakMap(), _AssetsController_rpcDataSource = new WeakMap(), _AssetsController_stakedBalanceDataSource = new WeakMap(), _AssetsController_priceDataSource = new WeakMap(), _AssetsController_detectionMiddleware = new WeakMap(), _AssetsController_tokenDataSource = new WeakMap(), _AssetsController_unsubscribeBasicFunctionality = new WeakMap(), _AssetsController_onActiveChainsUpdated = new WeakMap(), _AssetsController_instances = new WeakSet(), _AssetsController_selectedAccounts_get = function _AssetsController_selectedAccounts_get() {
+_AssetsController_isEnabled = new WeakMap(), _AssetsController_isBasicFunctionality = new WeakMap(), _AssetsController_defaultUpdateInterval = new WeakMap(), _AssetsController_trace = new WeakMap(), _AssetsController_firstInitFetchReported = new WeakMap(), _AssetsController_stateSizeReported = new WeakMap(), _AssetsController_uiOpen = new WeakMap(), _AssetsController_keyringUnlocked = new WeakMap(), _AssetsController_controllerMutex = new WeakMap(), _AssetsController_activeSubscriptions = new WeakMap(), _AssetsController_enabledChains = new WeakMap(), _AssetsController_backendWebsocketDataSource = new WeakMap(), _AssetsController_accountsApiDataSource = new WeakMap(), _AssetsController_snapDataSource = new WeakMap(), _AssetsController_rpcDataSource = new WeakMap(), _AssetsController_stakedBalanceDataSource = new WeakMap(), _AssetsController_priceDataSource = new WeakMap(), _AssetsController_detectionMiddleware = new WeakMap(), _AssetsController_tokenDataSource = new WeakMap(), _AssetsController_unsubscribeBasicFunctionality = new WeakMap(), _AssetsController_onActiveChainsUpdated = new WeakMap(), _AssetsController_instances = new WeakSet(), _AssetsController_emitTrace = function _AssetsController_emitTrace(name, data, tags = {
+    controller: 'AssetsController',
+}) {
+    if (!__classPrivateFieldGet(this, _AssetsController_trace, "f")) {
+        return;
+    }
+    try {
+        __classPrivateFieldGet(this, _AssetsController_trace, "f").call(this, { name, data, tags }, () => undefined).catch(() => {
+            // Telemetry failure must not break.
+        });
+    }
+    catch {
+        // Telemetry failure must not break.
+    }
+}, _AssetsController_emitStateSizeTrace = function _AssetsController_emitStateSizeTrace() {
+    if (!__classPrivateFieldGet(this, _AssetsController_trace, "f") || __classPrivateFieldGet(this, _AssetsController_stateSizeReported, "f")) {
+        return;
+    }
+    __classPrivateFieldSet(this, _AssetsController_stateSizeReported, true, "f");
+    const { assetsBalance: balances, customAssets, assetsInfo, assetsPrice, } = this.state;
+    let balanceEntries = 0;
+    for (const acct of Object.values(balances)) {
+        balanceEntries += Object.keys(acct).length;
+    }
+    let customAssetEntries = 0;
+    for (const ids of Object.values(customAssets)) {
+        if (Array.isArray(ids)) {
+            customAssetEntries += ids.length;
+        }
+    }
+    __classPrivateFieldGet(this, _AssetsController_instances, "m", _AssetsController_emitTrace).call(this, 'Assets State Size', {
+        balance_entries: balanceEntries,
+        balance_accounts: Object.keys(balances).length,
+        metadata_entries: Object.keys(assetsInfo).length,
+        price_entries: Object.keys(assetsPrice).length,
+        custom_asset_entries: customAssetEntries,
+    });
+}, _AssetsController_selectedAccounts_get = function _AssetsController_selectedAccounts_get() {
     return this.messenger.call('AccountTreeController:getAccountsFromSelectedAccountGroup');
 }, _AssetsController_allBalanceDataSources_get = function _AssetsController_allBalanceDataSources_get() {
     return [
@@ -896,11 +951,14 @@ async function _AssetsController_executeMiddlewares(sources, request, initialRes
             inclusive[i] = Date.now() - start;
         }
     }));
-    const chain = wrapped.reduceRight((next, middleware) => async (ctx) => {
+    const middlewareErrors = [];
+    const chain = wrapped.reduceRight((next, middleware, index) => async (ctx) => {
         try {
             return await middleware(ctx, next);
         }
         catch (error) {
+            const sourceName = names[index] ?? `middleware_${index}`;
+            middlewareErrors.push(sourceName);
             console.error('[AssetsController] Middleware failed:', error);
             return next(ctx);
         }
@@ -924,6 +982,23 @@ async function _AssetsController_executeMiddlewares(sources, request, initialRes
             durationByDataSource[key] = ms;
         }
     }
+    // Emit per-source timing traces for the Assets Health dashboard
+    for (const [sourceName, durationMs] of Object.entries(durationByDataSource)) {
+        __classPrivateFieldGet(this, _AssetsController_instances, "m", _AssetsController_emitTrace).call(this, 'Assets Data Source Timing', {
+            source: sourceName,
+            duration_ms: durationMs,
+            chain_count: request.chainIds.length,
+            account_count: request.accountsWithSupportedChains.length,
+        });
+    }
+    // Emit error traces for failed middlewares
+    if (middlewareErrors.length > 0) {
+        __classPrivateFieldGet(this, _AssetsController_instances, "m", _AssetsController_emitTrace).call(this, 'Assets Data Source Error', {
+            failed_sources: middlewareErrors.join(','),
+            error_count: middlewareErrors.length,
+            chain_count: request.chainIds.length,
+        });
+    }
     return { response: result.response, durationByDataSource };
 }, _AssetsController_resolveNativeAssetIds = function _AssetsController_resolveNativeAssetIds(chains) {
     const { nativeAssetIdentifiers } = this.messenger.call('NetworkEnablementController:getState');
@@ -1064,6 +1139,8 @@ async function _AssetsController_executeMiddlewares(sources, request, initialRes
                 }
             }
         });
+        // Emit state size trace (throttled to avoid JSON.stringify on every update)
+        __classPrivateFieldGet(this, _AssetsController_instances, "m", _AssetsController_emitStateSizeTrace).call(this);
         // Calculate changed prices
         const changedPriceAssets = normalizedResponse.assetsPrice
             ? Object.keys(normalizedResponse.assetsPrice).filter((assetId) => !(0, lodash_1.isEqual)(previousPrices[assetId], normalizedResponse.assetsPrice?.[assetId]))
@@ -1211,6 +1288,7 @@ async function _AssetsController_executeMiddlewares(sources, request, initialRes
         hasPriceSubscription: __classPrivateFieldGet(this, _AssetsController_activeSubscriptions, "f").has('ds:PriceDataSource'),
     });
     __classPrivateFieldSet(this, _AssetsController_firstInitFetchReported, false, "f");
+    __classPrivateFieldSet(this, _AssetsController_stateSizeReported, false, "f");
     // Stop price subscription first (uses direct messenger call)
     this.unsubscribeAssetsPrice();
     // Stop balance subscriptions by properly notifying data sources via messenger
@@ -1327,6 +1405,10 @@ async function _AssetsController_executeMiddlewares(sources, request, initialRes
     };
     source.subscribe(subscribeReq).catch((error) => {
         console.error(`[AssetsController] Failed to subscribe to '${sourceId}':`, error);
+        __classPrivateFieldGet(this, _AssetsController_instances, "m", _AssetsController_emitTrace).call(this, 'Assets Subscription Error', {
+            source: sourceId,
+            error_message: String(error),
+        });
     });
     // Track subscription
     const subscription = {