From b140d7f4d60b52e6a5ac9d9aae64fc7edf90d531 Mon Sep 17 00:00:00 2001
From: wangliwen <wangliwen@hisense.com>
Date: Thu, 23 Jun 2022 10:29:07 +0800
Subject: [PATCH] =?UTF-8?q?=E9=85=8D=E7=BD=AE=E6=96=87=E4=BB=B6=E5=86=85?=
 =?UTF-8?q?=E4=B8=8D=E9=9C=80=E7=BB=9F=E8=AE=A1=E7=9A=84=E8=B5=84=E6=BA=90?=
 =?UTF-8?q?=20=E6=8E=92=E9=99=A4=E6=8E=A5=E5=8F=A3=E5=87=BA=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../common/controller/CensusController.java   | 35 ++++++++-----------
 1 file changed, 14 insertions(+), 21 deletions(-)

diff --git a/renren-admin/src/main/java/io/renren/common/controller/CensusController.java b/renren-admin/src/main/java/io/renren/common/controller/CensusController.java
index b9be86de..203125c1 100644
--- a/renren-admin/src/main/java/io/renren/common/controller/CensusController.java
+++ b/renren-admin/src/main/java/io/renren/common/controller/CensusController.java
@@ -59,35 +59,28 @@ public class CensusController {
     @ApiOperation("各类资源数目")
     public Result<List<Map<String, Object>>> resourceAmount() {
         List<Map<String, Object>> dbAmount = resourceService.getAmountGroupByType();
-        List<String> temp = dbAmount.stream().map(index -> index.get("type").toString()).collect(Collectors.toList());
+        dbAmount = dbAmount.stream().filter(index -> Arrays.asList(censusTypes).contains(index.get("type").toString())).collect(Collectors.toList());
+        List<Map<String, Object>> finalDbAmount = dbAmount;
+
+        List<String> temp = finalDbAmount.stream().map(index -> index.get("type").toString()).collect(Collectors.toList());
         Arrays.stream(censusTypes).filter(index -> !temp.contains(index)).forEach(index -> { // 数据库内不存在的资源类型
-            if (index.equals("数据资源")) {
-                Map<String, Object> nullMap = new HashMap<String, Object>() {
-                    {
-                        put("amount", 10413);
-                        put("type", "数据资源");
-                    }
-                };
-                dbAmount.add(nullMap);
-            } else {
-                Map<String, Object> nullMap = new HashMap<String, Object>() {
-                    {
-                        put("amount", 0);
-                        put("type", index);
-                    }
-                };
-                dbAmount.add(nullMap);
-            }
+            Map<String, Object> nullMap = new HashMap<String, Object>() {
+                {
+                    put("amount", 0);
+                    put("type", index);
+                }
+            };
+            finalDbAmount.add(nullMap);
         });
-        Long sum = dbAmount.stream().mapToLong(index -> Long.valueOf(index.get("amount").toString())).sum();
+        Long sum = finalDbAmount.stream().mapToLong(index -> Long.valueOf(index.get("amount").toString())).sum();
         Map<String, Object> sumMap = new HashMap<String, Object>() {
             {
                 put("amount", sum);
                 put("type", "资源汇聚总量");
             }
         };
-        dbAmount.add(sumMap);
-        return new Result<List<Map<String, Object>>>().ok(dbAmount);
+        finalDbAmount.add(sumMap);
+        return new Result<List<Map<String, Object>>>().ok(finalDbAmount);
     }
 
     @GetMapping(value = "/whole_amount")