From 2a2a8a778b67539de2d86dfad4d6496335006167 Mon Sep 17 00:00:00 2001
From: qiaozhanwei <qiaozhanwei@analysys.com.cn>
Date: Wed, 24 Apr 2019 11:23:03 +0800
Subject: [PATCH] api token update

---
 .../api/service/AccessTokenService.java         | 17 +++++++++--------
 .../dao/mapper/AccessTokenMapper.java           |  6 ++++--
 .../dao/mapper/AccessTokenMapperProvider.java   |  6 ++++++
 3 files changed, 19 insertions(+), 10 deletions(-)

diff --git a/escheduler-api/src/main/java/cn/escheduler/api/service/AccessTokenService.java b/escheduler-api/src/main/java/cn/escheduler/api/service/AccessTokenService.java
index 4adde8ad60..df7990149a 100644
--- a/escheduler-api/src/main/java/cn/escheduler/api/service/AccessTokenService.java
+++ b/escheduler-api/src/main/java/cn/escheduler/api/service/AccessTokenService.java
@@ -58,15 +58,16 @@ public class AccessTokenService extends BaseService {
     public Map<String, Object> queryAccessTokenList(User loginUser, String searchVal, Integer pageNo, Integer pageSize) {
         Map<String, Object> result = new HashMap<>(5);
 
-        if (check(result, !isAdmin(loginUser), Status.USER_NO_OPERATION_PERM, Constants.STATUS)) {
-            return result;
-        }
-
-        Integer count = accessTokenMapper.countAccessTokenPaging(searchVal);
-
         PageInfo<AccessToken> pageInfo = new PageInfo<>(pageNo, pageSize);
-
-        List<AccessToken> accessTokenList = accessTokenMapper.queryAccessTokenPaging(searchVal, pageInfo.getStart(), pageSize);
+        Integer count;
+        List<AccessToken> accessTokenList;
+        if (loginUser.getUserType() == UserType.ADMIN_USER){
+             count = accessTokenMapper.countAccessTokenPaging(0,searchVal);
+            accessTokenList = accessTokenMapper.queryAccessTokenPaging(0,searchVal, pageInfo.getStart(), pageSize);
+        }else {
+            count = accessTokenMapper.countAccessTokenPaging(loginUser.getId(),searchVal);
+            accessTokenList = accessTokenMapper.queryAccessTokenPaging(loginUser.getId(),searchVal, pageInfo.getStart(), pageSize);
+        }
 
         pageInfo.setTotalCount(count);
         pageInfo.setLists(accessTokenList);
diff --git a/escheduler-dao/src/main/java/cn/escheduler/dao/mapper/AccessTokenMapper.java b/escheduler-dao/src/main/java/cn/escheduler/dao/mapper/AccessTokenMapper.java
index aa196561b5..a9a17b8b7d 100644
--- a/escheduler-dao/src/main/java/cn/escheduler/dao/mapper/AccessTokenMapper.java
+++ b/escheduler-dao/src/main/java/cn/escheduler/dao/mapper/AccessTokenMapper.java
@@ -74,7 +74,8 @@ public interface AccessTokenMapper {
             @Result(property = "updateTime", column = "update_time", javaType = Timestamp.class, jdbcType = JdbcType.DATE)
     })
     @SelectProvider(type = AccessTokenMapperProvider.class, method = "queryAccessTokenPaging")
-    List<AccessToken> queryAccessTokenPaging(@Param("searchVal") String searchVal,
+    List<AccessToken> queryAccessTokenPaging(@Param("userId") Integer userId,
+                                             @Param("searchVal") String searchVal,
                                              @Param("offset") Integer offset,
                                              @Param("pageSize") Integer pageSize);
 
@@ -84,5 +85,6 @@ public interface AccessTokenMapper {
      * @return
      */
     @SelectProvider(type = AccessTokenMapperProvider.class, method = "countAccessTokenPaging")
-    Integer countAccessTokenPaging(@Param("searchVal") String searchVal);
+    Integer countAccessTokenPaging(@Param("userId") Integer userId
+                            ,@Param("searchVal") String searchVal);
 }
diff --git a/escheduler-dao/src/main/java/cn/escheduler/dao/mapper/AccessTokenMapperProvider.java b/escheduler-dao/src/main/java/cn/escheduler/dao/mapper/AccessTokenMapperProvider.java
index 2e4f92f79b..7aa8e61b70 100644
--- a/escheduler-dao/src/main/java/cn/escheduler/dao/mapper/AccessTokenMapperProvider.java
+++ b/escheduler-dao/src/main/java/cn/escheduler/dao/mapper/AccessTokenMapperProvider.java
@@ -97,6 +97,9 @@ public class AccessTokenMapperProvider {
             FROM(TABLE_NAME + " t,t_escheduler_user u");
             Object searchVal = parameter.get("searchVal");
             WHERE("u.id = t.user_id");
+            if(parameter.get("userId") != null && (int)parameter.get("userId") != 0){
+                WHERE(" u.id = #{userId}");
+            }
             if(searchVal != null && StringUtils.isNotEmpty(searchVal.toString())){
                 WHERE(" u.user_name like concat('%', #{searchVal}, '%')");
             }
@@ -115,6 +118,9 @@ public class AccessTokenMapperProvider {
                 FROM(TABLE_NAME + " t,t_escheduler_user u");
                 Object searchVal = parameter.get("searchVal");
                 WHERE("u.id = t.user_id");
+                if(parameter.get("userId") != null && (int)parameter.get("userId") != 0){
+                    WHERE(" u.id = #{userId}");
+                }
                 if(searchVal != null && StringUtils.isNotEmpty(searchVal.toString())){
                     WHERE(" u.user_name like concat('%', #{searchVal}, '%') ");
                 }