Parcourir la source

add close stream

reiji-h il y a 1 an
Parent
commit
46d0148ee2

+ 26 - 3
apps/app/src/migrations/20211227060705-revision-path-to-page-id-schema-migration--fixed-7549.js

@@ -1,7 +1,6 @@
 import { Writable } from 'stream';
 import { Writable } from 'stream';
 
 
 import mongoose from 'mongoose';
 import mongoose from 'mongoose';
-import { pages } from 'next/dist/build/templates/app-page';
 import streamToPromise from 'stream-to-promise';
 import streamToPromise from 'stream-to-promise';
 
 
 import getPageModel from '~/server/models/page';
 import getPageModel from '~/server/models/page';
@@ -58,8 +57,20 @@ module.exports = {
     });
     });
 
 
     pagesStream
     pagesStream
+      .on('error', () => {
+        batchStrem.end();
+        migratePagesStream.end();
+      })
       .pipe(batchStrem)
       .pipe(batchStrem)
-      .pipe(migratePagesStream);
+      .on('error', () => {
+        pagesStream.destory();
+        migratePagesStream.end();
+      })
+      .pipe(migratePagesStream)
+      .on('error', () => {
+        pagesStream.destory();
+        batchStrem.destory();
+      });
 
 
     await streamToPromise(migratePagesStream);
     await streamToPromise(migratePagesStream);
 
 
@@ -109,8 +120,20 @@ module.exports = {
     });
     });
 
 
     pagesStream
     pagesStream
+      .on('error', () => {
+        batchStrem.end();
+        migratePagesStream.end();
+      })
       .pipe(batchStrem)
       .pipe(batchStrem)
-      .pipe(migratePagesStream);
+      .on('error', () => {
+        pagesStream.destory();
+        migratePagesStream.end();
+      })
+      .pipe(migratePagesStream)
+      .on('error', () => {
+        pagesStream.destory();
+        batchStrem.destory();
+      });
 
 
     await streamToPromise(migratePagesStream);
     await streamToPromise(migratePagesStream);