-
Notifications
You must be signed in to change notification settings - Fork 541
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #6007 from amanda-ariyaratne/issue-26702
In-Memory Claim Management
- Loading branch information
Showing
21 changed files
with
3,397 additions
and
139 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
148 changes: 93 additions & 55 deletions
148
.../java/org/wso2/carbon/identity/claim/metadata/mgt/ClaimMetadataManagementServiceImpl.java
Large diffs are not rendered by default.
Oops, something went wrong.
216 changes: 216 additions & 0 deletions
216
...rc/main/java/org/wso2/carbon/identity/claim/metadata/mgt/DBBasedClaimMetadataManager.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,216 @@ | ||
| /* | ||
| * Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com). | ||
| * | ||
| * WSO2 LLC. licenses this file to you under the Apache License, | ||
| * Version 2.0 (the "License"); you may not use this file except | ||
| * in compliance with the License. | ||
| * You may obtain a copy of the License at | ||
| * | ||
| * http://www.apache.org/licenses/LICENSE-2.0 | ||
| * | ||
| * Unless required by applicable law or agreed to in writing, | ||
| * software distributed under the License is distributed on an | ||
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | ||
| * KIND, either express or implied. See the License for the | ||
| * specific language governing permissions and limitations | ||
| * under the License. | ||
| */ | ||
|
|
||
| package org.wso2.carbon.identity.claim.metadata.mgt; | ||
|
|
||
| import org.apache.commons.lang.StringUtils; | ||
| import org.wso2.carbon.identity.claim.metadata.mgt.dao.CacheBackedClaimDialectDAO; | ||
| import org.wso2.carbon.identity.claim.metadata.mgt.dao.CacheBackedExternalClaimDAO; | ||
| import org.wso2.carbon.identity.claim.metadata.mgt.dao.CacheBackedLocalClaimDAO; | ||
| import org.wso2.carbon.identity.claim.metadata.mgt.dao.ClaimDialectDAO; | ||
| import org.wso2.carbon.identity.claim.metadata.mgt.dao.ExternalClaimDAO; | ||
| import org.wso2.carbon.identity.claim.metadata.mgt.dao.LocalClaimDAO; | ||
| import org.wso2.carbon.identity.claim.metadata.mgt.exception.ClaimMetadataException; | ||
| import org.wso2.carbon.identity.claim.metadata.mgt.exception.ClaimMetadataServerException; | ||
| import org.wso2.carbon.identity.claim.metadata.mgt.internal.ReadOnlyClaimMetadataManager; | ||
| import org.wso2.carbon.identity.claim.metadata.mgt.internal.ReadWriteClaimMetadataManager; | ||
| import org.wso2.carbon.identity.claim.metadata.mgt.model.Claim; | ||
| import org.wso2.carbon.identity.claim.metadata.mgt.model.ClaimDialect; | ||
| import org.wso2.carbon.identity.claim.metadata.mgt.model.ExternalClaim; | ||
| import org.wso2.carbon.identity.claim.metadata.mgt.model.LocalClaim; | ||
| import org.wso2.carbon.identity.claim.metadata.mgt.util.ClaimConstants; | ||
| import org.wso2.carbon.user.api.UserStoreException; | ||
|
|
||
| import java.util.List; | ||
| import java.util.Optional; | ||
|
|
||
| /** | ||
| * Database based claim metadata manager. | ||
| */ | ||
| public class DBBasedClaimMetadataManager implements ReadWriteClaimMetadataManager { | ||
|
|
||
| private final ClaimDialectDAO claimDialectDAO = new CacheBackedClaimDialectDAO(); | ||
| private final CacheBackedLocalClaimDAO localClaimDAO = new CacheBackedLocalClaimDAO(new LocalClaimDAO()); | ||
| private final CacheBackedExternalClaimDAO externalClaimDAO = new CacheBackedExternalClaimDAO(new ExternalClaimDAO()); | ||
|
|
||
| @Override | ||
| public List<ClaimDialect> getClaimDialects(int tenantId) throws ClaimMetadataException { | ||
|
|
||
| return this.claimDialectDAO.getClaimDialects(tenantId); | ||
| } | ||
|
|
||
| @Override | ||
| public Optional<ClaimDialect> getClaimDialect(String claimDialectURI, int tenantId) throws ClaimMetadataException { | ||
|
|
||
| if (StringUtils.isBlank(claimDialectURI)) { | ||
| throw new ClaimMetadataException("Invalid claim dialect URI: " + claimDialectURI); | ||
| } | ||
|
|
||
| return this.claimDialectDAO.getClaimDialects(tenantId).stream() | ||
| .filter(claimDialect -> claimDialectURI.equals(claimDialect.getClaimDialectURI())) | ||
| .findFirst(); | ||
| } | ||
|
|
||
| @Override | ||
| public void addClaimDialect(ClaimDialect claimDialect, int tenantId) throws ClaimMetadataException { | ||
|
|
||
| this.claimDialectDAO.addClaimDialect(claimDialect, tenantId); | ||
| } | ||
|
|
||
| @Override | ||
| public void removeClaimDialect(ClaimDialect claimDialect, int tenantId) throws ClaimMetadataException { | ||
|
|
||
| this.claimDialectDAO.removeClaimDialect(claimDialect, tenantId); | ||
| // When deleting a claim dialect the relevant external claim deletion is handled by the DB through | ||
| // ON DELETE CASCADE. Here we are removing the relevant cache entry. | ||
| externalClaimDAO.removeExternalClaimCache(claimDialect.getClaimDialectURI(), tenantId); | ||
| } | ||
|
|
||
| @Override | ||
| public List<LocalClaim> getLocalClaims(int tenantId) throws ClaimMetadataException { | ||
|
|
||
| return this.localClaimDAO.getLocalClaims(tenantId); | ||
| } | ||
|
|
||
| @Override | ||
| public Optional<LocalClaim> getLocalClaim(String localClaimURI , int tenantId) throws ClaimMetadataException { | ||
|
|
||
| if (StringUtils.isBlank(localClaimURI)) { | ||
| throw new ClaimMetadataException("Invalid local claim URI: " + localClaimURI); | ||
| } | ||
|
|
||
| List<LocalClaim> localClaims = this.localClaimDAO.getLocalClaims(tenantId); | ||
| return localClaims.stream() | ||
| .filter(localClaim -> localClaimURI.equals(localClaim.getClaimURI())) | ||
| .findFirst(); | ||
| } | ||
|
|
||
| @Override | ||
| public List<ExternalClaim> getExternalClaims(String externalClaimDialectURI, int tenantId) | ||
| throws ClaimMetadataException { | ||
|
|
||
| return this.externalClaimDAO.getExternalClaims(externalClaimDialectURI, tenantId); | ||
| } | ||
|
|
||
| @Override | ||
| public Optional<ExternalClaim> getExternalClaim(String externalClaimDialectURI, String claimURI, int tenantId) | ||
| throws ClaimMetadataException { | ||
|
|
||
| if (StringUtils.isBlank(externalClaimDialectURI) || StringUtils.isBlank(claimURI)) { | ||
| throw new ClaimMetadataException("Invalid external claim dialect URI or claim URI"); | ||
| } | ||
|
|
||
| return this.externalClaimDAO.getExternalClaims(externalClaimDialectURI, tenantId).stream() | ||
| .filter(externalClaim -> claimURI.equals(externalClaim.getClaimURI())) | ||
| .findFirst(); | ||
| } | ||
|
|
||
| @Override | ||
| public void addLocalClaim(LocalClaim localClaim, int tenantId) throws ClaimMetadataException { | ||
|
|
||
| this.localClaimDAO.addLocalClaim(localClaim, tenantId); | ||
| } | ||
|
|
||
| @Override | ||
| public void updateLocalClaim(LocalClaim localClaim, int tenantId) throws ClaimMetadataException { | ||
|
|
||
| this.localClaimDAO.updateLocalClaim(localClaim, tenantId); | ||
| } | ||
|
|
||
| @Override | ||
| public void updateLocalClaimMappings(List<LocalClaim> localClaims, int tenantId, String userStoreDomain) | ||
| throws ClaimMetadataException { | ||
|
|
||
| this.localClaimDAO.updateLocalClaimMappings(localClaims, tenantId, userStoreDomain); | ||
| } | ||
|
|
||
| @Override | ||
| public void removeLocalClaim(String localClaimURI, int tenantId) throws ClaimMetadataException { | ||
|
|
||
| this.localClaimDAO.removeLocalClaim(localClaimURI, tenantId); | ||
| } | ||
|
|
||
| @Override | ||
| public void removeClaimMappingAttributes(int tenantId, String userstoreDomain) throws ClaimMetadataException { | ||
|
|
||
| try { | ||
| this.localClaimDAO.removeClaimMappingAttributes(tenantId, userstoreDomain); | ||
| } catch (UserStoreException e) { | ||
| String errorMessage = String.format( | ||
| ClaimConstants.ErrorMessage.ERROR_CODE_SERVER_ERROR_DELETING_CLAIM_MAPPINGS.getMessage(), | ||
| tenantId, userstoreDomain); | ||
| throw new ClaimMetadataServerException( | ||
| ClaimConstants.ErrorMessage.ERROR_CODE_SERVER_ERROR_DELETING_CLAIM_MAPPINGS.getCode(), | ||
| errorMessage, e); | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public void addExternalClaim(ExternalClaim externalClaim, int tenantId) throws ClaimMetadataException { | ||
|
|
||
| this.externalClaimDAO.addExternalClaim(externalClaim, tenantId); | ||
| } | ||
|
|
||
| @Override | ||
| public void updateExternalClaim(ExternalClaim externalClaim, int tenantId) throws ClaimMetadataException { | ||
|
|
||
| this.externalClaimDAO.updateExternalClaim(externalClaim, tenantId); | ||
| } | ||
|
|
||
| @Override | ||
| public void removeExternalClaim(String externalClaimDialectURI, String externalClaimURI, int tenantId) | ||
| throws ClaimMetadataException { | ||
|
|
||
| this.externalClaimDAO.removeExternalClaim(externalClaimDialectURI, externalClaimURI, tenantId); | ||
| } | ||
|
|
||
| @Override | ||
| public List<Claim> getMappedExternalClaims(String localClaimURI, int tenantId) throws ClaimMetadataException { | ||
|
|
||
| return this.localClaimDAO.fetchMappedExternalClaims(localClaimURI, tenantId); | ||
| } | ||
|
|
||
| @Override | ||
| public void renameClaimDialect(ClaimDialect oldClaimDialect, ClaimDialect newClaimDialect, int tenantId) | ||
| throws ClaimMetadataException { | ||
|
|
||
| this.claimDialectDAO.renameClaimDialect(oldClaimDialect, newClaimDialect, tenantId); | ||
| externalClaimDAO.removeExternalClaimCache(oldClaimDialect.getClaimDialectURI(), tenantId); | ||
| } | ||
|
|
||
| @Override | ||
| public void removeAllClaimDialects(int tenantId) throws ClaimMetadataException { | ||
|
|
||
| // The relevant external claim deletions are handled by the DB through ON DELETE CASCADE. | ||
| this.claimDialectDAO.removeAllClaimDialects(tenantId); | ||
| } | ||
|
|
||
| @Override | ||
| public boolean isMappedLocalClaim(String localClaimURI, int tenantId) throws ClaimMetadataException { | ||
|
|
||
| return this.externalClaimDAO.isMappedLocalClaim(localClaimURI, tenantId); | ||
| } | ||
|
|
||
| @Override | ||
| public boolean isLocalClaimMappedWithinDialect(String mappedLocalClaim, String externalClaimDialectURI, | ||
| int tenantId) throws ClaimMetadataException { | ||
|
|
||
| return this.externalClaimDAO.isLocalClaimMappedWithinDialect(mappedLocalClaim, externalClaimDialectURI, | ||
| tenantId); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.