React

关注公众号 jb51net

关闭
首页 > 网络编程 > JavaScript > javascript类库 > React > React Antd Upload组件上传多个文件

React Antd Upload组件上传多个文件实现方式

作者:水果摊见

为实现多文件上传,需使用beforeUpload和customRequest替代onChange以避免多次调用问题,并处理文件路径以兼容Electron不同平台

前言

实现需求:

上传多个文件。其实就是获取多个文件的绝对路径交给后端接口去处理。

Upload组件

首先,这里不能调用Upload的onChange函数,因为上传中、完成、失败都会调用这个函数,在多个文件的情况下会出现多次调用问题。

改用beforeUpload和customRequest。

import React, { useRef, useState, useEffect } from 'react';
import { UploadOutlined } from '@ant-design/icons';
import type { UploadFile, UploadProps } from 'antd';
import { Button, Upload } from 'antd';
import { useTranslation } from 'react-i18next';
import Console from '../../../../util/Console';
import { UploadChangeParam } from 'antd/es/upload';
import { sendMsgToMain } from 'electron-prokit';

type Props = {
    name?: string;
    onChangeFilePath: (path: string | null | unknown, info?: UploadChangeParam<UploadFile<any>>) => void;
    accept: Array<string>;
    headers?: any;
    progress?: any;
    buttonTitle?: string;
    rest?: UploadProps;
    action?: string | ((file: any) => Promise<string>);
};

const FileMultiSelectButton: React.FC<Props> = ({
    name,
    onChangeFilePath,
    accept,
    headers,
    progress = null,
    buttonTitle,
    action,
    ...rest
}) => {
    const { t } = useTranslation();
    const fileState: any = useRef();
    const [uploadFiles, setUploadFiles] = useState<UploadFile[]>([]);
    const updateFiles = (function () {
        let fileList: UploadFile[] | null = null;
        return function (list: UploadFile[], setState: React.Dispatch<React.SetStateAction<UploadFile[]>>) {
            if (!fileList) {
                fileList = list;
                setState && setState(list);
            }
            return {
                fileList,
                reset() {
                    fileList = null;
                }
            };
        };
    })();

    const beforeUpload = (_: any, fileList: UploadFile[]) => {
        fileState.current = updateFiles(fileList, setUploadFiles);
        return false;
    }

    const customRequest = () => {
        if (uploadFiles.length > 0) {
            const path = uploadFiles.map(item => (item as any).path);
            // 这是electron的ipc处理函数,在node中复制文件
            sendMsgToMain({ key: 'fileMultiSelectPath', data: path }).then(filePath => {
                if (typeof onChangeFilePath === 'function') {
                    onChangeFilePath(filePath);
                } else {
                    Console.handleErrorMsg('onChangeFilePath is not a function');
                }
            });
        } else {
            Console.handleWarningMsg('no file uploaded');
        }

    }

    useEffect(() => {
        if (uploadFiles.length > 0) {
            customRequest();
            fileState.current.reset();
        }
    }, [uploadFiles]);

    return (
        <Upload
            name={name || 'file'}
            accept={accept.join(',')}
            progress={progress}
            showUploadList={false}
            headers={headers}
            action={action}
            multiple={true}
            beforeUpload={beforeUpload}
            customRequest={customRequest}
            {...rest}
        >
            <Button icon={<UploadOutlined />}> {buttonTitle || t('Select')} </Button>
        </Upload>
    );
};

export default FileMultiSelectButton;

在node中处理(不必须),这里主要因为electron需要兼容不同的平台,需要处理文件路径

function generateRandomString(length: number) {
    let result = '';
    const characters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
    const charactersLength = characters.length;
    for (let i = 0; i < length; i++) {
        result += characters.charAt(Math.floor(Math.random() * charactersLength));
    }
    return result;
}
const fileSelectPath = (sourcePath: string) => {
    return new Promise((resolve, reject) => {
        const userDataPath = app.getPath('userData');

        // 获取用户数据目录,并在该目录下创建文件夹,用于保存用户数据,需要判断是否存在,如果存在则直接使用,否则创建,并返回该目录,用于后续操作,并需要兼容跨平台,比如windows和mac
        // 拼接出 files 目录的完整路径
        const filesDirPath = path.join(userDataPath, 'files');

        fs.mkdir(filesDirPath, { recursive: true }, err => {
            if (err) {
                // 如果目录创建失败,返回错误信息
                // 如果目录创建失败,处理错误
                console.error('Failed to create files directory:', err);
            }

            // 生成一个随机数
            const randomBytes = generateRandomString(8); // 生成8字节的随机数

            // 获取当前日期和时间
            const date = new Date();
            const formattedDate = date.toISOString().split('T')[0].replace(/-/g, ''); // 格式化为不含破折号的日期
            const formattedTime = date.toTimeString().split(' ')[0].replace(/:/g, ''); // 格式化为不含冒号的时间

            // 提取源文件名和扩展名
            const { name, ext } = path.parse(sourcePath);

            // 根据日期、时间和随机数生成一个新的文件名
            const newName = `${name}_${formattedDate}_${formattedTime}_${randomBytes}${ext}`;
            // 构造目标路径
            const destinationPath = path.join(filesDirPath, newName);

            // 使用fs模块的copy方法复制文件
            fs.copyFile(sourcePath, destinationPath, err => {
                if (err) {
                    // 如果文件已存在或复制失败,返回错误信息
                    reject(err);
                } else {
                    // 复制成功,返回新的文件路径
                    resolve(destinationPath);
                }
            });
        });
    });
};

export const fileMultiSelectPath = (sourcePath: Array<string>) => {
    return Promise.all(sourcePath.map(async (sourcePath) => {
        return await fileSelectPath(sourcePath);
    }));
};

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:
阅读全文