如何使用 React Query 做下拉數據自動刷新?
useInfiniteQuery() API
看名字就能猜出來,useInfiniteQuery() 是專門用來應付無限查詢場景的。不僅如此,useInfiniteQuery() API 能力也是基于 useQuery() 的。
之前的文章中我們介紹了 useQuery() 的核心 API,為了找回印象,我們在此貼出來:
import { useQuery } from 'react-query'
const {
data,
error,
isError,
isFetching,
isLoading,
isRefetching,
isSuccess,
refetch,
} = useQuery(queryKey, queryFn?, {
enabled,
onError,
onSuccess,
refetchOnWindowFocus,
retry,
staleTime,
})
如果我們把這些 API 簡化如下:
const {
...result,
} = useQuery(queryKey, queryFn?, {
...options,
})
useInfiniteQuery() 其實就是在 useQuery() 基礎之上增添一些無限查詢場景的參數:
const {
fetchNextPage,
fetchPreviousPage,
hasNextPage,
hasPreviousPage,
isFetchingNextPage,
isFetchingPreviousPage,
...result
} = useInfiniteQuery(queryKey, ({ pageParam = 1 }) => fetchPage(pageParam), {
...options,
getNextPageParam: (lastPage, allPages) => lastPage.nextCursor,
getPreviousPageParam: (firstPage, allPages) => firstPage.prevCursor,
})
如你所見,增加的 API 其實就是跟上一頁/下一頁查詢動作相關的參數,相比較于自己組裝 的分頁查詢能力的 useQuery(),useInfiniteQuery() 需要配置上一頁/下一頁的參數獲取函數,并提供了相應的查詢調用能力,更加自動化和便捷。
當然,增加的不只是參數,還有 2 處:
一個是 queryFn 參數的入參,多了一個名為 pageParam 的參數。
pageParam 表示當前頁數。這個值是每次 useInfiniteQuery() 調用時,通過 getNextPageParam()/getPreviousPageParam() 返回值自動獲取并傳入 queryFn 的。
第二個還有返回值的數據結構,即 data。
const { data } = useInfiniteQuery()
原來 data 就是表示內部請求方法的返回值。而 useInfiniteQuery() 的返回 data 因為要包含多頁數據(展示舊數據時,還要持有舊數據),因此 data 變更為:
data: { pages: TData[], pageParams: unknown[] }
pages 很好理解,就是用來承載過程中請求到的多頁數據;pageParams 則是每個頁面當時在做數據獲取時使用的查詢參數。
簡單一例
當然語言上說得再多,也是蒼白無力的,實踐出真知。這里我們就舉一個簡單的例子說明 useInfiniteQuery() 的使用。
首先,我們先創建一個獲取數據的請求函數(使用 Fetch API)。
const getPosts = async (pageParam) => {
return fetch(`https://jsonplaceholder.typicode.com/posts?_page=${pageParam.page}&_limit=${pageParam.size}`).then(res => res.json())
}
接著,使用 useInfiniteQuery() 請求數據:
function Example() {
const {
isLoading,
isError,
error,
data,
} = useInfiniteQuery(
'posts',
({ pageParam }) => getPosts(pageParam),
{
getNextPageParam: (lastPage, allPages) => ({ page: allPages.length + 1, size: 6 }),
refetchOnWindowFocus: false, // Prevent refetching on window focus
}
)
// ...
}
增加下加載中或出現異常時的處理邏輯。
function Example() {
// ...
if (isLoading) {
return <div>Loading...</div>;
}
if (isError) {
return <div>Error: {error.message}</div>
}
// ...
}
最后渲染分頁數據。
function Example() {
// ...
return (
<div>
<ol>
{/* (1) */}
{data.pages.map((page) => (
{page.map((post) => (
<li key={post.id}>{post.title}</li>
))}
))}
</ol>
{/* (2) */}
<button onClick={() => fetchNextPage()}>More</button>
</div>
)
}
- 遍歷 data.pages 中所有頁面數據,渲染出來
- 使用 fetchNextPage() 函數加載更多(實際上即“下一頁”)數據
瀏覽器訪問,不幸運是,報錯了。
圖片
沒關系,點擊查看報錯處。
圖片
發現是是由于首次請求接口時,pageParam 參數是 undefined,這就需要我們給 pageParam 一個默認值。
- const getPosts = async (pageParam) => {
+ const getPosts = async (pageParam = { page: 1, size: 6 }) => {
return fetch(`https://jsonplaceholder.typicode.com/posts?_page=${pageParam.page}&_limit=${pageParam.size}`).then(res => res.json())
}
再次訪問頁面,發現我們成功請求到了第一頁數據。
圖片
接下來點擊“More”按鈕,發現發起了到第二頁數據的請求。
圖片
圖片
這里邏輯是先調用 getNextPageParam() 函數獲得請求參數。getNextPageParam() 函數的類型聲明如下:
getNextPageParam: (lastPage, allPages) => unknown | undefined
第一個參數就是上一頁數據,第二個參數即到目前為止請求到的所有數據。對照本例,當我們第一次點擊“More”按鈕時,返回的值為 { page:2, size:6 }。然后,再發起 getPosts 調用并這個參數,于是我們便發起了第二頁的數據請求,并最終得到了數據。
下面,我們再給“More”按鈕增加一個加載狀態——我們可以使用通用的 isFetching 狀態,但這里我們使用更具體的 isFetchingNextPage 狀態。
function Example() {
const {
isLoading,
+ isFetchingNextPage,
isError,
error,
data,
} = useInfiniteQuery()
添加給按鈕。
- <button notallow={() => fetchNextPage()}>More</button>
+ <button disabled={isFetchingNextPage} notallow={() => fetchNextPage()}>More</button>
查看效果。
圖片
這樣,我們就能在請求下一頁數據的時候禁用按鈕點擊,避免用戶誤觸。
再來一例
以上案例中,我們是通過 getNextPageParam() 所提供的第二個參數 allPages 來獲得下一頁頁碼的。
getNextPageParam: (lastPage, allPages) => ({ page: allPages.length + 1, size: 6 })
這是因為返回的第一頁數據總并未包含下一頁信息。
圖片
我們再稍微修改下
const getPosts = async (pageParam = { page: 1, size: 6 }) => {
return fetch(`https://jsonplaceholder.typicode.com/posts?_page=${pageParam.page}&_limit=${pageParam.size}`).then(res => {
const total = res.headers.get('X-Total-Count')
return res.json().then(data => {
return {
total,
data,
hasMore: pageParam.page * pageParam.size < total
}
})
})
}
我們將返回數據放在 data 屬性中,并返回數據總數(total)以及是否還有數據(hasMore)。
有了這些信息,我們就可以有了更加細膩的交互了。
下面,修改渲染邏輯。
function Example() {
// ...
return (
<div>
<ol>
{data.pages.map((page) => (
<>
{page.data.map((post) => (
<li key={post.id}>{post.title}</li>
))}
</>
))}
</ol>
<p>總共 <strong>{data.pages[0].total}</strong> 條數據</p>
{
data.pages[data.pages.length - 1].hasMore ? (
<button disabled={isFetchingNextPage} onClick={() => fetchNextPage()}>More</button>
) : <span>--- 我是有底線的 ---</span>
}
</div>
)
}
查看初始加載效果。
圖片
查看最終效果。
圖片
不過,按鈕顯隱邏輯還是有些冗余。
{
data.pages[data.pages.length - 1].hasMore ? (
<button disabled={isFetchingNextPage} onClick={() => fetchNextPage()}>More</button>
) : <span>--- 我是有底線的 ---</span>
}
這一點 useInfiniteQuery 也幫我們想到了,便提供了一個 hasNextPage 供我們使用。
{
- data.pages[data.pages.length - 1].hasMore ? (
+ hasNextPage ? (
<button disabled={isFetchingNextPage} notallow={() => fetchNextPage()}>More</button>
) : <span>--- 我是有底線的 ---</span>
}
hasNextPage 為 true 表示有下一頁數據,為 false 表示沒有數據了。
不過 hasNextPage 的值是受到 getNextPageParam() 返回值影響的——當 getNextPageParam() 返回 undefined 時,hasNextPage 值為 false,否則為 true。
因此,我們修改下 getNextPageParam() 的判斷邏輯。
getNextPageParam: (lastPage, allPages) => {
return lastPage.hasMore
? { page: allPages.length + 1, size: 50 }
: undefined
},
如此,我們再來查看下效果。
圖片
發現沒有數據的時候,同樣也不會展示“More”按鈕了。大功告成!
下拉拉取新數據
當然,以上通過點擊“More”按鈕加載新數據的方式還是太麻煩了一些。如果我們在網頁下拉到底部的時候自動獲取數據不是更好嗎?
下面就來實現一下。
我們將借助 IntersectionObserver API[4] + 底部一個 .loadMore 元素來實現。
首先,將 “More” 按鈕部分的代碼替換成:
<div className="loadMore" style={{ height: '30px', lineHeight: '30px' }} ref={loadMoreRef}>
{
isFetchingNextPage
? <span>Loading...</span>
: <span>--- 我是有底線的 ---</span>
}
</div>
注意,我們同時通過 loadMoreRef 拿到 DOM 元素。
然后,我們監聽 .loadMore 元素,一旦出現在屏幕中,就調用 fetchNextPage() 獲取下一頁數據。
useEffect(() => {
const observer = new IntersectionObserver((entries) => {
if (entries[0].isIntersecting && hasNextPage) {
fetchNextPage();
}
});
if (loadMoreRef.current) {
observer.observe(loadMoreRef.current);
}
return () => observer.disconnect();
}, [hasNextPage, fetchNextPage]);
來查看效果。
圖片
完美。
最后,再把完整代碼貼出來,方便大家學習。
import { useEffect, useRef } from 'react'
import { QueryClient, QueryClientProvider, useInfiniteQuery } from 'react-query'
// Create a client
const queryClient = new QueryClient()
export default function App() {
return (
// Provide the client to your App
<QueryClientProvider client={queryClient}>
<Example />
</QueryClientProvider>
)
}
const getPosts = async (pageParam = { page: 1, size: 25 }) => {
return fetch(`https://jsonplaceholder.typicode.com/posts?_page=${pageParam.page}&_limit=${pageParam.size}`).then(res => {
const total = res.headers.get('X-Total-Count')
return res.json().then(data => {
return {
total,
data,
hasMore: pageParam.page * pageParam.size < total
}
})
})
}
function Example() {
const {
isLoading,
isFetchingNextPage,
hasNextPage,
isError,
error,
data,
fetchNextPage
} = useInfiniteQuery(
'posts',
({ pageParam }) => getPosts(pageParam),
{
getNextPageParam: (lastPage, allPages) => {
return lastPage.hasMore ? { page: allPages.length + 1, size: 25 } : undefined
},
refetchOnWindowFocus: false, // Prevent refetching on window focus
}
)
const loadMoreRef = useRef(null);
useEffect(() => {
const observer = new IntersectionObserver((entries) => {
if (entries[0].isIntersecting && hasNextPage) {
fetchNextPage();
}
});
if (loadMoreRef.current) {
observer.observe(loadMoreRef.current);
}
return () => observer.disconnect();
}, [hasNextPage, fetchNextPage]);
if (isLoading) {
return <div>Loading...</div>;
}
if (isError) {
return <div>Error: {error.message}</div>
}
return (
<div>
<p>總共 <strong>{data.pages[0].total}</strong> 條數據</p>
<ol>
{data.pages.map((page) => (
<>
{page.data.map((post) => (
<li key={post.id}>{post.title}</li>
))}
</>
))}
</ol>
<div className="loadMore" style={{ height: '30px', lineHeight: '30px' }} ref={loadMoreRef}>
{
isFetchingNextPage ? <span>Loading...</span> : <span>--- 我是有底線的 ---</span>
}
</div>
</div>
)
}
總結
本文我們講述了 React Query 中用于無限查詢 API useInfiniteQuery() 的使用。
通過循序漸進的 3 個案例,最終實現了一個下拉到底后自動新數據的交互效果,還是比較好實現的。
當然,本文只是以“下一頁”舉例,“上一頁”與此同理。
希望本位講述的內容能夠對你的工作有所幫助。感謝閱讀,再見。
參考資料
[1]React Query 是做什么的?: https://juejin.cn/post/7378015213348257855
[2]一個數據獲竟然被 React Query 玩出這么多花樣來!: https://juejin.cn/post/7380342160581918731
[3]React Query 的 useQuery 竟也內置了分頁查詢支持!: https://juejin.cn/post/7380569775686746151
[4]IntersectionObserver API: https://ruanyifeng.com/blog/2016/11/intersectionobserver_api.html