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

Merge pull request #703 from mysema/i702

Browse Source 
Add Comparable support
This commit is contained in:
Timo Westkämper 2014-04-09 12:57:19 +03:00
commit e53b05cf71
1 changed files with 52 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

65
querydsl-core/src/main/java/com/mysema/query/types/expr/CaseBuilder.java
View File

@ -13,18 +13,17 @@
*/
package com.mysema.query.types.expr;
import java.sql.Time;
import java.sql.Timestamp;
import java.util.ArrayList;
import java.util.List;
import javax.annotation.Nullable;
import com.mysema.query.types.ConstantImpl;
import com.mysema.query.types.Expression;
import com.mysema.query.types.NullExpression;
import com.mysema.query.types.Ops;
import javax.annotation.Nullable;
import java.sql.Time;
import java.sql.Timestamp;
import java.util.ArrayList;
import java.util.List;
/**
* CaseBuilder enables the construction of typesafe case-when-then-else
* constructs :
@ -164,8 +163,28 @@ public final class CaseBuilder {
this.when = b;
}
@SuppressWarnings("unchecked")
public <A> Cases<A, SimpleExpression<A>> then(Expression<A> expr) {
if (expr instanceof BooleanExpression) {
return (Cases) then((BooleanExpression) expr);
} else if (expr instanceof StringExpression) {
return (Cases) then((StringExpression) expr);
} else if (expr instanceof NumberExpression) {
return then((NumberExpression) expr);
} else if (expr instanceof DateExpression) {
return then((DateExpression) expr);
} else if (expr instanceof DateTimeExpression) {
return then((DateTimeExpression) expr);
} else if (expr instanceof TimeExpression) {
return then((TimeExpression) expr);
} else if (expr instanceof ComparableExpression) {
return then((ComparableExpression) expr);
} else {
return thenSimple(expr);
}
}
@SuppressWarnings("unchecked")
private <A> Cases<A, SimpleExpression<A>> thenSimple(Expression<A> expr) {
return new Cases<A, SimpleExpression<A>>((Class)expr.getType()) {
@Override
protected SimpleExpression<A> createResult(Class<A> type, Expression<A> last) {
@ -176,12 +195,12 @@ public final class CaseBuilder {
}
public <A> Cases<A, SimpleExpression<A>> then(A constant) {
return then(ConstantImpl.create(constant));
return thenSimple(ConstantImpl.create(constant));
}
// Boolean
public Cases<Boolean,BooleanExpression> then(BooleanExpression expr) {
public Cases<Boolean, BooleanExpression> then(BooleanExpression expr) {
return thenBoolean(expr);
}
@ -200,6 +219,26 @@ public final class CaseBuilder {
return thenBoolean(ConstantImpl.create(b));
}
// Comparable
public <T extends Comparable> Cases<T, ComparableExpression<T>> then(ComparableExpression<T> expr) {
return thenComparable(expr);
}
private <T extends Comparable> Cases<T, ComparableExpression<T>> thenComparable(Expression<T> expr) {
return new Cases<T, ComparableExpression<T>>((Class)expr.getType()) {
@Override
protected ComparableExpression<T> createResult(Class<T> type, Expression<T> last) {
return ComparableOperation.create(type, Ops.CASE, last);
}
}.addCase(when, expr);
}
public <A extends Comparable> Cases<A, ComparableExpression<A>> then(A arg) {
return thenComparable(ConstantImpl.create(arg));
}
// Date
public <T extends Comparable> Cases<T, DateExpression<T>> then(DateExpression<T> expr) {
@ -216,7 +255,7 @@ public final class CaseBuilder {
}.addCase(when, expr);
}
public Cases<java.sql.Date, DateExpression<java.sql.Date>> thenDate(java.sql.Date date) {
public Cases<java.sql.Date, DateExpression<java.sql.Date>> then(java.sql.Date date) {
return thenDate(ConstantImpl.create(date));
}
@ -236,11 +275,11 @@ public final class CaseBuilder {
}.addCase(when, expr);
}
public Cases<Timestamp, DateTimeExpression<Timestamp>> thenDateTime(Timestamp ts) {
public Cases<Timestamp, DateTimeExpression<Timestamp>> then(Timestamp ts) {
return thenDateTime(ConstantImpl.create(ts));
}
public Cases<java.util.Date, DateTimeExpression<java.util.Date>> thenDateTime(java.util.Date date) {
public Cases<java.util.Date, DateTimeExpression<java.util.Date>> then(java.util.Date date) {
return thenDateTime(ConstantImpl.create(date));
}