From 53d8c84762c714e7f4ca52881cd8aa06f45bd578 Mon Sep 17 00:00:00 2001
From: Akshat Jain <akjn11@gmail.com>
Date: Fri, 2 Aug 2024 08:00:41 +0530
Subject: [PATCH] Fix issues with fetching task reports in SQL statements
 endpoint for middlemanager

---
 .../apache/druid/msq/sql/resources/SqlStatementResource.java    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/extensions-core/multi-stage-query/src/main/java/org/apache/druid/msq/sql/resources/SqlStatementResource.java b/extensions-core/multi-stage-query/src/main/java/org/apache/druid/msq/sql/resources/SqlStatementResource.java
index 1ee5f1030a40..322727aea92d 100644
--- a/extensions-core/multi-stage-query/src/main/java/org/apache/druid/msq/sql/resources/SqlStatementResource.java
+++ b/extensions-core/multi-stage-query/src/main/java/org/apache/druid/msq/sql/resources/SqlStatementResource.java
@@ -595,7 +595,7 @@ private Optional<SqlStatementResult> getStatementStatus(
         ));
       }
       catch (DruidException e) {
-        if (e.getErrorCode().equals("notFound")) {
+        if (e.getErrorCode().equals("notFound") || e.getMessage().contains("Unable to contact overlord")) {
           return Optional.empty();
         }
         throw e;