Picketlink SVN: r1214 - in idm/trunk: example/idm-servlet and 7 other directories.
by picketlink-commits@lists.jboss.org
Author: bdaw
Date: 2011-09-12 17:25:13 -0400 (Mon, 12 Sep 2011)
New Revision: 1214
Removed:
idm/trunk/picketlink-idm-jpa/
Modified:
idm/trunk/example/idm-servlet/jboss-idm-servlet.iml
idm/trunk/picketlink-idm-api/src/main/java/org/picketlink/idm/api/RelationshipManager.java
idm/trunk/picketlink-idm-api/src/main/java/org/picketlink/idm/api/RoleManager.java
idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/api/session/managers/RelationshipManagerImpl.java
idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/api/session/managers/RoleManagerImpl.java
idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/cache/AbstractSPISearchImpl.java
idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/cache/IdentityObjectRelationshipSearchImpl.java
idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/cache/JBossCacheIdentityStoreWrapper.java
idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/repository/FallbackIdentityStoreRepository.java
idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/repository/WrapperIdentityStoreRepository.java
idm/trunk/picketlink-idm-hibernate/src/main/java/org/picketlink/idm/impl/store/hibernate/HibernateIdentityStoreImpl.java
idm/trunk/picketlink-idm-ldap/src/main/java/org/picketlink/idm/impl/store/ldap/LDAPIdentityStoreImpl.java
idm/trunk/picketlink-idm-spi/src/main/java/org/picketlink/idm/spi/store/IdentityStore.java
idm/trunk/pom.xml
Log:
- remove jpa store for now as it is not maintained
- add new methods for better paginated searches
- performance tweaks for hibernate queries
Modified: idm/trunk/example/idm-servlet/jboss-idm-servlet.iml
===================================================================
--- idm/trunk/example/idm-servlet/jboss-idm-servlet.iml 2011-09-12 12:09:55 UTC (rev 1213)
+++ idm/trunk/example/idm-servlet/jboss-idm-servlet.iml 2011-09-12 21:25:13 UTC (rev 1214)
@@ -16,11 +16,12 @@
<component name="NewModuleRootManager" LANGUAGE_LEVEL="JDK_1_6" inherit-compiler-output="false">
<output url="file://$MODULE_DIR$/target/classes" />
<output-test url="file://$MODULE_DIR$/target/test-classes" />
+ <exclude-output />
<content url="file://$MODULE_DIR$">
- <sourceFolder url="file://$MODULE_DIR$/target/generated-sources/annotations" isTestSource="false" />
<sourceFolder url="file://$MODULE_DIR$/src/main/java" isTestSource="false" />
<sourceFolder url="file://$MODULE_DIR$/src/main/resources" isTestSource="false" />
<sourceFolder url="file://$MODULE_DIR$/src/test/java" isTestSource="true" />
+ <excludeFolder url="file://$MODULE_DIR$/target" />
</content>
<orderEntry type="inheritedJdk" />
<orderEntry type="sourceFolder" forTests="false" />
Modified: idm/trunk/picketlink-idm-api/src/main/java/org/picketlink/idm/api/RelationshipManager.java
===================================================================
--- idm/trunk/picketlink-idm-api/src/main/java/org/picketlink/idm/api/RelationshipManager.java 2011-09-12 12:09:55 UTC (rev 1213)
+++ idm/trunk/picketlink-idm-api/src/main/java/org/picketlink/idm/api/RelationshipManager.java 2011-09-12 21:25:13 UTC (rev 1214)
@@ -311,7 +311,28 @@
boolean parent,
boolean cascade) throws IdentityException;
+
/**
+ * Get count of groups that are associated with given group.
+ * If 'parent' parameter is set to false, all parent group will be returned. If parent parameter is
+ * set to true and 'cascade' is set to true all nested subgroubs will be returned.
+ *
+ * @param group parent group
+ * @param groupType can be null
+ * @param parent defines if given identity is parent or child side in the
+ * relationship - default is true (parent)
+ * @param cascade if true also identities from subgroubs will be retreived. Matters only when parent is set to true.
+ * Default is false
+ * @return
+ * @throws org.picketlink.idm.common.exception.IdentityException
+ */
+ int getAssociatedGroupsCount(Group group,
+ String groupType,
+ boolean parent,
+ boolean cascade,
+ IdentitySearchCriteria criteria) throws IdentityException;
+
+ /**
* Find groups that are associated with given group.
* If 'parent' parameter is set to false, all parent group will be returned. If parent parameter is
* set to true and 'cascade' is set to true all nested subgroubs will be returned.
Modified: idm/trunk/picketlink-idm-api/src/main/java/org/picketlink/idm/api/RoleManager.java
===================================================================
--- idm/trunk/picketlink-idm-api/src/main/java/org/picketlink/idm/api/RoleManager.java 2011-09-12 12:09:55 UTC (rev 1213)
+++ idm/trunk/picketlink-idm-api/src/main/java/org/picketlink/idm/api/RoleManager.java 2011-09-12 21:25:13 UTC (rev 1214)
@@ -344,7 +344,32 @@
<T extends IdentityType> Collection<Role> findRoles(T identityType,
RoleType roleType) throws IdentityException;
+
/**
+ * Find Role objects count with a given RoleType associated with a given IdentityType
+ * @param identityType
+ * @param roleType
+ * @param criteria
+ * @return
+ * @throws IdentityException
+ */
+ int getRolesCount(IdentityType identityType,
+ RoleType roleType,
+ IdentitySearchCriteria criteria) throws IdentityException;
+
+ /**
+ * Find Role objects with a given RoleType associated with a given IdentityType
+ * @param identityType
+ * @param roleType
+ * @param criteria
+ * @return
+ * @throws IdentityException
+ */
+ <T extends IdentityType> Collection<Role> findRoles(T identityType,
+ RoleType roleType,
+ IdentitySearchCriteria criteria) throws IdentityException;
+
+ /**
* Find Role objects with a given RoleType name associated with IdentityType for a given key
* @param key
* @param roleTypeName
@@ -354,7 +379,20 @@
<T extends IdentityType> Collection<Role> findRoles(String key,
String roleTypeName) throws IdentityException;
+
/**
+ * Find Role objects with a given RoleType name associated with IdentityType for a given key
+ * @param key
+ * @param roleTypeName
+ * @param criteria
+ * @return
+ * @throws IdentityException
+ */
+ <T extends IdentityType> Collection<Role> findRoles(String key,
+ String roleTypeName,
+ IdentitySearchCriteria criteria) throws IdentityException;
+
+ /**
*
* @param role
* @return
Modified: idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/api/session/managers/RelationshipManagerImpl.java
===================================================================
--- idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/api/session/managers/RelationshipManagerImpl.java 2011-09-12 12:09:55 UTC (rev 1213)
+++ idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/api/session/managers/RelationshipManagerImpl.java 2011-09-12 21:25:13 UTC (rev 1214)
@@ -856,6 +856,8 @@
private Collection<Group> findAssociatedGroupsCascaded(Collection<Group> previous, Group group, String groupType, boolean parent, IdentitySearchCriteria criteria) throws IdentityException
{
+ getIdentitySession().save();
+
Collection<Group> results = findAssociatedGroups(group, groupType, parent, false, criteria);
List<Group> newResults = new LinkedList<Group>();
@@ -868,6 +870,9 @@
{
newResults.add(result);
previous.add(result);
+
+ getIdentitySession().save();
+
newResults.addAll(findAssociatedGroupsCascaded(previous, result, groupType, parent, criteria));
}
}
@@ -876,6 +881,14 @@
}
+ public int getAssociatedGroupsCount(Group group, String groupType, boolean parent, boolean cascade, IdentitySearchCriteria criteria) throws IdentityException
+ {
+
+ //TODO: implement!
+ return findAssociatedGroups(group, groupType, parent, cascade, criteria).size();
+
+ }
+
public Collection<Group> findAssociatedGroups(Group group, String groupType, boolean parent, boolean cascade, IdentitySearchCriteria criteria) throws IdentityException
{
@@ -930,7 +943,6 @@
else
{
-
Collection<IdentityObject> ios = getRepository().
findIdentityObject(
getInvocationContext(),
Modified: idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/api/session/managers/RoleManagerImpl.java
===================================================================
--- idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/api/session/managers/RoleManagerImpl.java 2011-09-12 12:09:55 UTC (rev 1213)
+++ idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/api/session/managers/RoleManagerImpl.java 2011-09-12 21:25:13 UTC (rev 1214)
@@ -842,9 +842,73 @@
return findGroupsWithRelatedRole(user, groupType, criteria);
}
- public Collection<Role> findRoles(IdentityType identityType, RoleType roleType) throws IdentityException
+ public int getRolesCount(IdentityType identityType, RoleType roleType, IdentitySearchCriteria criteria) throws IdentityException
{
+ //TODO: cache support
+
checkNotNullArgument(identityType, "IdentityType");
+
+
+// if (cache != null)
+// {
+// RoleSearchImpl search = new RoleSearchImpl();
+// search.setIdentityTypeId(identityType.getKey());
+// search.setRoleType(roleType);
+//
+// Collection<Role> results = cache.getRoleSearch(cacheNS, search);
+// if (results != null)
+// {
+// return results;
+// }
+// }
+
+ int relationshipsCount = 0;
+
+ // If Identity then search for parent relationships
+ if (identityType instanceof User)
+ {
+ relationshipsCount = getRepository().
+ getRelationshipsCount(
+ getInvocationContext(),
+ createIdentityObject(identityType),
+ ROLE,
+ false,
+ true,
+ null,
+ convertSearchControls(criteria));
+ }
+ // If Group then search for child relationships
+ else
+ {
+ relationshipsCount = getRepository().
+ getRelationshipsCount(
+ getInvocationContext(),
+ createIdentityObject(identityType),
+ ROLE,
+ true,
+ true,
+ null,
+ convertSearchControls(criteria));
+ }
+
+
+// if (cache != null)
+// {
+// RoleSearchImpl search = new RoleSearchImpl();
+// search.setIdentityTypeId(identityType.getKey());
+// search.setRoleType(roleType);
+//
+// cache.putRoleSearch(cacheNS, search, roles);
+//
+// }
+
+ return relationshipsCount;
+
+ }
+
+ public Collection<Role> findRoles(IdentityType identityType, RoleType roleType, IdentitySearchCriteria criteria) throws IdentityException
+ {
+ checkNotNullArgument(identityType, "IdentityType");
//checkNotNullArgument(roleType, "RoleType");
Set<Role> roles = new HashSet<Role>();
@@ -867,12 +931,28 @@
// If Identity then search for parent relationships
if (identityType instanceof User)
{
- relationships = getRepository().resolveRelationships(getInvocationContext(), createIdentityObject(identityType), ROLE, false, true, null);
+ relationships = getRepository().
+ resolveRelationships(
+ getInvocationContext(),
+ createIdentityObject(identityType),
+ ROLE,
+ false,
+ true,
+ null,
+ convertSearchControls(criteria));
}
// If Group then search for child relationships
else
{
- relationships = getRepository().resolveRelationships(getInvocationContext(), createIdentityObject(identityType), ROLE, true, true, null);
+ relationships = getRepository().
+ resolveRelationships(
+ getInvocationContext(),
+ createIdentityObject(identityType),
+ ROLE,
+ true,
+ true,
+ null,
+ convertSearchControls(criteria));
}
for (IdentityObjectRelationship relationship : relationships)
@@ -904,14 +984,24 @@
}
- public Collection<Role> findRoles(String id, String roleTypeName) throws IdentityException
+ public Collection<Role> findRoles(IdentityType identityType, RoleType roleType) throws IdentityException
{
+ return findRoles(identityType, roleType, null);
+ }
+
+ public Collection<Role> findRoles(String id, String roleTypeName, IdentitySearchCriteria criteria) throws IdentityException
+ {
checkNotNullArgument(id, "Group id or User name");
RoleType roleType = roleTypeName != null ? new SimpleRoleType(roleTypeName) : null;
- return findRoles(createIdentityTypeFromId(id), roleType);
+ return findRoles(createIdentityTypeFromId(id), roleType, criteria);
}
+ public <T extends IdentityType> Collection<Role> findRoles(String key, String roleTypeName) throws IdentityException
+ {
+ return findRoles(key, roleTypeName, null);
+ }
+
public Map<String, String> getProperties(RoleType roleType) throws IdentityException
{
checkNotNullArgument(roleType, "RoleType name");
Modified: idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/cache/AbstractSPISearchImpl.java
===================================================================
--- idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/cache/AbstractSPISearchImpl.java 2011-09-12 12:09:55 UTC (rev 1213)
+++ idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/cache/AbstractSPISearchImpl.java 2011-09-12 21:25:13 UTC (rev 1214)
@@ -39,4 +39,32 @@
{
this.identityObjectSearchCriteria = identityObjectSearchCriteria;
}
+
+ @Override
+ public boolean equals(Object o)
+ {
+ if (this == o)
+ {
+ return true;
+ }
+ if (!(o instanceof AbstractSPISearchImpl))
+ {
+ return false;
+ }
+
+ AbstractSPISearchImpl that = (AbstractSPISearchImpl)o;
+
+ if (identityObjectSearchCriteria != null ? !identityObjectSearchCriteria.equals(that.identityObjectSearchCriteria) : that.identityObjectSearchCriteria != null)
+ {
+ return false;
+ }
+
+ return true;
+ }
+
+ @Override
+ public int hashCode()
+ {
+ return identityObjectSearchCriteria != null ? identityObjectSearchCriteria.hashCode() : 0;
+ }
}
Modified: idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/cache/IdentityObjectRelationshipSearchImpl.java
===================================================================
--- idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/cache/IdentityObjectRelationshipSearchImpl.java 2011-09-12 12:09:55 UTC (rev 1213)
+++ idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/cache/IdentityObjectRelationshipSearchImpl.java 2011-09-12 21:25:13 UTC (rev 1214)
@@ -206,4 +206,21 @@
return true;
}
+
+ @Override
+ public int hashCode()
+ {
+ int result = super.hashCode();
+ result = 31 * result + (fromIOName != null ? fromIOName.hashCode() : 0);
+ result = 31 * result + (fromIOType != null ? fromIOType.hashCode() : 0);
+ result = 31 * result + (toIOName != null ? toIOName.hashCode() : 0);
+ result = 31 * result + (toIOType != null ? toIOType.hashCode() : 0);
+ result = 31 * result + (relationshipType != null ? relationshipType.hashCode() : 0);
+ result = 31 * result + (ioName != null ? ioName.hashCode() : 0);
+ result = 31 * result + (ioType != null ? ioType.hashCode() : 0);
+ result = 31 * result + (parent ? 1 : 0);
+ result = 31 * result + (named ? 1 : 0);
+ result = 31 * result + (name != null ? name.hashCode() : 0);
+ return result;
+ }
}
Modified: idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/cache/JBossCacheIdentityStoreWrapper.java
===================================================================
--- idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/cache/JBossCacheIdentityStoreWrapper.java 2011-09-12 12:09:55 UTC (rev 1213)
+++ idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/cache/JBossCacheIdentityStoreWrapper.java 2011-09-12 21:25:13 UTC (rev 1214)
@@ -245,6 +245,38 @@
}
+ public int getIdentityObjectCount(IdentityStoreInvocationContext invocationCxt,
+ IdentityObject identity,
+ IdentityObjectRelationshipType relationshipType,
+ boolean parent,
+ IdentityObjectSearchCriteria criteria) throws IdentityException
+ {
+ return identityStore.getIdentityObjectCount(
+ invocationCxt,
+ identity,
+ relationshipType,
+ parent,
+ criteria
+ );
+ }
+
+ public int getIdentityObjectCount(IdentityStoreInvocationContext ctx,
+ IdentityObject identity,
+ IdentityObjectRelationshipType relationshipType,
+ Collection<IdentityObjectType> excludes,
+ boolean parent,
+ IdentityObjectSearchCriteria criteria) throws IdentityException
+ {
+ return identityStore.getIdentityObjectCount(
+ ctx,
+ identity,
+ relationshipType,
+ excludes,
+ parent,
+ criteria
+ );
+ }
+
public Collection<IdentityObject> findIdentityObject(IdentityStoreInvocationContext invocationCxt,
IdentityObject identity,
IdentityObjectRelationshipType relationshipType,
@@ -381,12 +413,36 @@
}
+ public int getRelationshipsCount(IdentityStoreInvocationContext ctx,
+ IdentityObject identity,
+ IdentityObjectRelationshipType type,
+ boolean parent,
+ boolean named,
+ String name,
+ IdentityObjectSearchCriteria searchCriteria) throws IdentityException
+ {
+
+ //TODO implement cache
+
+ return identityStore.getRelationshipsCount(
+ ctx,
+ identity,
+ type,
+ parent,
+ named,
+ name,
+ searchCriteria
+ );
+ }
+
+
public Set<IdentityObjectRelationship> resolveRelationships(IdentityStoreInvocationContext invocationCtx,
IdentityObject identity,
IdentityObjectRelationshipType relationshipType,
boolean parent,
boolean named,
- String name) throws IdentityException
+ String name,
+ IdentityObjectSearchCriteria criteria) throws IdentityException
{
IdentityObjectRelationshipSearchImpl search = new IdentityObjectRelationshipSearchImpl();
@@ -402,6 +458,7 @@
search.setParent(parent);
search.setNamed(named);
search.setName(name);
+ search.setIdentityObjectSearchCriteria(criteria);
Set<IdentityObjectRelationship> results = cacheSupport.getIdentityObjectRelationshipSearch(getCacheNS(invocationCtx), search);
@@ -411,7 +468,7 @@
}
results = identityStore.resolveRelationships(invocationCtx,
- identity, relationshipType, parent, named, name);
+ identity, relationshipType, parent, named, name, criteria);
cacheSupport.putIdentityObjectRelationshipSearch(getCacheNS(invocationCtx), search, results);
Modified: idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/repository/FallbackIdentityStoreRepository.java
===================================================================
--- idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/repository/FallbackIdentityStoreRepository.java 2011-09-12 12:09:55 UTC (rev 1213)
+++ idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/repository/FallbackIdentityStoreRepository.java 2011-09-12 21:25:13 UTC (rev 1214)
@@ -828,6 +828,62 @@
}
+ public int getIdentityObjectCount(IdentityStoreInvocationContext invocationCxt,
+ IdentityObject identity,
+ IdentityObjectRelationshipType relationshipType,
+ boolean parent,
+ IdentityObjectSearchCriteria criteria) throws IdentityException
+ {
+ //TODO: merge vs sum strategy
+
+ int count = 0;
+
+ Collection<IdentityStore> mappedStores = getConfiguredIdentityStores();
+
+ for (IdentityStore mappedStore : mappedStores)
+ {
+ count += mappedStore.getIdentityObjectCount(
+ invocationCxt,
+ identity,
+ relationshipType,
+ parent,
+ criteria
+ );
+
+ }
+
+ return count;
+ }
+
+ public int getIdentityObjectCount(IdentityStoreInvocationContext ctx,
+ IdentityObject identity,
+ IdentityObjectRelationshipType relationshipType,
+ Collection<IdentityObjectType> excludes,
+ boolean parent,
+ IdentityObjectSearchCriteria criteria) throws IdentityException
+ {
+
+ //TODO: merge vs sum strategy
+
+ int count = 0;
+
+ Collection<IdentityStore> mappedStores = getConfiguredIdentityStores();
+
+ for (IdentityStore mappedStore : mappedStores)
+ {
+ count += mappedStore.getIdentityObjectCount(
+ ctx,
+ identity,
+ relationshipType,
+ excludes,
+ parent,
+ criteria
+ );
+ }
+ return count;
+ }
+
+
public Collection<IdentityObject> findIdentityObject(IdentityStoreInvocationContext invocationCxt,
IdentityObject identity,
IdentityObjectRelationshipType relationshipType,
@@ -1153,8 +1209,46 @@
}
}
- public Set<IdentityObjectRelationship> resolveRelationships(IdentityStoreInvocationContext ctx, IdentityObject identity, IdentityObjectRelationshipType relationshipType, boolean parent, boolean named, String name) throws IdentityException
+ public int getRelationshipsCount(IdentityStoreInvocationContext ctx,
+ IdentityObject identity,
+ IdentityObjectRelationshipType type,
+ boolean parent,
+ boolean named,
+ String name,
+ IdentityObjectSearchCriteria searchCriteria) throws IdentityException
{
+
+ //TODO: merge vs sum strategy
+
+ int count = 0;
+
+ Collection<IdentityStore> mappedStores = getConfiguredIdentityStores();
+
+ for (IdentityStore mappedStore : mappedStores)
+ {
+ count += mappedStore.getRelationshipsCount(
+ ctx,
+ identity,
+ type,
+ parent,
+ named,
+ name,
+ searchCriteria
+ );
+ }
+
+ return count;
+
+ }
+
+ public Set<IdentityObjectRelationship> resolveRelationships(IdentityStoreInvocationContext ctx,
+ IdentityObject identity,
+ IdentityObjectRelationshipType relationshipType,
+ boolean parent,
+ boolean named,
+ String name,
+ IdentityObjectSearchCriteria criteria) throws IdentityException
+ {
try
{
Set<IdentityObjectRelationship> relationships = new HashSet<IdentityObjectRelationship>();
@@ -1173,7 +1267,8 @@
if ((!named || (named && identityStore.getSupportedFeatures().isNamedRelationshipsSupported()))
&& hasIdentityObject(storeCtx, identityStore, identity))
{
- relationships.addAll(identityStore.resolveRelationships(storeCtx, identity, relationshipType, parent, named, name));
+ relationships.addAll(identityStore.
+ resolveRelationships(storeCtx, identity, relationshipType, parent, named, name, criteria));
}
}
Modified: idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/repository/WrapperIdentityStoreRepository.java
===================================================================
--- idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/repository/WrapperIdentityStoreRepository.java 2011-09-12 12:09:55 UTC (rev 1213)
+++ idm/trunk/picketlink-idm-core/src/main/java/org/picketlink/idm/impl/repository/WrapperIdentityStoreRepository.java 2011-09-12 21:25:13 UTC (rev 1214)
@@ -184,6 +184,38 @@
return defaultIdentityStore.findIdentityObject(resolveIdentityStoreInvocationContext(invocationCtx), identityType, criteria);
}
+ public int getIdentityObjectCount(IdentityStoreInvocationContext invocationCxt,
+ IdentityObject identity,
+ IdentityObjectRelationshipType relationshipType,
+ boolean parent,
+ IdentityObjectSearchCriteria criteria) throws IdentityException
+ {
+ return defaultIdentityStore.getIdentityObjectCount(
+ invocationCxt,
+ identity,
+ relationshipType,
+ parent,
+ criteria
+ );
+ }
+
+ public int getIdentityObjectCount(IdentityStoreInvocationContext ctx,
+ IdentityObject identity,
+ IdentityObjectRelationshipType relationshipType,
+ Collection<IdentityObjectType> excludes,
+ boolean parent,
+ IdentityObjectSearchCriteria criteria) throws IdentityException
+ {
+ return defaultIdentityStore.getIdentityObjectCount(
+ ctx,
+ identity,
+ relationshipType,
+ excludes,
+ parent,
+ criteria
+ );
+ }
+
public Collection<IdentityObject> findIdentityObject(IdentityStoreInvocationContext invocationCtx,
IdentityObject identity,
IdentityObjectRelationshipType relationshipType,
@@ -239,15 +271,37 @@
return defaultIdentityStore.resolveRelationships(resolveIdentityStoreInvocationContext(invocationCxt), fromIdentity, toIdentity, relationshipType);
}
+
+
+ public int getRelationshipsCount(IdentityStoreInvocationContext ctx,
+ IdentityObject identity,
+ IdentityObjectRelationshipType type,
+ boolean parent,
+ boolean named,
+ String name,
+ IdentityObjectSearchCriteria searchCriteria) throws IdentityException
+ {
+ return defaultIdentityStore.getRelationshipsCount(
+ ctx,
+ identity,
+ type,
+ parent,
+ named,
+ name,
+ searchCriteria
+ );
+ }
+
public Set<IdentityObjectRelationship> resolveRelationships(IdentityStoreInvocationContext invocationCtx,
IdentityObject identity,
IdentityObjectRelationshipType relationshipType,
boolean parent,
boolean named,
- String name) throws IdentityException
+ String name,
+ IdentityObjectSearchCriteria criteria) throws IdentityException
{
return defaultIdentityStore.resolveRelationships(resolveIdentityStoreInvocationContext(invocationCtx),
- identity, relationshipType, parent, named, name);
+ identity, relationshipType, parent, named, name, criteria);
}
public String createRelationshipName(IdentityStoreInvocationContext ctx,
Modified: idm/trunk/picketlink-idm-hibernate/src/main/java/org/picketlink/idm/impl/store/hibernate/HibernateIdentityStoreImpl.java
===================================================================
--- idm/trunk/picketlink-idm-hibernate/src/main/java/org/picketlink/idm/impl/store/hibernate/HibernateIdentityStoreImpl.java 2011-09-12 12:09:55 UTC (rev 1213)
+++ idm/trunk/picketlink-idm-hibernate/src/main/java/org/picketlink/idm/impl/store/hibernate/HibernateIdentityStoreImpl.java 2011-09-12 21:25:13 UTC (rev 1214)
@@ -795,17 +795,63 @@
return findIdentityObject(ctx, identityType, null);
}
+ public int getIdentityObjectCount(IdentityStoreInvocationContext invocationCxt,
+ IdentityObject identity,
+ IdentityObjectRelationshipType relationshipType,
+ boolean parent,
+ IdentityObjectSearchCriteria criteria) throws IdentityException
+ {
+ return getIdentityObjectCount(invocationCxt, identity, relationshipType, null, parent, criteria);
+ }
@SuppressWarnings("unchecked")
public Collection<IdentityObject> findIdentityObject(IdentityStoreInvocationContext invocationCxt,
IdentityObject identity,
IdentityObjectRelationshipType relationshipType,
boolean parent,
- IdentityObjectSearchCriteria criteria) throws IdentityException
+ IdentityObjectSearchCriteria criteria) throws IdentityException
{
return findIdentityObject(invocationCxt, identity, relationshipType, null, parent, criteria);
}
+
+ public int getIdentityObjectCount(IdentityStoreInvocationContext ctx,
+ IdentityObject identity,
+ IdentityObjectRelationshipType relationshipType,
+ Collection<IdentityObjectType> excludes,
+ boolean parent,
+ IdentityObjectSearchCriteria criteria) throws IdentityException
+ {
+ int result;
+
+ try
+ {
+ org.hibernate.Query q = prepareIdentityObjectQuery(
+ ctx,
+ identity,
+ relationshipType,
+ excludes,
+ parent,
+ criteria,
+ false);
+
+
+ result = ((Integer)q.iterate().next() ).intValue();
+ }
+ catch (Exception e)
+ {
+ if (log.isLoggable(Level.FINER))
+ {
+ log.log(Level.FINER, "Exception occurred: ", e);
+ }
+
+ throw new IdentityException("Cannot get IdentityObject count", e);
+ }
+
+ return result;
+
+ }
+
@SuppressWarnings("unchecked")
public Collection<IdentityObject> findIdentityObject(IdentityStoreInvocationContext ctx,
IdentityObject identity,
@@ -814,12 +860,61 @@
boolean parent,
IdentityObjectSearchCriteria criteria) throws IdentityException
{
+
+ List<IdentityObject> results;
+
+ try
+ {
+ org.hibernate.Query q = prepareIdentityObjectQuery(
+ ctx,
+ identity,
+ relationshipType,
+ excludes,
+ parent,
+ criteria,
+ false);
+
+
+ results = q.list();
+ Hibernate.initialize(results);
+
+
+ }
+ catch (Exception e)
+ {
+ if (log.isLoggable(Level.FINER))
+ {
+ log.log(Level.FINER, "Exception occurred: ", e);
+ }
+
+ throw new IdentityException("Cannot find IdentityObjects", e);
+ }
+
+ if (criteria != null && criteria.isFiltered())
+ {
+ filterByAttributesValues(results, criteria.getValues());
+ if (criteria.isPaged())
+ {
+ results = (LinkedList)cutPageFromResults(results, criteria);
+ }
+ }
+
+ return results;
+ }
+
+ @SuppressWarnings("unchecked")
+ public org.hibernate.Query prepareIdentityObjectQuery(IdentityStoreInvocationContext ctx,
+ IdentityObject identity,
+ IdentityObjectRelationshipType relationshipType,
+ Collection<IdentityObjectType> excludes,
+ boolean parent,
+ IdentityObjectSearchCriteria criteria,
+ boolean count) throws IdentityException
+ {
//TODO:test
HibernateIdentityObject hibernateObject = safeGet(ctx, identity);
- List<IdentityObject> results;
-
boolean orderByName = false;
boolean ascending = true;
@@ -829,10 +924,12 @@
ascending = criteria.isAscending();
}
+ org.hibernate.Query q = null;
+
try
{
- org.hibernate.Query q = null;
+
StringBuilder hqlString = new StringBuilder("");
@@ -841,16 +938,24 @@
if (parent)
{
+ if (count)
+ {
+ hqlString.append("select count(distinct ior.toIdentityObject) from HibernateIdentityObjectRelationship ior where ");
+
+ }
+ else
+ {
+ hqlString.append("select distinct ior.toIdentityObject from HibernateIdentityObjectRelationship ior where ");
+ }
+
if (relationshipType != null)
{
- hqlString.append("select distinct ior.toIdentityObject from HibernateIdentityObjectRelationship ior where " +
- "ior.toIdentityObject.name like :nameFilter and ior.type.name = :relType and ior.fromIdentityObject = :identity");
+ hqlString.append("ior.toIdentityObject.name like :nameFilter and ior.type.name = :relType and ior.fromIdentityObject = :identity");
}
else
{
- hqlString.append("select distinct ior.toIdentityObject from HibernateIdentityObjectRelationship ior where " +
- "ior.toIdentityObject.name like :nameFilter and ior.fromIdentityObject = :identity");
+ hqlString.append("ior.toIdentityObject.name like :nameFilter and ior.fromIdentityObject = :identity");
}
if (excludes != null && excludes.size() > 0)
@@ -859,7 +964,7 @@
int i = 0;
for (IdentityObjectType exclude : excludes)
{
- hqlString.append(" and ior.toIdentityObject.identityType.name != ")
+ hqlString.append(" and ior.toIdentityObject.identityType.id <> ")
.append(":exclude" + i++);
}
}
@@ -875,15 +980,24 @@
}
else
{
+
+ if (count)
+ {
+ hqlString.append("select count(distinct ior.fromIdentityObject) from HibernateIdentityObjectRelationship ior where ");
+
+ }
+ else
+ {
+ hqlString.append("select distinct ior.fromIdentityObject from HibernateIdentityObjectRelationship ior where ");
+ }
+
if (relationshipType != null)
{
- hqlString.append("select distinct ior.fromIdentityObject from HibernateIdentityObjectRelationship ior where " +
- "ior.fromIdentityObject.name like :nameFilter and ior.type.name = :relType and ior.toIdentityObject = :identity");
+ hqlString.append("ior.fromIdentityObject.name like :nameFilter and ior.type.name = :relType and ior.toIdentityObject = :identity");
}
else
{
- hqlString.append("select distinct ior.fromIdentityObject from HibernateIdentityObjectRelationship ior where " +
- "ior.fromIdentityObject.name like :nameFilter and ior.toIdentityObject = :identity");
+ hqlString.append("ior.fromIdentityObject.name like :nameFilter and ior.toIdentityObject = :identity");
}
if (excludes != null && excludes.size() > 0)
@@ -891,7 +1005,7 @@
int i = 0;
for (IdentityObjectType exclude : excludes)
{
- hqlString.append(" and ior.fromIdentityObject.identityType.name != ")
+ hqlString.append(" and ior.fromIdentityObject.identityType.id <> ")
.append(":exclude" + i++);
}
}
@@ -930,7 +1044,8 @@
int i = 0;
for (IdentityObjectType exclude : excludes)
{
- q.setParameter("exclude" + i++, exclude.getName());
+ HibernateIdentityObjectType exType = getHibernateIdentityObjectType(ctx, exclude);
+ q.setParameter("exclude" + i++, exType.getId());
}
}
if (criteria != null && criteria.isPaged() && !criteria.isFiltered())
@@ -944,10 +1059,6 @@
q.setCacheable(true);
- results = q.list();
- Hibernate.initialize(results);
-
-
}
catch (Exception e)
{
@@ -956,19 +1067,10 @@
log.log(Level.FINER, "Exception occurred: ", e);
}
- throw new IdentityException("Cannot find IdentityObjects", e);
+ throw new IdentityException("Cannot prepare hibernate query", e);
}
- if (criteria != null && criteria.isFiltered())
- {
- filterByAttributesValues(results, criteria.getValues());
- if (criteria.isPaged())
- {
- results = (LinkedList)cutPageFromResults(results, criteria);
- }
- }
-
- return results;
+ return q;
}
public Collection<IdentityObject> findIdentityObject(IdentityStoreInvocationContext ctx,
@@ -1219,16 +1321,80 @@
return new HashSet<IdentityObjectRelationship>(results);
}
+
+ public int getRelationshipsCount(IdentityStoreInvocationContext ctx,
+ IdentityObject identity,
+ IdentityObjectRelationshipType type,
+ boolean parent,
+ boolean named,
+ String name,
+ IdentityObjectSearchCriteria searchCriteria) throws IdentityException
+ {
+
+ Criteria criteria = prepareResolveRelationshipsCriteria(
+ ctx,
+ identity,
+ type,
+ parent,
+ named,
+ name,
+ searchCriteria
+ );
+
+ criteria.setProjection(Projections.rowCount());
+
+ Iterator result = criteria.list().iterator();
+
+
+ if (!result.hasNext())
+ {
+ return 0;
+ }
+
+ return (Integer)result.next();
+
+ }
+
+
public Set<IdentityObjectRelationship> resolveRelationships(IdentityStoreInvocationContext ctx,
IdentityObject identity,
IdentityObjectRelationshipType type,
boolean parent,
boolean named,
- String name) throws IdentityException
+ String name,
+ IdentityObjectSearchCriteria searchCriteria) throws IdentityException
{
HibernateIdentityObject hio = safeGet(ctx, identity);
+ Criteria criteria = prepareResolveRelationshipsCriteria(
+ ctx,
+ identity,
+ type,
+ parent,
+ named,
+ name,
+ searchCriteria
+ );
+
+ List<HibernateIdentityObjectRelationship> results = criteria.list();
+
+ Hibernate.initialize(results);
+
+ return new HashSet<IdentityObjectRelationship>(results);
+ }
+
+ public Criteria prepareResolveRelationshipsCriteria(IdentityStoreInvocationContext ctx,
+ IdentityObject identity,
+ IdentityObjectRelationshipType type,
+ boolean parent,
+ boolean named,
+ String name,
+ IdentityObjectSearchCriteria searchCriteria) throws IdentityException
+ {
+ HibernateIdentityObject hio = safeGet(ctx, identity);
+
+
Criteria criteria = getHibernateSession(ctx).createCriteria(HibernateIdentityObjectRelationship.class);
criteria.setCacheable(true);
@@ -1264,11 +1430,17 @@
criteria.setFetchMode("fromIdentityObject", FetchMode.JOIN);
criteria.setFetchMode("toIdentityObject", FetchMode.JOIN);
- List<HibernateIdentityObjectRelationship> results = criteria.list();
+ if (searchCriteria != null && searchCriteria.isPaged() && !searchCriteria.isFiltered())
+ {
+ if (searchCriteria.getMaxResults() > 0)
+ {
+ criteria.setMaxResults(searchCriteria.getMaxResults());
+ }
+ criteria.setFirstResult(searchCriteria.getFirstResult());
+ }
- Hibernate.initialize(results);
- return new HashSet<IdentityObjectRelationship>(results);
+ return criteria;
}
public String createRelationshipName(IdentityStoreInvocationContext ctx, String name) throws IdentityException, OperationNotSupportedException
Modified: idm/trunk/picketlink-idm-ldap/src/main/java/org/picketlink/idm/impl/store/ldap/LDAPIdentityStoreImpl.java
===================================================================
--- idm/trunk/picketlink-idm-ldap/src/main/java/org/picketlink/idm/impl/store/ldap/LDAPIdentityStoreImpl.java 2011-09-12 12:09:55 UTC (rev 1213)
+++ idm/trunk/picketlink-idm-ldap/src/main/java/org/picketlink/idm/impl/store/ldap/LDAPIdentityStoreImpl.java 2011-09-12 21:25:13 UTC (rev 1214)
@@ -1120,7 +1120,37 @@
return findIdentityObject(invocationCxt, identity, relationshipType, null, parent, criteria);
}
+ public int getIdentityObjectCount(IdentityStoreInvocationContext invocationCxt,
+ IdentityObject identity,
+ IdentityObjectRelationshipType relationshipType,
+ boolean parent,
+ IdentityObjectSearchCriteria criteria) throws IdentityException
+ {
+ return findIdentityObject(
+ invocationCxt,
+ identity,
+ relationshipType,
+ parent,
+ criteria).size();
+ }
+ public int getIdentityObjectCount(IdentityStoreInvocationContext ctx,
+ IdentityObject identity,
+ IdentityObjectRelationshipType relationshipType,
+ Collection<IdentityObjectType> excludes,
+ boolean parent,
+ IdentityObjectSearchCriteria criteria) throws IdentityException
+ {
+ return findIdentityObject(
+ ctx,
+ identity,
+ relationshipType,
+ excludes,
+ parent,
+ criteria
+ ).size();
+ }
+
public Collection<IdentityObject> findIdentityObject(IdentityStoreInvocationContext ctx,
IdentityObject identity,
IdentityObjectRelationshipType relationshipType,
@@ -1531,12 +1561,18 @@
return objects;
}
+ public int getRelationshipsCount(IdentityStoreInvocationContext ctx, IdentityObject identity, IdentityObjectRelationshipType type, boolean parent, boolean named, String name, IdentityObjectSearchCriteria searchCriteria) throws IdentityException
+ {
+ return resolveRelationships(ctx, identity, type, parent, named, name, searchCriteria).size();
+ }
+
public Set<IdentityObjectRelationship> resolveRelationships(IdentityStoreInvocationContext ctx,
IdentityObject identity,
IdentityObjectRelationshipType type,
boolean parent,
boolean named,
- String name) throws IdentityException
+ String name,
+ IdentityObjectSearchCriteria searchCriteria) throws IdentityException
{
if (log.isLoggable(Level.FINER))
Modified: idm/trunk/picketlink-idm-spi/src/main/java/org/picketlink/idm/spi/store/IdentityStore.java
===================================================================
--- idm/trunk/picketlink-idm-spi/src/main/java/org/picketlink/idm/spi/store/IdentityStore.java 2011-09-12 12:09:55 UTC (rev 1213)
+++ idm/trunk/picketlink-idm-spi/src/main/java/org/picketlink/idm/spi/store/IdentityStore.java 2011-09-12 21:25:13 UTC (rev 1214)
@@ -182,12 +182,51 @@
* @throws IdentityException
*/
Collection<IdentityObject> findIdentityObject(IdentityStoreInvocationContext invocationCxt, IdentityObject identity,
- IdentityObjectRelationshipType relationshipType,
- Collection<IdentityObjectType> excludes,
- boolean parent,
- IdentityObjectSearchCriteria criteria) throws IdentityException;
+ IdentityObjectRelationshipType relationshipType,
+ Collection<IdentityObjectType> excludes,
+ boolean parent,
+ IdentityObjectSearchCriteria criteria) throws IdentityException;
+ /**
+ * Get count of relationship with given identity. Relationships are directional (from parent to child).
+ *
+ * @param invocationCxt
+ * @param identity
+ * @param relationshipType
+ * @param parent defines if given identity is parent or child side in the
+ * relationship - default is true (parent)
+ * @param criteria
+ * @return
+ * @throws IdentityException
+ */
+ int getIdentityObjectCount(IdentityStoreInvocationContext invocationCxt,
+ IdentityObject identity,
+ IdentityObjectRelationshipType relationshipType,
+ boolean parent,
+ IdentityObjectSearchCriteria criteria) throws IdentityException;
+
+ /**
+ * Get count of relationship with given identity. Relationships are directional (from parent to child).
+ *
+ * @param ctx
+ * @param identity
+ * @param relationshipType
+ * @param excludes list of object types to not include in search
+ * @param parent defines if given identity is parent or child side in the
+ * relationship - default is true (parent)
+ * @param criteria
+ * @return
+ * @throws IdentityException
+ */
+ public int getIdentityObjectCount(IdentityStoreInvocationContext ctx,
+ IdentityObject identity,
+ IdentityObjectRelationshipType relationshipType,
+ Collection<IdentityObjectType> excludes,
+ boolean parent,
+ IdentityObjectSearchCriteria criteria) throws IdentityException;
+
+
/**
* Create directional relationship of a given type between identities
*
@@ -252,12 +291,32 @@
/**
+ * Get count of relationships for a given IdentityObject. Relationships can be directional and parent switch defines which
+ * role identity play in it.
+ * @param ctx
+ * @param identity
+ * @param named
+ * @param name - can be null
+ * @param searchCriteria
+ * @return
+ * @throws IdentityException
+ */
+ public int getRelationshipsCount(IdentityStoreInvocationContext ctx,
+ IdentityObject identity,
+ IdentityObjectRelationshipType type,
+ boolean parent,
+ boolean named,
+ String name,
+ IdentityObjectSearchCriteria searchCriteria) throws IdentityException;
+
+ /**
* Resolve relationships for a given IdentityObject. Relationships can be directional and parent switch defines which
* role identity play in it.
* @param invocationCxt
* @param identity
* @param named
- * @param name - can be null
+ * @param name - can be null
+ * @param criteria
* @return
* @throws IdentityException
*/
@@ -266,10 +325,12 @@
IdentityObjectRelationshipType relationshipType,
boolean parent,
boolean named,
- String name)
+ String name,
+ IdentityObjectSearchCriteria criteria)
throws IdentityException;
+
// Named relationships
/**
Modified: idm/trunk/pom.xml
===================================================================
--- idm/trunk/pom.xml 2011-09-12 12:09:55 UTC (rev 1213)
+++ idm/trunk/pom.xml 2011-09-12 21:25:13 UTC (rev 1214)
@@ -35,13 +35,10 @@
<module>picketlink-idm-core</module>
<module>picketlink-idm-testsuite/common</module>
<module>picketlink-idm-hibernate</module>
- <module>picketlink-idm-jpa</module>
<module>picketlink-idm-ldap</module>
<module>picketlink-idm-cache</module>
<module>picketlink-idm-auth</module>
<module>picketlink-idm-testsuite/integration</module>
- <module>integration</module>
- <module>picketlink-idm-docs</module>
</modules>
</profile>
<profile>
@@ -57,7 +54,6 @@
<module>picketlink-idm-core</module>
<module>picketlink-idm-testsuite/common</module>
<module>picketlink-idm-hibernate</module>
- <module>picketlink-idm-jpa</module>
<module>picketlink-idm-ldap</module>
<module>picketlink-idm-cache</module>
<module>picketlink-idm-auth</module>
@@ -74,7 +70,6 @@
<module>picketlink-idm-core</module>
<module>picketlink-idm-testsuite/common</module>
<module>picketlink-idm-hibernate</module>
- <module>picketlink-idm-jpa</module>
<module>picketlink-idm-ldap</module>
<module>picketlink-idm-cache</module>
<module>picketlink-idm-auth</module>
@@ -107,7 +102,6 @@
<module>picketlink-idm-core</module>
<module>picketlink-idm-testsuite/common</module>
<module>picketlink-idm-hibernate</module>
- <module>picketlink-idm-jpa</module>
<module>picketlink-idm-ldap</module>
<module>picketlink-idm-cache</module>
<module>picketlink-idm-auth</module>