Local Gateway: Move shard state to be stored under each shard, and no…

…t globally under _state, closes #1618.
elastic · Jan 17, 2012 · 534f487 · 534f487
1 parent 801c709
commit 534f487
Show file tree

Hide file tree

Showing 9 changed files with 568 additions and 435 deletions.
diff --git a/src/main/java/org/elasticsearch/env/NodeEnvironment.java b/src/main/java/org/elasticsearch/env/NodeEnvironment.java
@@ -19,6 +19,7 @@
 
 package org.elasticsearch.env;
 
+import com.google.common.collect.Sets;
 import org.apache.lucene.store.Lock;
 import org.apache.lucene.store.NativeFSLockFactory;
 import org.elasticsearch.ElasticSearchIllegalStateException;
@@ -34,6 +35,7 @@
 import java.io.File;
 import java.io.IOException;
 import java.util.Arrays;
+import java.util.Set;
 
 /**
  *
@@ -175,6 +177,36 @@ public File[] shardLocations(ShardId shardId) {
         return shardLocations;
     }
 
+    public Set<ShardId> findAllShardIds() throws Exception {
+        if (nodeFiles == null || locks == null) {
+            throw new ElasticSearchIllegalStateException("node is not configured to store local location");
+        }
+        Set<ShardId> shardIds = Sets.newHashSet();
+        for (File indicesLocation : nodeIndicesLocations) {
+            File[] indicesList = indicesLocation.listFiles();
+            if (indicesList == null) {
+                continue;
+            }
+            for (File indexLocation : indicesList) {
+                if (!indexLocation.isDirectory()) {
+                    continue;
+                }
+                String indexName = indexLocation.getName();
+                File[] shardsList = indexLocation.listFiles();
+                if (shardsList == null) {
+                    continue;
+                }
+                for (File shardLocation : shardsList) {
+                    if (!shardLocation.isDirectory()) {
+                        continue;
+                    }
+                    shardIds.add(new ShardId(indexName, Integer.parseInt(shardLocation.getName())));
+                }
+            }
+        }
+        return shardIds;
+    }
+
     public void close() {
         if (locks != null) {
             for (Lock lock : locks) {