Sửa lỗi tạo docker

2026-06-12 07:59:10 +07:00
parent d7556aa087
commit 377c4d41d8
13 changed files with 342 additions and 26 deletions
@@ -0,0 +1,93 @@
+const mongoose = require('mongoose');
+const fs = require('fs');
+const path = require('path');
+require('dotenv').config({ path: path.resolve(__dirname, '../../.env') }); // Load .env for UPLOAD_DIR
+
+const connectDB = require('../config/db');
+const Asset = require('../models/Asset');
+const User = require('../models/User'); // Required for Asset model's 'uploadedBy' reference
+
+// Xác định thư mục uploads gốc
+const uploadDir = process.env.UPLOAD_DIR ? path.resolve(process.env.UPLOAD_DIR) : path.join(__dirname, '../../uploads');
+
+const migrateAssetsToUserFolders = async () => {
+    try {
+        console.log('--- Bắt đầu quy trình di chuyển Assets vào thư mục người dùng ---');
+        await connectDB();
+
+        const allAssets = await Asset.find({});
+        console.log(`Tìm thấy ${allAssets.length} Assets cần kiểm tra.`);
+
+        let movedCount = 0;
+        let updatedDbCount = 0;
+        let skippedCount = 0;
+        let errorCount = 0;
+
+        for (const asset of allAssets) {
+            const currentFilePath = asset.filePath;
+            const userId = asset.uploadedBy ? asset.uploadedBy.toString() : null;
+
+            if (!userId) {
+                console.warn(`[WARN] Asset ${asset._id}: Không có thông tin người tải lên. Bỏ qua.`);
+                skippedCount++;
+                continue;
+            }
+
+            // Kiểm tra xem đường dẫn hiện tại đã có userId trong cấu trúc chưa
+            // Ví dụ: uploads/654321abcdef/processed_123.jpg
+            const relativePathSegments = path.relative(uploadDir, currentFilePath).split(path.sep);
+            if (relativePathSegments.length > 1 && relativePathSegments[0] === userId) {
+                console.log(`[SKIP] Asset ${asset._id}: Đường dẫn đã ở đúng định dạng người dùng (${currentFilePath}).`);
+                skippedCount++;
+                continue;
+            }
+
+            const fileName = path.basename(currentFilePath);
+            const userUploadSubDir = path.join(uploadDir, userId); // e.g., /app/uploads/654321abcdef
+            const newFilePath = path.join(userUploadSubDir, fileName); // e.g., /app/uploads/654321abcdef/processed_123.jpg
+
+            if (!fs.existsSync(userUploadSubDir)) {
+                console.log(`[MKDIR] Tạo thư mục: ${userUploadSubDir}`);
+                fs.mkdirSync(userUploadSubDir, { recursive: true });
+            }
+
+            try {
+                // Kiểm tra sự tồn tại của tệp tin vật lý trước khi di chuyển
+                if (fs.existsSync(currentFilePath)) {
+                    fs.renameSync(currentFilePath, newFilePath);
+                    movedCount++;
+                    console.log(`[MOVE] Asset ${asset._id}: Di chuyển từ ${currentFilePath} sang ${newFilePath}`);
+                } else {
+                    console.warn(`[WARN] Tệp tin vật lý không tồn tại: ${currentFilePath} cho Asset ${asset._id}.`);
+                }
+                
+                // Cập nhật đường dẫn trong bản ghi Asset trong cơ sở dữ liệu
+                asset.filePath = newFilePath;
+                await asset.save();
+                updatedDbCount++;
+
+            } catch (fileError) {
+                console.error(`[ERROR] Lỗi khi xử lý file cho Asset ${asset._id} (${currentFilePath}): ${fileError.message}`);
+                errorCount++;
+            }
+        }
+
+        console.log('--- Hoàn tất quy trình di chuyển Assets ---');
+        console.log(`Tổng Assets kiểm tra: ${allAssets.length}`);
+        console.log(`Assets đã di chuyển file: ${movedCount}`);
+        console.log(`Assets đã cập nhật DB: ${updatedDbCount}`);
+        console.log(`Assets đã bỏ qua (đã đúng định dạng hoặc không có userId): ${skippedCount}`);
+        console.log(`Assets gặp lỗi: ${errorCount}`);
+
+    } catch (dbError) {
+        console.error('Lỗi kết nối hoặc truy vấn Database:', dbError.message);
+        process.exit(1);
+    } finally {
+        if (mongoose.connection) {
+            await mongoose.connection.close();
+        }
+        process.exit(0);
+    }
+};
+
+migrateAssetsToUserFolders();