Get totalResults value by user count

core/org.wso2.carbon.user.core/src/main/java/org/wso2/carbon/user/core/common/AbstractUserStoreManager.java

@@ -9481,6 +9481,22 @@ protected List<User> doGetUserListOfRoleWithID(String roleName, String filter) t
                 "doGetUserListOfRoleWithID operation is not implemented in: " + this.getClass());
     }
 
+    /**
+     * Return the count of users belong to the given role for the given filter.
+     * @param roleName role name.
+     * @param filter filter.
+     * @return
+     * @throws UserStoreException
+     */
+    protected int doGetUserCountOfRole(String roleName, String filter) throws UserStoreException {
+
+        if (log.isDebugEnabled()) {
+            log.debug("doGetUserCountOfRole operation is not implemented in: " + this.getClass());
+        }
+        throw new NotImplementedException(
+                "doGetUserCountOfRole operation is not implemented in: " + this.getClass());
+    }
+
     /**
      * Return the list of users belong to the given role for the given filter.
      *
@@ -19161,4 +19177,37 @@ public UserUniqueIDDomainResolver getUserUniqueIDDomainResolver() {
 
         return userUniqueIDDomainResolver;
     }
+
+    public int getUserCountForRole(String  roleName) throws UserStoreException {
+
+        if (!isSecureCall.get()) {
+            Class argTypes[] = new Class[] { String.class, String.class };
+            Object object = callSecure("getUserCountByRole", new Object[] { roleName, QUERY_FILTER_STRING_ANY }, argTypes);
+            return (int) object;
+        }
+
+        return getUserCountByRole(roleName, QUERY_FILTER_STRING_ANY);
+    }
+
+    public int getUserCountByRole(String roleName, String filter) throws UserStoreException {
+
+        int count = 0;
+
+        if (!isSecureCall.get()) {
+            Class argTypes[] = new Class[] { String.class, String.class };
+            Object object = callSecure("getUserCountByRole", new Object[] { roleName, filter }, argTypes);
+            return (int) object;
+        }
+
+        // If role does not exit, just return
+        if (!isExistingRole(roleName)) {
+            return count;
+        }
+
+        if (readGroupsEnabled) {
+            count += doGetUserCountOfRole(roleName, filter);
+        }
+
+        return count;
+    }
 }

core/org.wso2.carbon.user.core/src/main/java/org/wso2/carbon/user/core/jdbc/JDBCRealmConstants.java

@@ -30,6 +30,7 @@ public final class JDBCRealmConstants {
     public static final String GET_ROLE_LIST_H2 = "GetRoleListSQLH2";
     public static final String GET_ROLE_LIST_WITH_ESCAPE = "GetRoleListSQLWithEscape";
     public static final String GET_ROLE_LIST_WITH_ESCAPE_H2 = "GetRoleListSQLWithEscapeH2";
+    public static final String GET_ROLE_ID_BY_NAME = "GetRoleIDByNameSQL";
     public static final String GET_SHARED_ROLE_LIST = "GetSharedRoleListSQL";
     public static final String GET_SHARED_ROLE_LIST_H2 = "GetSharedRoleListSQLH2";
     public static final String GET_USER_FILTER = "UserFilterSQL";
@@ -55,6 +56,7 @@ public final class JDBCRealmConstants {
     public static final String GET_USERS_IN_ROLE = "GetUserListOfRoleSQL";
     public static final String GET_USERS_IN_ROLE_FILTER = "GetUserListOfRoleFilterSQL";
     public static final String GET_USERS_IN_ROLE_FILTER_WITH_ID = "GetUserListOfRoleFilterWithIDSQL";
+    public static final String GET_USERS_COUNT_WITH_FILTER_ROLE = "GetUserCountWithFilterRoleSQL";
     public static final String GET_USERS_IN_SHARED_ROLE = "GetUserListOfSharedRoleSQL";
     public static final String GET_USERS_IN_SHARED_ROLE_FILTER = "GetUserListOfSharedRoleFilterSQL";
     public static final String GET_USERS_IN_SHARED_ROLE_FILTER_WITH_ID = "GetUserListOfSharedRoleFilterWithIDSQL";
@@ -267,6 +269,13 @@ public final class JDBCRealmConstants {
                     + ".UM_USER_NAME LIKE ? AND UM_ROLE.UM_ROLE_NAME=? AND UM_USER.UM_ID=UM_USER_ROLE.UM_USER_ID AND "
                     + "UM_ROLE.UM_ID=UM_USER_ROLE.UM_ROLE_ID AND UM_USER_ROLE.UM_TENANT_ID=? AND UM_ROLE"
                     + ".UM_TENANT_ID=? AND UM_USER.UM_TENANT_ID=?";
+
+    public static final String GET_USERS_COUNT_WITH_FILTER_ROLE_SQL = "SELECT count(UM_ID) FROM UM_USER_ROLE " +
+            "WHERE UM_ROLE_ID = ? AND UM_TENANT_ID = ?";
+
+    public static final String GET_ROLE_ID_BY_NAME_SQL =  "SELECT UM_ID FROM UM_ROLE " +
+            "WHERE UM_ROLE_NAME = ? AND UM_TENANT_ID = ? ";
+
     public static final String GET_USERS_IN_SHARED_ROLE_SQL =
             "SELECT UM_USER_NAME FROM UM_SHARED_USER_ROLE INNER JOIN UM_USER ON "
                     + "UM_SHARED_USER_ROLE.UM_USER_ID = UM_USER.UM_ID INNER JOIN UM_ROLE ON "

core/org.wso2.carbon.user.core/src/main/java/org/wso2/carbon/user/core/jdbc/UniqueIDJDBCUserStoreManager.java

@@ -351,6 +351,14 @@ public List<User> doGetUserListOfRoleWithID(String roleName, String filter) thro
         return getUserListOfJDBCRoleWithID(roleContext, filter);
     }
 
+    @Override
+    public int doGetUserCountOfRole(String roleName, String filter)
+                throws UserStoreException {
+
+            RoleContext roleContext = createRoleContext(roleName);
+            return getUserCountByRole(roleContext, filter);
+    }
+
     public List<User> getUserListOfJDBCRoleWithID(RoleContext ctx, String filter) throws UserStoreException {
 
         return getUserListOfJDBCRoleWithID(ctx, filter, QUERY_MAX_ITEM_LIMIT_ANY);
@@ -417,6 +425,94 @@ public List<User> getUserListOfJDBCRoleWithID(RoleContext ctx, String filter, in
         return users;
     }
 
+    /**
+     * Return the count of users belong to the given role for the given {@link RoleContext} and filter.
+     *
+     * @param ctx    {@link RoleContext} corresponding to the role.
+     * @param filter String filter for the users.
+     */
+    public int getUserCountByRole(RoleContext ctx, String filter) throws UserStoreException {
+
+        String roleName = ctx.getRoleName();
+
+        if (StringUtils.isNotEmpty(filter)) {
+            filter = filter.trim();
+            filter = filter.replace("*", "%");
+            filter = filter.replace("?", "_");
+        } else {
+            filter = "%";
+        }
+        return getUserCountByRoleFromDatabase(roleName, filter);
+    }
+
+    /**
+     * Return the count of users belong to the given role for the given {@link RoleContext} and filter.
+     *
+     * @param roleName Name of the role.
+     * @param filter   String filter for the users.
+     * @return The count of users matching the provided constraints.
+     * @throws UserStoreException
+     */
+    public int getUserCountByRoleFromDatabase(String roleName, String filter)
+                throws UserStoreException {
+
+        String roleId = getRoleIdByName(roleName, tenantId);
+        Connection dbConnection = null;
+        PreparedStatement prepStmt = null;
+        ResultSet rs = null;
+        int count = 0 ;
+        String sqlStmt = realmConfig.getUserStoreProperty(JDBCRealmConstants.GET_USERS_COUNT_WITH_FILTER_ROLE);
+        try {
+            dbConnection = getDBConnection();
+            prepStmt = dbConnection.prepareStatement(sqlStmt);
+            prepStmt.setString(1, roleId);
+            prepStmt.setInt(2, tenantId);
+            rs = prepStmt.executeQuery();
+            while (rs.next()) {
+                count = rs.getInt(1);
+            }
+            return count;
+        } catch (SQLException e) {
+            String errorMessage =
+                    "Error occurred while getting the count of users in the role : " + roleName;
+            if (log.isDebugEnabled()) {
+                log.debug(errorMessage, e);
+            }
+            throw new UserStoreException(errorMessage, e);
+        } finally {
+            DatabaseUtil.closeAllConnections(dbConnection, rs, prepStmt);
+        }
+    }
+
+    private String getRoleIdByName(String roleName, int tenantId) throws UserStoreException {
+
+        String roleID = null;
+        Connection dbConnection = null;
+        PreparedStatement prepStmt = null;
+        ResultSet rs = null;
+        String sqlStmt = realmConfig.getUserStoreProperty(JDBCRealmConstants.GET_ROLE_ID_BY_NAME);
+        try {
+            dbConnection = getDBConnection();
+            prepStmt = dbConnection.prepareStatement(sqlStmt);
+            prepStmt.setString(1, roleName);
+            prepStmt.setInt(2, tenantId);
+            rs = prepStmt.executeQuery();
+            while (rs.next()) {
+                roleID = rs.getString(1);
+            }
+            return roleID;
+        } catch (SQLException e) {
+            String errorMessage =
+                    "Error occurred while getting the role id for the role : " + roleName;
+            if (log.isDebugEnabled()) {
+                log.debug(errorMessage, e);
+            }
+            throw new UserStoreException(errorMessage, e);
+        } finally {
+            DatabaseUtil.closeAllConnections(dbConnection, rs, prepStmt);
+        }
+    }
+
     @Override
     public String[] getProfileNames(String userName) throws UserStoreException {
 

core/org.wso2.carbon.user.core/src/main/java/org/wso2/carbon/user/core/util/JDBCRealmUtil.java

@@ -535,6 +535,14 @@ public static Map<String, String> getSQL(Map<String, String> properties) {
                     JDBCRealmConstants.GET_USERS_PROPS_FOR_PROFILE_WITH_ID_SQL);
         }
 
+        if (!properties.containsKey(JDBCRealmConstants.GET_USERS_COUNT_WITH_FILTER_ROLE)) {
+            properties.put(JDBCRealmConstants.GET_USERS_COUNT_WITH_FILTER_ROLE, JDBCRealmConstants.GET_USERS_COUNT_WITH_FILTER_ROLE_SQL);
+        }
+
+        if (!properties.containsKey(JDBCRealmConstants.GET_ROLE_ID_BY_NAME)) {
+            properties.put(JDBCRealmConstants.GET_ROLE_ID_BY_NAME, JDBCRealmConstants.GET_ROLE_ID_BY_NAME_SQL);
+        }
+
         if (!properties.containsKey(JDBCCaseInsensitiveConstants.GET_USERS_PROPS_FOR_PROFILE_CASE_INSENSITIVE)) {
             properties.put(JDBCCaseInsensitiveConstants.GET_USERS_PROPS_FOR_PROFILE_CASE_INSENSITIVE,
                     JDBCCaseInsensitiveConstants.GET_USERS_PROPS_FOR_PROFILE_SQL_CASE_INSENSITIVE);