useLoadMore.ts 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  1. import { getToken } from "@/utils/Cookies";
  2. import { useState } from "react";
  3. /**
  4. * @description 加载更多 hooks
  5. * @param {Function} func 请求方法
  6. * @param {Object} params 请求参数
  7. * @return haMore 是否有更多
  8. * @return loadMore 加载更多方法
  9. * @return getData 获取数据方法
  10. */
  11. export default function useLoadMore(func: Function, params: any) {
  12. const token = getToken();
  13. const [hasMore, setHasMore] = useState(true);
  14. const loadMore = async () => {
  15. await func({
  16. ...params.current,
  17. });
  18. };
  19. const getData = async (params: any) => {
  20. if (!token) {
  21. setHasMore(false);
  22. return;
  23. }
  24. try {
  25. const res = await func(params);
  26. // setDataSource((val) => ({ ...val, list: [...val.list, ...res.data] }));
  27. setHasMore(!res.page.is_end!);
  28. } catch (error) {
  29. setHasMore(false);
  30. return Promise.reject();
  31. }
  32. };
  33. return {
  34. hasMore,
  35. loadMore,
  36. getData,
  37. };
  38. }