adding more logs if query fails

airbnb · Nov 15, 2024 · fb105c9 · fb105c9
1 parent f874f8b
commit fb105c9
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/astra/src/main/java/com/slack/astra/logstore/rocksdb/query/RocksdbIndexSearcherImpl.java b/astra/src/main/java/com/slack/astra/logstore/rocksdb/query/RocksdbIndexSearcherImpl.java
@@ -1,14 +1,5 @@
 package com.slack.astra.logstore.rocksdb.query;
 
-import static java.util.stream.Collectors.toList;
-
-import com.fasterxml.jackson.databind.JsonNode;
-import com.fasterxml.jackson.databind.ObjectMapper;
-import com.google.common.base.Stopwatch;
-import com.slack.astra.logstore.LogMessage;
-import com.slack.astra.logstore.search.LogIndexSearcher;
-import com.slack.astra.logstore.search.SearchResult;
-import com.slack.astra.logstore.search.SourceFieldFilter;
 import java.io.IOException;
 import java.nio.ByteBuffer;
 import java.nio.file.Files;
@@ -21,7 +12,9 @@
 import java.util.List;
 import java.util.concurrent.TimeUnit;
 import java.util.stream.Collectors;
+import static java.util.stream.Collectors.toList;
 import java.util.stream.Stream;
+
 import org.opensearch.index.query.QueryBuilder;
 import org.opensearch.search.aggregations.AggregatorFactories;
 import org.rocksdb.IngestExternalFileOptions;
@@ -31,6 +24,14 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.google.common.base.Stopwatch;
+import com.slack.astra.logstore.LogMessage;
+import com.slack.astra.logstore.search.LogIndexSearcher;
+import com.slack.astra.logstore.search.SearchResult;
+import com.slack.astra.logstore.search.SourceFieldFilter;
+
 public class RocksdbIndexSearcherImpl implements LogIndexSearcher<LogMessage> {
   private static final Logger LOG = LoggerFactory.getLogger(RocksdbIndexSearcherImpl.class);
 
@@ -159,6 +160,7 @@ public SearchResult<LogMessage> search(
       return new SearchResult<>(
           results, elapsedTime.elapsed(TimeUnit.MICROSECONDS), 0, 0, 1, 1, null);
     } catch (IOException ex) {
+      LOG.error("Error fetching and parsing a result from rocksdb index for key: {}", dataset, ex);
       throw new IllegalStateException(
           "Error fetching and parsing a result from rocksdb index for key: " + dataset, ex);
     }