Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
76 changes: 30 additions & 46 deletions packages/query-core/src/queriesObserver.ts
Original file line number Diff line number Diff line change
Expand Up @@ -207,59 +207,43 @@
#findMatchingObservers(
queries: Array<QueryObserverOptions>,
): Array<QueryObserverMatch> {
const prevObservers = this.#observers
const prevObserversMap = new Map(
prevObservers.map((observer) => [observer.options.queryHash, observer]),
this.#observers.map((observer) => [observer.options.queryHash, observer]),
)

const defaultedQueryOptions = queries.map((options) =>
this.#client.defaultQueryOptions(options),
)

const matchingObservers: Array<QueryObserverMatch> =
defaultedQueryOptions.flatMap((defaultedOptions) => {
const match = prevObserversMap.get(defaultedOptions.queryHash)
if (match != null) {
return [{ defaultedQueryOptions: defaultedOptions, observer: match }]
}
return []
})

const matchedQueryHashes = new Set(
matchingObservers.map((match) => match.defaultedQueryOptions.queryHash),
)
const unmatchedQueries = defaultedQueryOptions.filter(
(defaultedOptions) => !matchedQueryHashes.has(defaultedOptions.queryHash),
)
const observers: Array<QueryObserverMatch> = []

const getObserver = (options: QueryObserverOptions): QueryObserver => {
queries.forEach((options) => {
const defaultedOptions = this.#client.defaultQueryOptions(options)
const currentObserver = this.#observers.find(
(o) => o.options.queryHash === defaultedOptions.queryHash,
)
const match = prevObserversMap.get(defaultedOptions.queryHash)
if (match) {
observers.push({
defaultedQueryOptions: defaultedOptions,
observer: match,
})
} else {
const existingObserver = this.#observers.find(
(o) => o.options.queryHash === defaultedOptions.queryHash,

Check warning on line 226 in packages/query-core/src/queriesObserver.ts

View check run for this annotation

Codecov / codecov/patch

packages/query-core/src/queriesObserver.ts#L226

Added line #L226 was not covered by tests
)
observers.push({
defaultedQueryOptions: defaultedOptions,
observer:
existingObserver ??
new QueryObserver(this.#client, defaultedOptions),
})
}
})

return observers.sort((a, b) => {
return (
currentObserver ?? new QueryObserver(this.#client, defaultedOptions)
queries.findIndex(
(q) => q.queryHash === a.defaultedQueryOptions.queryHash,
) -
queries.findIndex(
(q) => q.queryHash === b.defaultedQueryOptions.queryHash,
)
)
}

const newOrReusedObservers: Array<QueryObserverMatch> =
unmatchedQueries.map((options) => {
return {
defaultedQueryOptions: options,
observer: getObserver(options),
}
})

const sortMatchesByOrderOfQueries = (
a: QueryObserverMatch,
b: QueryObserverMatch,
): number =>
defaultedQueryOptions.indexOf(a.defaultedQueryOptions) -
defaultedQueryOptions.indexOf(b.defaultedQueryOptions)

return matchingObservers
.concat(newOrReusedObservers)
.sort(sortMatchesByOrderOfQueries)
})
}

#onUpdate(observer: QueryObserver, result: QueryObserverResult): void {
Expand Down
Loading