Skip to content

Commit 1244e5a

Browse files
author
Attila Cseh
committed
getQueueItemIds cache invalidation added
1 parent 360ff17 commit 1244e5a

File tree

2 files changed

+10
-4
lines changed

2 files changed

+10
-4
lines changed

invokeai/frontend/web/src/services/api/endpoints/queue.ts

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@ export const queueApi = api.injectEndpoints({
8282
'CurrentSessionQueueItem',
8383
'NextSessionQueueItem',
8484
'QueueCountsByDestination',
85+
'SessionQueueItemIdList',
8586
{ type: 'SessionQueueItem', id: LIST_TAG },
8687
{ type: 'SessionQueueItem', id: LIST_ALL_TAG },
8788
],
@@ -132,6 +133,7 @@ export const queueApi = api.injectEndpoints({
132133
invalidatesTags: [
133134
'SessionQueueStatus',
134135
'BatchStatus',
136+
'SessionQueueItemIdList',
135137
{ type: 'SessionQueueItem', id: LIST_TAG },
136138
{ type: 'SessionQueueItem', id: LIST_ALL_TAG },
137139
],
@@ -160,6 +162,7 @@ export const queueApi = api.injectEndpoints({
160162
'CurrentSessionQueueItem',
161163
'NextSessionQueueItem',
162164
'QueueCountsByDestination',
165+
'SessionQueueItemIdList',
163166
{ type: 'SessionQueueItem', id: LIST_TAG },
164167
{ type: 'SessionQueueItem', id: LIST_ALL_TAG },
165168
],
@@ -279,6 +282,7 @@ export const queueApi = api.injectEndpoints({
279282
return [];
280283
}
281284
const tags: ApiTagDescription[] = [
285+
'SessionQueueItemIdList',
282286
{ type: 'SessionQueueItem', id: result.item_id },
283287
{ type: 'BatchStatus', id: result.batch_id },
284288
];
@@ -314,6 +318,7 @@ export const queueApi = api.injectEndpoints({
314318
'SessionQueueStatus',
315319
'BatchStatus',
316320
'QueueCountsByDestination',
321+
'SessionQueueItemIdList',
317322
{ type: 'SessionQueueItem', id: LIST_TAG },
318323
{ type: 'SessionQueueItem', id: LIST_ALL_TAG },
319324
...batch_ids.map((id) => ({ type: 'BatchStatus', id }) satisfies ApiTagDescription),
@@ -378,6 +383,7 @@ export const queueApi = api.injectEndpoints({
378383
'CurrentSessionQueueItem',
379384
'NextSessionQueueItem',
380385
'QueueCountsByDestination',
386+
'SessionQueueItemIdList',
381387
{ type: 'SessionQueueItem', id: LIST_TAG },
382388
{ type: 'SessionQueueItem', id: LIST_ALL_TAG },
383389
...item_ids.map((id) => ({ type: 'SessionQueueItem', id }) satisfies ApiTagDescription),
@@ -441,10 +447,10 @@ export const queueApi = api.injectEndpoints({
441447
url: buildQueueUrl(`item_ids?${queryString.stringify(queryArgs)}`),
442448
method: 'GET',
443449
}),
444-
providesTags: (result, error, queryArgs) => [
445-
'QueueItemIdList',
450+
providesTags: (queryArgs) => [
446451
'FetchOnReconnect',
447-
{ type: 'QueueItemIdList', id: stableHash(queryArgs) },
452+
'SessionQueueItemIdList',
453+
{ type: 'SessionQueueItemIdList', id: stableHash(queryArgs) },
448454
],
449455
}),
450456
getQueueItemDTOsByItemIds: build.mutation<GetQueueItemDTOsByItemIdsResult, GetQueueItemDTOsByItemIdsArgs>({

invokeai/frontend/web/src/services/api/index.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ const tagTypes = [
3030
'ImageCollection',
3131
'ImageMetadataFromFile',
3232
'IntermediatesCount',
33+
'SessionQueueItemIdList',
3334
'SessionQueueItem',
3435
'SessionQueueStatus',
3536
'SessionProcessorStatus',
@@ -55,7 +56,6 @@ const tagTypes = [
5556
'StylePreset',
5657
'Schema',
5758
'QueueCountsByDestination',
58-
'QueueItemIdList',
5959
'Video',
6060
'VideoMetadata',
6161
'VideoList',

0 commit comments

Comments
 (0)