1.1 restoration
[mir.git] / source / mircoders / producer / IndexingProducerNode.java
index 685a2c1..db12ed7 100755 (executable)
@@ -34,6 +34,7 @@ import java.util.Calendar;
 import java.util.Date;
 import java.util.GregorianCalendar;
 import java.util.Map;
+import java.io.File;
 
 import mir.entity.Entity;
 import mir.entity.adapter.EntityAdapter;
@@ -42,6 +43,7 @@ import mir.misc.StringUtil;
 import mir.producer.ProducerFailure;
 import mir.producer.ProducerNode;
 import mir.util.ParameterExpander;
+import mir.util.FileFunctions;
 import mircoders.entity.EntityContent;
 import mircoders.search.AudioSearchTerm;
 import mircoders.search.ContentSearchTerm;
@@ -63,10 +65,12 @@ import org.apache.lucene.store.FSDirectory;
 public class IndexingProducerNode implements ProducerNode {
   private String contentKey;
   private String indexPath;
+  private File indexBasePath;
 
-  public IndexingProducerNode(String aContentKey, String pathToIndex) {
+  public IndexingProducerNode(File anIndexBasePath, String aContentKey, String pathToIndex) {
     contentKey = aContentKey;
     indexPath = pathToIndex;
+    indexBasePath = anIndexBasePath;
   }
 
   public void produce(Map aValueMap, String aVerb, LoggerWrapper aLogger)
@@ -74,14 +78,14 @@ public class IndexingProducerNode implements ProducerNode {
     IndexWriter indexWriter = null;
     Object data;
     Entity entity;
-    String index = null;
     long startTime;
     long endTime;
+    File indexFile = null;
 
     startTime = System.currentTimeMillis();
 
     try {
-      index = ParameterExpander.expandExpression(aValueMap, indexPath);
+      indexFile = FileFunctions.getAbsoluteOrRelativeFile(indexBasePath, ParameterExpander.expandExpression(aValueMap, indexPath));
       data = ParameterExpander.findValueForKey(aValueMap, contentKey);
 
       if (!(data instanceof EntityAdapter)) {
@@ -98,22 +102,19 @@ public class IndexingProducerNode implements ProducerNode {
           entity.getClass().getName() + " adapter", null);
       }
 
-      aLogger.info("Indexing " + (String) entity.getFieldValue("id") + " into " +
-        index);
+      aLogger.info("Indexing " + entity.getFieldValue("id") + " into " +  indexFile.getAbsolutePath());
 
       // create an index here if one did not already exist
-      if (!(IndexReader.indexExists(index))) {
-        aLogger.error("Didn't find existing index, so I'm making one in " +
-          index);
+      if (!(IndexReader.indexExists(indexFile))) {
+        aLogger.error("Didn't find existing index, so I'm making one in " + indexFile.getAbsolutePath());
 
-        IndexWriter indexCreator =
-          new IndexWriter(index, new StandardAnalyzer(), true);
+        IndexWriter indexCreator = new IndexWriter(indexFile, new StandardAnalyzer(), true);
         indexCreator.close();
       }
 
-      IndexUtil.unindexEntity((EntityContent) entity, index);
+      IndexUtil.unindexEntity((EntityContent) entity, indexFile);
 
-      indexWriter = new IndexWriter(index, new StandardAnalyzer(), false);
+      indexWriter = new IndexWriter(indexFile, new StandardAnalyzer(), false);
 
       Document theDoc = new Document();
 
@@ -136,21 +137,21 @@ public class IndexingProducerNode implements ProducerNode {
 
       if (textValue!=null) {
         try {
-          year = Integer.parseInt(textValue.substring(0,4));
-          month = Integer.parseInt(textValue.substring(5,7));
-          day = Integer.parseInt(textValue.substring(8,10));
-         hours = Integer.parseInt(textValue.substring(11,13));
-          minutes = Integer.parseInt(textValue.substring(14,16));
+          year = Integer.parseInt(textValue.substring(0, 4));
+          month = Integer.parseInt(textValue.substring(5, 7));
+          day = Integer.parseInt(textValue.substring(8, 10));
+          hours = Integer.parseInt(textValue.substring(11, 13));
+          minutes = Integer.parseInt(textValue.substring(14, 16));
 
-          calendar.set(year, month-1, day, hours, minutes);
+          calendar.set(year, month - 1, day, hours, minutes);
           date = calendar.getTime();
-         SimpleDateFormat formatter = new SimpleDateFormat ("yyyy.MM.dd hh:mm");
-         formattedDate=formatter.format(date);
+          SimpleDateFormat formatter = new SimpleDateFormat("yyyy.MM.dd hh:mm");
+          formattedDate = formatter.format(date);
+        }
+        catch (Throwable t) {
+          aLogger.error("Error while generating content date to index: " + t.getMessage());
+          t.printStackTrace(aLogger.asPrintWriter(LoggerWrapper.DEBUG_MESSAGE));
         }
-       catch (Throwable t){
-         aLogger.error("Error while generating content date to index: " + t.getMessage());
-         t.printStackTrace(aLogger.asPrintWriter(LoggerWrapper.DEBUG_MESSAGE));
-       }
       }
       (new KeywordSearchTerm("webdb_create_formatted", "search_date",
         "webdb_create_formatted", "webdb_create_formatted",
@@ -205,19 +206,23 @@ public class IndexingProducerNode implements ProducerNode {
       if (indexWriter != null) {
         try {
           indexWriter.close();
-        } catch (Throwable t) {
+        }
+        catch (Throwable t) {
           aLogger.warn("Error while closing indexWriter: " + t.getMessage());
         }
       }
 
-      try {
-        FSDirectory theIndexDir = FSDirectory.getDirectory(index, false);
+      if (indexFile!=null) {
+        try {
+          FSDirectory theIndexDir = FSDirectory.getDirectory(indexFile, false);
 
-        if (IndexReader.isLocked(theIndexDir)) {
-          IndexReader.unlock(theIndexDir);
+          if (IndexReader.isLocked(theIndexDir)) {
+            IndexReader.unlock(theIndexDir);
+          }
+        }
+        catch (Throwable t) {
+          aLogger.warn("Error while unlocking index: " + t.getMessage());
         }
-      } catch (Throwable t) {
-        aLogger.warn("Error while unlocking index: " + t.getMessage());
       }
     }