From 9df23d32703d6e7e6bf4d64113487aed3db58328 Mon Sep 17 00:00:00 2001
From: zhangjinli <zhangjinli_808@163.com>
Date: Wed, 22 Feb 2023 16:39:29 +0800
Subject: [PATCH] =?UTF-8?q?fixed=20=E9=A6=96=E9=A1=B5=E4=BB=BB=E5=8A=A1?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=AF=BC=E5=87=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../manage/StatisticsController.java          | 105 +++++++-----------
 .../zh/project0512/service/IAdminService.java |   5 +-
 .../serviceImpl/AdminServiceImpl.java         |  83 +++++++++++++-
 3 files changed, 126 insertions(+), 67 deletions(-)

diff --git a/src/main/java/com/zh/project0512/controller/manage/StatisticsController.java b/src/main/java/com/zh/project0512/controller/manage/StatisticsController.java
index 9b4dce6..8baff51 100644
--- a/src/main/java/com/zh/project0512/controller/manage/StatisticsController.java
+++ b/src/main/java/com/zh/project0512/controller/manage/StatisticsController.java
@@ -8,16 +8,20 @@ import com.fasterxml.jackson.annotation.JsonFormat;
 import com.zh.project0512.annotation.AdminTokenValid;
 import com.zh.project0512.mapper.AdminMapper;
 import com.zh.project0512.mapper.UserEffectRecordsMapper;
+import com.zh.project0512.model.Task;
 import com.zh.project0512.model.UserEffectRecords;
 import com.zh.project0512.model.Video;
 import com.zh.project0512.model.dto.RankListDTO;
+import com.zh.project0512.service.IAdminService;
 import com.zh.project0512.service.IRankService;
+import com.zh.project0512.utils.ExcelUtil;
 import com.zh.project0512.utils.MybatisPlusUtil;
 import com.zh.project0512.utils.result.Result;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.media.Schema;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import lombok.Data;
+import org.apache.poi.ss.usermodel.Workbook;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.PostMapping;
@@ -26,6 +30,11 @@ import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
 import javax.annotation.Resource;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.io.OutputStream;
+import java.io.UnsupportedEncodingException;
+import java.net.URLEncoder;
 import java.time.DayOfWeek;
 import java.time.LocalDate;
 import java.time.LocalDateTime;
@@ -35,6 +44,7 @@ import java.time.temporal.ChronoUnit;
 import java.time.temporal.TemporalAdjusters;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Map;
 import java.util.stream.Stream;
 
 /**
@@ -44,6 +54,8 @@ import java.util.stream.Stream;
 @RestController
 @RequestMapping("/manage/statistics")
 public class StatisticsController {
+    @Autowired
+    private IAdminService adminService;
     @Autowired
     private AdminMapper adminMapper;
     @Autowired
@@ -57,7 +69,7 @@ public class StatisticsController {
     }
 
     @Data
-    static class SSParam {
+    public static class SSParam {
         @Schema(title = "任务Id")
         private int taskId;
         @Schema(title = "员工Id")
@@ -76,70 +88,7 @@ public class StatisticsController {
     @PostMapping("/data2")
     @AdminTokenValid
     public Result data2(@Validated @RequestBody SSParam param) {
-        int condition = param.getDateCondition();
-        QueryWrapper<Video> qw = new QueryWrapper<>();
-        DateTimeFormatter df = DateTimeFormatter.ofPattern("yyyy-MM-dd");
-        DateTimeFormatter df2 = DateTimeFormatter.ofPattern("yyyy-MM");
-        LocalDate now = LocalDateTime.now().toLocalDate();
-        LocalDate before = now.with(TemporalAdjusters.firstDayOfMonth());
-        switch (condition) {
-            case 1:
-                before = now;
-                break;
-            case 2:
-                before = now.with(DayOfWeek.MONDAY);
-                break;
-            case 3:
-                before = now.with(TemporalAdjusters.firstDayOfMonth());
-                break;
-            case 4:
-                before = now.with(TemporalAdjusters.firstDayOfYear());
-                break;
-            default:
-                before = param.getBeginTime() != null ? param.getBeginTime().toLocalDate() : before;
-                now = param.getEndTime() != null ? param.getEndTime().toLocalDate() : now;
-        }
-        if(param.getTaskId()>0){
-            qw.eq("taskId", param.getTaskId());
-        }
-        if(param.getUserId()>0){
-            qw.eq("userId", param.getUserId());
-        }
-        ArrayList<JSONObject> res = new ArrayList<>();
-        if (4 == condition) {
-            qw.between("date_format(creatAt, '%Y-%m')", before.format(df2), now.plusMonths(1).format(df2));
-            List<JSONObject> videoDataBydDay = adminMapper.statistics3(qw);
-            for (int i = 0; i <= ChronoUnit.MONTHS.between(before, now); i++) {
-                LocalDate today = before.plusMonths(i);
-                JSONObject obj = null;
-//            Stream<JSONObject> dat1 = videoDataBydDay.stream().filter(x -> LocalDate.parse((CharSequence) x.get("dat"), df) == today);
-                for (JSONObject o : videoDataBydDay) {
-                    if (o.get("dat").equals(today.format(df2))) {
-                        obj = o;
-                        break;
-                    }
-                }
-                res.add(obj == null ? new JSONObject().fluentPut("dat", today.format(df2)).fluentPut("playNumSum", 0).fluentPut("collectionNumSum", 0).fluentPut("reSendNumSum", 0)
-                    .fluentPut("recommendNumSum", 0).fluentPut("commentNumSum", 0).fluentPut("commendNumSum", 0) : obj);
-            }
-        } else {
-            qw.between("date_format(creatAt, '%Y-%m-%d')", before, now.plusDays(1));
-            List<JSONObject> videoDataBydDay = adminMapper.statistics2(qw);
-            for (int i = 0; i <= ChronoUnit.DAYS.between(before, now); i++) {
-                LocalDate today = before.plusDays(i);
-                JSONObject obj = null;
-//            Stream<JSONObject> dat1 = videoDataBydDay.stream().filter(x -> LocalDate.parse((CharSequence) x.get("dat"), df) == today);
-                for (JSONObject o : videoDataBydDay) {
-                    if (LocalDate.parse((CharSequence) o.get("dat"), df).equals(today)) {
-                        obj = o;
-                        break;
-                    }
-                }
-                res.add(obj == null ? new JSONObject().fluentPut("dat", today).fluentPut("playNumSum", 0).fluentPut("collectionNumSum", 0).fluentPut("reSendNumSum", 0)
-                    .fluentPut("recommendNumSum", 0).fluentPut("commentNumSum", 0).fluentPut("commendNumSum", 0) : obj);
-            }
-        }
-        return Result.success(res, "请求成功");
+        return Result.success(adminService.videoData(param), "请求成功");
     }
 
     @Data
@@ -183,4 +132,30 @@ public class StatisticsController {
         IPage page = (param.getType() != null && param.getType() == 1) ? userEffectRecordsMapper.rankByDepartment(pg, qw) : userEffectRecordsMapper.rankByUser(pg, qw);
         return Result.success(page, "请求成功");
     }
+
+    @Operation(summary = "导出excel", description = "视频数据导出")
+    @PostMapping("/excel2")
+    @AdminTokenValid
+    public void excel(HttpServletResponse response, @Validated @RequestBody SSParam param) throws IOException {
+        //这是表头及格式
+        String[][] array = {
+            {"时间", "播放", "点赞", "收藏", "评论", "转发", "推荐"},
+            {"dat", "playNumSum", "commendNumSum", "collectionNumSum", "commentNumSum", "reSendNumSum", "recommendNumSum"},
+            {"String", "int", "int", "int", "int", "int", "int"}
+        };
+        List<Map> l = adminService.videoData(param);
+
+        Workbook wb = ExcelUtil.writeToExcelByList(array, l);
+        OutputStream output = response.getOutputStream();
+        String fileName = "视频数据.xlsx";
+        try {
+            fileName = URLEncoder.encode(fileName, "UTF-8");
+        } catch (UnsupportedEncodingException e) {
+            e.printStackTrace();
+        }
+        response.setHeader("Content-disposition", "attachment;filename=" + fileName + ";" + "filename*=utf-8''" + fileName);
+        wb.write(output);
+        output.close();
+    }
+
 }
diff --git a/src/main/java/com/zh/project0512/service/IAdminService.java b/src/main/java/com/zh/project0512/service/IAdminService.java
index e14ef73..e67d1cc 100644
--- a/src/main/java/com/zh/project0512/service/IAdminService.java
+++ b/src/main/java/com/zh/project0512/service/IAdminService.java
@@ -1,8 +1,11 @@
 package com.zh.project0512.service;
 
+import com.zh.project0512.controller.manage.StatisticsController;
 import com.zh.project0512.model.Admin;
 import com.baomidou.mybatisplus.extension.service.IService;
 
+import java.util.ArrayList;
+
 /**
  * <p>
  *  服务类
@@ -12,5 +15,5 @@ import com.baomidou.mybatisplus.extension.service.IService;
  * @since 2022-06-10
  */
 public interface IAdminService extends IService<Admin> {
-
+    ArrayList videoData(StatisticsController.SSParam params);
 }
diff --git a/src/main/java/com/zh/project0512/serviceImpl/AdminServiceImpl.java b/src/main/java/com/zh/project0512/serviceImpl/AdminServiceImpl.java
index 5268c7a..2e0fbb6 100644
--- a/src/main/java/com/zh/project0512/serviceImpl/AdminServiceImpl.java
+++ b/src/main/java/com/zh/project0512/serviceImpl/AdminServiceImpl.java
@@ -1,11 +1,25 @@
 package com.zh.project0512.serviceImpl;
 
+import com.alibaba.fastjson.JSONObject;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.zh.project0512.controller.manage.StatisticsController;
 import com.zh.project0512.model.Admin;
 import com.zh.project0512.mapper.AdminMapper;
+import com.zh.project0512.model.Video;
 import com.zh.project0512.service.IAdminService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
+import java.time.DayOfWeek;
+import java.time.LocalDate;
+import java.time.LocalDateTime;
+import java.time.format.DateTimeFormatter;
+import java.time.temporal.ChronoUnit;
+import java.time.temporal.TemporalAdjusters;
+import java.util.ArrayList;
+import java.util.List;
+
 /**
  * <p>
  *  服务实现类
@@ -16,5 +30,72 @@ import org.springframework.stereotype.Service;
  */
 @Service
 public class AdminServiceImpl extends ServiceImpl<AdminMapper, Admin> implements IAdminService {
-
+    @Autowired
+    private AdminMapper adminMapper;
+    public ArrayList videoData(StatisticsController.SSParam params){
+        int condition = params.getDateCondition();
+        QueryWrapper<Video> qw = new QueryWrapper<>();
+        DateTimeFormatter df = DateTimeFormatter.ofPattern("yyyy-MM-dd");
+        DateTimeFormatter df2 = DateTimeFormatter.ofPattern("yyyy-MM");
+        LocalDate now = LocalDateTime.now().toLocalDate();
+        LocalDate before = now.with(TemporalAdjusters.firstDayOfMonth());
+        switch (condition) {
+            case 1:
+                before = now;
+                break;
+            case 2:
+                before = now.with(DayOfWeek.MONDAY);
+                break;
+            case 3:
+                before = now.with(TemporalAdjusters.firstDayOfMonth());
+                break;
+            case 4:
+                before = now.with(TemporalAdjusters.firstDayOfYear());
+                break;
+            default:
+                before = params.getBeginTime() != null ? params.getBeginTime().toLocalDate() : before;
+                now = params.getEndTime() != null ? params.getEndTime().toLocalDate() : now;
+        }
+        if(params.getTaskId()>0){
+            qw.eq("taskId", params.getTaskId());
+        }
+        if(params.getUserId()>0){
+            qw.eq("userId", params.getUserId());
+        }
+        ArrayList<JSONObject> res = new ArrayList<>();
+        if (4 == condition) {
+            qw.between("date_format(creatAt, '%Y-%m')", before.format(df2), now.plusMonths(1).format(df2));
+            List<JSONObject> videoDataBydDay = adminMapper.statistics3(qw);
+            for (int i = 0; i <= ChronoUnit.MONTHS.between(before, now); i++) {
+                LocalDate today = before.plusMonths(i);
+                JSONObject obj = null;
+//            Stream<JSONObject> dat1 = videoDataBydDay.stream().filter(x -> LocalDate.parse((CharSequence) x.get("dat"), df) == today);
+                for (JSONObject o : videoDataBydDay) {
+                    if (o.get("dat").equals(today.format(df2))) {
+                        obj = o;
+                        break;
+                    }
+                }
+                res.add(obj == null ? new JSONObject().fluentPut("dat", today.format(df2)).fluentPut("playNumSum", 0).fluentPut("collectionNumSum", 0).fluentPut("reSendNumSum", 0)
+                    .fluentPut("recommendNumSum", 0).fluentPut("commentNumSum", 0).fluentPut("commendNumSum", 0) : obj);
+            }
+        } else {
+            qw.between("date_format(creatAt, '%Y-%m-%d')", before, now.plusDays(1));
+            List<JSONObject> videoDataBydDay = adminMapper.statistics2(qw);
+            for (int i = 0; i <= ChronoUnit.DAYS.between(before, now); i++) {
+                LocalDate today = before.plusDays(i);
+                JSONObject obj = null;
+//            Stream<JSONObject> dat1 = videoDataBydDay.stream().filter(x -> LocalDate.parse((CharSequence) x.get("dat"), df) == today);
+                for (JSONObject o : videoDataBydDay) {
+                    if (LocalDate.parse((CharSequence) o.get("dat"), df).equals(today)) {
+                        obj = o;
+                        break;
+                    }
+                }
+                res.add(obj == null ? new JSONObject().fluentPut("dat", today).fluentPut("playNumSum", 0).fluentPut("collectionNumSum", 0).fluentPut("reSendNumSum", 0)
+                    .fluentPut("recommendNumSum", 0).fluentPut("commentNumSum", 0).fluentPut("commendNumSum", 0) : obj);
+            }
+        }
+        return res;
+    }
 }