@prometheus-ags/prometheus-entity-management 1.2.0 → 1.2.1

package/dist/index.js

@@ -1180,10 +1180,11 @@ function useEntity(opts) {
   fetchRef.current = opts.fetch;
   const normalizeRef = React6.useRef(opts.normalize);
   normalizeRef.current = opts.normalize;
-  const data = zustand.useStore(useGraphStore, shallow.useShallow((state) => {
+  const dataSelector = React6.useCallback((state) => {
     if (!id) return null;
     return state.readEntitySnapshot(type, id);
-  }));
+  }, [id, type]);
+  const data = zustand.useStore(useGraphStore, shallow.useShallow(dataSelector));
   const entityState = zustand.useStore(useGraphStore, React6.useCallback(
     (state) => state.entityStates[`${type}:${id}`] ?? EMPTY_ENTITY_STATE,
     [type, id]
@@ -1216,13 +1217,11 @@ function useEntityList(opts) {
   const normalizeRef = React6.useRef(opts.normalize);
   normalizeRef.current = opts.normalize;
   const listState = zustand.useStore(useGraphStore, React6.useCallback((state) => state.lists[key] ?? EMPTY_LIST_STATE, [key]));
-  const items = zustand.useStore(
-    useGraphStore,
-    shallow.useShallow((state) => {
-      const ids = state.lists[key]?.ids ?? EMPTY_IDS;
-      return ids.map((id) => state.readEntitySnapshot(type, id)).filter((x) => x !== null);
-    })
-  );
+  const itemsSelector = React6.useCallback((state) => {
+    const ids = state.lists[key]?.ids ?? EMPTY_IDS;
+    return ids.map((id) => state.readEntitySnapshot(type, id)).filter((x) => x !== null);
+  }, [key, type]);
+  const items = zustand.useStore(useGraphStore, shallow.useShallow(itemsSelector));
   const doFetch = React6.useCallback((params = {}) => {
     if (!enabled) return;
     fetchList({ type, queryKey, mode, fetch: fetchRef.current, normalize: normalizeRef.current }, params, getEngineOptions(), false);