From 766dea9c243a96f382144b81b25133a07ad1f2ea Mon Sep 17 00:00:00 2001
From: simple321vip <simple321@vip.qq.com>
Date: Sun, 11 Sep 2022 22:24:22 +0800
Subject: [PATCH] add element icon, add flask client, add strategy route, api
 modify

---
 src/api/cta_strategy.ts               | 116 +++++++++++++------
 src/main.ts                           |   8 ++
 src/router/index.ts                   |  14 +++
 src/utils/request.ts                  |   6 +-
 src/view/trader/cta_strategy.vue      | 157 ++++++++++++++------------
 src/view/trader/cta_strategy_file.vue | 126 +++++++++++++++++++++
 6 files changed, 314 insertions(+), 113 deletions(-)
 create mode 100644 src/view/trader/cta_strategy_file.vue

diff --git a/src/api/cta_strategy.ts b/src/api/cta_strategy.ts
index 64a4e9e..c119629 100644
--- a/src/api/cta_strategy.ts
+++ b/src/api/cta_strategy.ts
@@ -1,69 +1,113 @@
-import { service2 } from '../utils/request'
+import { trader_service } from '../utils/request'
 
 "CTA strategy"
 
-"strategy list"
-const get_strategy_list = () => {
-  return service2({
-    url: '/auth/api/v1/trader',
+"strategy file list"
+const get_strategy_files = () => {
+  return trader_service({
+    url: '/trader/api/v1/strategy_file',
+    method: 'GET',
+  })
+}
+
+"upload a strategy file"
+const upload_strategy_file = (file: any) => {
+  const formData = new FormData();
+  formData.append("file", file)
+  return trader_service({
+    url: '/trader/api/v1/strategy_file',
+    method: 'POST',
+    data: formData,
+    headers: {
+      'Content-Type': 'multipart/form-data',
+    }
+  })
+}
+
+"load a strategy file"
+const load_strategy_file = (file_name: string) => {
+  return trader_service({
+    url: '/trader/api/v1/strategy_file/' + file_name,
     method: 'PUT',
   })
 }
 
-"upload a strategy"
-const upload_strategy = () => {
-  return service2({
-    url: '/auth/api/v1/trader',
-    method: 'PUT',
+"unload a strategy file"
+const unload_strategy_file = (class_name: string) => {
+  return trader_service({
+    url: '/trader/api/v1/strategy_file/' + class_name,
+    method: 'PATCH',
   })
 }
 
-"load a strategy"
-const load_strategy = (strategy_id: string) => {
-  return service2({
-    url: '/auth/api/v1/trader',
-    method: 'PUT',
+"remove a strategy file"
+const remove_strategy_file = (file_name: string) => {
+  return trader_service({
+    url: '/trader/api/v1/strategy_file/' + file_name,
+    method: 'DELETE',
   })
 }
 
-"unload a strategy"
-const unload_strategy = (strategy_id: string) => {
-  return service2({
-    url: '/auth/api/v1/trader',
-    method: 'PUT',
+
+"get strategy list"
+const get_strategies = () => {
+  return trader_service({
+    url: '/trader/api/v1/strategies',
+    method: 'GET',
+  })
+}
+
+"create a strategy"
+const create_strategy = () => {
+  return trader_service({
+    url: '/trader/api/v1/strategy',
+    method: 'POST',
+    data: {}
   })
 }
 
 "start a strategy"
-const start_strategy = (strategy_id: string) => {
-  return service2({
-    url: '/auth/api/v1/trader',
+const start_strategy = (strategy_name: string) => {
+  return trader_service({
+    url: '/trader/api/v1/strategy/' + strategy_name,
     method: 'PUT',
   })
 }
 
 "stop a strategy"
-const stop_strategy = (strategy_id: string) => {
-  return service2({
-    url: '/auth/api/v1/trader',
-    method: 'PUT',
+const stop_strategy = (strategy_name: string) => {
+  return trader_service({
+    url: '/trader/api/v1/strategy/' + strategy_name,
+    method: 'PATCH',
   })
 }
 
 "remove a strategy"
-const remove_strategy = (strategy_id: string) => {
-  return service2({
-    url: '/auth/api/v1/trader',
-    method: 'PUT',
+const remove_strategy = (strategy_name: string) => {
+  return trader_service({
+    url: '/trader/api/v1/strategy/' + strategy_name,
+    method: 'DELETE',
+  })
+}
+
+"get a strategy status"
+const get_strategy_status = (strategy_name: string) => {
+  return trader_service({
+    url: '/trader/api/v1/strategy/status/' + strategy_name,
+    method: 'GET',
   })
 }
 
 export {
-  get_strategy_list,
-  upload_strategy,
-  load_strategy,
-  unload_strategy,
+  get_strategy_files,
+  upload_strategy_file,
+  load_strategy_file,
+  unload_strategy_file,
+  remove_strategy_file,
+  get_strategies,
+  create_strategy,
   start_strategy,
   stop_strategy,
-  remove_strategy
+  remove_strategy,
+  get_strategy_status
 }
\ No newline at end of file
diff --git a/src/main.ts b/src/main.ts
index a7c7e61..aa1f634 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -7,8 +7,16 @@ import router from './router'
 import { createPinia } from 'pinia'
 const store = createPinia()
 
+import * as ElementPlusIconsVue from '@element-plus/icons-vue'
+
 let app = createApp(App)
 
+
+for (const [key, component] of Object.entries(ElementPlusIconsVue)) {
+  app.component(key, component)
+}
+
+
 app.use(router)
 app.use(ElementPlus)
 app.use(store)
diff --git a/src/router/index.ts b/src/router/index.ts
index c2c10fb..8bcab45 100644
--- a/src/router/index.ts
+++ b/src/router/index.ts
@@ -68,6 +68,20 @@ const routes: Array<RouteRecordRaw> = [
           name: 'trader'
         }
       },
+      {
+        path: '/strategy',
+        component: () => import('../view/trader/cta_strategy_file.vue'),
+        meta: {
+          name: '策略文件管理'
+        }
+      },
+      {
+        path: '/strategy_instance',
+        component: () => import('../view/trader/cta_strategy.vue'),
+        meta: {
+          name: '策略管理'
+        }
+      },
     ]
   },
   {
diff --git a/src/utils/request.ts b/src/utils/request.ts
index cf19357..16921c5 100644
--- a/src/utils/request.ts
+++ b/src/utils/request.ts
@@ -10,7 +10,7 @@ const service = axios.create(
 )
 
 // violin-trader
-const service2 = axios.create(
+const trader_service = axios.create(
   {
     baseURL: import.meta.env.VITE_APP_URL2 as string,
     timeout: 5000
@@ -35,7 +35,7 @@ service.interceptors.request.use(
   }
 )
 
-service2.interceptors.request.use(
+trader_service.interceptors.request.use(
   config => {
     if (typeof getToken() === 'string') {
       (config as any).headers['Authorization'] = 'Bearer' + getToken()
@@ -82,4 +82,4 @@ service2.interceptors.request.use(
 //   }
 // )
 
-export { service, service2 }
\ No newline at end of file
+export { service, trader_service }
\ No newline at end of file
diff --git a/src/view/trader/cta_strategy.vue b/src/view/trader/cta_strategy.vue
index 3b1920e..7ecae50 100644
--- a/src/view/trader/cta_strategy.vue
+++ b/src/view/trader/cta_strategy.vue
@@ -11,132 +11,141 @@
           <el-button type="primary" @click="doSearch">检索</el-button>
         </el-col>
       </el-row>
-      <div class="tag_list">
-        <el-tag class="ml-2 click-icon" :type="item.clicked ? 'danger' : 'info'" v-for="(item) in bookmark_types"
-          @click="handleTags(item)">{{ item.bk_type_name }}</el-tag>
-        <el-button type="primary" @click="handleAddType" v-show="Tenant.account">增加</el-button>
-        <el-button type="primary" @click="handleManageType" v-show="Tenant.account">自定义</el-button>
-      </div>
-    </el-form>
+    </el-form> -->
     <div class="create_dialog">
-      <el-button type="primary" @click="handleInsert" v-show="Tenant.account">新建</el-button>
+      <el-button type="primary" @click="on_create">新建</el-button>
     </div>
 
-    <el-table ref="multipleTableRef" :data="strategy_list" @selection-change="handleSelectionChange" style="width: 100%">
+    <el-table ref="multipleTableRef" :data="strategy_list" style="width: 100%">
       <el-table-column type="selection" width="55" />
       <el-table-column type="index" label="index" width="80" />
-      <el-table-column prop="bk_type_name" label="分类" width="100" />
-      <el-table-column prop="comment" label="名称" width="180" />
-      <el-table-column prop="facvion" label="图标" width="60" />
-      <el-table-column class="click-icon" prop="url" label="url">
+      <el-table-column prop="strategy_name" label="策略名" width="200" />
+      <el-table-column prop="class_name" label="策略类名" width="200" />
+      <el-table-column prop="vt_symbol" label="合约" width="100" />
+      <el-table-column prop="status" label="状态" width="60">
         <template #default="scope">
-          <span style="cursor: pointer" @click="openUrl(scope.row.url)">{{ scope.row.url }}</span>
+          <el-icon v-if="scope.row.status == 0" :size="20">
+            <Cloudy />
+          </el-icon>
+          <el-icon v-if="scope.row.status == 1" :size="20" color="yellow">
+            <Sunrise />
+          </el-icon>
+          <el-icon v-if="scope.row.status == 2" :size="20" color="gold">
+            <Sunny />
+          </el-icon>
+          <el-icon v-if="scope.row.status == 3" :size="20" color="blue">
+            <Moon />
+          </el-icon>
         </template>
       </el-table-column>
-      <el-table-column label="操作">
+      <el-table-column prop="url" label="开始" width="60">
         <template #default="scope">
-          <el-icon :size="20" @click="copyNumber(scope.row)" class="click-icon">
-            <CopyDocument />
-          </el-icon>
-          <el-button class="click-icon" size="small" @click="handleEdit(scope.$index, scope.row)"
-            v-show="Tenant.account">编辑
-          </el-button>
-          <el-button size="small" type="danger" @click="handleDelete(scope.$index, scope.row)" v-show="Tenant.account">
-            删除
-          </el-button>
+          <span v-if="scope.row.status == 0" style="cursor: pointer" @click="on_remove(scope.row)">
+            <el-icon :size="20" @click="on_start(scope.row)">
+              <VideoPlay />
+            </el-icon>
+          </span>
+        </template>
+      </el-table-column>
+      <el-table-column prop="url" label="停止" width="60">
+        <template #default="scope">
+          <span v-if="scope.row.status == 2" style="cursor: pointer" @click="on_remove(scope.row)">
+            <el-icon :size="20" @click="on_stop(scope.row)">
+              <VideoPause />
+            </el-icon>
+          </span>
+        </template>
+      </el-table-column>
+      <el-table-column prop="url" label="删除" width="60">
+        <template #default="scope">
+          <span v-if="scope.row.status == 0" style="cursor: pointer" @click="on_remove(scope.row)">
+            <el-icon :size="20" @click="on_remove(scope.row)">
+              <Delete />
+            </el-icon>
+          </span>
         </template>
       </el-table-column>
     </el-table>
-    <div style="margin-top: 20px">
-      <el-button @click="toggleSelection([tableData[1], tableData[2]])">反选</el-button>
-      <el-button @click="toggleSelection()">清除</el-button>
-    </div> -->
-
-    <!-- <el-dialog v-model="dialogFormVisible" title="添加书签">
-      <bookmark_dialog :dialog_form="currentDialogData" :operate_code="operate" :types="bookmark_types"
-        @on-concel="closeDialog" @on-submit="doSubmit"></bookmark_dialog>
-    </el-dialog>
-
-    <el-dialog v-model="dialogVisible">
-      <delete_dialog :delete_id="currentDialogData.bk_id" @on-submit="doDelete"></delete_dialog>
-    </el-dialog>
-
-    <el-dialog v-model="createTypeDialogVisible">
-      <createType_dialog @on-update="doUpdate"></createType_dialog>
-    </el-dialog> -->
-
-    <!-- <el-dialog v-model="typeDialogVisible">
-      <bookmarkType_dialog :delete_id="currentDialogData.bk_id" @on-submit="doDelete"></bookmarkType_dialog>
-    </el-dialog> -->
   </div>
 </template>
 
 <script setup lang="ts">
-import { reactive } from 'vue';
+import { reactive, ref } from 'vue';
 import {
-  get_strategy_list,
-  upload_strategy,
-  load_strategy,
-  unload_strategy,
+  get_strategies,
+  create_strategy,
   start_strategy,
   stop_strategy,
-  remove_strategy
+  get_strategy_status,
 } from '../../api/cta_strategy'
+import { genFileId } from 'element-plus'
+import type { UploadInstance, UploadProps, UploadRawFile } from 'element-plus'
+import { h } from 'vue'
+import { ElMessage } from 'element-plus'
+
+const upload = ref<UploadInstance>()
+const handleExceed: UploadProps['onExceed'] = (files) => {
+  upload.value!.clearFiles()
+  const file = files[0] as UploadRawFile
+  file.uid = genFileId()
+  upload.value!.handleStart(file)
+}
 
 
 type Strategy = {
-  strategy_id: string,
   strategy_name: string,
+  class_name: string,
+  vt_symbol: string,
+  setting: Object,
+  status: number,
 }
 
 const strategy_list = reactive<Strategy[]>([])
 
 const on_list = () => {
-  get_strategy_list().then(res => {
+  get_strategies().then(res => {
+    strategy_list.length = 0
     res.data.forEach((item: Strategy, index: number) => {
       strategy_list.push(item)
     })
   })
 }
 
-const on_upload = () => {
-  upload_strategy().then(res => {
+const on_create = () => {
+
+
+}
+
+const on_start = (row: any) => {
+  start_strategy(row.strategy_name).then(res => {
 
   })
 }
 
-const on_load = (strategy_id: string) => {
-  load_strategy(strategy_id).then(res => {
+const on_stop = (row: any) => {
+  stop_strategy(row.strategy_name).then(res => {
 
   })
 }
 
-const on_unload = (strategy_id: string) => {
-  unload_strategy(strategy_id).then(res => {
-
+const on_remove = (row: any) => {
+  stop_strategy(row.strategy_name).then(res => {
+    on_list()
   })
 }
 
-const on_start = (strategy_id: string) => {
-  start_strategy(strategy_id).then(res => {
-
+const on_status = (row: any) => {
+  get_strategy_status(row.strategy_name).then(res => {
+    row.status = res.data.status
   })
 }
 
-const on_stop = (strategy_id: string) => {
-  stop_strategy(strategy_id).then(res => {
-
-  })
-}
-
-const on_remove = (strategy_id: string) => {
-  remove_strategy(strategy_id).then(res => {
-
-  })
-}
+// init process
+on_list()
 
 
 </script>
 
 <style scoped>
+
 </style>
\ No newline at end of file
diff --git a/src/view/trader/cta_strategy_file.vue b/src/view/trader/cta_strategy_file.vue
new file mode 100644
index 0000000..ec87e8d
--- /dev/null
+++ b/src/view/trader/cta_strategy_file.vue
@@ -0,0 +1,126 @@
+<template>
+  <div class="bookmark">
+    <el-upload ref="upload" class="upload-demo" action="http://localhost:5000/trader/api/v1/strategy_file" :limit="1"
+      :on-exceed="handleExceed" :auto-upload="false">
+      <template #trigger>
+        <el-button type="primary">select file</el-button>
+      </template>
+      <el-button class="ml-3" type="success" @click="on_upload">
+        upload to server
+      </el-button>
+      <template #tip>
+        <div class="el-upload__tip text-red">
+          limit 1 file, new file will cover the old file
+        </div>
+      </template>
+    </el-upload>
+
+    <el-table ref="multipleTableRef" :data="strategy_file_list" style="width: 100%">
+      <el-table-column type="selection" width="55" />
+      <el-table-column type="index" label="index" width="80" />
+      <el-table-column prop="file_name" label="策略文件名" width="200" />
+      <el-table-column prop="class_name" label="策略类名" width="200" />
+      <el-table-column prop="status" label="状态" width="180" />
+      <el-table-column prop="facvion" label="图标" width="60" />
+      <el-table-column class="click-icon" prop="url" label="操作">
+        <template #default="scope">
+          <span v-if="scope.row.status == 0" style="cursor: pointer" @click="on_load(scope.row)">{{ "加载图标" }}</span>
+          <span v-if="scope.row.status == 0" style="cursor: pointer" @click="on_remove(scope.row)">{{ "删除" }}</span>
+          <span v-if="scope.row.status == 2" style="cursor: pointer">{{ "加载中图标" }}</span>
+          <span v-if="scope.row.status == 1" style="cursor: pointer" @click="on_unload(scope.row)">{{
+          "卸载图标" }}</span>
+        </template>
+      </el-table-column>
+    </el-table>
+  </div>
+</template>
+
+<script setup lang="ts">
+import { reactive, ref } from 'vue';
+import {
+  get_strategy_files,
+  load_strategy_file,
+  unload_strategy_file,
+  remove_strategy_file
+} from '../../api/cta_strategy'
+import { genFileId } from 'element-plus'
+import type { UploadInstance, UploadProps, UploadRawFile } from 'element-plus'
+import { h } from 'vue'
+import { ElMessage } from 'element-plus'
+
+const upload = ref<UploadInstance>()
+const handleExceed: UploadProps['onExceed'] = (files) => {
+  upload.value!.clearFiles()
+  const file = files[0] as UploadRawFile
+  file.uid = genFileId()
+  upload.value!.handleStart(file)
+}
+
+
+type StrategyFile = {
+  file_name: string,
+  class_name: string,
+  status: string
+}
+
+const strategy_file_list = reactive<StrategyFile[]>([])
+
+const on_list = () => {
+  get_strategy_files().then(res => {
+    strategy_file_list.length = 0
+    res.data.forEach((item: StrategyFile, index: number) => {
+      strategy_file_list.push(item)
+    })
+  })
+}
+
+const on_upload = () => {
+
+  upload.value!.submit()
+  on_list()
+}
+
+const on_load = (row: any) => {
+  row.status = 2
+  load_strategy_file(row.file_name).then(res => {
+    if (res.status == 200) {
+      row.status = 1
+    }
+    if (res.status == 500) {
+      row.status = 0
+    }
+  })
+}
+
+const on_unload = (row: any) => {
+  unload_strategy_file(row.class_name).then(res => {
+    if (res.status == 200) {
+      row.status = 0
+    }
+    if (res.status == 500) {
+      row.status = 1
+    }
+  })
+}
+
+const on_remove = (row: any) => {
+  remove_strategy_file(row.file_name).then(res => {
+    on_list()
+  }).catch(error => {
+    ElMessage({
+      message: h('p', null, [
+        h('i', { style: 'color: teal' }, error),
+      ]),
+    })
+  })
+}
+
+// init process
+on_list()
+
+
+</script>
+
+<style scoped>
+
+</style>
\ No newline at end of file