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

package/dist/index.mjs

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