Skip to content

Commit ac85675

Browse files
authored
[b/402436948] Create a lite version of WarehouseMeteringHistory (#784)
1 parent ad8e73e commit ac85675

File tree

1 file changed

+19
-4
lines changed
  • dumper/app/src/main/java/com/google/edwmigration/dumper/application/dumper/connector/snowflake

1 file changed

+19
-4
lines changed

dumper/app/src/main/java/com/google/edwmigration/dumper/application/dumper/connector/snowflake/SnowflakePlanner.java

Lines changed: 19 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -205,9 +205,24 @@ Task<?> warehouseEventsTask() {
205205
}
206206

207207
Task<?> warehouseMeteringTask() {
208-
String view = "SNOWFLAKE.ACCOUNT_USAGE.WAREHOUSE_METERING_HISTORY";
209-
String query = String.format("SELECT warehouse_name, count(1) FROM %s GROUP BY ALL", view);
210-
ImmutableList<String> header = ImmutableList.of("Name", "Count");
211-
return new LiteTimeSeriesTask("warehouse_metering.csv", query, header);
208+
ImmutableList<String> selectList =
209+
ImmutableList.of(
210+
"start_time",
211+
"end_time",
212+
"warehouse_name",
213+
"credits_used_compute",
214+
"credits_used_cloud_services");
215+
String query =
216+
String.format(
217+
"SELECT %s FROM SNOWFLAKE.ACCOUNT_USAGE.WAREHOUSE_METERING_HISTORY",
218+
String.join(", ", selectList));
219+
ImmutableList<String> header =
220+
ImmutableList.of(
221+
"StartTime",
222+
"EndTime",
223+
"WarehouseName",
224+
"CreditsUsedCompute",
225+
"CreditsUsedCloudServices");
226+
return new LiteTimeSeriesTask("warehouse_metering_lite.csv", query, header);
212227
}
213228
}

0 commit comments

Comments
 (0)