account-manager/app/controller/SNContest.js

'use strict';

const Controller = require('../core/base_controller');

const fs = require('fs');

const fileName = "SNContest";

class SNContestController extends Controller {
    // 获取SN号竞赛消息
    async getSNContestMsg() {
        const { ctx } = this;
        try {
            let rdmcode = await this.checkUid(ctx);
            if (!rdmcode) {
                this.setResponse(ctx, 401, 'token过期或错误', "none");
                return;
            }
            const body = ctx.request.body;
            const result = await ctx.service.sNContest.getSNContestMsg(body);
            if (result == "error") {
                this.setResponse(ctx, 500, '服务器错误', "none");
                return;
            }
            if (result) {
                let snContestMsg = [];
                result.rows.forEach(element => {
                    snContestMsg.push(element.dataValues);
                });
                const data = {
                    snContestMsg: snContestMsg,
                    count: result.count
                };
                this.setResponse(ctx, 200, '获取SN号竞赛消息成功', "deconstruction", data);
            } else {
                this.setResponse(ctx, 404, '获取SN号竞赛消息失败', "none");
            }
        } catch (error) {
            const blockName = fileName + "--" + "getSNContestMsg";
            ctx.app.middleware.log('../../controllerLog', blockName + ">>>>" + error);
            this.setResponse(ctx, 500, '服务器错误', "none");
        }
    }

    // 管理系统导出SN号竞赛消息
    async downloadSNConest() {
        const { ctx } = this;
        try {
            let rdmcode = await this.checkUid(ctx);
            if (!rdmcode) {
                this.setResponse(ctx, 401, 'token过期或错误', "none");
                return;
            }
            const uid = JSON.parse(rdmcode).uid;
            const { form, searchMsg, dataHeaders } = ctx.request.body;
            const result = await ctx.service.sNContest.downloadSNConest(uid, searchMsg);
            if (result == "error") {
                this.setResponse(ctx, 500, '服务器错误', "none");
                return;
            }
            if (result.code) {
                const file = this.downloadData(form, result.data, dataHeaders);
                if (file) {
                    const { filePath, completeName, buf } = file;
                    try {
                        fs.writeFileSync(filePath, Buffer.from(buf));
                        const data = {
                            // 本机服务器地址
                            // 'http://127.0.0.1:8012'
                            // 'http://192.168.31.28:8012'
                            // 公司服务器地址
                            // 'http://123.57.205.46:8012'
                            // 'http://manager.bdrd.hwasmart.com//accountManager'
                            fileUrl: 'http://127.0.0.1:8012' + `/public/${completeName}?time=${new Date().getTime()}`,
                            fileName: completeName,
                        }
                        this.setResponse(ctx, 200, '文件已成功写入', "deconstruction", data);
                        setTimeout(() => {
                            fs.unlinkSync(filePath);
                        }, 600000);
                    } catch (error) {
                        this.setResponse(ctx, 500, '写入文件时发生错误', "none");
                    }
                } else {
                    this.setResponse(ctx, 404, '操作失败', "none");
                }
            } else {
                this.setResponse(ctx, 404, '操作失败', "none");
            }
        } catch (error) {
            const blockName = fileName + "--" + "downloadSNConest";
            ctx.app.middleware.log('../../controllerLog', blockName + ">>>>" + error);
            this.setResponse(ctx, 500, '服务器错误', "none");
        }
    }
}

module.exports = SNContestController;