新增功能

app/controller/Organization.js

@@ -2,6 +2,8 @@
 
 const Controller = require('../core/base_controller');
 
+const fs = require('fs');
+
 const fileName = "Organization";
 
 class OrganizationController extends Controller {
@@ -175,6 +177,58 @@ class OrganizationController extends Controller {
       this.setResponse(ctx, 500, '服务器错误', "none");
     }
   }
+
+  // 管理系统导出B端用户单位
+  async downloadOrganization() {
+    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.organization.downloadOrganization(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 + "--" + "downloadOrganization";
+      ctx.app.middleware.log('../../controllerLog', blockName + ">>>>" + error);
+      this.setResponse(ctx, 500, '服务器错误', "none");
+    }
+  }
 }
 
 module.exports = OrganizationController;