如何查看RTK Query中同一端点的订阅者引用计数

方法1：读取RTK Query内部状态 #

RTK Query会在Redux store的api切片中维护所有端点的订阅状态，你可以直接从中读取对应端点的订阅者数量：

• 访问store.getState().api.subscriptions，该对象的键为端点的缓存标识，值包含subscribers数组，数组长度即为当前订阅该端点的组件数量。
• 示例代码：

// 假设API切片名称为'api'
const getSubscriberCount = (store, endpointName, cacheKey) => {
  const subscriptions = store.getState().api.subscriptions;
  const targetEntry = Object.values(subscriptions).find(
    entry => entry.endpointName === endpointName && entry.cacheKey === cacheKey
  );
  return targetEntry ? targetEntry.subscribers.length : 0;
};

• 说明：cacheKey是调用端点时传入参数生成的标识，比如调用useGetPostsQuery({page: 1})，对应的cacheKey通常为['getPosts', {page: 1}]（具体格式取决于配置）。

方法2：自定义订阅追踪逻辑 #

如果需要更灵活的计数管理，可以在API切片的端点定义中，利用onCacheEntryAdded生命周期钩子手动维护计数：

• 示例代码：

import { createApi, fetchBaseQuery } from '@reduxjs/toolkit/query/react';

// 全局存储各端点的订阅计数
const subscriberCounts = {};

export const api = createApi({
  baseQuery: fetchBaseQuery({ baseUrl: '/' }),
  endpoints: (builder) => ({
    getPosts: builder.query({
      query: (page) => `/posts?page=${page}`,
      async onCacheEntryAdded(
        arg,
        { cacheEntryRemoved }
      ) {
        const countKey = `getPosts-${JSON.stringify(arg)}`;
        // 订阅新增时计数+1
        subscriberCounts[countKey] = (subscriberCounts[countKey] || 0) + 1;
        
        // 订阅移除时计数-1
        await cacheEntryRemoved;
        subscriberCounts[countKey] -= 1;
        if (subscriberCounts[countKey] === 0) delete subscriberCounts[countKey];
      },
    }),
  }),
});

// 暴露获取计数的方法
export const getEndpointSubscriberCount = (endpointName, arg) => {
  const countKey = `${endpointName}-${JSON.stringify(arg)}`;
  return subscriberCounts[countKey] || 0;
};

注意事项 #

• 方法1依赖RTK Query内部状态结构，虽当前稳定，但需注意版本更新可能带来的结构变化。
• 方法2需手动处理订阅生命周期，确保计数逻辑正确，避免内存泄漏。

内容的提问来源于stack exchange，提问作者alithecodeguy