Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Fix multi-attribute user filter with count provided. #555

Merged
merged 4 commits into from
Jul 21, 2024
Merged

Fix multi-attribute user filter with count provided. #555

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
987416e
Fix multi-attribute user filter with count provided.
Thisara-Welmilla Jun 11, 2024
74f70aa
Improve tenant qualified URL feature.
Thisara-Welmilla Jul 8, 2024
925a1b5
Merge branch 'wso2-extensions:master' into fix-users-filter
Thisara-Welmilla Jul 9, 2024
8851256
Improve tenant qualified URL feature.
Thisara-Welmilla Jul 17, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
69 changes: 58 additions & 11 deletions ...cim2.common/src/main/java/org/wso2/carbon/identity/scim2/common/impl/SCIMUserManager.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1881,6 +1881,39 @@ private int calculateLimit(int limit, int numberOfFilteredUsers) {
return newLimit;
}

/**
* Method to update the offset when iterating a filter across all domains with multi attribute filter.
*
* @param node Condition node of the multi attribute filter
* @param offset Starting index
* @param sortBy Sort by
* @param sortOrder Sort order
* @param domainName Domain to be filtered
* @return New calculated offset
* @throws CharonException Error while filtering the domain from index 1 to offset
*/
private int calculateOffsetForMultiAttributeFilter(Node node, int offset, String sortBy, String sortOrder,
String domainName) throws CharonException, BadRequestException {

if (log.isDebugEnabled()) {
log.debug(String.format("Checking for number of matches from the beginning to the original offset: %d for "
+ "the same filter and updating the new offset.", offset));
}
// Starting index of the filter
int initialOffset = 1;

// Checking the number of matches till the original offset.
boolean paginationRequest = false;
Set<org.wso2.carbon.user.core.common.User> skippedUsers =
getFilteredUsersFromMultiAttributeFiltering(node, initialOffset, offset, sortBy, sortOrder, domainName,
paginationRequest);

int skippedUserCount = skippedUsers.size();

// Calculate the new offset and return
return offset - skippedUserCount;
}

/**
* Method to update the offset when iterating a filter across all domains.
*
Expand Down Expand Up @@ -2125,14 +2158,14 @@ private UsersGetResponse getMultiAttributeFilteredUsers(Node node, Map<String, B
int totalResults = 0;
// Handle pagination.
if (limit > 0) {
users = getFilteredUsersFromMultiAttributeFiltering(node, offset, limit, sortBy, sortOrder, domainName,
users = getMultiAttributeFilteredUsersWithMaxLimit(node, offset, sortBy, sortOrder, domainName, limit,
true);
filteredUsers.addAll(getFilteredUserDetails(users, requiredAttributes));
} else {
int maxLimit = getMaxLimit(domainName);
users = getMultiAttributeFilteredUsersWithMaxLimit(node, offset, sortBy, sortOrder, domainName, maxLimit);
filteredUsers.addAll(getFilteredUserDetails(users, requiredAttributes));
users = getMultiAttributeFilteredUsersWithMaxLimit(node, offset, sortBy, sortOrder, domainName, maxLimit,
false);
}
filteredUsers.addAll(getFilteredUserDetails(users, requiredAttributes));
// Check that total user count matching the client query needs to be calculated.
if (isJDBCUSerStore(domainName) || isAllConfiguredUserStoresJDBC() ||
SCIMCommonUtils.isConsiderTotalRecordsForTotalResultOfLDAPEnabled()) {
Expand All @@ -2142,7 +2175,7 @@ private UsersGetResponse getMultiAttributeFilteredUsers(Node node, Map<String, B
}
// Get total users based on the filter query.
totalResults += getMultiAttributeFilteredUsersWithMaxLimit(node, 1, sortBy,
sortOrder, domainName, maxLimit).size();
sortOrder, domainName, maxLimit, false).size();
} else {
totalResults += filteredUsers.size();
if (totalResults == 0 && filteredUsers.size() > 1) {
Expand All @@ -2154,27 +2187,41 @@ private UsersGetResponse getMultiAttributeFilteredUsers(Node node, Map<String, B


private Set<org.wso2.carbon.user.core.common.User> getMultiAttributeFilteredUsersWithMaxLimit(Node node, int offset,
String sortBy, String sortOrder, String domainName, int maxLimit)
String sortBy, String sortOrder, String domainName, int maxLimit, boolean paginationRequested)
throws CharonException, BadRequestException {

Set<org.wso2.carbon.user.core.common.User> users = null;
Set<org.wso2.carbon.user.core.common.User> users = new LinkedHashSet<>();
if (StringUtils.isNotEmpty(domainName)) {
users = getFilteredUsersFromMultiAttributeFiltering(node, offset, maxLimit, sortBy, sortOrder, domainName,
false);
return users;
} else {
AbstractUserStoreManager tempCarbonUM = carbonUM;
// If pagination and domain name are not given, then perform filtering on all available user stores.
while (tempCarbonUM != null) {
// If domain name are not given, then perform filtering on all available user stores.
while (tempCarbonUM != null && maxLimit > 0) {
// If carbonUM is not an instance of Abstract User Store Manger we can't get the domain name.
if (tempCarbonUM instanceof AbstractUserStoreManager) {
domainName = tempCarbonUM.getRealmConfiguration().getUserStoreProperty("DomainName");
users = getFilteredUsersFromMultiAttributeFiltering(node, offset, maxLimit, sortBy, sortOrder,
domainName, false);
users.addAll(getFilteredUsersFromMultiAttributeFiltering(node, offset, maxLimit, sortBy, sortOrder,
domainName, paginationRequested));
}
// If secondary user store manager assigned to carbonUM then global variable carbonUM will contains the
// secondary user store manager.
tempCarbonUM = (AbstractUserStoreManager) tempCarbonUM.getSecondaryUserStoreManager();

// Calculating new offset and limit parameters.
int numberOfFilteredUsers = users.size();
if (numberOfFilteredUsers <= 0 && offset > 1) {
if (log.isDebugEnabled()) {
log.debug(String.format("Filter returned no results for original offset: %d.", offset));
}
offset = calculateOffsetForMultiAttributeFilter(node, offset, sortBy, sortOrder, domainName);
} else {
// Returned usernames size > 0 implies there are users in that domain which is larger than
// the offset.
offset = 1;
maxLimit = calculateLimit(maxLimit, numberOfFilteredUsers);
}
}
return users;
}
Expand Down
Loading