KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > apache > roller > business > search > operations > RebuildWebsiteIndexOperation


1 /*
2  * Licensed to the Apache Software Foundation (ASF) under one or more
3  * contributor license agreements. The ASF licenses this file to You
4  * under the Apache License, Version 2.0 (the "License"); you may not
5  * use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License. For additional information regarding
15  * copyright in this work, please see the NOTICE file in the top level
16  * directory of this distribution.
17  */
18 /* Created on Jul 16, 2003 */
19 package org.apache.roller.business.search.operations;
20
21 import java.io.IOException JavaDoc;
22 import java.text.MessageFormat JavaDoc;
23 import java.util.Date JavaDoc;
24 import java.util.Iterator JavaDoc;
25 import java.util.List JavaDoc;
26
27 import org.apache.commons.logging.Log;
28 import org.apache.commons.logging.LogFactory;
29 import org.apache.lucene.index.IndexReader;
30 import org.apache.lucene.index.IndexWriter;
31 import org.apache.lucene.index.Term;
32 import org.apache.roller.RollerException;
33 import org.apache.roller.business.IndexManagerImpl;
34 import org.apache.roller.business.search.FieldConstants;
35 import org.apache.roller.business.search.IndexUtil;
36 import org.apache.roller.model.Roller;
37 import org.apache.roller.model.RollerFactory;
38 import org.apache.roller.model.WeblogManager;
39 import org.apache.roller.pojos.WeblogEntryData;
40 import org.apache.roller.pojos.WebsiteData;
41
42 /**
43  * An index operation that rebuilds a given users index (or all indexes).
44  * @author Mindaugas Idzelis (min@idzelis.com)
45  */
46 public class RebuildWebsiteIndexOperation extends WriteToIndexOperation {
47     
48     //~ Static fields/initializers =============================================
49
50     private static Log mLogger =
51         LogFactory.getFactory().getInstance(RebuildWebsiteIndexOperation.class);
52     
53     //~ Instance fields ========================================================
54
55     private WebsiteData website;
56     
57     //~ Constructors ===========================================================
58
59     /**
60      * Create a new operation that will recreate an index.
61      *
62      * @param website The website to rebuild the index for, or null for all users.
63      */
64     public RebuildWebsiteIndexOperation(IndexManagerImpl mgr, WebsiteData website) {
65         super(mgr);
66         this.website = website;
67     }
68     
69     //~ Methods ================================================================
70
71     public void doRun() {
72         Date JavaDoc start = new Date JavaDoc();
73         
74         IndexReader reader = beginDeleting();
75         
76         try {
77             if (reader != null) {
78                 Term tWebsite = null;
79                 if (website != null) {
80                     tWebsite = IndexUtil.getTerm(
81                             FieldConstants.WEBSITE_HANDLE, website.getHandle());
82                 }
83                 if (tWebsite != null) {
84                     reader.delete(tWebsite);
85                 } else {
86                     Term all =
87                             IndexUtil.getTerm(FieldConstants.CONSTANT,
88                             FieldConstants.CONSTANT_V);
89                     reader.delete(all);
90                 }
91             }
92         } catch (IOException JavaDoc e) {
93             mLogger.info("Problems deleting doc from index", e);
94         } finally {
95             endDeleting();
96         }
97         
98         IndexWriter writer = beginWriting();
99         
100         Roller roller = RollerFactory.getRoller();
101         try {
102             if (writer != null) {
103                 WeblogManager weblogManager = roller.getWeblogManager();
104                 
105                 List JavaDoc entries = weblogManager .getWeblogEntries(
106                         website, // website
107 null,
108                         null, // startDate
109 new Date JavaDoc(), // endDate (don't index 'future' entries)
110 null, // catName
111 WeblogEntryData.PUBLISHED, // status
112 null, // sortby (null means pubTime)
113 null, 0, -1); // offset, length, locale
114
115                 for (Iterator JavaDoc wbItr = entries.iterator(); wbItr.hasNext();) {
116                     WeblogEntryData entry = (WeblogEntryData) wbItr.next();
117                     writer.addDocument(getDocument(entry));
118                     mLogger.debug(
119                             MessageFormat.format("Indexed entry {0}: {1}",
120                             new Object JavaDoc[] {entry.getPubTime(), entry.getAnchor()}));
121                 }
122                 // release the database connection
123 roller.release();
124             }
125         } catch (Exception JavaDoc e) {
126             mLogger.error("ERROR adding doc to index", e);
127         } finally {
128             endWriting();
129             if (roller != null) roller.release();
130         }
131         
132         Date JavaDoc end = new Date JavaDoc();
133         double length = (end.getTime() - start.getTime()) / (double) 1000;
134         
135         if (website == null) {
136             mLogger.info(
137                     "Completed rebuilding index for all users in '" + length + "' secs");
138         } else {
139             mLogger.info("Completed rebuilding index for website handle: '" +
140                     website.getHandle() + "' in '" + length + "' seconds");
141         }
142     }
143 }
144
Popular Tags