Added DepartmentManagement

Pinnacle/src/main/java/com/cfive/pinnacle/controller/DepartmentController.java

@@ -1,14 +1,14 @@
 package com.cfive.pinnacle.controller;
 
+import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.cfive.pinnacle.entity.Department;
 import com.cfive.pinnacle.entity.common.ResponseCode;
 import com.cfive.pinnacle.entity.common.ResponseResult;
 import com.cfive.pinnacle.service.IDepartmentService;
+import io.swagger.v3.oas.annotations.tags.Tag;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.CrossOrigin;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.security.access.prepost.PreAuthorize;
+import org.springframework.web.bind.annotation.*;
 
 import java.util.List;
 
@@ -22,18 +22,58 @@ import java.util.List;
  */
 @RestController
 @RequestMapping("/department")
-@CrossOrigin
+@Tag(name = "部门", description = "部门相关接口")
 public class DepartmentController {
+    private IDepartmentService departmentService;
+
     @Autowired
-    IDepartmentService departmentService;
+    public void setDepartmentService(IDepartmentService departmentService) {
+        this.departmentService = departmentService;
+    }
 
     //获取所有部门及其各部门所属成员
-    @GetMapping
-    public ResponseResult getDepartAndUser(){
+    @GetMapping("/user")
+    @Deprecated
+    public ResponseResult getDepartAndUser() {
         List<Department> getDepartAndUser = departmentService.getDepartAndUser();
         Integer code = getDepartAndUser != null ? ResponseCode.DATABASE_SELECT_OK : ResponseCode.DATABASE_SELECT_ERROR;
         String msg = getDepartAndUser != null ? "" : "数据查询失败，请尝试！";
         return ResponseResult.build(code, msg, getDepartAndUser);
     }
 
+    @GetMapping
+    @PreAuthorize("hasAuthority('department:admin:get')")
+    public ResponseResult<IPage<Department>> getAllDepartment(Long currentPage, Long pageSize, Integer searchType, String searchInput) {
+        return ResponseResult.databaseSelectSuccess(departmentService.getAllDepartment(currentPage, pageSize, searchType, searchInput));
+    }
+
+    @PostMapping
+    @PreAuthorize("hasAuthority('department:admin:add')")
+    public ResponseResult<?> addDepartment(@RequestBody Department department) {
+        if (departmentService.save(department)) {
+            return ResponseResult.databaseSaveSuccess(null);
+        } else {
+            return ResponseResult.build(ResponseCode.DATABASE_SAVE_ERROR, "error", null);
+        }
+    }
+
+    @PutMapping
+    @PreAuthorize("hasAuthority('department:admin:modify')")
+    public ResponseResult<?> modifyDepartment(@RequestBody Department department) {
+        if (departmentService.updateById(department)) {
+            return ResponseResult.databaseUpdateSuccess(null);
+        } else {
+            return ResponseResult.build(ResponseCode.DATABASE_SAVE_ERROR, "error", null);
+        }
+    }
+
+    @DeleteMapping("/{id}")
+    @PreAuthorize("hasAuthority('department:admin:delete')")
+    public ResponseResult<?> deleteDepartment(@PathVariable Long id) {
+        if (departmentService.removeById(id)) {
+            return ResponseResult.databaseDeleteSuccess();
+        } else {
+            return ResponseResult.build(ResponseCode.DATABASE_DELETE_ERROR, "error", null);
+        }
+    }
 }