소스 검색

use getModelSafely

yusuketk 6 년 전
부모
커밋
31089123df
1개의 변경된 파일5개의 추가작업 그리고 2개의 파일을 삭제
  1. 5 2
      src/migrations/20200414164011-add-image-attachment-path-to-user.js

+ 5 - 2
src/migrations/20200414164011-add-image-attachment-path-to-user.js

@@ -4,13 +4,16 @@ const logger = require('@alias/logger')('growi:migrate:add-image-attachment-parh
 const mongoose = require('mongoose');
 const config = require('@root/config/migrate');
 
+const { getModelSafely } = require('@commons/util/mongoose-utils');
+
 module.exports = {
 
   async up(db) {
     logger.info('Apply migration');
     mongoose.connect(config.mongoUri, config.mongodb.options);
-    const User = require('@server/models/user')();
-    require('@server/models/attachment')();
+
+    const User = getModelSafely('User') || require('@server/models/user')();
+    require('@server/models/attachment')(); // for populating imageAttachment
 
     const users = await User.find({ imageAttachment: { $exists: true } })
       .populate({