初次提交
This commit is contained in:
293
services/minioService.js
Normal file
293
services/minioService.js
Normal file
@@ -0,0 +1,293 @@
|
||||
const { createMinioClient, minioConfig, getPublicUrl } = require('../config/minio');
|
||||
const path = require('path');
|
||||
const crypto = require('crypto');
|
||||
|
||||
/**
|
||||
* MinIO 文件服务
|
||||
* 提供文件上传、删除、获取等功能
|
||||
*/
|
||||
class MinioService {
|
||||
constructor() {
|
||||
this.client = createMinioClient();
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成唯一文件名
|
||||
* @param {string} originalName - 原始文件名
|
||||
* @returns {string} 唯一文件名
|
||||
*/
|
||||
generateUniqueFileName(originalName) {
|
||||
const now = new Date();
|
||||
const year = now.getFullYear();
|
||||
const month = String(now.getMonth() + 1).padStart(2, '0');
|
||||
const day = String(now.getDate()).padStart(2, '0');
|
||||
const timestamp = Date.now();
|
||||
const randomString = crypto.randomBytes(8).toString('hex');
|
||||
const ext = path.extname(originalName);
|
||||
return `${year}/${month}/${day}/${timestamp}_${randomString}${ext}`;
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据文件类型获取存储桶名称
|
||||
* @param {string} type - 文件类型 (avatar, product, document)
|
||||
* @returns {string} 存储桶名称
|
||||
*/
|
||||
getBucketName(type = 'document') {
|
||||
const bucketMap = {
|
||||
'avatar': minioConfig.buckets.avatars,
|
||||
'product': minioConfig.buckets.products,
|
||||
'document': minioConfig.buckets.documents
|
||||
};
|
||||
return bucketMap[type] || minioConfig.buckets.documents;
|
||||
}
|
||||
|
||||
/**
|
||||
* 上传单个文件
|
||||
* @param {Buffer} fileBuffer - 文件缓冲区
|
||||
* @param {string} originalName - 原始文件名
|
||||
* @param {string} mimeType - 文件MIME类型
|
||||
* @param {string} type - 文件类型
|
||||
* @returns {Promise<Object>} 上传结果
|
||||
*/
|
||||
async uploadFile(fileBuffer, originalName, mimeType, type = 'document') {
|
||||
try {
|
||||
const bucketName = this.getBucketName(type);
|
||||
const fileName = this.generateUniqueFileName(originalName);
|
||||
|
||||
// 设置文件元数据
|
||||
const metaData = {
|
||||
'Content-Type': mimeType,
|
||||
'Original-Name': encodeURIComponent(originalName),
|
||||
'Upload-Time': new Date().toISOString()
|
||||
};
|
||||
|
||||
// 上传文件到MinIO
|
||||
await this.client.putObject(bucketName, fileName, fileBuffer, fileBuffer.length, metaData);
|
||||
|
||||
// 生成访问URL
|
||||
const url = getPublicUrl(bucketName, fileName);
|
||||
|
||||
return {
|
||||
success: true,
|
||||
data: {
|
||||
filename: fileName,
|
||||
originalname: originalName,
|
||||
mimetype: mimeType,
|
||||
size: fileBuffer.length,
|
||||
bucket: bucketName,
|
||||
path: `${bucketName}/${fileName}`,
|
||||
url: url
|
||||
}
|
||||
};
|
||||
} catch (error) {
|
||||
console.error('MinIO文件上传失败:', error);
|
||||
throw new Error(`文件上传失败: ${error.message}`);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 迁移专用：上传文件到指定存储桶和路径
|
||||
* @param {string} bucketName - 存储桶名称
|
||||
* @param {string} filePath - 文件路径
|
||||
* @param {Buffer} fileBuffer - 文件缓冲区
|
||||
* @param {string} mimeType - 文件MIME类型
|
||||
* @returns {Promise<Object>} 上传结果
|
||||
*/
|
||||
async uploadFileForMigration(bucketName, filePath, fileBuffer, mimeType) {
|
||||
try {
|
||||
// 设置文件元数据
|
||||
const metaData = {
|
||||
'Content-Type': mimeType,
|
||||
'Upload-Time': new Date().toISOString()
|
||||
};
|
||||
|
||||
// 上传文件到MinIO
|
||||
await this.client.putObject(bucketName, filePath, fileBuffer, fileBuffer.length, metaData);
|
||||
|
||||
// 生成访问URL
|
||||
const url = getPublicUrl(bucketName, filePath);
|
||||
|
||||
return {
|
||||
success: true,
|
||||
data: {
|
||||
filename: filePath,
|
||||
mimetype: mimeType,
|
||||
size: fileBuffer.length,
|
||||
bucket: bucketName,
|
||||
path: `${bucketName}/${filePath}`,
|
||||
url: url
|
||||
}
|
||||
};
|
||||
} catch (error) {
|
||||
console.error('MinIO文件迁移上传失败:', error);
|
||||
throw new Error(`文件迁移上传失败: ${error.message}`);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 上传多个文件
|
||||
* @param {Array} files - 文件数组，每个文件包含 {buffer, originalName, mimeType}
|
||||
* @param {string} type - 文件类型
|
||||
* @returns {Promise<Array>} 上传结果数组
|
||||
*/
|
||||
async uploadMultipleFiles(files, type = 'document') {
|
||||
try {
|
||||
const uploadPromises = files.map(file =>
|
||||
this.uploadFile(file.buffer, file.originalName, file.mimeType, type)
|
||||
);
|
||||
|
||||
const results = await Promise.all(uploadPromises);
|
||||
const uploadedFiles = results.map(result => result.data);
|
||||
|
||||
return {
|
||||
success: true,
|
||||
data: {
|
||||
files: uploadedFiles,
|
||||
urls: uploadedFiles.map(file => file.url),
|
||||
count: uploadedFiles.length
|
||||
}
|
||||
};
|
||||
} catch (error) {
|
||||
console.error('MinIO多文件上传失败:', error);
|
||||
throw new Error(`多文件上传失败: ${error.message}`);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除文件
|
||||
* @param {string} bucketName - 存储桶名称
|
||||
* @param {string} fileName - 文件名
|
||||
* @returns {Promise<boolean>} 删除结果
|
||||
*/
|
||||
async deleteFile(bucketName, fileName) {
|
||||
try {
|
||||
await this.client.removeObject(bucketName, fileName);
|
||||
console.log(`✅ 文件删除成功: ${bucketName}/${fileName}`);
|
||||
return true;
|
||||
} catch (error) {
|
||||
console.error('MinIO文件删除失败:', error);
|
||||
throw new Error(`文件删除失败: ${error.message}`);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量删除文件
|
||||
* @param {string} bucketName - 存储桶名称
|
||||
* @param {Array<string>} fileNames - 文件名数组
|
||||
* @returns {Promise<Object>} 删除结果
|
||||
*/
|
||||
async deleteMultipleFiles(bucketName, fileNames) {
|
||||
try {
|
||||
const deletePromises = fileNames.map(fileName =>
|
||||
this.deleteFile(bucketName, fileName)
|
||||
);
|
||||
|
||||
await Promise.all(deletePromises);
|
||||
|
||||
return {
|
||||
success: true,
|
||||
deletedCount: fileNames.length,
|
||||
message: `成功删除${fileNames.length}个文件`
|
||||
};
|
||||
} catch (error) {
|
||||
console.error('MinIO批量删除失败:', error);
|
||||
throw new Error(`批量删除失败: ${error.message}`);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查文件是否存在
|
||||
* @param {string} bucketName - 存储桶名称
|
||||
* @param {string} fileName - 文件名
|
||||
* @returns {Promise<boolean>} 文件是否存在
|
||||
*/
|
||||
async fileExists(bucketName, fileName) {
|
||||
try {
|
||||
await this.client.statObject(bucketName, fileName);
|
||||
return true;
|
||||
} catch (error) {
|
||||
if (error.code === 'NotFound') {
|
||||
return false;
|
||||
}
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取文件信息
|
||||
* @param {string} bucketName - 存储桶名称
|
||||
* @param {string} fileName - 文件名
|
||||
* @returns {Promise<Object>} 文件信息
|
||||
*/
|
||||
async getFileInfo(bucketName, fileName) {
|
||||
try {
|
||||
const stat = await this.client.statObject(bucketName, fileName);
|
||||
return {
|
||||
size: stat.size,
|
||||
lastModified: stat.lastModified,
|
||||
etag: stat.etag,
|
||||
contentType: stat.metaData['content-type'],
|
||||
originalName: decodeURIComponent(stat.metaData['original-name'] || fileName)
|
||||
};
|
||||
} catch (error) {
|
||||
console.error('获取文件信息失败:', error);
|
||||
throw new Error(`获取文件信息失败: ${error.message}`);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成预签名URL（用于临时访问）
|
||||
* @param {string} bucketName - 存储桶名称
|
||||
* @param {string} fileName - 文件名
|
||||
* @param {number} expiry - 过期时间（秒），默认7天
|
||||
* @returns {Promise<string>} 预签名URL
|
||||
*/
|
||||
async getPresignedUrl(bucketName, fileName, expiry = 7 * 24 * 60 * 60) {
|
||||
try {
|
||||
const url = await this.client.presignedGetObject(bucketName, fileName, expiry);
|
||||
return url;
|
||||
} catch (error) {
|
||||
console.error('生成预签名URL失败:', error);
|
||||
throw new Error(`生成预签名URL失败: ${error.message}`);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 列出存储桶中的文件
|
||||
* @param {string} bucketName - 存储桶名称
|
||||
* @param {string} prefix - 文件前缀
|
||||
* @param {number} limit - 限制数量
|
||||
* @returns {Promise<Array>} 文件列表
|
||||
*/
|
||||
async listFiles(bucketName, prefix = '', limit = 100) {
|
||||
try {
|
||||
const files = [];
|
||||
const stream = this.client.listObjects(bucketName, prefix, true);
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
stream.on('data', (obj) => {
|
||||
if (files.length < limit) {
|
||||
files.push({
|
||||
name: obj.name,
|
||||
size: obj.size,
|
||||
lastModified: obj.lastModified,
|
||||
etag: obj.etag,
|
||||
url: getPublicUrl(bucketName, obj.name)
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
stream.on('end', () => resolve(files));
|
||||
stream.on('error', reject);
|
||||
});
|
||||
} catch (error) {
|
||||
console.error('列出文件失败:', error);
|
||||
throw new Error(`列出文件失败: ${error.message}`);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 创建单例实例
|
||||
const minioService = new MinioService();
|
||||
|
||||
module.exports = minioService;
|
||||
Reference in New Issue
Block a user