NazerkeBS commented on a change in pull request #1527: URL: https://github.com/apache/lucene-solr/pull/1527#discussion_r435401394
########## File path: solr/core/src/java/org/apache/solr/request/SolrRequestInfo.java ########## @@ -52,35 +56,60 @@ private static final Logger log = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass()); public static SolrRequestInfo getRequestInfo() { - return threadLocal.get(); + Deque<SolrRequestInfo> stack = threadLocal.get(); + if (stack.isEmpty()) return null; + return stack.peek(); } + /** Adds the SolrRequestInfo onto the stack provided that the stack is not reached MAX_STACK_SIZE */ public static void setRequestInfo(SolrRequestInfo info) { - // TODO: temporary sanity check... this can be changed to just an assert in the future - SolrRequestInfo prev = threadLocal.get(); - if (prev != null) { - log.error("Previous SolrRequestInfo was not closed! req={}", prev.req.getOriginalParams()); - log.error("prev == info : {}", prev.req == info.req, new RuntimeException()); + Deque<SolrRequestInfo> stack = threadLocal.get(); + if (info == null) { + throw new IllegalArgumentException("SolrRequestInfo is null"); + } else { + if (stack.size() <= MAX_STACK_SIZE) { + stack.push(info); + } else { + assert true : "SolrRequestInfo Stack is full"; + log.error("SolrRequestInfo Stack is full"); + } } - assert prev == null; - - threadLocal.set(info); } + /** Removes the most recent SolrRequestInfo from the stack */ public static void clearRequestInfo() { - try { - SolrRequestInfo info = threadLocal.get(); - if (info != null && info.closeHooks != null) { - for (Closeable hook : info.closeHooks) { - try { - hook.close(); - } catch (Exception e) { - SolrException.log(log, "Exception during close hook", e); - } + Deque<SolrRequestInfo> stack = threadLocal.get(); + if (stack.isEmpty()) { + log.error("clearRequestInfo called too many times"); + } else { + SolrRequestInfo info = stack.pop(); + closeHooks(info); + } + } + + /** + * This reset method is more of a protection mechanism as + * we expect it to be empty by now because all "set" calls need to be balanced with a "clear". + */ + public static void reset() { + Deque<SolrRequestInfo> stack = threadLocal.get(); + boolean isEmpty = stack.isEmpty(); + while (!stack.isEmpty()) { + SolrRequestInfo info = stack.pop(); + closeHooks(info); + } + assert isEmpty : "SolrRequestInfo Stack should have been cleared."; + } + + private static void closeHooks(SolrRequestInfo info) { + if (info != null && info.closeHooks != null) { Review comment: it can be null; when getRequestInfo is called, in case of the stack.isEmpty(), it returns null; ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: issues-unsubscr...@lucene.apache.org For additional commands, e-mail: issues-h...@lucene.apache.org