public CustomLoader(ICustomQuery customQuery, ISessionFactoryImplementor factory) : base(factory)
{
sql = customQuery.SQL;
querySpaces.UnionWith(customQuery.QuerySpaces);
if (querySpaces.Count > 0)
{
customPersisters = factory
.GetEntityPersisters(querySpaces).OfType <IPersister>()
.Concat(factory.GetCollectionPersisters(querySpaces).OfType <IPersister>())
.ToArray();
supportsQueryCache = customPersisters.All(x => x.SupportsQueryCache);
}
parametersSpecifications = customQuery.CollectedParametersSpecifications.ToList();
List <IQueryable> entitypersisters = new List <IQueryable>();
List <int> entityowners = new List <int>();
List <IEntityAliases> entityaliases = new List <IEntityAliases>();
List <IQueryableCollection> collectionpersisters = new List <IQueryableCollection>();
List <int> collectionowners = new List <int>();
List <ICollectionAliases> collectionaliases = new List <ICollectionAliases>();
List <LockMode> lockmodes = new List <LockMode>();
List <IResultColumnProcessor> resultColumnProcessors = new List <IResultColumnProcessor>();
List <IReturn> nonScalarReturnList = new List <IReturn>();
List <IType> resulttypes = new List <IType>();
List <string> specifiedAliases = new List <string>();
List <bool> includeInResultRowList = new List <bool>();
int returnableCounter = 0;
bool hasScalars = false;
foreach (IReturn rtn in customQuery.CustomQueryReturns)
{
if (rtn is ScalarReturn)
{
ScalarReturn scalarRtn = (ScalarReturn)rtn;
resulttypes.Add(scalarRtn.Type);
specifiedAliases.Add(scalarRtn.ColumnAlias);
resultColumnProcessors.Add(new ScalarResultColumnProcessor(scalarRtn.ColumnAlias, scalarRtn.Type));
includeInResultRowList.Add(true);
hasScalars = true;
}
else if (rtn is RootReturn)
{
RootReturn rootRtn = (RootReturn)rtn;
IQueryable persister = (IQueryable)factory.GetEntityPersister(rootRtn.EntityName);
entitypersisters.Add(persister);
lockmodes.Add(rootRtn.LockMode);
resultColumnProcessors.Add(new NonScalarResultColumnProcessor(returnableCounter++));
nonScalarReturnList.Add(rtn);
entityowners.Add(-1);
resulttypes.Add(persister.Type);
specifiedAliases.Add(rootRtn.Alias);
entityaliases.Add(rootRtn.EntityAliases);
querySpaces.UnionWith(persister.QuerySpaces);
// 6.0 TODO: Use IPersister.SupportsQueryCache property once IPersister's todo is done.
supportsQueryCache = supportsQueryCache && persister.SupportsQueryCache();
includeInResultRowList.Add(true);
}
else if (rtn is CollectionReturn)
{
CollectionReturn collRtn = (CollectionReturn)rtn;
string role = collRtn.OwnerEntityName + "." + collRtn.OwnerProperty;
IQueryableCollection persister = (IQueryableCollection)factory.GetCollectionPersister(role);
collectionpersisters.Add(persister);
lockmodes.Add(collRtn.LockMode);
resultColumnProcessors.Add(new NonScalarResultColumnProcessor(returnableCounter++));
nonScalarReturnList.Add(rtn);
collectionowners.Add(-1);
resulttypes.Add(persister.Type);
specifiedAliases.Add(collRtn.Alias);
collectionaliases.Add(collRtn.CollectionAliases);
// determine if the collection elements are entities...
IType elementType = persister.ElementType;
if (elementType.IsEntityType)
{
IQueryable elementPersister = (IQueryable)((EntityType)elementType).GetAssociatedJoinable(factory);
entitypersisters.Add(elementPersister);
entityowners.Add(-1);
entityaliases.Add(collRtn.ElementEntityAliases);
querySpaces.UnionWith(elementPersister.QuerySpaces);
// 6.0 TODO: Use IPersister.SupportsQueryCache property once IPersister's todo is done.
supportsQueryCache = supportsQueryCache && elementPersister.SupportsQueryCache();
}
includeInResultRowList.Add(true);
}
else if (rtn is EntityFetchReturn)
{
EntityFetchReturn fetchRtn = (EntityFetchReturn)rtn;
NonScalarReturn ownerDescriptor = fetchRtn.Owner;
int ownerIndex = nonScalarReturnList.IndexOf(ownerDescriptor);
entityowners.Add(ownerIndex);
lockmodes.Add(fetchRtn.LockMode);
IQueryable ownerPersister = DetermineAppropriateOwnerPersister(ownerDescriptor);
EntityType fetchedType = (EntityType)ownerPersister.GetPropertyType(fetchRtn.OwnerProperty);
string entityName = fetchedType.GetAssociatedEntityName(Factory);
IQueryable persister = (IQueryable)factory.GetEntityPersister(entityName);
entitypersisters.Add(persister);
nonScalarReturnList.Add(rtn);
specifiedAliases.Add(fetchRtn.Alias);
entityaliases.Add(fetchRtn.EntityAliases);
querySpaces.UnionWith(persister.QuerySpaces);
// 6.0 TODO: Use IPersister.SupportsQueryCache property once IPersister's todo is done.
supportsQueryCache = supportsQueryCache && persister.SupportsQueryCache();
includeInResultRowList.Add(false);
}
else if (rtn is CollectionFetchReturn)
{
CollectionFetchReturn fetchRtn = (CollectionFetchReturn)rtn;
NonScalarReturn ownerDescriptor = fetchRtn.Owner;
int ownerIndex = nonScalarReturnList.IndexOf(ownerDescriptor);
collectionowners.Add(ownerIndex);
lockmodes.Add(fetchRtn.LockMode);
IQueryable ownerPersister = DetermineAppropriateOwnerPersister(ownerDescriptor);
string role = ownerPersister.EntityName + '.' + fetchRtn.OwnerProperty;
IQueryableCollection persister = (IQueryableCollection)factory.GetCollectionPersister(role);
collectionpersisters.Add(persister);
nonScalarReturnList.Add(rtn);
specifiedAliases.Add(fetchRtn.Alias);
collectionaliases.Add(fetchRtn.CollectionAliases);
// determine if the collection elements are entities...
IType elementType = persister.ElementType;
if (elementType.IsEntityType)
{
IQueryable elementPersister = (IQueryable)((EntityType)elementType).GetAssociatedJoinable(factory);
entitypersisters.Add(elementPersister);
entityowners.Add(ownerIndex);
entityaliases.Add(fetchRtn.ElementEntityAliases);
querySpaces.UnionWith(elementPersister.QuerySpaces);
// 6.0 TODO: Use IPersister.SupportsQueryCache property once IPersister's todo is done.
supportsQueryCache = supportsQueryCache && elementPersister.SupportsQueryCache();
}
includeInResultRowList.Add(false);
}
else
{
throw new HibernateException("unexpected custom query return type : " + rtn.GetType().FullName);
}
}
entityPersisters = entitypersisters.ToArray();
entityOwners = entityowners.ToArray();
entityAliases = entityaliases.ToArray();
collectionPersisters = collectionpersisters.ToArray();
collectionOwners = collectionowners.ToArray();
collectionAliases = collectionaliases.ToArray();
lockModes = lockmodes.ToArray();
ResultTypes = resulttypes.ToArray();
transformerAliases = specifiedAliases.ToArray();
rowProcessor = new ResultRowProcessor(hasScalars, resultColumnProcessors.ToArray());
includeInResultRow = includeInResultRowList.ToArray();
ResultRowAliases = transformerAliases.Where((a, i) => includeInResultRowList[i]).ToArray();
}
