const express = require('express');
const multer = require('multer');
const fs = require('fs');
const path = require('path');
const app = express();
const uploadDir = path.join(__dirname, 'uploads');
// 确保上传目录存在
fs.existsSync(uploadDir) || fs.mkdirSync(uploadDir);
const storage = multer.diskStorage({
destination: (req, file, cb) => {
cb(null, uploadDir);
},
filename: (req, file, cb) => {
const filename = req.body.filename;
const chunkIndex = req.body.chunkIndex;
cb(null, `${filename}.part_${chunkIndex}`); // 每个 chunk 保存为一个文件
}
});
const upload = multer({ storage: storage }).single('chunk');
app.post('/upload', (req, res) => {
upload(req, res, (err) => {
if (err) {
console.error(err);
return res.status(500).send('上传失败');
}
const filename = req.body.filename;
const chunkIndex = parseInt(req.body.chunkIndex);
const totalChunks = parseInt(req.body.totalChunks);
console.log(`Received chunk ${chunkIndex + 1} of ${totalChunks} for ${filename}`);
// 检查是否所有 chunk 都已上传
const allChunksUploaded = Array.from({ length: totalChunks }, (_, i) => {
return fs.existsSync(path.join(uploadDir, `${filename}.part_${i}`));
}).every(Boolean);
if (allChunksUploaded) {
console.log(`All chunks uploaded for ${filename}, merging...`);
mergeChunks(filename, totalChunks, uploadDir)
.then(() => {
console.log(`Successfully merged ${filename}`);
res.status(200).send('上传成功');
})
.catch(err => {
console.error(`Error merging chunks for ${filename}:`, err);
res.status(500).send('合并失败');
});
} else {
res.status(200).send('Chunk uploaded');
}
});
});
async function mergeChunks(filename, totalChunks, uploadDir) {
const finalFilePath = path.join(uploadDir, filename);
const writeStream = fs.createWriteStream(finalFilePath);
for (let i = 0; i < totalChunks; i++) {
const chunkPath = path.join(uploadDir, `${filename}.part_${i}`);
const readStream = fs.createReadStream(chunkPath);
await new Promise((resolve, reject) => {
readStream.pipe(writeStream, { end: false }); // end: false 很重要,避免提前关闭 writeStream
readStream.on('end', () => {
fs.unlinkSync(chunkPath); // 删除已合并的 chunk
resolve();
});
readStream.on('error', reject);
writeStream.on('error', reject);
});
}
writeStream.end(); // 关闭 writeStream
}
app.listen(3000, () => {
console.log('Server listening on port 3000');
});登录后复制