1 package org.apache.lucene.queryParser.standard.builders;
4 * Licensed to the Apache Software Foundation (ASF) under one or more
5 * contributor license agreements. See the NOTICE file distributed with
6 * this work for additional information regarding copyright ownership.
7 * The ASF licenses this file to You under the Apache License, Version 2.0
8 * (the "License"); you may not use this file except in compliance with
9 * the License. You may obtain a copy of the License at
11 * http://www.apache.org/licenses/LICENSE-2.0
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS,
15 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
20 import java.util.LinkedList;
21 import java.util.List;
22 import java.util.TreeMap;
24 import org.apache.lucene.index.Term;
25 import org.apache.lucene.queryParser.core.QueryNodeException;
26 import org.apache.lucene.queryParser.core.builders.QueryTreeBuilder;
27 import org.apache.lucene.queryParser.core.nodes.FieldQueryNode;
28 import org.apache.lucene.queryParser.core.nodes.QueryNode;
29 import org.apache.lucene.queryParser.standard.nodes.MultiPhraseQueryNode;
30 import org.apache.lucene.search.MultiPhraseQuery;
31 import org.apache.lucene.search.TermQuery;
34 * Builds a {@link MultiPhraseQuery} object from a {@link MultiPhraseQueryNode}
37 public class MultiPhraseQueryNodeBuilder implements StandardQueryBuilder {
39 public MultiPhraseQueryNodeBuilder() {
43 public MultiPhraseQuery build(QueryNode queryNode) throws QueryNodeException {
44 MultiPhraseQueryNode phraseNode = (MultiPhraseQueryNode) queryNode;
46 MultiPhraseQuery phraseQuery = new MultiPhraseQuery();
48 List<QueryNode> children = phraseNode.getChildren();
50 if (children != null) {
51 TreeMap<Integer, List<Term>> positionTermMap = new TreeMap<Integer, List<Term>>();
53 for (QueryNode child : children) {
54 FieldQueryNode termNode = (FieldQueryNode) child;
55 TermQuery termQuery = (TermQuery) termNode
56 .getTag(QueryTreeBuilder.QUERY_TREE_BUILDER_TAGID);
57 List<Term> termList = positionTermMap.get(termNode
58 .getPositionIncrement());
60 if (termList == null) {
61 termList = new LinkedList<Term>();
62 positionTermMap.put(termNode.getPositionIncrement(), termList);
66 termList.add(termQuery.getTerm());
70 for (int positionIncrement : positionTermMap.keySet()) {
71 List<Term> termList = positionTermMap.get(positionIncrement);
73 phraseQuery.add(termList.toArray(new Term[termList.size()]),