From b191cbae6bfaf22695f574aa486aa0c45cd0e1e0 Mon Sep 17 00:00:00 2001
From: nineven <nineven@qq.com>
Date: Sat, 14 May 2022 14:19:52 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E4=BB=A3=E7=A0=81=EF=BC=8C?=
 =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=AD=98=E5=82=A8=E8=BF=81=E7=A7=BB=E9=94=81?=
 =?UTF-8?q?=E8=A7=A3=E9=99=A4=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../src/components/user/FirUserStorage.vue    | 51 ++++++++++++++++---
 fir_client/src/utils/index.js                 |  8 +++
 fir_ser/api/utils/modelutils.py               |  3 +-
 fir_ser/api/utils/utils.py                    |  5 +-
 fir_ser/api/views/storage.py                  | 11 +++-
 fir_ser/common/cache/state.py                 |  3 ++
 6 files changed, 70 insertions(+), 11 deletions(-)

diff --git a/fir_client/src/components/user/FirUserStorage.vue b/fir_client/src/components/user/FirUserStorage.vue
index 0365c31..74e910f 100644
--- a/fir_client/src/components/user/FirUserStorage.vue
+++ b/fir_client/src/components/user/FirUserStorage.vue
@@ -669,8 +669,6 @@
         <el-tag style="margin: 20px 0"> 应用版本数设置，当应用历史版本超过该限制，将会自动清理较老的版本</el-tag>
         <div style="margin: auto;width: 700px;height: 100%">
           <el-form ref="form" :model="storage_config" label-width="180px">
-
-
             <el-form-item label="应用历史版本数" style="text-align: left">
               <el-input-number v-model="storage_config.user_history_limit" :min="1"
                                style="width: 300px;margin: 0 10px"></el-input-number>
@@ -685,6 +683,18 @@
                 {{ diskSize(storage_config.user_max_storage_capacity - storage_config.user_used_storage_capacity) }}
               </el-tag>
             </el-form-item>
+
+
+            <el-form-item label="存储迁移状态" style="text-align: left">
+              <el-tag v-if="!storage_config.storage_status">未迁移，状态正常</el-tag>
+              <div v-else>
+                <el-tag>迁移中，迁移时间 {{ getFormatDate(storage_config.storage_status) }}</el-tag>
+                <el-tag type="warning">若存储状态长时间处于迁移中，可能是迁移卡死了，可以尝试
+                  <el-button plain size="mini" type="text" @click="cancelStorage">解除迁移锁</el-button>
+                  进行解除
+                </el-tag>
+              </div>
+            </el-form-item>
             <el-form-item label="清理所有应用数据" style="text-align: left">
               <el-button style="margin-left: 10px"
                          type="danger"
@@ -692,7 +702,6 @@
                 清理所有应用数据
               </el-button>
             </el-form-item>
-
             <el-form-item label="清理应用历史数据" style="text-align: left">
               <el-button style="margin-left: 10px"
                          type="danger"
@@ -700,7 +709,6 @@
                 清理应用历史版本数据，只保留最新版本数据
               </el-button>
             </el-form-item>
-
           </el-form>
         </div>
 
@@ -724,7 +732,7 @@
 
 <script>
 import {cleanStorageData, configStorageData, getStorageinfo, shareStorageData} from "@/restful";
-import {deepCopy, diskSize, getUserInfoFun} from "@/utils";
+import {deepCopy, diskSize, getFormatDate, getUserInfoFun} from "@/utils";
 import {format_time} from "@/utils/base/utils";
 
 export default {
@@ -734,7 +742,12 @@ export default {
       pagination: {"currentPage": 1, "total": 0, "pagesize": 10},
       target_uid: '',
       shareInfo: '',
-      storage_config: {'user_history_limit': 0, 'user_max_storage_capacity': 0, 'user_used_storage_capacity': 0},
+      storage_config: {
+        'user_history_limit': 0,
+        'user_max_storage_capacity': 0,
+        'user_used_storage_capacity': 0,
+        'storage_status': 0
+      },
       target_number: '',
       fstorage_lists: [],
       share_bill_lists: [],
@@ -765,6 +778,29 @@ export default {
   }, methods: {
     format_time,
     diskSize,
+    getFormatDate,
+    cancelStorage() {
+
+      this.$confirm("确定要解除迁移锁么，在未完成迁移状态下解除迁移锁之后，可能会导致数据异常?", '提示', {
+        confirmButtonText: '确定',
+        cancelButtonText: '取消',
+        type: 'warning'
+      }).then(() => {
+        configStorageData(res => {
+          if (res.code === 1000) {
+            this.$message.success("操作成功")
+            this.refreshactiveFun()
+          } else {
+            this.$message.error("操作失败了，" + res.msg)
+          }
+        }, {"methods": "POST", "data": this.storage_config})
+      }).catch(() => {
+        this.$message({
+          type: 'info',
+          message: '已取消撤回操作'
+        });
+      });
+    },
     storage_usedColor(percentage) {
       if (percentage < 20) {
         return '#6f7ad3';
@@ -792,7 +828,7 @@ export default {
         loading.close()
         if (res.code === 1000) {
           this.$message.success("操作成功")
-          this.storage_config = res.data
+          this.refreshactiveFun()
         } else {
           this.$message.error("操作失败了，" + res.msg)
         }
@@ -1048,6 +1084,7 @@ export default {
           loading.close();
           if (data.code === 1000) {
             this.$message.success('清理成功');
+            this.refreshactiveFun();
           } else {
             this.$message.error(data.msg)
           }
diff --git a/fir_client/src/utils/index.js b/fir_client/src/utils/index.js
index da1b12d..5bd8acd 100644
--- a/fir_client/src/utils/index.js
+++ b/fir_client/src/utils/index.js
@@ -662,3 +662,11 @@ export function format_choices(key, obj) {
     }
     return "未知"
 }
+
+export function getFormatDate(timestamp) {
+    let now = new Date(timestamp * 1000)
+    let y = now.getFullYear()
+    let m = now.getMonth() + 1
+    let d = now.getDate()
+    return y + "-" + (m < 10 ? "0" + m : m) + "-" + (d < 10 ? "0" + d : d) + " " + now.toTimeString().substr(0, 8);
+}
diff --git a/fir_ser/api/utils/modelutils.py b/fir_ser/api/utils/modelutils.py
index 79c077b..510223c 100644
--- a/fir_ser/api/utils/modelutils.py
+++ b/fir_ser/api/utils/modelutils.py
@@ -126,7 +126,8 @@ def base_random_weight(obj, key):
 
 
 def ad_random_weight(user_obj):
-    ad_info_list = UserAdDisplayInfo.objects.filter(user_id=user_obj, is_enable=True).order_by('-created_time')
+    ad_info_list = UserAdDisplayInfo.objects.filter(user_id=user_obj, is_enable=True, ad_pic__regex='.+').order_by(
+        '-created_time')
     return base_random_weight(ad_info_list, 'weight')
 
 
diff --git a/fir_ser/api/utils/utils.py b/fir_ser/api/utils/utils.py
index 3d27f12..ec74e67 100644
--- a/fir_ser/api/utils/utils.py
+++ b/fir_ser/api/utils/utils.py
@@ -77,8 +77,9 @@ def change_storage_and_change_head_img(user_obj, new_storage_obj, clean_old_data
 
 
 def change_storage_and_change_advert_img(user_obj, new_storage_obj, clean_old_data=True):
-    for user_advert_obj in UserAdDisplayInfo.objects.filter(user_id=user_obj):
-        migrating_storage_file_data(user_obj, user_advert_obj.ad_pic, new_storage_obj, clean_old_data)
+    for user_advert_obj in UserAdDisplayInfo.objects.filter(user_id=user_obj, ad_pic__regex='.+'):
+        if len(user_advert_obj.ad_pic) > 5:
+            migrating_storage_file_data(user_obj, user_advert_obj.ad_pic, new_storage_obj, clean_old_data)
 
 
 def download_files_form_oss(storage_obj, org_file, force=False):
diff --git a/fir_ser/api/views/storage.py b/fir_ser/api/views/storage.py
index 1097970..1714067 100644
--- a/fir_ser/api/views/storage.py
+++ b/fir_ser/api/views/storage.py
@@ -422,10 +422,12 @@ class StorageConfigView(APIView):
             'user_max_storage_capacity': get_user_storage_capacity(request.user),
             'user_used_storage_capacity': get_user_storage_used(request.user),
             'user_history_limit': UserInfo.objects.filter(pk=request.user.pk).first().history_release_limit,
+            'storage_status': MigrateStorageState(request.user.uid).get_state()
         }
         return Response(res.dict)
 
     def put(self, request):
+        res = BaseResponse()
         history_release_limit = request.data.get('user_history_limit')
         if history_release_limit:
             try:
@@ -440,4 +442,11 @@ class StorageConfigView(APIView):
             for app_obj in app_obj_lists:
                 clean_history_apps(app_obj, request.user, abs(history_release_limit))
 
-        return self.get(request)
+        return Response(res.dict)
+
+    def post(self, request):
+        res = BaseResponse()
+        migrate_obj = MigrateStorageState(request.user.uid)
+        if migrate_obj.get_state() == request.data.get('storage_status'):
+            migrate_obj.del_state()
+        return Response(res.dict)
diff --git a/fir_ser/common/cache/state.py b/fir_ser/common/cache/state.py
index ef4c4e9..978ced5 100644
--- a/fir_ser/common/cache/state.py
+++ b/fir_ser/common/cache/state.py
@@ -22,6 +22,9 @@ class CacheBaseState(object):
     def get_state(self):
         return cache.get(self.key)
 
+    def del_state(self):
+        return cache.delete(self.key)
+
     def __enter__(self):
         if cache.get(self.key):
             return False