001    /*
002     * Licensed to the Apache Software Foundation (ASF) under one or more
003     * contributor license agreements.  See the NOTICE file distributed with
004     * this work for additional information regarding copyright ownership.
005     * The ASF licenses this file to You under the Apache License, Version 2.0
006     * (the "License"); you may not use this file except in compliance with
007     * the License.  You may obtain a copy of the License at
008     *
009     *      http://www.apache.org/licenses/LICENSE-2.0
010     *
011     * Unless required by applicable law or agreed to in writing, software
012     * distributed under the License is distributed on an "AS IS" BASIS,
013     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014     * See the License for the specific language governing permissions and
015     * limitations under the License.
016     */
017    package org.apache.servicemix.jbi.audit.lucene;
018    
019    import java.io.IOException;
020    
021    import org.apache.lucene.analysis.standard.StandardAnalyzer;
022    import org.apache.lucene.document.Document;
023    import org.apache.lucene.queryParser.ParseException;
024    import org.apache.lucene.queryParser.QueryParser;
025    import org.apache.lucene.search.Hits;
026    import org.apache.lucene.search.IndexSearcher;
027    import org.apache.lucene.search.Query;
028    
029    /**
030     * Default Lucene Callback implementation. Used on LuceneAuditor
031     * 
032     * @author George Gastaldi (gastaldi)
033     * @since 2.1
034     * @version $Revision: 34165 $
035     */
036    public class DefaultLuceneCallback implements LuceneCallback {
037        private String field;
038    
039        private String query;
040    
041        public DefaultLuceneCallback(String field, String query) {
042            this.field = field;
043            this.query = query;
044        }
045    
046        public Object doCallback(IndexSearcher is) throws IOException {
047            try {
048                Query queryObj = QueryParser.parse(query, field, new StandardAnalyzer());
049                Hits hits = is.search(queryObj);
050                int total = hits.length();
051                String[] ids = new String[total];
052                for (int i = 0; i < total; i++) {
053                    Document d = hits.doc(i);
054                    ids[i] = d.get("org.apache.servicemix.exchangeid");
055                }
056                return ids;
057            } catch (ParseException pe) {
058                return new String[0];
059            }
060        }
061    
062    }