93 lines
2.2 KiB
TypeScript
93 lines
2.2 KiB
TypeScript
import {onUnmounted, ref} from 'vue'
|
|
import type { Ref } from 'vue'
|
|
import {isFunction, get, isArray} from 'lodash-es'
|
|
import type { AxiosResponseRewrite } from '@/utils/request'
|
|
|
|
interface RequestOptions<T, S> {
|
|
immediate: boolean
|
|
/**
|
|
* 成功回调
|
|
* @param data
|
|
* @returns
|
|
*/
|
|
onSuccess: (data: AxiosResponseRewrite<S>) => S | void
|
|
/**
|
|
* 返回参数处理
|
|
* @returns
|
|
*/
|
|
formatName: string | [string]
|
|
onError: (e: any) => void
|
|
|
|
defaultParams: S | any | any[]
|
|
|
|
handleResponse: (data: any) => any
|
|
}
|
|
|
|
const defaultOptions: any = {
|
|
immediate: true,
|
|
formatName: 'result'
|
|
}
|
|
|
|
type Run = (...args: any[]) => void
|
|
|
|
export const useRequest = <T = any, S = any>(
|
|
request: (...args: any[]) => Promise<AxiosResponseRewrite<T>>,
|
|
options: Partial<RequestOptions<T, S>> = defaultOptions,
|
|
): {
|
|
data: Ref<S | undefined>,
|
|
loading: Ref<boolean>,
|
|
run: Run
|
|
} => {
|
|
const data = ref<S>()
|
|
const loading = ref(false)
|
|
const _options = {
|
|
...defaultOptions,
|
|
...options
|
|
}
|
|
|
|
async function run(...arg: any[]) {
|
|
if (request && isFunction(request)) {
|
|
loading.value = true
|
|
try {
|
|
// @ts-ignore
|
|
const resp = await request.apply(this, arg)
|
|
|
|
loading.value = false
|
|
|
|
if (resp?.success) {
|
|
const successData = await _options.onSuccess?.(resp)
|
|
data.value = successData || get(resp, _options.formatName!)
|
|
// console.log(data.value)
|
|
} else {
|
|
_options.onError?.(resp)
|
|
}
|
|
} catch (e) {
|
|
loading.value = false
|
|
_options.onError?.(e)
|
|
}
|
|
}
|
|
}
|
|
|
|
if (_options.immediate) { // 主动触发
|
|
if (_options.defaultParams) {
|
|
isArray(_options.defaultParams) ? run(..._options.defaultParams) : run(_options.defaultParams)
|
|
} else {
|
|
run()
|
|
}
|
|
}
|
|
|
|
onUnmounted(() => {
|
|
// 销毁时，撤销该请求
|
|
if (request && isFunction(request)) {
|
|
|
|
}
|
|
// request()
|
|
})
|
|
|
|
return {
|
|
data,
|
|
loading,
|
|
run
|
|
}
|
|
}
|