CodeCow/querydsl
1
0
Fork 0
mirror of https://github.com/querydsl/querydsl.git synced 2026-06-27 21:01:15 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

#799254 : improved query instance type check

Browse Source
This commit is contained in:
Timo Westkämper 2011-07-01 12:58:26 +00:00
parent d0ee87c943
commit f60f91e3b2
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
querydsl-jpa/src/main/java/com/mysema/query/jpa/impl/AbstractJPAQuery.java
View File

@ -49,7 +49,9 @@ public abstract class AbstractJPAQuery<Q extends AbstractJPAQuery<Q>> extends JP
@Nullable
private LockModeType lockMode;
private Class<?> hibernateQueryClass;
public AbstractJPAQuery(EntityManager em) {
this(new DefaultSessionHolder(em), HQLTemplates.DEFAULT, new DefaultQueryMetadata());
}
@ -57,6 +59,11 @@ public abstract class AbstractJPAQuery<Q extends AbstractJPAQuery<Q>> extends JP
public AbstractJPAQuery(JPASessionHolder sessionHolder, JPQLTemplates patterns, QueryMetadata metadata) {
super(metadata, patterns);
this.sessionHolder = sessionHolder;
try {
this.hibernateQueryClass = Class.forName("org.hibernate.ejb.HibernateQuery");
} catch (ClassNotFoundException e) {
// do nothing
}
}
public long count() {
@ -132,7 +139,8 @@ public abstract class AbstractJPAQuery<Q extends AbstractJPAQuery<Q>> extends JP
if (projection.size() == 1){
Expression<?> expr = projection.get(0);
if (expr instanceof FactoryExpression<?>){
if (query.getClass().getName().startsWith("org.hibernate")){
// if (query.getClass().getName().startsWith("org.hibernate")){
if (hibernateQueryClass != null && hibernateQueryClass.isInstance(query)) {
try {
Class<?> cl = Class.forName("com.mysema.query.jpa.impl.JPAQueryTransformerTask");
cl.getConstructor(Query.class, FactoryExpression.class).newInstance(query, expr);