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

added support for in-operation

Browse Source
This commit is contained in:
Timo Westkämper 2010-04-05 11:16:59 +00:00
parent bee87d573d
commit 2677d957e7
1 changed files with 20 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
querydsl-lucene/src/main/java/com/mysema/query/lucene/LuceneSerializer.java
View File

@ -6,6 +6,7 @@
package com.mysema.query.lucene;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Locale;
@ -74,6 +75,8 @@ public class LuceneSerializer {
return stringContains(operation);
} else if (op == Ops.BETWEEN) {
return between(operation);
} else if (op == Ops.IN){
return in(operation);
}
throw new UnsupportedOperationException("Illegal operation " + operation);
}
@ -106,6 +109,10 @@ public class LuceneSerializer {
verifyArguments(operation);
String field = toField(operation.getArg(0));
String[] terms = createTerms(operation.getArg(1));
return eq(field, terms);
}
private Query eq(String field, String[] terms){
if (terms.length > 1) {
PhraseQuery pq = new PhraseQuery();
for (String s : terms) {
@ -113,9 +120,21 @@ public class LuceneSerializer {
}
return pq;
}
return new TermQuery(new Term(field, normalize(terms[0])));
return new TermQuery(new Term(field, normalize(terms[0])));
}
@SuppressWarnings("unchecked")
private Query in(Operation<?> operation){
String field = toField(operation.getArg(0));
Collection values = (Collection) ((Constant)operation.getArg(1)).getConstant();
BooleanQuery bq = new BooleanQuery();
for (Object value : values){
bq.add(eq(field, StringUtils.split(value.toString())), Occur.SHOULD);
}
return bq;
}
private Query ne(Operation<?> operation) {
BooleanQuery bq = new BooleanQuery();
bq.add(new BooleanClause(eq(operation), Occur.MUST_NOT));