1   /**
2    * Copyright (c) 2000-2009 Liferay, Inc. All rights reserved.
3    *
4    * The contents of this file are subject to the terms of the Liferay Enterprise
5    * Subscription License ("License"). You may not use this file except in
6    * compliance with the License. You can obtain a copy of the License by
7    * contacting Liferay, Inc. See the License for the specific language governing
8    * permissions and limitations under the License, including but not limited to
9    * distribution rights of the Software.
10   *
11   * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
12   * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
13   * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
14   * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
15   * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
16   * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
17   * SOFTWARE.
18   */
19  
20  package com.liferay.portal.search.lucene.messaging;
21  
22  import com.liferay.portal.kernel.log.Log;
23  import com.liferay.portal.kernel.log.LogFactoryUtil;
24  import com.liferay.portal.kernel.messaging.Message;
25  import com.liferay.portal.kernel.messaging.MessageListener;
26  import com.liferay.portal.kernel.search.Document;
27  import com.liferay.portal.kernel.search.messaging.SearchRequest;
28  import com.liferay.portal.search.lucene.LuceneSearchEngineUtil;
29  
30  /**
31   * <a href="LuceneWriterMessageListener.java.html"><b><i>View Source</i></b></a>
32   *
33   * @author Bruno Farache
34   *
35   */
36  public class LuceneWriterMessageListener implements MessageListener {
37  
38      public void receive(Message message) {
39          try {
40              doReceive(message);
41          }
42          catch (Exception e) {
43              _log.error("Unable to process message " + message, e);
44          }
45      }
46  
47      public void doReceive(Message message) throws Exception {
48          Object payload = message.getPayload();
49  
50          if (!(payload instanceof SearchRequest)) {
51              return;
52          }
53  
54          SearchRequest searchRequest = (SearchRequest)payload;
55  
56          String command = searchRequest.getCommand();
57  
58          if (!LuceneSearchEngineUtil.isRegistered() &&
59              !command.equals(SearchRequest.COMMAND_REGISTER)) {
60  
61              return;
62          }
63  
64          long companyId = searchRequest.getCompanyId();
65          String id = searchRequest.getId();
66          Document doc = searchRequest.getDocument();
67  
68          if (command.equals(SearchRequest.COMMAND_ADD)) {
69              LuceneSearchEngineUtil.addDocument(companyId, doc);
70          }
71          else if (command.equals(SearchRequest.COMMAND_DELETE)) {
72              LuceneSearchEngineUtil.deleteDocument(companyId, id);
73          }
74          else if (command.equals(SearchRequest.COMMAND_DELETE_PORTLET_DOCS)) {
75              LuceneSearchEngineUtil.deletePortletDocuments(companyId, id);
76          }
77          else if (command.equals(SearchRequest.COMMAND_REGISTER)) {
78              LuceneSearchEngineUtil.register(id);
79          }
80          else if (command.equals(SearchRequest.COMMAND_UNREGISTER)) {
81              LuceneSearchEngineUtil.unregister(id);
82          }
83          else if (command.equals(SearchRequest.COMMAND_UPDATE)) {
84              LuceneSearchEngineUtil.updateDocument(companyId, id, doc);
85          }
86      }
87  
88      private static Log _log =
89          LogFactoryUtil.getLog(LuceneWriterMessageListener.class);
90  
91  }