Kaynağa Gözat

对四维看看提供接口-管理员放到token中

wuweihao 3 yıl önce
ebeveyn
işleme
7b238fffda

+ 36 - 0
720yun_fd_manage/gis_common/src/main/java/com/gis/common/service/IBaseService.java

@@ -0,0 +1,36 @@
+package com.gis.common.service;
+
+import com.gis.common.shiro.JwtUtil;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import javax.servlet.http.HttpServletRequest;
+
+/**
+ * Created by owen on 2021/12/29 0029 14:12
+ */
+@Component
+public class IBaseService {
+
+    @Autowired
+    HttpServletRequest request;
+
+
+
+    public String getToken(){
+        return request.getHeader("token");
+    }
+
+    /**登录用户id*/
+    public Long getUserId(){
+       return JwtUtil.getUserId(getToken());
+    }
+
+
+    public Integer getIsAdmin(){
+        return JwtUtil.getIsAdmin(getToken());
+    }
+
+
+
+}

+ 1 - 1
720yun_fd_manage/gis_service/src/main/java/com/gis/service/WorkService.java

@@ -17,7 +17,7 @@ import java.util.List;
 public interface WorkService extends IBaseService<WorkEntity, String> {
 
 
-    Result search(PageDto param);
+    Result search(PageDto param, boolean isAdmin);
 
     Result<WorkEntity> entitySave(WorkDto param);
 

+ 2 - 3
720yun_fd_manage/gis_service/src/main/java/com/gis/service/impl/WorkServiceImpl.java

@@ -78,11 +78,10 @@ public class WorkServiceImpl extends IBaseStrServiceImpl<WorkEntity, String> imp
 
 
     @Override
-    public Result search(PageDto param) {
+    public Result search(PageDto param, boolean isAdmin) {
         startPage(param);
         String userNameForToken = getUserNameForToken();
-
-        if (1 == getIsAdmin()){
+        if (isAdmin){
             userNameForToken = null;
         }
         List<WorkEntity> search = entityMapper.search(param, userNameForToken);

+ 10 - 1
720yun_fd_manage/gis_web/src/main/java/com/gis/web/controller/AgeController.java

@@ -1,5 +1,7 @@
 package com.gis.web.controller;
 
+import com.gis.common.exception.BaseRuntimeException;
+import com.gis.common.service.IBaseService;
 import com.gis.common.util.Result;
 import com.gis.domain.dto.PageDto;
 import com.gis.domain.entity.WorkEntity;
@@ -21,6 +23,9 @@ public class AgeController {
     @Autowired
     WorkService workService;
 
+
+    @Autowired
+    IBaseService iBaseService;
     /**
      * 获取所有场景
      * 使用超级管理员权限的
@@ -30,12 +35,16 @@ public class AgeController {
     @ApiOperation(value = "场景-列表", notes = "超级管理员权限token")
     @PostMapping("/list")
     public Result<WorkEntity> list(@RequestBody PageDto param) {
-        return workService.search(param);
+        Integer isAdmin = iBaseService.getIsAdmin();
+        BaseRuntimeException.isTrue(isAdmin != 1, 3100, "非法用户");
+        return workService.search(param, true);
     }
 
     @ApiOperation(value = "场景-删除")
     @PostMapping("/remove/{id}")
     public Result remove( @PathVariable String id) {
+        Integer isAdmin = iBaseService.getIsAdmin();
+        BaseRuntimeException.isTrue(isAdmin != 1, 3100, "非法用户");
         return workService.remove(id);
     }
 

+ 1 - 1
720yun_fd_manage/gis_web/src/main/java/com/gis/web/controller/WorkController.java

@@ -51,7 +51,7 @@ public class WorkController extends BaseController {
     @ApiOperation(value = "列表", position = 1)
     @PostMapping("list")
     public Result<WorkEntity> list(@RequestBody PageDto param) {
-        return workService.search(param);
+        return workService.search(param, false);
     }