From c72b8f2aba59413806668103e97d8c98018ae16a Mon Sep 17 00:00:00 2001
From: cjw <miko20001@163.com>
Date: Wed, 26 Jun 2024 14:36:42 +0800
Subject: [PATCH] =?UTF-8?q?=E9=97=A8=E6=88=B7=E8=B5=84=E6=BA=90=E5=88=86?=
 =?UTF-8?q?=E9=A1=B5=E6=9F=A5=E8=AF=A2=E4=BC=98=E5=8C=96=EF=BC=9B=E8=B5=84?=
 =?UTF-8?q?=E6=BA=90=E5=AE=B9=E9=87=8F=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../system/domain/vo/SysOssVolumeVo.java      |  2 +
 .../mapper/SysCatalogResourceMapper.java      |  5 ++-
 .../impl/SysCatalogResourceServiceImpl.java   |  4 ++
 .../service/impl/SysOssServiceImpl.java       | 12 +++++-
 .../system/SysCatalogResourceMapper.xml       | 39 ++++++++++---------
 5 files changed, 40 insertions(+), 22 deletions(-)

diff --git a/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/domain/vo/SysOssVolumeVo.java b/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/domain/vo/SysOssVolumeVo.java
index 3f5872b..07a0259 100644
--- a/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/domain/vo/SysOssVolumeVo.java
+++ b/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/domain/vo/SysOssVolumeVo.java
@@ -43,4 +43,6 @@ public class SysOssVolumeVo implements Serializable {
      */
     private Long useVolume;
     private String useVolumename;
+
+    private String volumePercent;
 }
diff --git a/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/mapper/SysCatalogResourceMapper.java b/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/mapper/SysCatalogResourceMapper.java
index 8d8d3ad..8cb89ca 100644
--- a/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/mapper/SysCatalogResourceMapper.java
+++ b/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/mapper/SysCatalogResourceMapper.java
@@ -17,5 +17,8 @@ import org.dromara.system.domain.vo.SysCatalogResourceVo;
 @Mapper
 public interface SysCatalogResourceMapper extends BaseMapperPlus<SysCatalogResource, SysCatalogResourceVo> {
     Page<SysCatalogResourceVo> selectPageList(@Param("page") Page<SysCatalogResource> page, @Param("catalogId") Long catalogId);
-    Page<SysCatalogResourceVo> selectProtalPageList(@Param("page") Page<SysCatalogResource> page,@Param("tenantId") String tenantId);
+
+    Page<SysCatalogResourceVo> selectProtalPageList(@Param("page") Page<SysCatalogResource> page, @Param("tenantId") String tenantId);
+
+    long selectChildFileNumById(@Param("tenantId") String tenantId, @Param("catalogId") Long catalogId);
 }
diff --git a/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/service/impl/SysCatalogResourceServiceImpl.java b/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/service/impl/SysCatalogResourceServiceImpl.java
index aca6351..34f5e6f 100644
--- a/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/service/impl/SysCatalogResourceServiceImpl.java
+++ b/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/service/impl/SysCatalogResourceServiceImpl.java
@@ -98,6 +98,10 @@ public class SysCatalogResourceServiceImpl implements ISysCatalogResourceService
         //临时存放
         String tenantId = bo.getAncestors();
         IPage<SysCatalogResourceVo> result = baseMapper.selectProtalPageList(pageQuery.build(), tenantId);
+        for (SysCatalogResourceVo record : result.getRecords()) {
+            long count = baseMapper.selectChildFileNumById(tenantId, record.getCatalogId());
+            record.setResourceNum(record.getResourceNum() + count);
+        }
         return TableDataInfo.build(result);
     }
 
diff --git a/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/service/impl/SysOssServiceImpl.java b/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/service/impl/SysOssServiceImpl.java
index da6cc35..fd9903f 100644
--- a/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/service/impl/SysOssServiceImpl.java
+++ b/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/service/impl/SysOssServiceImpl.java
@@ -43,6 +43,8 @@ import org.springframework.stereotype.Service;
 import org.springframework.web.multipart.MultipartFile;
 
 import java.io.*;
+import java.math.BigDecimal;
+import java.math.RoundingMode;
 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.List;
@@ -377,10 +379,16 @@ public class SysOssServiceImpl implements ISysOssService, OssService {
     @Override
     public SysOssVolumeVo getVolumeVoByType(Integer type) {
         SysOssVolumeVo ossVolumeVo = volumeMapper.selectVoOne(new LambdaQueryWrapper<SysOssVolume>().eq(SysOssVolume::getType, type));
-        String totalName = DataSizeUtil.format(ossVolumeVo.getTotalVolume());
-        String useName = DataSizeUtil.format(ossVolumeVo.getUseVolume());
+        Long totalVolume = ossVolumeVo.getTotalVolume();
+        Long useVolume = ossVolumeVo.getUseVolume();
+        String totalName = DataSizeUtil.format(totalVolume);
+        String useName = DataSizeUtil.format(useVolume);
         ossVolumeVo.setTotalVolumeName(totalName);
         ossVolumeVo.setUseVolumename(useName);
+        int i = BigDecimal.valueOf(useVolume)
+            .divide(BigDecimal.valueOf(totalVolume), 2, RoundingMode.HALF_UP)
+            .multiply(BigDecimal.valueOf(100)).intValue();
+        ossVolumeVo.setVolumePercent(i + "%");
         return ossVolumeVo;
     }
 
diff --git a/ruoyi-modules/ruoyi-system/src/main/resources/mapper/system/SysCatalogResourceMapper.xml b/ruoyi-modules/ruoyi-system/src/main/resources/mapper/system/SysCatalogResourceMapper.xml
index beea35b..d026868 100644
--- a/ruoyi-modules/ruoyi-system/src/main/resources/mapper/system/SysCatalogResourceMapper.xml
+++ b/ruoyi-modules/ruoyi-system/src/main/resources/mapper/system/SysCatalogResourceMapper.xml
@@ -15,24 +15,25 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
     </select>
 
     <select id="selectProtalPageList" resultType="org.dromara.system.domain.vo.SysCatalogResourceVo">
-        with recursive cte as (select *
-                               from sys_catalog_resource
-                               where type = 2
-                                 and del_flag = 0
-                                 and tenant_id = #{tenantId}
-                               union all
-                               select d.*
-                               from sys_catalog_resource d
-                                        inner join cte on d.parent_id = cte.catalog_id
-                               where d.del_flag = 0
-                                 and d.tenant_id = #{tenantId})
-        select t.catalog_id              as catalogId,
-               any_value(t.catalog_name) as catalogName,
-               any_value(t.create_time)  as createTime,
-               any_value(t.cover)        as coverUrl,
-               count(r.id)               as resourceNum
-        from cte t
-                 left join sys_oss_resource r on r.catalog_id = t.catalog_id and r.status = 1
-        group by t.catalog_id
+        select cr.catalog_id   as catalogId,
+               cr.catalog_name as catalogName,
+               cr.create_time  as createTime,
+               cr.cover        as coverUrl,
+               count(r.id)     as resourceNum
+        from sys_catalog_resource cr
+                 left join sys_oss_resource r on r.catalog_id = cr.catalog_id and r.status = 1
+        where cr.type = 2
+          and cr.del_flag = 0
+          and cr.tenant_id = #{tenantId}
+        group by cr.catalog_id
+    </select>
+
+    <select id="selectChildFileNumById" resultType="long">
+        select count(*)
+        from sys_catalog_resource cr
+                 left join sys_oss_resource r on r.catalog_id = cr.catalog_id
+        where cr.tenant_id = #{tenantId}
+          and r.status = 1
+          and FIND_IN_SET(#{catalogId}, cr.ancestors)
     </select>
 </mapper>