soybean-admin/src/utils/service/transform.ts
2022-07-16 02:04:59 +08:00

62 lines
1.5 KiB
TypeScript

import qs from 'qs';
import FormData from 'form-data';
import { EnumContentType } from '@/enum';
import { isArray, isFile } from '../common';
/**
* 请求数据的转换
* @param requestData - 请求数据
* @param contentType - 请求头的Content-Type
*/
export async function transformRequestData(requestData: any, contentType?: string) {
// application/json类型不处理
let data = requestData;
// form类型转换
if (contentType === EnumContentType.formUrlencoded) {
data = qs.stringify(requestData);
}
// form-data类型转换
if (contentType === EnumContentType.formData) {
data = await handleFormData(requestData);
}
return data;
}
async function handleFormData(data: Record<string, any>) {
const formData = new FormData();
const entries = Object.entries(data);
entries.forEach(async ([key, value]) => {
const isFileType = isFile(value) || (isArray(value) && value.length && isFile(value[0]));
if (isFileType) {
await transformFile(formData, key, value);
} else {
formData.append(key, value);
}
});
return formData;
}
/**
* 接口为上传文件的类型时数据转换
* @param key - 文件的属性名
* @param file - 单文件或多文件
*/
async function transformFile(formData: FormData, key: string, file: File[] | File) {
if (isArray(file)) {
// 多文件
await Promise.all(
(file as File[]).map(item => {
formData.append(key, item);
return true;
})
);
} else {
// 单文件
formData.append(key, file);
}
}