Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -290,6 +290,8 @@ private String createExtendedQueryFromAccountUsage(ConnectorArguments arguments)
String overrideQuery = getOverrideQuery(arguments);
if (overrideQuery != null) return overrideQuery;

String overrideWhere = getOverrideWhere(arguments);

@SuppressWarnings("OrphanedFormatString")
StringBuilder queryBuilder =
new StringBuilder(
Expand Down Expand Up @@ -341,7 +343,12 @@ private String createExtendedQueryFromAccountUsage(ConnectorArguments arguments)
+ "AND end_time <= to_timestamp_ltz('%s')\n"
+ "AND is_client_generated_statement = FALSE\n");

queryBuilder.append(getOverrideWhere(arguments));
if (!StringUtils.isBlank(arguments.getQueryLogEarliestTimestamp()))
queryBuilder
.append("AND start_time >= ")
.append(arguments.getQueryLogEarliestTimestamp())
.append("\n");
if (overrideWhere != null) queryBuilder.append(" AND ").append(overrideWhere);
return queryBuilder.toString().replace('\n', ' ');
}

Expand All @@ -359,25 +366,16 @@ private String getOverrideQuery(@Nonnull ConnectorArguments arguments)
return null;
}

@Nonnull
@CheckForNull
private String getOverrideWhere(@Nonnull ConnectorArguments arguments)
throws MetadataDumperUsageException {
ConnectorProperty property = SnowflakeLogConnectorProperties.OVERRIDE_WHERE;
String overrideWhere = arguments.getDefinition(property);
if (overrideWhere != null) {
return String.format(" AND %s", overrideWhere);
} else {
return "";
}
return arguments.getDefinition(SnowflakeLogConnectorProperties.OVERRIDE_WHERE);
}

@Override
public final void addTasksTo(
@Nonnull List<? super Task<?>> out, @Nonnull ConnectorArguments arguments)
throws MetadataDumperUsageException {

boolean isAssessment = arguments.isAssessment();

out.add(new DumpMetadataTask(arguments, FORMAT_NAME));
out.add(new FormatTask(FORMAT_NAME));

Expand All @@ -392,7 +390,7 @@ public final void addTasksTo(
arguments, rotationDuration, IntervalExpander.createBasedOnDuration(rotationDuration));
logger.info("Exporting query log for " + queryLogIntervals);

if (!isAssessment) {
if (!arguments.isAssessment()) {
TaskDescription queryHistoryTask =
new TaskDescription(ZIP_ENTRY_PREFIX, newQueryFormat(arguments), Header.class);
queryLogIntervals.forEach(interval -> addJdbcTask(out, interval, queryHistoryTask));
Expand Down
Loading