@nger/fk-upload 1.0.201 → 1.0.204

package/dist/services/upload.service.js

@@ -0,0 +1,100 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.UploadService = void 0;
+const tslib_1 = require("tslib");
+const upload_task_1 = require("./../entities/upload-task");
+const core_1 = require("@nger/core");
+const typeorm_1 = require("@nger/typeorm");
+const rabbitmq_1 = require("@nger/rabbitmq");
+const actions_1 = require("../tasks/actions");
+let UploadService = class UploadService {
+    injector;
+    db;
+    constructor(injector, db) {
+        this.injector = injector;
+        this.db = db;
+    }
+    async startUpload(id) {
+        const download = await this.db.manager.findOneOrFail(upload_task_1.FkUploadTask, { where: { id } });
+        if (download.status === '1') {
+            // no start
+            await this.db.manager.update(upload_task_1.FkUploadTask, id, { status: `2` });
+            await this.start(id);
+        }
+        else if (download.status === '2') {
+            // wait
+            await this.start(id);
+        }
+        else if (download.status === '3') {
+            // uploading
+            await this.db.manager.update(upload_task_1.FkUploadTask, id, { status: `5` });
+            await this.continue(id);
+        }
+        else if (download.status === '4') {
+            // stop
+            await this.db.manager.update(upload_task_1.FkUploadTask, id, { status: `5` });
+            await this.continue(id);
+        }
+        else if (download.status === '5') {
+            // wait continue
+            await this.continue(id);
+        }
+        else if (download.status === '6') {
+            // error
+            await this.db.manager.update(upload_task_1.FkUploadTask, id, { status: `5` });
+            await this.continue(id);
+        }
+        else if (download.status === '7') {
+        }
+        else {
+            await this.continue(id);
+        }
+    }
+    async start(id) {
+        const db = this.injector.get(typeorm_1.Db);
+        const upload = await db.manager.findOneOrFail(upload_task_1.FkUploadTask, {
+            where: { id }
+        });
+        const taskManager = this.injector.get(rabbitmq_1.TaskManager);
+        const t = new rabbitmq_1.TaskEntity();
+        t.topic = actions_1.UPLOAD_ACTION;
+        t.data = { id: upload.id };
+        await taskManager.send(t);
+    }
+    async starts(ids) {
+        return Promise.all(ids.map(id => this.startUpload(id)));
+    }
+    async continue(id) {
+        const db = this.injector.get(typeorm_1.Db);
+        const upload = await db.manager.findOneOrFail(upload_task_1.FkUploadTask, {
+            where: { id }
+        });
+        const taskManager = this.injector.get(rabbitmq_1.TaskManager);
+        const t = new rabbitmq_1.TaskEntity();
+        t.topic = actions_1.UPLOAD_CONTINUE_ACTION;
+        t.data = { id: upload.id };
+        await taskManager.send(t);
+    }
+    async stop(id) {
+        const db = this.injector.get(typeorm_1.Db);
+        const upload = await db.manager.findOneOrFail(upload_task_1.FkUploadTask, {
+            where: { id }
+        });
+        if (upload.status === '7') { }
+        else if (upload.status === '1') {
+        }
+        else if (upload.status === '2') {
+        }
+        else {
+            await db.manager.update(upload_task_1.FkUploadTask, id, { status: '4' });
+        }
+    }
+    async stops(ids) {
+        return Promise.all((ids || []).map(id => this.stop(id)));
+    }
+};
+UploadService = tslib_1.__decorate([
+    (0, core_1.Injectable)(),
+    tslib_1.__metadata("design:paramtypes", [core_1.Injector, typeorm_1.Db])
+], UploadService);
+exports.UploadService = UploadService;

package/dist/tasks/actions.d.ts

@@ -0,0 +1,28 @@
+import { StringToken } from "@nger/core";
+export interface IDownloadTask {
+    id: number;
+}
+export declare const DOWNLOAD_ACTION: StringToken<IDownloadTask>;
+export declare const DOWNLOAD_CONTINUE_ACTION: StringToken<IDownloadTask>;
+export interface IScheduleTask {
+    name: string;
+    rule: string;
+}
+export declare const SCHEDULE_ACTION: StringToken<IScheduleTask>;
+export interface IUploadTask {
+    id: number;
+}
+export declare const UPLOAD_ACTION: StringToken<IUploadTask>;
+export declare const UPLOAD_CONTINUE_ACTION: StringToken<IUploadTask>;
+export declare const YASUO_ACTION: StringToken<IUploadTask>;
+export interface IEffectTask {
+    id: number;
+}
+export declare const EFFECT_ACTION: StringToken<IEffectTask>;
+export interface ICheckTask {
+    id: number;
+    to: string;
+    from: string;
+}
+export declare const CHECK_ACTION: StringToken<ICheckTask>;
+export declare const CHECK_CONTINUE_ACTION: StringToken<ICheckTask>;

package/dist/tasks/actions.js

@@ -0,0 +1,12 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CHECK_CONTINUE_ACTION = exports.CHECK_ACTION = exports.EFFECT_ACTION = exports.YASUO_ACTION = exports.UPLOAD_CONTINUE_ACTION = exports.UPLOAD_ACTION = exports.SCHEDULE_ACTION = exports.DOWNLOAD_CONTINUE_ACTION = exports.DOWNLOAD_ACTION = void 0;
+exports.DOWNLOAD_ACTION = `@nger/fk-upload/download-task`;
+exports.DOWNLOAD_CONTINUE_ACTION = `@nger/fk-upload/download-continue-task`;
+exports.SCHEDULE_ACTION = `@nger/fk-upload/schedule-task`;
+exports.UPLOAD_ACTION = `@nger/fk-upload/upload-task`;
+exports.UPLOAD_CONTINUE_ACTION = `@nger/fk-upload/upload-continue-task`;
+exports.YASUO_ACTION = `@nger/fk-upload/yasuo-task`;
+exports.EFFECT_ACTION = `@nger/fk-upload/effect-task`;
+exports.CHECK_ACTION = `@nger/fk-upload/check-task`;
+exports.CHECK_CONTINUE_ACTION = `@nger/fk-upload/check-continue-task`;

package/dist/{templates/tasks/schedule-task.d.ts → tasks/check-task.d.ts}

@@ -1,9 +1,7 @@
-import { Task } from "@nger/rabbitmq";
 import { Injector, Next } from "@nger/core";
-/**
- * finish
- */
-export declare class ScheduleTask extends Task {
+import { Task } from "@nger/rabbitmq";
+export declare class CheckTask extends Task {
     constructor();
+    ids: Set<any>;
     handle(injector: Injector, next?: Next): Promise<void>;
 }

package/dist/tasks/check-task.js

@@ -0,0 +1,205 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CheckTask = void 0;
+const effect_service_1 = require("./../services/effect.service");
+const effect_task_1 = require("./../entities/effect-task");
+const typeorm_1 = require("@nger/typeorm");
+const rabbitmq_1 = require("@nger/rabbitmq");
+const actions_1 = require("./actions");
+const check_task_1 = require("../entities/check-task");
+const schedule_1 = require("@nger/schedule");
+const typeorm_2 = require("typeorm");
+const w7_chat_topic_1 = require("../entities/w7_chat_topic");
+const w7_uni_account_1 = require("../entities/w7_uni_account");
+const download_task_1 = require("../entities/download-task");
+const upload_task_1 = require("../entities/upload-task");
+const date_fns_1 = require("date-fns");
+const download_service_1 = require("../services/download.service");
+const upload_service_1 = require("../services/upload.service");
+class CheckTask extends rabbitmq_1.Task {
+    constructor() {
+        super(actions_1.CHECK_ACTION);
+        this.count = 10;
+    }
+    ids = new Set();
+    async handle(injector, next) {
+        const db = injector.get(typeorm_1.Db);
+        const task = injector.get(rabbitmq_1.DATA);
+        const complete = injector.get(rabbitmq_1.COMPLETE);
+        const fail = injector.get(rabbitmq_1.FAIL);
+        const { id, from, to } = task;
+        const check = await db.manager.findOne(check_task_1.FkCheckTask, { where: { id: task.id }, relations: ['w7Account', 'w7Account.fkAccount', 'w7Account.w7'] });
+        if (!check) {
+            complete();
+            return next && next();
+        }
+        console.log({ from, to, id, status: check.status });
+        const manager = injector.get(rabbitmq_1.TaskManager);
+        if (this.ids.has(id)) {
+            complete();
+            return next && next();
+        }
+        if (check.status === '1') {
+            let job;
+            console.log(`work start ------- \n\n\n`);
+            this.ids.add(task.id);
+            const finish = async (reson) => {
+                this.ids.delete(check.id);
+                await db.manager.update(check_task_1.FkCheckTask, check.id, { status: `1`, desc: reson });
+                complete();
+                schedule.cancelJob(check.title);
+            };
+            const create = async () => {
+                const item = new rabbitmq_1.TaskEntity();
+                item.topic = actions_1.CHECK_ACTION;
+                item.data = task;
+                await manager.send(item);
+            };
+            await db.manager.update(check_task_1.FkCheckTask, check.id, { status: `3`, desc: `` });
+            const schedule = injector.get(schedule_1.ScheduleService);
+            const type = check.type;
+            const w7Account = check.w7Account;
+            if (!w7Account) {
+                await finish(`绑定错误`);
+                return next && next();
+            }
+            const w7 = w7Account.w7;
+            if (!w7) {
+                await finish(`绑定错误`);
+                return next && next();
+            }
+            const fkAccount = w7Account.fkAccount;
+            if (!fkAccount) {
+                await finish(`绑定错误`);
+                complete();
+                return next && next();
+            }
+            const ds = new typeorm_2.DataSource({
+                type: 'mysql',
+                host: w7.host,
+                port: w7.port,
+                password: w7.password,
+                database: w7.database,
+                username: w7.username,
+                synchronize: false,
+                entities: [
+                    w7_chat_topic_1.W7ChatTopicEntity,
+                    w7_uni_account_1.W7UniAccount
+                ]
+            });
+            await ds.initialize();
+            const w7UniAccount = await ds.manager.findOne(w7_uni_account_1.W7UniAccount, { where: { name: w7Account.username } });
+            if (!w7UniAccount) {
+                await finish(`绑定错误`);
+                return next && next();
+            }
+            const uniacid = w7UniAccount.uniacid;
+            const id = check.id;
+            const count = check.count;
+            async function run(time) {
+                console.log(time);
+                const runtimeCheck = await db.manager.findOne(check_task_1.FkCheckTask, { where: { id } });
+                if (!runtimeCheck) {
+                    await finish(``);
+                    return;
+                }
+                if (runtimeCheck.status !== `3`) {
+                    await finish(`人为关闭`);
+                    return;
+                }
+                if (type === 'create_download') {
+                    const now = new Date().getTime() - 1000 * 60 * 60 * 24 * 8;
+                    (0, date_fns_1.format)(now, 'yyyy');
+                    const _from = (0, date_fns_1.format)(now, 'yyyy-MM-dd HH:mm:ss');
+                    const nowTime = Math.floor(now / 1000);
+                    const downloads = await db.manager.find(download_task_1.FkDownloadTask, { where: { createDate: (0, typeorm_2.MoreThan)(_from) } });
+                    console.log(downloads.length);
+                    const ids = downloads.map(it => it.topicId).filter(it => !!it);
+                    const where = { uniacid, createTime: (0, typeorm_2.MoreThan)(nowTime) };
+                    if (ids.length > 0) {
+                        where.id = (0, typeorm_2.Not)((0, typeorm_2.In)(ids));
+                    }
+                    const topics = await ds.manager.find(w7_chat_topic_1.W7ChatTopicEntity, { where });
+                    topics.filter(it => it.thirdUrl && it.thirdUrl.length > 0 && it.thirdUrl.endsWith('.mp4') && it.topicStyle === 6)
+                        .map(async (topic) => {
+                        const downloadTask = new download_task_1.FkDownloadTask();
+                        downloadTask.topicId = topic.id;
+                        downloadTask.url = topic.thirdUrl;
+                        downloadTask.w7AccountId = w7Account.id;
+                        await db.manager.save(download_task_1.FkDownloadTask, downloadTask);
+                    });
+                    // start download
+                    const one = await db.manager.findOne(download_task_1.FkDownloadTask, { where: { status: `1` } });
+                    if (one) {
+                        const downloadService = injector.get(download_service_1.DownloadService);
+                        await downloadService.startDownload(one.id);
+                    }
+                    await db.manager.update(check_task_1.FkCheckTask, id, { count: count + 1 });
+                    await finish(`执行完成`);
+                    await create();
+                }
+                else if (type === 'create_upload') {
+                    // create upload
+                    const list = await db.manager.find(download_task_1.FkDownloadTask, { where: { status: '7', isUpload: false } });
+                    list.map(async (download) => {
+                        const uploadTask = new upload_task_1.FkUploadTask();
+                        uploadTask.downloadTaskId = download.id;
+                        db.tryRun(async (runner) => {
+                            await runner.manager.save(upload_task_1.FkUploadTask, uploadTask);
+                            await runner.manager.update(download_task_1.FkDownloadTask, download.id, { isUpload: true });
+                        });
+                    });
+                    // start upload
+                    const waitUploads = await db.manager.find(upload_task_1.FkUploadTask, { where: { status: '1' } });
+                    const uploadService = injector.get(upload_service_1.UploadService);
+                    if (waitUploads.length > 0) {
+                        await uploadService.starts(waitUploads.map(it => it.id));
+                    }
+                    // downloadService.startDownload()
+                    await db.manager.update(check_task_1.FkCheckTask, id, { count: count + 1 });
+                    await finish(`执行完成`);
+                    await create();
+                }
+                else if (type === 'create_effect') {
+                    const list = await db.manager.find(upload_task_1.FkUploadTask, { where: { status: '7', isEffect: false } });
+                    list.map(async (upload) => {
+                        db.tryRun(async (runner) => {
+                            const effectTask = new effect_task_1.FkEffectTask();
+                            effectTask.uploadId = upload.id;
+                            await runner.manager.save(effect_task_1.FkEffectTask, effectTask);
+                            await runner.manager.update(upload_task_1.FkUploadTask, upload.id, { isEffect: true });
+                        });
+                    });
+                    // start effect
+                    const waitEffects = await db.manager.find(effect_task_1.FkEffectTask, { where: { status: '1' } });
+                    const effectService = injector.get(effect_service_1.EffectService);
+                    if (waitEffects.length > 0) {
+                        await effectService.starts(waitEffects.map(it => it.id));
+                    }
+                    await db.manager.update(check_task_1.FkCheckTask, id, { count: count + 1 });
+                    await finish(`执行完成`);
+                    await create();
+                }
+                else {
+                    await finish(`未知任务`);
+                }
+            }
+            job = schedule.scheduleJob(check.title, check.time, async (time) => {
+                run(time);
+                /***
+                const manager = injector.get(TaskManager)
+                const i = new TaskEntity()
+                i.data = task;
+                i.topic = CHECK_ACTION;
+                await manager.send(i)
+                 */
+            });
+        }
+        else {
+            await db.manager.update(check_task_1.FkCheckTask, task.id, { status: `1` });
+            complete();
+            return next && next();
+        }
+    }
+}
+exports.CheckTask = CheckTask;

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

@@ -0,0 +1,9 @@
+import { Task } from "@nger/rabbitmq";
+import { Injector, Next } from "@nger/core";
+/**
+ * upload and task
+ */
+export declare class DownloadContinueTask extends Task {
+    constructor();
+    handle(injector: Injector, next?: Next): Promise<void>;
+}

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

@@ -0,0 +1,127 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DownloadContinueTask = void 0;
+const tslib_1 = require("tslib");
+const rabbitmq_1 = require("@nger/rabbitmq");
+const core_1 = require("@nger/core");
+const actions_1 = require("./actions");
+const request_1 = require("request");
+const fs_1 = require("fs");
+const md5_file_1 = tslib_1.__importDefault(require("md5-file"));
+const typeorm_1 = require("@nger/typeorm");
+const rxjs_1 = require("rxjs");
+const operators_1 = require("rxjs/operators");
+const download_task_1 = require("../entities/download-task");
+const client_service_1 = require("../services/client.service");
+const fk_account_1 = require("../entities/fk-account");
+/**
+ * upload and task
+ */
+class DownloadContinueTask extends rabbitmq_1.Task {
+    constructor() {
+        super(actions_1.DOWNLOAD_CONTINUE_ACTION);
+        this.count = 1;
+    }
+    async handle(injector, next) {
+        const db = injector.get(typeorm_1.Db);
+        const task = injector.get(rabbitmq_1.DATA);
+        const complete = injector.get(rabbitmq_1.COMPLETE);
+        const fail = injector.get(rabbitmq_1.FAIL);
+        const download = await db.manager.findOne(download_task_1.FkDownloadTask, {
+            where: { id: task.id }, relations: [
+                'jiqun',
+                'w7Account',
+                `w7Account.fkAccount`
+            ]
+        });
+        if (!download || (download && download.status !== `5`)) {
+            await complete();
+            return next && next();
+        }
+        if (!download.filename) {
+            await db.manager.update(download_task_1.FkDownloadTask, download.id, { status: `1`, desc: `文件名为空` });
+            await complete();
+            return next && next();
+        }
+        const client = injector.get(client_service_1.ClientService);
+        const clientId = client.createClientId();
+        if (download.jiqun && download.jiqun.clientId !== clientId) {
+            await fail();
+            return next && next();
+        }
+        const filePath = download.path;
+        await db.manager.update(download_task_1.FkDownloadTask, download.id, { status: '3', desc: '' });
+        const writeStream = (0, fs_1.createWriteStream)(filePath);
+        const url = download.url.startsWith('http') ? download.url : `https://${download.url}`;
+        const req = (0, request_1.get)(url, {
+            headers: {
+                'Content-Type': 'application/octet-stream'
+            }
+        });
+        let total = 0;
+        let length = 0;
+        const sub = new rxjs_1.Subject();
+        sub.pipe((0, operators_1.throttleTime)(400), (0, operators_1.switchMap)(async (size) => {
+            console.log(`download continue--${size}`);
+            const download = await db.manager.findOne(download_task_1.FkDownloadTask, { where: { id: task.id } });
+            if (!download || download.status !== '3') {
+                sub.error(new Error(`人为中断`));
+                return;
+            }
+            return db.manager.update(download_task_1.FkDownloadTask, download.id, { size: total, downloadSize: size });
+        })).subscribe({
+            next: () => { },
+            complete: async () => {
+                const fileMd5 = await (0, md5_file_1.default)(filePath);
+                await db.manager.update(download_task_1.FkDownloadTask, download.id, { md5: fileMd5, status: `7` });
+                await complete();
+            },
+            error: async (err) => {
+                req.abort();
+                await db.manager.update(download_task_1.FkDownloadTask, download.id, { isBigFile: true, status: '6', desc: err.message });
+                await fail();
+            }
+        });
+        // get split size
+        const _defer = (0, core_1.defer)();
+        req.on('data', async (buf) => {
+            length += buf.length;
+            writeStream.write(buf, (err) => {
+                if (err) {
+                    sub.error(err);
+                    return;
+                }
+                if (total === length) {
+                    _defer.resolve();
+                }
+                sub.next(length);
+            });
+        });
+        req.on('end', async () => {
+            await _defer;
+            sub.complete();
+        });
+        req.on('error', async (err) => {
+            sub.error(err);
+        });
+        req.on('response', async (resp) => {
+            const headers = resp.headers;
+            total = Number(headers["content-length"]);
+            if (total > 1024 * 1024 * 1024) {
+                sub.error(new Error(`文件超出1G`));
+                await _defer;
+                sub.complete();
+                return;
+            }
+            const useSize = download.w7Account.fkAccount.usedSize + total;
+            await db.manager.update(fk_account_1.FkAccount, download.w7Account.fkAccount.id, { usedSize: useSize });
+            if (download.w7Account.fkAccount.totalSize <= useSize + 1024) {
+                await db.manager.update(fk_account_1.FkAccount, download.w7Account.fkAccount.id, { warning: `2` });
+                req.abort();
+            }
+        });
+        if (next)
+            await next();
+    }
+}
+exports.DownloadContinueTask = DownloadContinueTask;

package/dist/{templates/tasks → tasks}/download-task.d.ts

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

package/dist/tasks/download-task.js

@@ -0,0 +1,128 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DownloadTask = void 0;
+const tslib_1 = require("tslib");
+const fk_account_1 = require("./../entities/fk-account");
+const rabbitmq_1 = require("@nger/rabbitmq");
+const core_1 = require("@nger/core");
+const actions_1 = require("./actions");
+const request_1 = require("request");
+const fs_1 = require("fs");
+const md5_file_1 = tslib_1.__importDefault(require("md5-file"));
+const typeorm_1 = require("@nger/typeorm");
+const rxjs_1 = require("rxjs");
+const operators_1 = require("rxjs/operators");
+const path_1 = require("path");
+const download_task_1 = require("../entities/download-task");
+const download_service_1 = require("../services/download.service");
+const client_service_1 = require("../services/client.service");
+const fk_jiqun_1 = require("../entities/fk-jiqun");
+/**
+ * upload and task
+ */
+class DownloadTask extends rabbitmq_1.Task {
+    constructor() {
+        super(actions_1.DOWNLOAD_ACTION);
+        this.count = 1;
+    }
+    async handle(injector, next) {
+        const db = injector.get(typeorm_1.Db);
+        const task = injector.get(rabbitmq_1.DATA);
+        const complete = injector.get(rabbitmq_1.COMPLETE);
+        const fail = injector.get(rabbitmq_1.FAIL);
+        const download = await db.manager.findOne(download_task_1.FkDownloadTask, {
+            where: { id: task.id }, relations: [
+                'w7Account',
+                `w7Account.fkAccount`
+            ]
+        });
+        if (!download || (download && download.status !== `2`)) {
+            await complete();
+            return next && next();
+        }
+        process.on('exit', async (code) => {
+            await db.manager.update(download_task_1.FkDownloadTask, task.id, { status: '6', desc: `异常退出:${code}` });
+            fail();
+        });
+        const downloadService = injector.get(download_service_1.DownloadService);
+        const dir = downloadService.createDir();
+        const filename = downloadService.createFilename(download.url);
+        const filePath = (0, path_1.join)(dir, filename);
+        const client = injector.get(client_service_1.ClientService);
+        const clientId = client.createClientId();
+        const jiqun = await db.manager.findOne(fk_jiqun_1.FkJiQun, { where: { clientId } });
+        await db.manager.update(download_task_1.FkDownloadTask, download.id, { status: '3', desc: '', path: filePath, filename, jiqunId: jiqun.id });
+        const writeStream = (0, fs_1.createWriteStream)(filePath);
+        const url = download.url.startsWith('http') ? download.url : `https://${download.url}`;
+        const req = (0, request_1.get)(url, {
+            headers: {
+                'Content-Type': 'application/octet-stream'
+            }
+        });
+        let total = 0;
+        let length = 0;
+        const sub = new rxjs_1.Subject();
+        sub.pipe((0, operators_1.throttleTime)(1000), (0, operators_1.switchMap)(async (size) => {
+            console.log(`download--${size}`);
+            const download = await db.manager.findOne(download_task_1.FkDownloadTask, { where: { id: task.id } });
+            if (!download || download.status !== '3') {
+                sub.error(new Error(`人为中断`));
+                return;
+            }
+            return db.manager.update(download_task_1.FkDownloadTask, download.id, { size: total, downloadSize: size });
+        })).subscribe({
+            next: () => { },
+            complete: async () => {
+                const fileMd5 = await (0, md5_file_1.default)(filePath);
+                await db.manager.update(download_task_1.FkDownloadTask, download.id, { downloadSize: total, md5: fileMd5, status: `7` });
+                await complete();
+            },
+            error: async (err) => {
+                req.abort();
+                await db.manager.update(download_task_1.FkDownloadTask, download.id, { status: '6', desc: err.message });
+                await fail();
+            }
+        });
+        // get split size
+        const _defer = (0, core_1.defer)();
+        req.on('data', async (buf) => {
+            length += buf.length;
+            writeStream.write(buf, (err) => {
+                if (err) {
+                    sub.error(err);
+                    return;
+                }
+                if (total === length) {
+                    _defer.resolve();
+                }
+                sub.next(length);
+            });
+        });
+        req.on('end', async () => {
+            await _defer;
+            sub.complete();
+        });
+        req.on('error', async (err) => {
+            sub.error(err);
+        });
+        req.on('response', async (resp) => {
+            const headers = resp.headers;
+            total = Number(headers["content-length"]);
+            if (total > 1024 * 1024 * 1024) {
+                sub.error(new Error(`文件超出1G`));
+                await _defer;
+                await db.manager.update(download_task_1.FkDownloadTask, download.id, { isBigFile: true, desc: '文件超出1G', status: `6` });
+                return;
+            }
+            const useSize = download.w7Account.fkAccount.usedSize + total;
+            await db.manager.update(fk_account_1.FkAccount, download.w7Account.fkAccount.id, { usedSize: useSize });
+            if (download.w7Account.fkAccount.totalSize <= useSize + 1024) {
+                await db.manager.update(fk_account_1.FkAccount, download.w7Account.fkAccount.id, { warning: `2` });
+                req.abort();
+            }
+        });
+        if (next)
+            await next();
+    }
+}
+exports.DownloadTask = DownloadTask;

package/dist/{templates/tasks → tasks}/effect-task.d.ts

@@ -1,5 +1,5 @@
 import { Task } from "@nger/rabbitmq";
-import { Next, Injector } from "@nger/core";
+import { Injector, Next } from "@nger/core";
 export declare class EffectTask extends Task {
     constructor();
     handle(injector: Injector, next?: Next): Promise<void>;