Fix null (#3258)

finos · Nov 20, 2024 · f8f951e · f8f951e
1 parent 6c5d72e
commit f8f951e
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/...va/org/finos/legend/engine/persistence/components/relational/snowflake/SnowflakeSink.java b/...va/org/finos/legend/engine/persistence/components/relational/snowflake/SnowflakeSink.java
@@ -620,10 +620,26 @@ private void appendLoadQueryStats(Executor<SqlGen, TabularData, SqlPlan> executo
                     switch ((String) queryStats.get(QueryStatsLogicalPlanUtils.OPERATOR_TYPE_ALIAS))
                     {
                         case QueryStatsLogicalPlanUtils.EXTERNAL_SCAN_STAGE:
-                            stats.put(StatisticName.INPUT_FILES_BYTES_SCANNED, queryStats.get(QueryStatsLogicalPlanUtils.EXTERNAL_BYTES_SCANNED_ALIAS));
+                            Object externalScan = queryStats.get(QueryStatsLogicalPlanUtils.EXTERNAL_BYTES_SCANNED_ALIAS);
+                            if (externalScan != null)
+                            {
+                                stats.put(StatisticName.INPUT_FILES_BYTES_SCANNED, externalScan);
+                            }
+                            else
+                            {
+                                stats.put(StatisticName.INPUT_FILES_BYTES_SCANNED, 0);
+                            }
                             break;
                         case QueryStatsLogicalPlanUtils.INSERT_STAGE:
-                            stats.put(StatisticName.INCOMING_RECORD_COUNT, queryStats.get(QueryStatsLogicalPlanUtils.INPUT_ROWS_ALIAS));
+                            Object insert = queryStats.get(QueryStatsLogicalPlanUtils.INPUT_ROWS_ALIAS);
+                            if (insert != null)
+                            {
+                                stats.put(StatisticName.INCOMING_RECORD_COUNT, insert);
+                            }
+                            else
+                            {
+                                stats.put(StatisticName.INCOMING_RECORD_COUNT, 0);
+                            }
                             break;
                     }
                 });