From b7a5a955d710a64d8170e5d7b7576d6620a0ceff Mon Sep 17 00:00:00 2001
From: Adrian Mroz <adrian.mroz@allegro.pl>
Date: Tue, 6 Dec 2022 12:47:00 +0100
Subject: [PATCH] handle error in simulateQueryPlan for modal with Druid query

---
 .../modals/druid-query-modal/druid-query-modal.tsx     | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/client/modals/druid-query-modal/druid-query-modal.tsx b/src/client/modals/druid-query-modal/druid-query-modal.tsx
index 0c1331611..41db3b83d 100644
--- a/src/client/modals/druid-query-modal/druid-query-modal.tsx
+++ b/src/client/modals/druid-query-modal/druid-query-modal.tsx
@@ -34,12 +34,16 @@ export const DruidQueryModal: React.FunctionComponent<DruidQueryModalProps> = ({
   const queryFn = visualization.name === "grid" ? gridQuery : standardQuery;
   const query = queryFn(essence, timekeeper);
   const external = External.fromJS({ engine: "druid", attributes, source, customAggregations, customTransforms });
-  const plan = query.simulateQueryPlan({ main: external });
-  const planSource = JSON.stringify(plan, null, 2);
+  let plan;
+  try {
+    plan = JSON.stringify(query.simulateQueryPlan({ main: external }), null, 2);
+  } catch (e) {
+    plan = "Couldn't create Druid Query Plan.";
+  }
 
   return <SourceModal
     onClose={onClose}
     title="Druid query"
     copyLabel="Copy query"
-    source={planSource} />;
+    source={plan} />;
 };