correct collections from mongoose.connection instance

src/server/routes/apiv3/export.js

@@ -42,7 +42,7 @@ module.exports = (crowi) => {
   const adminRequired = require('../../middleware/admin-required')(crowi);
   const csrf = require('../../middleware/csrf')(crowi);
 
-  const { growiBridgeService, exportService } = crowi;
+  const { exportService } = crowi;
 
   this.adminEvent = crowi.event('admin');
 
@@ -103,10 +103,8 @@ module.exports = (crowi) => {
     // TODO: add express validator
     try {
       const { collections } = req.body;
-      // get model for collection
-      const models = collections.map(collectionName => growiBridgeService.getModelFromCollectionName(collectionName));
 
-      exportService.export(models);
+      exportService.export(collections);
 
       const status = await exportService.getStatus();
 

src/server/routes/apiv3/mongo.js

@@ -3,6 +3,7 @@ const loggerFactory = require('@alias/logger');
 const logger = loggerFactory('growi:routes:apiv3:mongo'); // eslint-disable-line no-unused-vars
 
 const express = require('express');
+const mongoose = require('mongoose');
 
 const router = express.Router();
 
@@ -33,12 +34,12 @@ module.exports = (crowi) => {
    *                      type: string
    */
   router.get('/collections', async(req, res) => {
-    const collections = Object.values(crowi.models).map(model => model.collection.name);
+    const collections = Object.keys(mongoose.connection.collections);
 
     // TODO: use res.apiv3
     return res.json({
       ok: true,
-      collections: [...new Set(collections)], // remove duplicates
+      collections,
     });
   });
 

src/server/service/export.js

@@ -1,9 +1,13 @@
 const logger = require('@alias/logger')('growi:services:ExportService'); // eslint-disable-line no-unused-vars
+
 const fs = require('fs');
 const path = require('path');
+const mongoose = require('mongoose');
 const { Transform } = require('stream');
 const streamToPromise = require('stream-to-promise');
 const archiver = require('archiver');
+
+
 const toArrayIfNot = require('../../lib/util/toArrayIfNot');
 
 
@@ -26,10 +30,10 @@ class ExportingStatus {
     this.progressMap = {};
   }
 
-  async init(models) {
-    const promisesForCreatingInstance = models.map(async(Model) => {
-      const collectionName = Model.collection.name;
-      const totalCount = await Model.countDocuments();
+  async init(collections) {
+    const promisesForCreatingInstance = collections.map(async(collectionName) => {
+      const collection = mongoose.connection.collection(collectionName);
+      const totalCount = await collection.count();
       return new ExportingProgress(collectionName, totalCount);
     });
     this.progressList = await Promise.all(promisesForCreatingInstance);
@@ -177,15 +181,13 @@ class ExportService {
    * dump a mongodb collection into json
    *
    * @memberOf ExportService
-   * @param {object} Model instance of mongoose model
+   * @param {string} collectionName collection name
    * @return {string} path to zip file
    */
-  async exportCollectionToJson(Model) {
-    const collectionName = Model.collection.name;
+  async exportCollectionToJson(collectionName) {
+    const collection = mongoose.connection.collection(collectionName);
 
-    // get native Cursor instance
-    //  cz: Mongoose cursor might cause memory leak
-    const nativeCursor = Model.collection.find();
+    const nativeCursor = collection.find();
     const readStream = nativeCursor
       .snapshot()
       .stream({ transform: JSON.stringify });
@@ -215,14 +217,14 @@ class ExportService {
    * export multiple Collections into json and Zip
    *
    * @memberOf ExportService
-   * @param {Array.<object>} models array of instances of mongoose model
+   * @param {Array.<string>} collections array of collection name
    * @return {Array.<string>} paths to json files created
    */
-  async exportCollectionsToZippedJson(models) {
+  async exportCollectionsToZippedJson(collections) {
     const metaJson = await this.createMetaJson();
 
-    const promisesForModels = models.map(Model => this.exportCollectionToJson(Model));
-    const jsonFiles = await Promise.all(promisesForModels);
+    const promises = collections.map(collectionName => this.exportCollectionToJson(collectionName));
+    const jsonFiles = await Promise.all(promises);
 
     // zip json
     const configs = jsonFiles.map((jsonFile) => { return { from: jsonFile, as: path.basename(jsonFile) } });
@@ -240,16 +242,16 @@ class ExportService {
     // TODO: remove broken zip file
   }
 
-  async export(models) {
+  async export(collections) {
     if (this.currentExportingStatus != null) {
       throw new Error('There is an exporting process running.');
     }
 
     this.currentExportingStatus = new ExportingStatus();
-    await this.currentExportingStatus.init(models);
+    await this.currentExportingStatus.init(collections);
 
     try {
-      await this.exportCollectionsToZippedJson(models);
+      await this.exportCollectionsToZippedJson(collections);
     }
     finally {
       this.currentExportingStatus = null;