作者:天气好
https://juejin.cn/post/7299849645206781963
本文将通过介绍什么是 Hook、如何在 Vue 使用 Hook,以及在实践场景中如何封装自己的 Vue Hook,带你走进 Hook 的世界,写出更优雅的代码。如果你觉得这篇文章写的不错,可以点赞支持一下,如果文章中存在不足(代码量多,难免出现 bug,咳咳),欢迎在评论区指出!
什么是 Hook
Vue3 官方文档是这样定义组合式函数的。A "composable" is a function that leverages Vue's Composition API to encapsulate and reuse stateful logic.
,一个利用 Vue 的组合式 API 来封装和复用具有状态逻辑的函数。
这个概念借鉴自 React 的 Hook。在 16.8 的版本中,React 引入了 React Hook。这是一项特别强大的技术,通过封装有状态的函数,极大提高了组件的编写效率和维护性。在下文中也是使用 Hook 来替代“组合式函数”进行叙述。
在开发中,我们经常会发现一些可以重复利用的代码段,于是我们将其封装成函数以供调用。这类函数包括工具函数,但是又不止工具函数,因为我们可能也会封装一些重复的业务逻辑。以往,在前端原生开发中,我们封装的这些函数都是“无状态”的。为了建立数据与视图之间的联系,基于 MVC 架构的 React 框架和基于 MVVM 的 Vue 框架都引入了“状态”这一概念,状态是特殊的 JavaScript 变量,它的变化会引起视图的变化。在这类框架中,如果一个变量的变化不会引起视图的变化,那么它就是普通变量,如果一个变量已经被框架注册为状态,那么这个变量的变化就会引发视图的变化,我们称之为响应式变量。如果一个函数包含了状态(响应式变量),那么它就是一个 Hook 函数。
在具备“状态”的框架的基础上,才有 Hook 这一说。Hook 函数与普通函数的本质区别在于是否具备“状态”。
比如,在一个 Vue 项目中,我们可能同时引入了 lodash 库和 VueUse 库,这两个库都是提供一些方便的工具函数。工具函数库只引入一个不行吗,不会重复吗?或许不行,因为 lodash 的函数是无状态的,用来处理普通变量或者响应式变量中的数据部分,而 VueUse 提供的 api 都是 Hook。如果你的项目中既有普通变量又有响应式变量,你或许就会在同一个项目中同时接触到这两个库。
React 官方为我们提供了一些非常方便的 Hook 函数,比如 useState、useEffect(我们通常使用 use 作为前缀来标识 Hook 函数),但是这远远不够,或者说,它们足够通用但是不够具体。为了在具体业务下复用某些逻辑,我们往往会封装自己的 Hook,即自定义 Hook。为什么这里会反复提到 React 中呢?因为提到 Hook,就不可能避开 React。Hook 是 React 发扬光大的,使用 Hook 已经是 React 社区的主流。然而,只要框架具备“状态”这一概念,都可以使用 Hook 技术!下面文章将会介绍如何将 Hook 应用到 Vue 当中。
在 Vue 中使用Hook
下面我们来看一个简单的自定义 Hook(来自 Vue 官方文档):
需求:在页面实时显示鼠标的坐标。 实现:没有使用 Hook。
<script setup>
import { ref, onMounted, onUnmounted } from 'vue'
const x = ref(0)
const y = ref(0)
function update(event) {
x.value = event.pageX
y.value = event.pageY
}
onMounted(() => window.addEventListener('mousemove', update))
onUnmounted(() => window.removeEventListener('mousemove', update))
</script>
<template>Mouse position is at: {{ x }}, {{ y }}</template>
在没有封装的情况下,如果我们在另一个页面也需要这个功能,我们需要将代码复制过去。另外,可以看出,它声明了两个变量,并且在生命周期钩子 onMounted
和 onUnmounted
中书写了一些代码,如果这个页面需要更多的功能,那么会出现代码中存在很多变量、生命周期中存在很多逻辑写在一起的现象,使得这些逻辑混杂在一起,而使用 Hook 可以将其分隔开来(这也是为什么会有很多人使用 Hook 的原因,分离代码,提高可维护性!)
使用 Hook:
<script setup>
import { useMouse } from './mouse.js'
const { x, y } = useMouse()
</script>
<template>Mouse position is at: {{ x }}, {{ y }}</template>
可以发现,比原来的代码更加简洁,这时如果加入其它功能的变量,也不会觉得眼花缭乱了。
当然,我们需要在外部定义这个 Hook:
// mouse.js
import { ref, onMounted, onUnmounted } from 'vue'
// 按照惯例,组合式函数名以“use”开头
export function useMouse() {
// 被组合式函数封装和管理的状态
const x = ref(0)
const y = ref(0)
// 组合式函数可以随时更改其状态。
function update(event) {
x.value = event.pageX
y.value = event.pageY
}
// 一个组合式函数也可以挂靠在所属组件的生命周期上
// 来启动和卸载副作用
onMounted(() => window.addEventListener('mousemove', update))
onUnmounted(() => window.removeEventListener('mousemove', update))
// 通过返回值暴露所管理的状态
return { x, y }
}
或许,你可以试着去 VueUse 库找到别人封装好的 useMouse!
import { useMouse } from 'VueUse'
恭喜你,掌握了 VueUse 库的使用方法。如果需要其它 Hook,你可以先试着去官方文档VueUse | VueUse(https://vueuse.org/)查找,使用现成的函数,而不是自己去封装。
封装一(入门级的表格 Hook)
在前面,我们介绍完了 Hook 的概念,完成了一个简单的自定义 Hook,还学会了使用社区提供的大量现成的 Hook 函数(VueUse 库),接下来,我们将结合实际业务,完成我们自己的 Hook 函数!
场景分析
首先定义一个表格:
<template>
<el-table :data="tableData" style="width: 100%">
<el-table-column prop="date" label="Date" width="180" />
<el-table-column prop="name" label="Name" width="180" />
<el-table-column prop="address" label="Address" />
</el-table>
<button @click="refresh">refresh</button>
</template>
表格的数据通过 api 获取(一般写法):
<script lang="ts" setup>
import { onMounted, ref } from "vue";
import { getTableDataApi } from "./api.ts";
const tableData = ref([]);
const refresh=async () => {
const data = await getTableDataApi();
tableData.value = data;
}
onMounted(refresh);
</script>
模拟 api:
// api.ts
export const getTableDataApi = () => {
const data = [
{
date: '2016-05-03',
name: 'Tom',
address: 'No. 189, Grove St, Los Angeles',
},
{
date: '2016-05-02',
name: 'Tom',
address: 'No. 189, Grove St, Los Angeles',
},
{
date: '2016-05-04',
name: 'Tom',
address: 'No. 189, Grove St, Los Angeles',
},
{
date: '2016-05-01',
name: 'Tom',
address: 'No. 189, Grove St, Los Angeles',
},
]
return new Promise(resolve => {
setTimeout(() => {
resolve(data)
}, 100);
})
}
如果存在多个表格,我们的 js 代码会变得比较复杂:
<script lang="ts" setup>
import { onMounted, ref } from "vue";
import { getTableDataApi1, getTableDataApi2, getTableDataApi3 } from "./api.ts";
const tableData1 = ref([]);
const refresh1=async () => {
const data = await getTableDataApi1();
tableData1.value = data;
}
const tableData2 = ref([]);
const refresh2=async () => {
const data = await getTableDataApi2();
tableData2.value = data;
}
const tableData3 = ref([]);
const refresh3=async () => {
const data = await getTableDataApi3();
tableData3.value = data;
}
onMounted(refresh1);
</script>
封装实例
封装我们的 useTable:
// useTable.ts
import { ref } from 'vue'
export function useTable(api) {
const data = ref([])
const refresh = () => { api().then(res => data.value = res) };
refresh()
return [data, refresh]
}
改造代码:
<script lang="ts" setup>
import { onMounted, ref } from "vue";
import { getTableDataApi1, getTableDataApi2, getTableDataApi3 } from "./api.ts";
import { useTable } from './useTable.ts'
const [tableData1, refresh1] = useTable(getTableDataApi1);
const [tableData2, refresh2] = useTable(getTableDataApi2);
const [tableData3, refresh3] = useTable(getTableDataApi3);
onMounted(refresh1);
</script>
封装技巧 - Hook 返回值
一般自定义 Hook 有返回数组的,也有返回对象的,上面 useTable 使用了返回数组的写法,useMouse 使用了返回对象的写法。数组是对应位置命名的,可以方便重命名,对象对于类型和语法提示更加友好。两种写法都是可以替换的。
因为 Hook 返回对象或者数组,那么它一定是一个非 async 函数(async 函数一定返回 Promise),所以在 Hook 中,一般使用 then 而不是 await 来处理异步请求。
返回值如果是对象,一般在函数中通过 reactive 创建一个对象,最后通过 toRefs 导出,这样做的原因是可以产生批量的可以解构的 Ref 对象,以免在解构返回值时丢失响应性。
// 使用 reactive 和 toRefs 可以快速创建多个ref对象,并在解构后使用时不丢失其响应性和与原先数据的关联性
function usePaginaion(){
const pagination = reactive({
current: 1,
total: 0,
sizeOption,
size: sizeOption[0]
})
...
return {...toRefs(pagination)}
}
const { current,total } = usePagination()
封装二(支持分页查询)
需求分析
上面我们封装了一个简单的 hook,但是实际应用中并不会如此简单,下面我列出一个比较完整的 useTable 在实践中应该具备的功能,并在后续的文章部分完成它。
封装表格组件逻辑:
维护 api 的调用和刷新(已完成)
支持分页查询(页数、总条数、每页大小等)
支持 api 参数。
增加辅助功能(loading、立即执行等)
下面我们将对 useTable 进行改造,使其支持分页器。
先改造一些我们的 api,使其支持分页查询:
export const getTableDataApi = (page, limit) => {
const data = [
{
date: '2016-05-03',
name: 'Tom',
address: 'No. 189, Grove St, Los Angeles',
},
{
date: '2016-05-02',
name: 'Tom',
address: 'No. 189, Grove St, Los Angeles',
},
{
date: '2016-05-04',
name: 'Tom',
address: 'No. 189, Grove St, Los Angeles',
},
{
date: '2016-05-01',
name: 'Tom',
address: 'No. 189, Grove St, Los Angeles',
},
{
date: '2017-05-03',
name: 'Tom',
address: 'No. 189, Grove St, Los Angeles',
},
{
date: '2017-05-02',
name: 'Tom',
address: 'No. 189, Grove St, Los Angeles',
},
{
date: '2017-05-04',
name: 'Tom',
address: 'No. 189, Grove St, Los Angeles',
},
{
date: '2017-05-01',
name: 'Tom',
address: 'No. 189, Grove St, Los Angeles',
},
]
return new Promise(resolve => {
setTimeout(() => {
resolve({
total: data.length,
data: data.slice((page - 1) * limit, (page - 1) * limit + limit)
})
}, 100);
})
}
如果没有使用 Hook,我们的 vue 文件应该是这样的:
<template>
<el-table :data="tableData" style="width: 100%">
<el-table-column prop="date" label="Date" width="180" />
<el-table-column prop="name" label="Name" width="180" />
<el-table-column prop="address" label="Address" />
</el-table>
<button @click="refresh">refresh</button>
<!-- 分页器 -->
<el-pagination
v-model:current-page="current"
:page-size="size"
layout="total, prev, pager, next"
:page-sizes="sizeOption"
:total="total"
@size-change="handleSizeChange"
@current-change="handleCurrentChange"
/>
</template>
<script lang="ts" setup>
import { onMounted, ref } from "vue";
import { getTableDataApi } from "./api.ts";
const tableData = ref([]); // 表格数据
const current = ref(1); // 当前页数
const sizeOption = [10, 20, 50, 100, 200]; // 每页大小选项
const size = ref(sizeOption[0]); //每页大小
const total = ref(0); // 总条数
// 每页大小变化
const handleSizeChange = (size: number) => {
size.value = size;
current.value = 1;
// total.value = 0;
refresh();
};
// 页数变化
const handleCurrentChange = (page: number) => {
current.value = page;
// total.value = 0;
refresh();
};
const refresh = async () => {
const result = await getTableDataApi({
page: current.value,
limit: size.value,
});
tableData.value = result.data || [];
total.value = result.total || 0;
};
onMounted(refresh);
</script>
可以看出,如果存在多个表格,会创建很多套变量和重复的代码。
封装实例
先写个 usePagination:该钩子接受一个回调函数,当页数改变时就会调用该函数。
import { reactive } from "vue";
export function usePagination(
cb: any,
sizeOption: Array<number> = [10, 20, 50, 100, 200]
): any {
const pagination = reactive({
current: 1,
total: 0,
sizeOption,
size: sizeOption[0],
// 维护page和size(一般是主动触发)
onPageChange: (page: number) => {
pagination.current = page;
return cb();
},
onSizeChange: (size: number) => {
pagination.current = 1;
pagination.size = size;
return cb();
},
// 一般调用cb后会还会修改total(一般是被动触发)
setTotal: (total: number) => {
pagination.total = total;
},
reset() {
pagination.current = 1;
pagination.total = 0;
pagination.size = pagination.sizeOption[0];
},
});
return [
pagination,
pagination.onPageChange,
pagination.onSizeChange,
pagination.setTotal,
];
}
与 useTable 结合:代码非常简单,在调用 api 时传入参数,并在接受返回值时更新 data 和 total。这里我们的 refresh 函数是一个返回 Promise 的函数,能够支持在调用 refresh 处再链接 then 进行下一层处理。
export function useTable(api: (params: any) => Promise<T>) {
const [pagination, , , setTotal] = usePagination(() => refresh());
const data = ref([]);
const refresh = () => {
return api({ page: pagination.current, limit: pagination.size }).then(
(res) => {
data.value = res.data;
setTotal(res.total);
}
);
};
return [data, refresh, pagination];
}
注:我们新建一个文件 customHooks.js
并将 usePagination 和 useTable 放在里面。
使用 useTable:
<template>
<el-table :data="tableData" style="width: 100%">
<el-table-column prop="date" label="Date" width="180" />
<el-table-column prop="name" label="Name" width="180" />
<el-table-column prop="address" label="Address" />
</el-table>
<button @click="refresh">refresh</button>
<!-- 分页器 -->
<el-pagination
v-model:current-page="pagination.current"
:page-size="pagination.size"
layout="total, prev, pager, next"
:page-sizes="pagination.sizeOption"
:total="pagination.total"
@size-change="pagination.onSizeChange"
@current-change="pagination.onCurrentChange"
/>
</template>
<script lang="ts" setup>
import { onMounted, ref } from "vue";
import { getTableDataApi } from "./api.ts";
import { useTable } from './customHooks.ts'
const [tableData, refresh, pagination] = useTable(getTableDataApi);
onMounted(refresh);
</script>
封装三(支持不同接口字段)
封装分析
上面我们封装了一个“看起来”比较使用的 useTable 函数,但实际上,你会发现很多问题:
每次都要写 onMounted 来初始化数据。
接口接受的格式可能不一样,比如,页数的字段为"currentPage",而不是“page”。
接口返回的格式可能不一样,比如,返回的 data 并不在 refresh 方法定义的“data”上。
封装实例
接下来,我们通过增加 useTable 函数的参数,来解决上面所有问题!
import { get, has, defaults } from "lodash-es";
type keyPath = Array<string> | string;
export function useTable<T>(
api: (params: any) => Promise<T>,
options?: {
path?: { data?: keyPath; total?: keyPath; page?: string; size?: string };
immediate?: boolean;
}
) {
// 参数处理
defaults(options, {
path: { data: "data", total: "total", page: "page", size: "size" },
immediate: false,
});
const [pagination, , , setTotal] = () => refresh();
const data = ref([]);
const loading = ref(false)
const refresh = () => {
loading.value = true
return api({ [options?.path?.page]: pagination.current, [options?.path?.size]: pagination.size }).then(
(res) => {
data.value = get(res, options!.path?.data, []);
setTotal(get(res, options!.path?.total, 0));
// 友好提示
if (!has(res, options!.path?.data) || !has(res, options!.path?.total)) {
console.warn("useTable:响应数据缺少所需字段");
}
}.finally(() => {
loading.value = false
})
);
};
// 立即执行
options!.immediate && refresh();
return [data, refresh, loading, pagination];
}
这里引入了 lodash 库中的三个工具函数来辅助处理对象:
defaults,将后面参数的属性,赋值到第一个对象的值为 undefined 的属性上,用于初始化函数参数。
get,获取对象属性,如果为 undefined,使用第三个参数的值。
has,判断对象属性。
具体用法可以查看官方文档Lodash中文网(https://www.lodashjs.com/) 此外,还新增了 loading,可以挂载到 el-table 的 v-loading 上,展示数据加载中的效果。
<el-table v-loding="loading" ...>...</el-table>
改造后:不管接口接受的格式还是响应的格式字段是什么样的,都可以正常接收。设置 immediate 为 true,调用 useTable 时立即执行一遍 api,onMounted 都不用写了。
<script lang="ts" setup>
import { onMounted, ref } from "vue";
import { getTableDataApi } from "./api.ts";
import { useTable } from './customHooks.ts'
const [tableData, refresh, loading, pagination] = useTable(getTableDataApi, {
path: {
data: 'data',
total: 'total',
page: 'page',
size: 'limit'
},
immediate: true
});
// onMounted(refresh);
</script>
JavaScript 函数传参技巧
一般函数定义参数越少越好,最好不要超过两个,所以这里我只定义了两个参数 api 和options。
在函数头上可以给参数定义默认值,但是如果参数是一个对象,只要传入一个属性,就不会使用默认值,比如:
export function useTable<T>(
api: (params: any) => Promise<T>,
options: {
path?: { data?: keyPath; total?: keyPath; page?: string; size?: string };
immediate?: boolean;
} = {
path: { data: "data", total: "total", page: "page", size: "size" },
immediate: false,
}
){...函数体}
useTable(xxxApi,{immediate:false})
只要该位置的值非 undefined,那么 options 将不会使用默认值,这意味着,此时 options 的值为 {immediate:false}
,其它地方的默认值不会生效,{path:undefined,}
。 所以对于函数参数为对象的,我们往往通过在函数体内赋默认值,比如:
保证options只传入一个值,其它位置也会有默认值
{
options.path = options.path || {}
options.path.data = options.path.data || 'data'
options.path.total = options.path.total || 'total'
options.path.page = options.path.page || 'page'
options.path.size = options.path.size || 'size'
options.immediate = options.immediate ?? false
}
需要注意元素的层次,在不存在 path 时,给 path. data 赋值会出现错误,需要先保证 path 有值,才能给 path 的下一层赋值。
使用 defaults 可以快速给整个对象赋默认值:
defaults(options, {
path: { data: "data", total: "total", page: "page", size: "size" },
immediate: false,
});
封装四(接口传参-定义时)
封装分析
现在,我们的 useTable 趋近完整了:
维护 api 的调用和刷新(已完成)
支持分页查询(已完成)
支持 api 参数。
增加辅助功能 loading、立即执行等。(已完成)
我们还可以让我们的 api 接受参数。但是如何实现?还需要考虑一下。
首先我们想一想那里可以接受 api 的参数?
const params = {
id:2
}
// api本身
getTableDataApi({limit:3,page:2,...params})
// useTable也可以接受参数
const [data,refresh]=useTable(getTableDataApi,params,api)
// refresh也可以接受参数
refresh(params)
从使用上看,我们在 refresh 上接受参数,和我们在 getTableDataApi 的使用上感觉是最相似的,因为 refresh 本来就是在 api 的基础上增加 then 维护了页数而已。但是我们还是先从 useTable 传参开始讲起,最后我们两种方式都可以接受!
方案一:在调用 useTable 的时候就接受参数,在 useTable 内部将这个参数传给 refresh。 存在问题:如果我们传入的是值类型,那么这个值会被拷贝过去,并传给 refresh,后续调用 refresh,都是不变的参数。只适合需要传参但参数之后都不会变的接口,比如接受当前用户的 id。如果参数会变,这种方法是不行的。
function useTable(api,id,options){
...
const refresh=()=>api(id).then(res=>data=res)
return [data,refresh]
}
const [data,refresh]=useTable(api,id)
refresh()
refresh() // 都是id=2
如果我们传入的是引用类型,那么在后续调用中,我们可以通过改变对象的属性值来改变 refresh 的参数(但是需要一些技巧,因为我们需要和分页参数进行结合)。
const params = { id:12 }
function useTable(api,params,options){
...
// 错误,使用解构会丢失与原来对象的联系,导致原来的对象params更改,但这里仍使用旧值。
const refresh=()=>api({[options.path.size]:pagination.size,[options.path.page]:pagination.page,...params}).then(res=>data=res)
// 正确,可以保持与外部params的联系。
const refresh=()=>api(Object.assign(params,{[options.path.size]:pagination.size,[options.path.page]:pagination.page})).then(res=>data=res)
return [data,refresh]
}
const [data,refresh]=useTable(api,params)
refresh() // id=12
params.id = 10
refresh() // id=10
这样,我们就实现了 api 参数的传递,而且如果 params 的属性 id 是响应式的,还可以与页面结合,实现搜索功能!然而,使用同一个引用 params,可以解决传参问题,但是还是存在一些问题:在 refresh 中,Object. assign 会给原来的对象 params 增加两个属性,要注意避免在 params 中与这两个属性发生冲突。另外,我们可以看到这里的参数间存在了一种优先级,就是如果我们在 param 中也传入了分页参数,会在 refresh 中被 pagination 的分页参数覆盖调,pagination 的分页参数比 params 中的分页参数优先级更高,这样好吗?
第一个问题,在 refresh 中每次都会被 pagination 的属性覆盖,所以并不会出现什么问题,除非你在 params 上保存相同属性名的数据,这将被覆盖掉。第二个问题和第一个问题本质是一样的,就是覆盖问题。根本原因就是都是引用同一个对象。如果我们能够额外创建一个对象,就不会改变原来的对象,但是如何保持新创建对象能够动态变化呢?
方案二:试试 useTable 接受传入函数 params 如何?
const params={id:12}
const paramsFn =()=>{ id: params.id }
function useTable(api,paramsFn(),options){
...
const refresh=()=>api(Object.assign(paramsFn(),{[options.path.size]:pagination.size,[options.path.page]:pagination.page})).then(res=>data=res)
return [data,refresh]
}
const [data,refresh]=useTable(api,paramsFn)
refresh() // id=12
params.id = 10
refresh() // id=10
完美解决。
最后,兼容一下两种参数,让传入 useTable 的 api 参数既可以是函数,又可以是对象:
export function useTable<T>(
api: (params: any) => Promise<T>,
params?: object | (() => object),
options?: {
path?: { data?: keyPath; total?: keyPath; page?: string; size?: string }
immediate?: boolean
},
) {
// 参数处理
defaults(options, {
path: { data: 'data', total: 'total', page: 'page', size: 'size' },
immediate: false,
})
const [pagination, , , setTotal] = usePagination(() =>refresh())
const loading = ref(false)
const data = ref([])
const refresh = (extraData?: object | (() => object)) => {
const requestData = {
[options?.path?.page as string]: pagination.current,
[options?.path?.size as string]: pagination.size,
}
if (params) {
if (typeof params === 'function') {
Object.assign(requestData, params())
} else {
Object.assign(requestData, params)
}
}
loading.value = true
return api(requestData)
.then((res) => {
data.value = get(res, options!.path?.data, [])
setTotal(get(res, options!.path?.total, 0))
if (!has(res, options!.path?.data) || !has(res, options!.path?.total)) {
console.warn('useTable:响应数据缺少所需字段')
}
})
.finally(() => {
loading.value = false
})
}
options!.immediate && refresh()
return [data as T, refresh, loading, pagination]
}
这里代码主要新增了三处改变:
如果 params 是对象,直接使用,如果是函数,则读取其返回值。
优先级调整:paginaiton 的参数可以被 params 的同名属性覆盖,适用于开发者自己维护分页参数。
定义了返回值的类型。
使用示例
试想一个常见,点击列表的某一项,就展示列表对应 id 的表格,如何实现?
<template>
<ul>
// 自定义组件,点击时emit发送onClick事件并传入item的id
<Item v-for="item in list" :key="item.key" :label="item.label" @on-click="handleClick">
...
</ul>
</template>
<script>
...
// 这里接受item的id
const handleClick=(id:number)=>{
params.id=number;
refresh()
}
...
</script>
封装五(接口传参-调用时)
最后,来让 refresh 函数也能接受我们的传参。 先看效果:
<script>
...
// 这里接受item的id
const handleClick=(id:number)=>{
refresh({id})
}
...
</script>
可以省去 params 和 paramsFn 的定义了!
实现代码:在定义 refresh 时允许加入参数。
export function useTable<T>(
api: (params: any) => Promise<T>,
params?: object | (() => object),
options?: {
path?: { data?: keyPath; total?: keyPath; page?: string; size?: string }
immediate?: boolean
},
) {
defaults(options, {
path: { data: 'data', total: 'total', page: 'page', size: 'size' },
immediate: false,
})
// 使用()=>fn()而不是fn()区别在于后者只是一个值且立即执行
const [pagination, , , setTotal] = usePagination((extraData?: object) =>
extraData ? refresh(extraData) : refresh(),
)
const loading = ref(false)
const data = ref([])
const refresh = (extraData?: object | (() => object)) => {
const requestData = {
[options?.path?.page as string]: pagination.current,
[options?.path?.size as string]: pagination.size,
}
if (extraData) {
if (typeof extraData === 'function') {
Object.assign(requestData, extraData())
} else {
Object.assign(requestData, extraData)
}
}
if (params) {
if (typeof params === 'function') {
Object.assign(requestData, params())
} else {
Object.assign(requestData, params)
}
}
loading.value = true
return api(requestData)
.then((res) => {
// TODO 检查响应状态码
data.value = get(res, options!.path?.data, [])
setTotal(get(res, options!.path?.total, 0))
// 友好提示
if (!has(res, options!.path?.data) || !has(res, options!.path?.total)) {
console.warn('useTable:响应数据缺少所需字段')
}
})
.finally(() => {
loading.value = false
})
}
return[data,refresh,paginaiton,loading]
}
需要注意的是,usePagination 处接受的回调函数也要适当修改。当然,pagination 也是要修改的了(增加回调函数有参数的情况,之前回调是没有参数的)。这里还额外新增了一个 reset 方法,用于重置分页器状态,这或许会有用!
export function usePagination(
cb: any,
sizeOption: Array<number> = [10, 20, 50, 100, 200],
): any {
const pagination = reactive({
current: 1,
total: 0,
size: sizeOption[0],
sizeOption,
onPageChange: (page: number, extraData?: object) => {
pagination.current = page
return extraData ? cb(extraData) : cb()
},
onSizeChange: (size: number, extraData?: object) => {
pagination.current = 1
pagination.size = size
return extraData ? cb(extraData) : cb()
},
setTotal: (total: number) => {
pagination.total = total
},
reset() {
pagination.current = 1
pagination.total = 0
pagination.size = pagination.sizeOption[0]
},
})
return [
pagination,
pagination.onPageChange,
pagination.onSizeChange,
pagination.setTotal,
]
}
使用:
<!-- 分页器 -->
<el-pagination
v-model:current-page="current"
:page-size="size"
layout="total, prev, pager, next"
:page-sizes="sizeOption"
:total="total"
@size-change="(size)=>handleSizeChange(size,params.id)"
@current-change="(page)=>handleCurrentChange(page,params.id)"
/>
在此之前,需要保存 item. id 作为全局变量以供读取。
const handleClick=(id:number)=>{
params.id=id;
}
这样,我们就完成了一个功能相对完善的 Hook 函数。
总结
本文通过介绍 Hook 的概念和使用方法,并在实践的过程中封装了一个功能相对完善的 Hook 函数,但是它还有很多可以拓展的地方,比如 useTable 中可以再导出一个 clear 函数,用来将 data 赋值为空数组,以及对 data 数据的每一项进行查找、删除,或者新增一个 showData,用来过滤 data 并展示在视图上,总之,我们打开了 Hook 世界的大门,看到了 Hook 这项技术的强大之处:状态复用!
因为本文主要讲解 Hook 封装,所以比较少提及组件封装。如果代码需要复用,首先考虑组件封装,因为它可以对 html、css 和 javacript 代码进行复用,而 Hook 只是复用 JavaScript 代码。如果将二者结合,能够高效地提高你的开发效率,以及项目的可维护性,帮助你写出优雅的代码。