From 7feb722a12e26f5572c7df9ccc1ce8c1e26b9f94 Mon Sep 17 00:00:00 2001
From: Pu Zhibing <393733352@qq.com>
Date: 星期二, 22 四月 2025 23:30:07 +0800
Subject: [PATCH] 新增加功能
---
ManagementOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/ReportController.java | 30 +++++++++++++++++++++++++-----
1 files changed, 25 insertions(+), 5 deletions(-)
diff --git a/ManagementOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/ReportController.java b/ManagementOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/ReportController.java
index 52fbfa1..451c965 100644
--- a/ManagementOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/ReportController.java
+++ b/ManagementOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/ReportController.java
@@ -2,12 +2,12 @@
import com.baomidou.mybatisplus.mapper.EntityWrapper;
+import com.baomidou.mybatisplus.mapper.Wrapper;
+import com.stylefeng.guns.core.shiro.ShiroKit;
import com.stylefeng.guns.core.util.ToolUtil;
import com.stylefeng.guns.modular.system.model.TCompany;
-import com.stylefeng.guns.modular.system.service.ITCompanyService;
-import com.stylefeng.guns.modular.system.service.IUserActivityDiscount1Service;
-import com.stylefeng.guns.modular.system.service.IUserCouponRecordService;
-import com.stylefeng.guns.modular.system.service.IUserRedPacketRecordService;
+import com.stylefeng.guns.modular.system.model.User;
+import com.stylefeng.guns.modular.system.service.*;
import com.stylefeng.guns.modular.system.util.ResultUtil;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.springframework.beans.factory.annotation.Autowired;
@@ -42,6 +42,9 @@
@Autowired
private IUserActivityDiscount1Service userActivityDiscount1Service;
+
+ @Autowired
+ private IUserService userService;
@@ -52,7 +55,13 @@
*/
@RequestMapping("/showOperationalData")
public String showOperationalData(Model model){
- List<TCompany> tCompanies = companyService.selectList(new EntityWrapper<TCompany>().eq("state", 0).ne("flag", 3));
+ Wrapper<TCompany> wrapper = new EntityWrapper<TCompany>().eq("state", 0).ne("flag", 3);
+ Integer id = ShiroKit.getUser().getId();
+ User user = userService.selectById(id);
+ if(user.getRoleType() != 1){
+ wrapper.eq("id", user.getObjectId());
+ }
+ List<TCompany> tCompanies = companyService.selectList(wrapper);
model.addAttribute("company", tCompanies);
return PREFIX + "operationalData.html";
}
@@ -107,6 +116,11 @@
start = time.split(" - ")[0];
end = time.split(" - ")[1];
}
+ Integer id = ShiroKit.getUser().getId();
+ User user = userService.selectById(id);
+ if(user.getRoleType() != 1){
+ companyId = user.getObjectId();
+ }
return companyService.queryOperationalData(type, start, end, companyId, offset, limit);
}catch (Exception e){
e.printStackTrace();
@@ -131,6 +145,12 @@
start = time.split(" - ")[0];
end = time.split(" - ")[1];
}
+
+ Integer id = ShiroKit.getUser().getId();
+ User user = userService.selectById(id);
+ if(user.getRoleType() != 1){
+ companyId = user.getObjectId();
+ }
HSSFWorkbook hssfWorkbook = companyService.downloadOperationalData(type, start, end, companyId);
response.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode("日常运营数据汇总表.xls", "utf-8"));
response.setContentType("application/vnd.ms-excel");
--
Gitblit v1.7.1