-
Notifications
You must be signed in to change notification settings - Fork 1.9k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(projects): @sa/axios: createRequest, createFlatRequest, createHo…
…okRequest
- Loading branch information
1 parent
fbf4cc4
commit bac1632
Showing
26 changed files
with
672 additions
and
75 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -9,7 +9,8 @@ export default defineConfig( | |
{ | ||
ignores: ['index', 'App', '[id]'] | ||
} | ||
] | ||
], | ||
'no-empty-function': 'off' | ||
} | ||
} | ||
); |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,17 @@ | ||
{ | ||
"name": "@sa/axios", | ||
"version": "1.0.0", | ||
"exports": { | ||
".": "./src/index.ts" | ||
}, | ||
"typesVersions": { | ||
"*": { | ||
"*": ["./src/*"] | ||
} | ||
}, | ||
"dependencies": { | ||
"@sa/utils": "workspace:*", | ||
"axios": "1.6.5", | ||
"axios-retry": "^4.0.0" | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
/** request id key */ | ||
export const REQUEST_ID_KEY = 'X-Request-Id'; | ||
|
||
/** the backend error code key */ | ||
export const BACKEND_ERROR_CODE = 'BACKEND_ERROR'; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,176 @@ | ||
import axios, { AxiosError } from 'axios'; | ||
import type { AxiosResponse, CancelTokenSource, CreateAxiosDefaults, InternalAxiosRequestConfig } from 'axios'; | ||
import axiosRetry from 'axios-retry'; | ||
import { nanoid } from '@sa/utils'; | ||
import { createAxiosConfig, createDefaultOptions, createRetryOptions } from './options'; | ||
import { BACKEND_ERROR_CODE, REQUEST_ID_KEY } from './constant'; | ||
import type { | ||
CustomAxiosRequestConfig, | ||
FlatRequestInstance, | ||
MappedType, | ||
RequestInstance, | ||
RequestOption, | ||
ResponseType | ||
} from './type'; | ||
|
||
function createCommonRequest<ResponseData = any>( | ||
axiosConfig?: CreateAxiosDefaults, | ||
options?: Partial<RequestOption<ResponseData>> | ||
) { | ||
const opts = createDefaultOptions<ResponseData>(options); | ||
|
||
const axiosConf = createAxiosConfig(axiosConfig); | ||
const instance = axios.create(axiosConf); | ||
|
||
const cancelTokenSourceMap = new Map<string, CancelTokenSource>(); | ||
|
||
// config axios retry | ||
const retryOptions = createRetryOptions(axiosConf); | ||
axiosRetry(instance, retryOptions); | ||
|
||
instance.interceptors.request.use(conf => { | ||
const config: InternalAxiosRequestConfig = { ...conf }; | ||
|
||
// set request id | ||
const requestId = nanoid(); | ||
config.headers.set(REQUEST_ID_KEY, requestId); | ||
|
||
// config cancel token | ||
const cancelTokenSource = axios.CancelToken.source(); | ||
config.cancelToken = cancelTokenSource.token; | ||
cancelTokenSourceMap.set(requestId, cancelTokenSource); | ||
|
||
// handle config by hook | ||
const handledConfig = opts.onRequest?.(config) || config; | ||
|
||
return handledConfig; | ||
}); | ||
|
||
instance.interceptors.response.use( | ||
async response => { | ||
if (opts.isBackendSuccess(response)) { | ||
return Promise.resolve(response); | ||
} | ||
|
||
const fail = await opts.onBackendFail(response, instance); | ||
if (fail) { | ||
return fail; | ||
} | ||
|
||
const backendError = new AxiosError<ResponseData>( | ||
'the backend request error', | ||
BACKEND_ERROR_CODE, | ||
response.config, | ||
response, | ||
response.request | ||
); | ||
|
||
await opts.onError(backendError); | ||
|
||
return Promise.reject(backendError); | ||
}, | ||
async (error: AxiosError<ResponseData>) => { | ||
await opts.onError(error); | ||
|
||
return Promise.reject(error); | ||
} | ||
); | ||
|
||
function cancelRequest(requestId: string) { | ||
const cancelTokenSource = cancelTokenSourceMap.get(requestId); | ||
if (cancelTokenSource) { | ||
cancelTokenSource.cancel(); | ||
cancelTokenSourceMap.delete(requestId); | ||
} | ||
} | ||
|
||
function cancelAllRequest() { | ||
cancelTokenSourceMap.forEach(cancelTokenSource => { | ||
cancelTokenSource.cancel(); | ||
}); | ||
cancelTokenSourceMap.clear(); | ||
} | ||
|
||
return { | ||
instance, | ||
opts, | ||
cancelRequest, | ||
cancelAllRequest | ||
}; | ||
} | ||
|
||
/** | ||
* create a request instance | ||
* | ||
* @param axiosConfig axios config | ||
* @param options request options | ||
*/ | ||
export function createRequest<ResponseData = any>( | ||
axiosConfig?: CreateAxiosDefaults, | ||
options?: Partial<RequestOption<ResponseData>> | ||
) { | ||
const { instance, opts, cancelRequest, cancelAllRequest } = createCommonRequest<ResponseData>(axiosConfig, options); | ||
|
||
const request: RequestInstance = async function request<T = any, R extends ResponseType = 'json'>( | ||
config: CustomAxiosRequestConfig | ||
) { | ||
const response: AxiosResponse<ResponseData> = await instance(config); | ||
|
||
const responseType = response.config?.responseType || 'json'; | ||
|
||
if (responseType === 'json') { | ||
return opts.transformBackendResponse(response); | ||
} | ||
|
||
return response.data as MappedType<R, T>; | ||
} as RequestInstance; | ||
|
||
request.cancelRequest = cancelRequest; | ||
request.cancelAllRequest = cancelAllRequest; | ||
|
||
return request; | ||
} | ||
|
||
/** | ||
* create a flat request instance | ||
* | ||
* The response data is a flat object: { data: any, error: AxiosError } | ||
* | ||
* @param axiosConfig axios config | ||
* @param options request options | ||
*/ | ||
export function createFlatRequest<ResponseData = any>( | ||
axiosConfig?: CreateAxiosDefaults, | ||
options?: Partial<RequestOption<ResponseData>> | ||
) { | ||
const { instance, opts, cancelRequest, cancelAllRequest } = createCommonRequest<ResponseData>(axiosConfig, options); | ||
|
||
const flatRequest: FlatRequestInstance = async function flatRequest<T = any, R extends ResponseType = 'json'>( | ||
config: CustomAxiosRequestConfig | ||
) { | ||
try { | ||
const response: AxiosResponse<ResponseData> = await instance(config); | ||
|
||
const responseType = response.config?.responseType || 'json'; | ||
|
||
if (responseType === 'json') { | ||
const data = opts.transformBackendResponse(response); | ||
|
||
return { data, error: null }; | ||
} | ||
|
||
return { data: response.data as MappedType<R, T>, error: null }; | ||
} catch (error) { | ||
return { data: null, error }; | ||
} | ||
} as FlatRequestInstance; | ||
|
||
flatRequest.cancelRequest = cancelRequest; | ||
flatRequest.cancelAllRequest = cancelAllRequest; | ||
|
||
return flatRequest; | ||
} | ||
|
||
export { BACKEND_ERROR_CODE, REQUEST_ID_KEY }; | ||
export type * from './type'; | ||
export type { CreateAxiosDefaults, AxiosError }; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,44 @@ | ||
import type { CreateAxiosDefaults } from 'axios'; | ||
import type { IAxiosRetryConfig } from 'axios-retry'; | ||
import { isHttpSuccess } from './shared'; | ||
import type { RequestOption } from './type'; | ||
|
||
export function createDefaultOptions<ResponseData = any>(options?: Partial<RequestOption<ResponseData>>) { | ||
const opts: RequestOption<ResponseData> = { | ||
onRequest: async config => config, | ||
isBackendSuccess: _response => true, | ||
onBackendFail: async () => {}, | ||
transformBackendResponse: async response => response.data, | ||
onError: async () => {} | ||
}; | ||
|
||
Object.assign(opts, options); | ||
|
||
return opts; | ||
} | ||
|
||
export function createRetryOptions(config?: Partial<CreateAxiosDefaults>) { | ||
const retryConfig: IAxiosRetryConfig = { | ||
retries: 3 | ||
}; | ||
|
||
Object.assign(retryConfig, config); | ||
|
||
return retryConfig; | ||
} | ||
|
||
export function createAxiosConfig(config?: Partial<CreateAxiosDefaults>) { | ||
const TEN_SECONDS = 10 * 1000; | ||
|
||
const axiosConfig: CreateAxiosDefaults = { | ||
timeout: TEN_SECONDS, | ||
headers: { | ||
'Content-Type': 'application/json' | ||
}, | ||
validateStatus: isHttpSuccess | ||
}; | ||
|
||
Object.assign(axiosConfig, config); | ||
|
||
return axiosConfig; | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,28 @@ | ||
import type { AxiosHeaderValue, AxiosResponse, InternalAxiosRequestConfig } from 'axios'; | ||
|
||
export function getContentType(config: InternalAxiosRequestConfig) { | ||
const contentType: AxiosHeaderValue = config.headers?.['Content-Type'] || 'application/json'; | ||
|
||
return contentType; | ||
} | ||
|
||
/** | ||
* check if http status is success | ||
* | ||
* @param status | ||
*/ | ||
export function isHttpSuccess(status: number) { | ||
const isSuccessCode = status >= 200 && status < 300; | ||
return isSuccessCode || status === 304; | ||
} | ||
|
||
/** | ||
* is response json | ||
* | ||
* @param response axios response | ||
*/ | ||
export function isResponseJson(response: AxiosResponse) { | ||
const { responseType } = response.config; | ||
|
||
return responseType === 'json' || responseType === undefined; | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,97 @@ | ||
import type { AxiosError, AxiosInstance, AxiosRequestConfig, AxiosResponse, InternalAxiosRequestConfig } from 'axios'; | ||
|
||
export type ContentType = | ||
| 'text/html' | ||
| 'text/plain' | ||
| 'multipart/form-data' | ||
| 'application/json' | ||
| 'application/x-www-form-urlencoded' | ||
| 'application/octet-stream'; | ||
|
||
export interface RequestOption<ResponseData = any> { | ||
/** | ||
* The hook before request | ||
* | ||
* For example: You can add header token in this hook | ||
* | ||
* @param config Axios config | ||
*/ | ||
onRequest: (config: InternalAxiosRequestConfig) => InternalAxiosRequestConfig | Promise<InternalAxiosRequestConfig>; | ||
/** | ||
* The hook to check backend response is success or not | ||
* | ||
* @param response Axios response | ||
*/ | ||
isBackendSuccess: (response: AxiosResponse<ResponseData>) => boolean; | ||
/** | ||
* The hook after backend request fail | ||
* | ||
* For example: You can handle the expired token in this hook | ||
* | ||
* @param response Axios response | ||
* @param instance Axios instance | ||
* @returns | ||
*/ | ||
onBackendFail: ( | ||
response: AxiosResponse<ResponseData>, | ||
instance: AxiosInstance | ||
) => Promise<AxiosResponse> | Promise<void>; | ||
/** | ||
* transform backend response when the responseType is json | ||
* | ||
* @param response Axios response | ||
*/ | ||
transformBackendResponse(response: AxiosResponse<ResponseData>): any | Promise<any>; | ||
/** | ||
* The hook to handle error | ||
* | ||
* For example: You can show error message in this hook | ||
* | ||
* @param error | ||
*/ | ||
onError: (error: AxiosError<ResponseData>) => void | Promise<void>; | ||
} | ||
|
||
interface ResponseMap { | ||
blob: Blob; | ||
text: string; | ||
arrayBuffer: ArrayBuffer; | ||
stream: ReadableStream<Uint8Array>; | ||
document: Document; | ||
} | ||
export type ResponseType = keyof ResponseMap | 'json'; | ||
|
||
export type MappedType<R extends ResponseType, JsonType = any> = R extends keyof ResponseMap | ||
? ResponseMap[R] | ||
: JsonType; | ||
|
||
export type CustomAxiosRequestConfig<R extends ResponseType = 'json'> = Omit<AxiosRequestConfig, 'responseType'> & { | ||
responseType?: R; | ||
}; | ||
|
||
/** The request instance */ | ||
export interface RequestInstance { | ||
<T = any, R extends ResponseType = 'json'>(config: CustomAxiosRequestConfig<R>): Promise<MappedType<R, T>>; | ||
cancelRequest: (requestId: string) => void; | ||
cancelAllRequest: () => void; | ||
} | ||
|
||
export type FlatResponseSuccessData<T = any> = { | ||
data: T; | ||
error: null; | ||
}; | ||
|
||
export type FlatResponseFailData<T = any> = { | ||
data: null; | ||
error: AxiosError<T>; | ||
}; | ||
|
||
export type FlatResponseData<T = any> = FlatResponseSuccessData<T> | FlatResponseFailData<T>; | ||
|
||
export interface FlatRequestInstance { | ||
<T = any, R extends ResponseType = 'json'>( | ||
config: CustomAxiosRequestConfig<R> | ||
): Promise<FlatResponseData<MappedType<R, T>>>; | ||
cancelRequest: (requestId: string) => void; | ||
cancelAllRequest: () => void; | ||
} |
File renamed without changes.
Oops, something went wrong.