From c66273ef272140f1e0482fc14b2433d7f66d101e Mon Sep 17 00:00:00 2001
From: Stephane Gamard <stephane.gamard@sonarsource.com>
Date: Tue, 9 Sep 2014 10:09:34 +0200
Subject: [PATCH] SONAR-5530 - Implemented getByKeys in BaseIndex

---
 .../src/main/java/org/sonar/server/search/BaseIndex.java  | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/server/sonar-server/src/main/java/org/sonar/server/search/BaseIndex.java b/server/sonar-server/src/main/java/org/sonar/server/search/BaseIndex.java
index 006869ce648..e5039c14c6b 100644
--- a/server/sonar-server/src/main/java/org/sonar/server/search/BaseIndex.java
+++ b/server/sonar-server/src/main/java/org/sonar/server/search/BaseIndex.java
@@ -21,6 +21,7 @@ package org.sonar.server.search;
 
 import com.google.common.collect.ArrayListMultimap;
 import com.google.common.collect.ImmutableMap;
+import com.google.common.collect.ImmutableSet;
 import com.google.common.collect.Multimap;
 import org.elasticsearch.action.admin.indices.exists.indices.IndicesExistsResponse;
 import org.elasticsearch.action.count.CountRequestBuilder;
@@ -400,8 +401,7 @@ public abstract class BaseIndex<DOMAIN, DTO extends Dto<KEY>, KEY extends Serial
     return null;
   }
 
-  public Collection<DOMAIN> getByKeys(KEY... keys) {
-
+  public Collection<DOMAIN> getByKeys(Collection<KEY> keys) {
     List<DOMAIN> results = new ArrayList<DOMAIN>();
     MultiGetRequestBuilder request = client.prepareMultiGet()
       .setPreference("_local");
@@ -421,6 +421,10 @@ public abstract class BaseIndex<DOMAIN, DTO extends Dto<KEY>, KEY extends Serial
     return results;
   }
 
+  public Collection<DOMAIN> getByKeys(KEY... keys) {
+    return getByKeys(ImmutableSet.<KEY>copyOf(keys));
+  }
+
   /* ES QueryHelper Methods */
 
   protected BoolFilterBuilder addTermFilter(BoolFilterBuilder filter, String field, @Nullable Collection<String> values) {
-- 
2.39.5