From 248469d61eeac3c6c90e9e562ff677baf04a9091 Mon Sep 17 00:00:00 2001
From: Mokaddem <mokaddem.sami@gmail.com>
Date: Tue, 14 Mar 2017 10:37:31 +0100
Subject: [PATCH] Indexer now create an index_dir when it became too large
 (search in index not updated yet)

---
 bin/Indexer.py | 58 ++++++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 51 insertions(+), 7 deletions(-)

diff --git a/bin/Indexer.py b/bin/Indexer.py
index 39f0fe51..70ca598a 100755
--- a/bin/Indexer.py
+++ b/bin/Indexer.py
@@ -16,9 +16,24 @@ from pubsublogger import publisher
 from whoosh.index import create_in, exists_in, open_dir
 from whoosh.fields import Schema, TEXT, ID
 import os
+from os.path import join, getsize
 
 from Helper import Process
 
+# Config variable
+INDEX_SIZE_THRESHOLD = 500 #Mb
+TIME_WAIT = 1.0 #sec
+
+# return in bytes
+def check_index_size(indexnum):
+    global baseindexpath
+    the_index_name = "index_"+str(indexnum) if indexnum != 0 else "old_index"
+    the_index_name = os.path.join(baseindexpath, the_index_name)
+    cur_sum = 0
+    for root, dirs, files in os.walk(the_index_name):
+        cur_sum += sum(getsize(join(root, name)) for name in files)
+    return cur_sum
+
 
 if __name__ == "__main__":
     publisher.port = 6380
@@ -29,19 +44,36 @@ if __name__ == "__main__":
     p = Process(config_section)
 
     # Indexer configuration - index dir and schema setup
-    indexpath = os.path.join(os.environ['AIL_HOME'],
+    baseindexpath = os.path.join(os.environ['AIL_HOME'],
                              p.config.get("Indexer", "path"))
+    indexRegister_path = os.path.join(os.environ['AIL_HOME'], 
+                             p.config.get("Indexer", "register"))
     indexertype = p.config.get("Indexer", "type")
     if indexertype == "whoosh":
         schema = Schema(title=TEXT(stored=True), path=ID(stored=True,
                                                          unique=True),
                         content=TEXT)
-        if not os.path.exists(indexpath):
-            os.mkdir(indexpath)
-        if not exists_in(indexpath):
-            ix = create_in(indexpath, schema)
-        else:
-            ix = open_dir(indexpath)
+        if not os.path.exists(baseindexpath):
+            os.mkdir(baseindexpath)
+
+        # create the index register if not present
+        if not os.path.isfile(indexRegister_path):
+            with open(indexRegister_path, 'w') as f:
+                f.write("1")
+
+        with open(indexRegister_path, "r") as f:
+            allIndex = f.read()
+            allIndex = allIndex.split(',')
+            allIndex.sort()
+            indexnum = int(allIndex[-1])
+
+            indexpath = os.path.join(baseindexpath, "index_"+str(indexnum))
+            if not exists_in(indexpath):
+                ix = create_in(indexpath, schema)
+            else:
+                ix = open_dir(indexpath)
+ 
+        last_refresh = time.time()
 
     # LOGGING #
     publisher.info("ZMQ Indexer is Running")
@@ -59,6 +91,18 @@ if __name__ == "__main__":
             docpath = message.split(" ", -1)[-1]
             paste = PST.get_p_content()
             print "Indexing :", docpath
+
+
+            if time.time() - last_refresh > TIME_WAIT: #avoid calculating the index's size at each message
+                last_refresh = time.time()
+                if check_index_size(indexnum) > INDEX_SIZE_THRESHOLD*(1000*1000):
+                    indexpath = os.path.join(baseindexpath, "index_"+str(indexnum+1))
+                    ix = create_in(indexpath, schema, indexname=str(indexnum+1))
+                    ## Correctly handle the file
+                    with open(indexRegister_path, "a") as f:
+                        f.write(","+str(indexnum))
+
+
             if indexertype == "whoosh":
                 indexwriter = ix.writer()
                 indexwriter.update_document(