AWS Amplify Storage 中获取文件类型的正确方法

在使用 AWS Amplify 的 Storage 模块时，开发者经常需要获取存储在 S3 中的文件类型信息，以便在前端进行适当的展示和处理。本文将详细介绍如何正确获取文件内容类型(contentType)的方法。

常见问题现象

许多开发者在使用 getProperties 方法获取文件属性时，会发现无论上传的是 PDF、JPEG、PNG 还是 TXT 文档，返回的 contentType 始终是 application/octet-stream。这显然不符合预期，因为开发者期望能够获取到精确的文件类型信息。

问题根源分析

经过深入研究发现，AWS Amplify 在上传文件到 S3 时有一个默认行为：如果没有显式指定文件的内容类型(contentType)，系统会自动使用 application/octet-stream 作为默认值。这种设计类似于 HTTP 协议中的通用二进制流类型，适用于未知或未指定的文件类型。

解决方案

要正确获取文件类型，开发者需要在上传文件时显式指定 contentType 参数。以下是两种实现方式：

1. 自动检测并设置文件类型

const uploadFileToS3 = async (file) => {
    try {
        await uploadData({
            path: `protected/${user.identityId}/documents/${file.name}`,
            data: file,
            options: {
                contentType: file.type // 使用文件自带的type属性
            }
        });
    } catch (error) {
        console.error('上传失败:', error);
    }
};

2. 手动指定文件类型

const uploadPdfToS3 = async (pdfFile) => {
    try {
        await uploadData({
            path: `protected/${user.identityId}/documents/report.pdf`,
            data: pdfFile,
            options: {
                contentType: 'application/pdf' // 明确指定PDF类型
            }
        });
    } catch (error) {
        console.error('PDF上传失败:', error);
    }
};

获取文件类型的正确方法

设置好 contentType 后，就可以使用 getProperties 方法正确获取文件类型了：

const getFileContentType = async (filePath) => {
    const properties = await getProperties({
        path: ({identityId}) => `protected/${identityId}/${filePath}`
    });
    return properties.contentType;
};

最佳实践建议

1. 始终指定contentType：上传文件时养成指定contentType的习惯，这不仅有助于后续检索，也能确保浏览器正确处理文件。

2. 文件类型验证：即使指定了contentType，也应在前端进行文件类型验证，防止恶意文件上传。

3. 错误处理：考虑添加默认contentType回退逻辑，处理未指定类型的情况。

4. 类型映射表：对于常见文件扩展名，可以维护一个扩展名到contentType的映射表，确保一致性。

通过遵循这些实践，开发者可以确保在AWS Amplify Storage中准确获取和使用文件类型信息，从而在前端实现更精确的文件处理和展示逻辑。