Fix stats type

finos · Dec 23, 2024 · 3cc10db · 3cc10db
1 parent b2fcba7
commit 3cc10db
Showing 1 changed file with 4 additions and 4 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
@@ -622,22 +622,22 @@ private void appendLoadQueryStats(Executor<SqlGen, TabularData, SqlPlan> executo
                             Object externalScan = queryStats.get(QueryStatsLogicalPlanUtils.EXTERNAL_BYTES_SCANNED_ALIAS);
                             if (externalScan != null)
                             {
-                                stats.put(StatisticName.INPUT_FILES_BYTES_SCANNED, externalScan);
+                                stats.put(StatisticName.INPUT_FILES_BYTES_SCANNED, Long.parseLong(String.valueOf(externalScan)));
                             }
                             else
                             {
-                                stats.put(StatisticName.INPUT_FILES_BYTES_SCANNED, 0);
+                                stats.put(StatisticName.INPUT_FILES_BYTES_SCANNED, 0L);
                             }
                             break;
                         case QueryStatsLogicalPlanUtils.INSERT_STAGE:
                             Object insert = queryStats.get(QueryStatsLogicalPlanUtils.INPUT_ROWS_ALIAS);
                             if (insert != null)
                             {
-                                stats.put(StatisticName.INCOMING_RECORD_COUNT, insert);
+                                stats.put(StatisticName.INCOMING_RECORD_COUNT, Long.parseLong(String.valueOf(insert)));
                             }
                             else
                             {
-                                stats.put(StatisticName.INCOMING_RECORD_COUNT, 0);
+                                stats.put(StatisticName.INCOMING_RECORD_COUNT, 0L);
                             }
                             break;
                     }