From 30e858fa504b268b9b436afca0a1259cf6e8c488 Mon Sep 17 00:00:00 2001
From: MrShi <1878285526@qq.com>
Date: 星期二, 19 八月 2025 11:01:40 +0800
Subject: [PATCH] 优化

---
 server/src/main/java/com/doumee/api/business/CompanyController.java |   38 ++++++++++++++++++++++++++++++++++----
 1 files changed, 34 insertions(+), 4 deletions(-)

diff --git a/server/src/main/java/com/doumee/api/business/CompanyController.java b/server/src/main/java/com/doumee/api/business/CompanyController.java
index 06d07ad..f9e1292 100644
--- a/server/src/main/java/com/doumee/api/business/CompanyController.java
+++ b/server/src/main/java/com/doumee/api/business/CompanyController.java
@@ -1,6 +1,8 @@
 package com.doumee.api.business;
 
 import com.doumee.api.BaseController;
+import com.doumee.config.annotation.EncryptionReq;
+import com.doumee.config.annotation.EncryptionResp;
 import com.doumee.core.annotation.excel.ExcelExporter;
 import com.doumee.core.annotation.pr.PreventRepeat;
 import com.doumee.core.model.ApiResponse;
@@ -9,15 +11,12 @@
 import com.doumee.dao.business.model.Company;
 import com.doumee.service.business.CompanyService;
 import io.swagger.annotations.Api;
-import io.swagger.annotations.ApiImplicitParam;
-import io.swagger.annotations.ApiImplicitParams;
 import io.swagger.annotations.ApiOperation;
 import org.apache.shiro.authz.annotation.RequiresPermissions;    
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 import javax.servlet.http.HttpServletResponse;
 
-import java.util.ArrayList;
 import java.util.List;
 
 /**
@@ -35,6 +34,8 @@
     @PreventRepeat
     @ApiOperation("鏂板缓")
     @PostMapping("/create")
+    @EncryptionResp
+    @EncryptionReq
     @RequiresPermissions("business:company:create")
     public ApiResponse create(@RequestBody Company company) {
         return ApiResponse.success(companyService.create(company));
@@ -42,6 +43,8 @@
     @PreventRepeat
     @ApiOperation("鍚屾閮ㄩ棬淇℃伅")
     @PostMapping("/syncAll")
+    @EncryptionResp
+    @EncryptionReq
     @RequiresPermissions("business:company:create")
     public ApiResponse syncAll(@RequestBody Company company) {
         return ApiResponse.success(companyService.syncAll(company));
@@ -49,14 +52,34 @@
 
     @ApiOperation("缁勭粐鏍戞煡璇�")
     @PostMapping("/tree")
+    @EncryptionResp
+    @EncryptionReq
     @RequiresPermissions("business:company:query")
     public ApiResponse<List<Company>> tree (@RequestBody Company company){
         return ApiResponse.success(companyService.companyTree());
+    }
+    @ApiOperation("鏍规嵁鐖剁骇鏌ヨ闆嗗悎")
+    @PostMapping("/list")
+    @EncryptionResp
+    @EncryptionReq
+    @RequiresPermissions("business:company:query")
+    public ApiResponse<List<Company>> list (@RequestBody Company company){
+        return ApiResponse.success(companyService.findList(company));
+    }
+    @ApiOperation("缁勭粐鏍戞煡璇�")
+    @PostMapping("/treeWithMember")
+    @EncryptionResp
+    @EncryptionReq
+    @RequiresPermissions("business:company:query")
+    public ApiResponse<List<Company>> treeWithMember (@RequestBody Company company){
+        return ApiResponse.success(companyService.companyTreeWithMember ());
     }
 
 
     @ApiOperation("鏍规嵁ID鍒犻櫎")
     @GetMapping("/delete/{id}")
+    @EncryptionResp
+    @EncryptionReq
     @RequiresPermissions("business:company:delete")
     public ApiResponse deleteById(@PathVariable Integer id) {
         companyService.deleteById(id);
@@ -66,6 +89,8 @@
     @ApiOperation("鎵归噺鍒犻櫎")
     @GetMapping("/delete/batch")
     @RequiresPermissions("business:company:delete")
+    @EncryptionResp
+    @EncryptionReq
     public ApiResponse deleteByIdInBatch(@RequestParam String ids) {
         companyService.deleteByIdInBatch(this.getIdList(ids));
         return ApiResponse.success(null);
@@ -81,6 +106,8 @@
 
     @ApiOperation("鍒嗛〉鏌ヨ")
     @PostMapping("/page")
+    @EncryptionResp
+    @EncryptionReq
     @RequiresPermissions("business:company:query")
     public ApiResponse<PageData<Company>> findPage (@RequestBody PageWrap<Company> pageWrap) {
         return ApiResponse.success(companyService.findPage(pageWrap));
@@ -88,13 +115,16 @@
 
     @ApiOperation("瀵煎嚭Excel")
     @PostMapping("/exportExcel")
+    @EncryptionReq
     @RequiresPermissions("business:company:exportExcel")
     public void exportExcel (@RequestBody PageWrap<Company> pageWrap, HttpServletResponse response) {
-        ExcelExporter.build(Company.class).exportData(companyService.findPage(pageWrap).getRecords(), "缁勭粐淇℃伅琛�", response);
+        ExcelExporter.build(Company.class).export(companyService.findPage(pageWrap).getRecords(), "缁勭粐淇℃伅琛�", response);
     }
 
     @ApiOperation("鏍规嵁ID鏌ヨ")
     @GetMapping("/{id}")
+    @EncryptionResp
+    @EncryptionReq
     @RequiresPermissions("business:company:query")
     public ApiResponse findById(@PathVariable Integer id) {
         return ApiResponse.success(companyService.findById(id));

--
Gitblit v1.9.3