account-manager/app/controller/Monitoring.js

'use strict';

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

const fs = require('fs');

const fileName = "Monitoring";

class MonitoringController extends Controller {
    // 查看危险操作记录
    async showMonitoring() {
        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.monitoring.showMonitoring(body);
            if (result == "error") {
                this.setResponse(ctx, 500, '服务器错误', "none");
                return;
            }
            if (result) {
                let MonitoringMsg = [];
                result.rows.forEach(element => {
                    MonitoringMsg.push(element.dataValues);
                });
                const data = {
                    MonitoringMsg: MonitoringMsg,
                    count: result.count
                };
                this.setResponse(ctx, 200, '搜索危险操作记录成功', "deconstruction", data);
            } else {
                this.setResponse(ctx, 404, '搜索危险操作记录失败', "none");
            }
        } catch (error) {
            const blockName = fileName + "--" + "showMonitoring";
            ctx.app.middleware.log('../../controllerLog', blockName + ">>>>" + error);
            this.setResponse(ctx, 500, '服务器错误', "none");
        }
    }

    // 删除危险操作记录
    async deleteMonitoring() {
        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 body = ctx.request.body;
            const { model, password, requestMsg } = body
            const modelType = this.decryptRequest(model);
            const deCodePassword = this.decryptRequest(password);
            const result = await ctx.service.monitoring.deleteMonitoring(uid, modelType, deCodePassword, requestMsg);
            if (result == "error") {
                this.setResponse(ctx, 500, '服务器错误', "none");
                return;
            }
            if (result) {
                this.setResponse(ctx, 200, '删除危险操作记录成功', "none");
            } else {
                this.setResponse(ctx, 404, '删除危险操作记录失败', "none");
            }
        } catch (error) {
            const blockName = fileName + "--" + "deleteMonitoring";
            ctx.app.middleware.log('../../controllerLog', blockName + ">>>>" + error);
            this.setResponse(ctx, 500, '服务器错误', "none");
        }
    }

    // 管理系统导出监控记录
    async downloadMonitoring() {
        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.monitoring.downloadMonitoring(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 + "--" + "downloadMonitoring";
            ctx.app.middleware.log('../../controllerLog', blockName + ">>>>" + error);
            this.setResponse(ctx, 500, '服务器错误', "none");
        }
    }
}

module.exports = MonitoringController;