소스 검색

added getModelSafety

ryuichi-e 6 년 전
부모
커밋
40ffd4a06b
1개의 변경된 파일4개의 추가작업 그리고 2개의 파일을 삭제
  1. 4 2
      src/migrations/2020040216038-remove-deleteduser-from-relationgroup.js

+ 4 - 2
src/migrations/2020040216038-remove-deleteduser-from-relationgroup.js

@@ -4,13 +4,15 @@ const logger = require('@alias/logger')('growi:migrate:adjust-pages-path');
 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')();
-    const UserGroupRelation = require('@server/models/user-group-relation')();
+    const User = getModelSafely('User') || require('@server/models/user')();
+    const UserGroupRelation = getModelSafely('UserGroupRelation') || require('@server/models/user-group-relation')();
 
     const deletedUsers = await User.find({ status: 4 }); // deleted user
     await UserGroupRelation.remove({ relatedUser: deletedUsers });