@nger/fk-upload 1.0.68 → 1.0.71

package/dist/templates/tasks/download-task.d.ts

@@ -5,6 +5,7 @@ import { Injector, Next } from "@nger/core";
  */
 export declare class DownloadTask extends Task {
     constructor();
+    downloadingCount: number;
     handle(injector: Injector, next?: Next): Promise<void>;
     private createUploadTask;
 }

package/dist/templates/tasks/download-task.js

@@ -21,7 +21,12 @@ class DownloadTask extends rabbitmq_1.Task {
     constructor() {
         super(actions_1.DOWNLOAD_ACTION);
     }
+    downloadingCount = 0;
     async handle(injector, next) {
+        if (this.downloadingCount > 3) {
+            throw new Error(`system is busy`);
+        }
+        this.downloadingCount += 1;
         const db = injector.get(typeorm_1.Db);
         const task = injector.get(rabbitmq_1.DATA);
         const complete = injector.get(rabbitmq_1.COMPLETE);
@@ -66,6 +71,11 @@ class DownloadTask extends rabbitmq_1.Task {
         req.on('response', async (resp) => {
             const headers = resp.headers;
             total = Number(headers["content-length"]);
+            if (total > 1024 * 1024 * 1024) {
+                console.log(`file is to large`);
+                req.destroy();
+                complete();
+            }
         });
         if (next)
             await next();

package/dist/templates/tasks/effect-task.js

@@ -20,6 +20,10 @@ class EffectTask extends rabbitmq_1.Task {
         await db.manager.update(entities_1.FkDownloadTaskEntity, task.filename, { status: 3 });
         if (task.topicId) {
             await (0, fs_extra_1.unlink)(task.path);
+            if (task.uploadUrl.startsWith('https://site')) {
+                complete();
+                return;
+            }
             await w7.manager.update(w7_1.W7ChatTopicEntity, task.topicId, { thirdUrl: task.uploadUrl }).then(() => complete()).catch(() => fail());
             complete();
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nger/fk-upload",
-  "version": "1.0.68",
+  "version": "1.0.71",
   "description": "",
   "main": "dist/core.js",
   "types": "dist/core.d.ts",