From e48a7653c7eca46abf036bffc55cfb27d27befe8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=A9=AC=E7=90=A6=E9=92=A7?= <wb-mqj919107@alibaba-inc.com>
Date: Sun, 8 Aug 2021 01:28:57 +0800
Subject: [PATCH] =?UTF-8?q?refactor:=20=E9=80=9A=E8=BF=87=E6=96=87?=
 =?UTF-8?q?=E4=BB=B6=E5=BD=A2=E5=BC=8F=E4=B8=8B=E8=BD=BD=E5=88=86=E7=B1=BB?=
 =?UTF-8?q?=E7=BB=93=E6=9E=9C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 fe/src/components/NER/index.vue | 28 ++++++++++++++++++++--------
 fe/src/js/utils.js              |  8 ++++++++
 2 files changed, 28 insertions(+), 8 deletions(-)
 create mode 100644 fe/src/js/utils.js

diff --git a/fe/src/components/NER/index.vue b/fe/src/components/NER/index.vue
index 27f7a55..a428f98 100644
--- a/fe/src/components/NER/index.vue
+++ b/fe/src/components/NER/index.vue
@@ -102,29 +102,39 @@
 
 <script>
 import { getColor } from '../../js/color.js'
+import { saveAsFile } from '../../js/utils.js'
 
 // 是否是单机版
 const isLocal = false
 
-function get (url, cb) {
-  query('GET', url, '', cb)
+function get (url, cb, config={}) {
+  query('GET', url, '', cb, config)
 }
-function post (url, data, cb) {
-  query('POST', url, data, cb)
+function post (url, data, cb, config={}) {
+  query('POST', url, data, cb, config)
 }
-function query (method, url, data = '', cb, tryTimes = 0) {
+function query (method, url, data = '', cb, config = {}) { 
+  console.log('config',config); 
+  config.tryTimes = config.tryTimes || 0
+  config.isDirect = config.isDirect || false
+
   var xhr = new XMLHttpRequest()
   xhr.open(method, url)
   xhr.setRequestHeader('content-type', 'application/json')
   xhr.onreadystatechange = function () {
     if (xhr.readyState === 4 && xhr.status === 200) {
+      if (config.isDirect) {
+        return cb(xhr.responseText) // 直接返回请求结果
+      }
       const result = JSON.parse(xhr.responseText)
+      console.log('result',result);
       if (result.errCode !== 0) {
-        if (tryTimes >= 2) {
+        if (config.tryTimes >= 2) {
           alert(result.errMsg)
         } else {
           setTimeout(() => {
-            query(method, url, data = '', cb, tryTimes + 1)
+            config.tryTimes += 1;
+            query(method, url, data = '', cb, config)
           }, 200)
         }
       } else {
@@ -527,7 +537,9 @@ export default {
         if (this.projectType === '命名实体识别')
           window.open(`/v1/files/get_json?projectName=${this.projectName}`, '_self')
         if (this.projectType === '文本分类')
-          window.open(`/v1/files/get_labels?projectName=${this.projectName}`, '_self')
+          get(`/v1/files/get_labels?projectName=${this.projectName}`, function(text){
+            saveAsFile(text, 'labels.json');
+          }, {isDirect : true})
         return true
       }
       this.nersCache[this.nowFile] = this.ners
diff --git a/fe/src/js/utils.js b/fe/src/js/utils.js
new file mode 100644
index 0000000..6330059
--- /dev/null
+++ b/fe/src/js/utils.js
@@ -0,0 +1,8 @@
+export function saveAsFile(data, name) {
+    var urlObject = window.URL || window.webkitURL || window;
+    var export_blob = new Blob([data]);
+    var save_link = document.createElementNS("http://www.w3.org/1999/xhtml", "a")
+    save_link.href = urlObject.createObjectURL(export_blob);
+    save_link.download = name;
+    save_link.click();
+}